main.cpp

#include "main.h"
// Load data from the database into in-memory containers
void retrieveDatabaseData() {
  customers.clear();
  getCustomers(); // retrieve the customers that are in the database
  staffs.clear();
  getStaffs(); // retrieve the staffs that are in the database
  roomtypes.clear();
  getRoomTypes(); // retrieve the room types that are in the database
  rooms.clear();
  getRooms(); // retrieve the rooms that are in the database
  bookings.clear();
  getBookings();   
  payments.clear();
  getPayments(); // retrieve the payments that are in the database
}

// Main program loop
void startProgram() {
    cout << "\n\n============================================\n";
    showHighlight("Welcome to the Hotel Management System");
    cout << "============================================\n\n";

  while (true) {
    if (current_user.customer_id == -1 && current_user.staff_id == -1) {
        showRegistrationMenu();
    } else {
      if (current_user.isCustomer) {
        showCustomerMenu(startProgram, retrieveDatabaseData);
      } 
      else {
        showStaffMenu(startProgram, retrieveDatabaseData);
      }
    }
  }
}

// Entry point of the program
int main() {
    connectToDatabase();       // Initialize DB connection
    createTables();            // Create required tables if they don't exist
    retrieveDatabaseData();    // Load initial data

    startProgram();            // Start the user interface loop

    return 0;
}