main.cpp

#include <iostream>
#include <string>

#include "SQLServerConn.h"
#include "DBConn.h"
#include "CustomerManager.h"
#include "SupplierManager.h"
#include "SupplierNameManager.h"
#include "ProductManager.h"
#include "CartItemManager.h"
#include "TransactionManager.h"
#include "OrderItemManager.h"

#include "RetailApp.h"

int main() {
    try {
        // Table names
        std::string connectionString = "DRIVER={SQL Server};SERVER=KN\\SQLEXPRESS;Trusted_Connection=yes;";
        std::string dbName = "sample_store";
        std::string customerTableName = "Customers";
        std::string supplierTableName = "Suppliers";
        std::string supplierNameTableName = "Supplier_Names";
        std::string productTableName = "Products";
        std::string cartItemTableName = "Cart_Items";
        std::string transactionTableName = "Transactions";
        std::string orderItemTableName = "Order_Items";


        // Connect to SQL Server instance on 
        SQLServerConn connector;
        connector.connect(connectionString);

        // Connect to database 
        DBConn dbConn(connector.getHDBC());
        if (!dbConn.dbExists(dbName)) {
            dbConn.createDatabase(dbName);
        }
        dbConn.useDatabase(dbName);


        // Create manager for 'customer'. Create table if it doesn't already exist
        CustomerManager customerManager(dbConn, customerTableName);
        if (!dbConn.tableExists(customerTableName)) {
            customerManager.initTable();
        }

        /*
        - Create managers for the SupplierName and Supplier.

        NOTE: Then after this you must initialize the Supplier table first, adn then the
        supplier name table. This is because the latter references the former, so we must make sure
        the former exists before the latter.
        */
        SupplierNameManager supplierNameManager(dbConn, supplierNameTableName, supplierTableName);
        SupplierManager supplierManager(dbConn, supplierTableName, supplierNameManager);

        if (!dbConn.tableExists(supplierTableName)) {
            supplierManager.initTable();
        }

        if (!dbConn.tableExists(supplierNameTableName)) {
            supplierNameManager.initTable();
        }

        // Create manager for products table
        ProductManager productManager(dbConn, productTableName, supplierTableName);
        if (!dbConn.tableExists(productTableName)) {
            productManager.initTable();
        }

        // Create manager for cart items table and initialize table if it doesn't already exist
        CartItemManager cartItemManager(dbConn, cartItemTableName, customerTableName, productTableName);
        if (!dbConn.tableExists(cartItemTableName)) {
            cartItemManager.initTable();
        }

        // Create manager for transactions table; needs for customer table to exist first
        TransactionManager transactionManager(dbConn, transactionTableName, customerTableName);
        if (!dbConn.tableExists(transactionTableName)) {
            transactionManager.initTable();
        }

        // Create manager for order items table; needs the transaction and product table to exist first.
        OrderItemManager orderItemManager(dbConn, orderItemTableName, transactionTableName, productTableName);
        if (!dbConn.tableExists(orderItemTableName)) {
            orderItemManager.initTable();
        }



        

        RetailApp myStore(customerManager, supplierManager, productManager, cartItemManager, transactionManager, orderItemManager);
        int choice;

        do {
            // Display main menu and prompt input
            std::cout << "Main Menu: " << std::endl;
            std::cout << "1. Customers" << std::endl;
            std::cout << "2. Suppliers" << std::endl;
            std::cout << "3. Products" << std::endl;
            std::cout << "4. Cart Items" << std::endl;
            std::cout << "5. Transactions" << std::endl;
            std::cout << "6. Quit" << std::endl;
            std::cout << "Please enter a number to continue: ";
            std::cin >> choice;

            if (std::cin.fail()) {
                std::cout << "Invalid input. Please enter a number!" << std::endl;
                std::cin.clear(); // Clear the error flag
                std::cin.ignore(64, '\n'); // Clear input buffer up to 64 characters or until newline is encountered
                continue; // Restart the loop
            }

            switch (choice) {
            case 1:
                myStore.handleCustomerMenu();
                break;
            case 2:
                myStore.handleSupplierMenu();
                break;
            case 3:
                myStore.handleProductMenu();
                break;
            case 4:
                myStore.handleCartMenu();
                break;
            case 5:
                myStore.handleTransactionMenu();
                break;
            case 6:
                std::cout << "Exiting Program!" << std::endl;
                break;
            default:
                std::cout << "Invalid choice. Please enter a number between 1 and 5." << std::endl;
            }

        } while (choice != 6);
        

        // Disconnect from SQL Server
        connector.disconnect();
    }
    catch (const std::exception& ex) {
        std::cerr << "Exception caught: " << ex.what() << std::endl;
    }
    return 0; // Don't forget to return a value from main
}