Bank_database.cpp

#include "BankDB.h"

int main()
{
	BankDB db;
	int choice;
	string path = "DB.text";

	while (true)
	{
		std::cout << "\n--- BANK DB MENU ---\n";
		std::cout << "1. Load DB from file\n";
		std::cout << "2. Save DB to file\n";
		std::cout << "3. Show DB\n";
		std::cout << "4. Add account\n";
		std::cout << "5. Remove account\n";
		std::cout << "6. Sort by account number\n";
		std::cout << "7. Find by account number\n";
		std::cout << "8. Filter by date range\n";
		std::cout << "9. Get list of clients for prolongation\n";
		std::cout << "10. Calculate future deposits\n";
		std::cout << "0. Exit\n";
		std::cout << "Choice: ";

		std::cin >> choice;

		switch (choice)
		{
		case 0:
			std::cout << "Exiting program...\n";
			return 0;  

		case 1:
		{
			std::string temp;

			std::cout << "Enter file name (empty = DB.txt): ";
			std::cin.ignore();
			std::getline(std::cin, temp);

			if (temp.empty())
			{
				path = "DB.txt";
			}
			else
			{
				temp += ".txt";
				path = temp;
			}

			db.loadFromFile(path);
			break;
		}
		case 2:
			db.saveToFile("DB.txt");
			break;  

		case 3:
			db.printAll();
			break;  

		case 4:
			db.addAccount();
			break;  

		case 5:
			db.removeAccount();
			break;  

		case 6:
			db.sortByAccountNumber();
			break;  

		case 7:
			db.findByNumber();
			break;  

		case 8:
		{
			unsigned short y1, m1, d1;
			unsigned short y2, m2, d2;

			std::cout << "From (y m d): ";
			std::cin >> y1 >> m1 >> d1;

			std::cout << "To (y m d): ";
			std::cin >> y2 >> m2 >> d2;

			CDate from{ y1, m1, d1 };
			CDate to{ y2, m2, d2 };

			db.filteredByDateRange(from, to);
			break;  
		}
		case 9:
		{
			db.getClientsForProlongation();
			break;
		}
		case 10: 
		{
			int m;
			std::cout << "How many months into the future: ";
			std::cin >> m;
			if (m < 1) {
				std::cout << "Can't do that.";
				break;
			}
			db.calculateFutureDeposits(m);
			break;
		}
		default:
			std::cout << "Invalid choice. Please enter a number between 0 and 10.\n";
			break;  
		}
	}

	return 0;
}