main.cpp

#include "common.h"
#include "RouteGraph.h"
#include "BusScheduler.h"
#include "DataLoader.h"
#include "BookingManager.h"
#include <iostream>
#include <string>

using namespace std;

// Function to print the main menu
void printMenu()
{
    cout << "\n======= Smart Bus System Menu =======" << endl;
    cout << "  1. Plan a New Journey" << endl;
    cout << "  2. View an Existing Ticket" << endl;
    cout << "  3. Cancel a Ticket" << endl;
    cout << "  4. List All Available Stops" << endl;
    cout << "  5. Quit" << endl;
    cout << "=======================================" << endl;
    cout << "Enter your choice (1-5): ";
}

// Function to handle planning and booking
void planAndBookJourney(RouteGraph &graph, BusScheduler &scheduler, BookingManager &booker)
{
    string startStop, endStop, startTimeStr, passengerName, choice;

    graph.printAllStops();

    cout << "\nEnter your starting stop: ";
    getline(cin, startStop);

    cout << "Enter your destination stop: ";
    getline(cin, endStop);

    cout << "Enter desired departure time (HH:MM, e.g., 09:30): ";
    getline(cin, startTimeStr);

    int startTimeMinutes = timeToMinutes(startTimeStr); // Validate time format
    if (startTimeMinutes == -1)
    {
        cout << "Error: Invalid time format. Please use HH:MM." << endl;
        return;
    }

    // Find Optimal Path -> fastest travel time
    RoutePlan optimalPath = graph.findShortestPath(startStop, endStop);
    if (optimalPath.totalTime == -1)
    {
        return;
    }

    // Find Actual Scheduled Journey
    JourneyPlan finalJourney = scheduler.findJourney(optimalPath, startTimeMinutes);
    if (finalJourney.finalArrivalTime == -1)
    {
        cout << "Tip: Try departing at a different time." << endl;
        return;
    }

    // Print the plan for user confirmation
    scheduler.printJourneyPlan(finalJourney);

    // Ask user to book
    cout << "\nDo you want to book this journey? (y/n): ";
    getline(cin, choice);

    if (normalizeString(choice) == "y")
    {
        cout << "Please enter passenger name: ";
        getline(cin, passengerName);

        // This is where we use the BookingManager
        // Add request to the queue
        booker.requestTicket(passengerName, finalJourney);
        // Process from queue, save to map, and save to CSV
        booker.processNextBooking();
    }
    else
    {
        cout << "Booking canceled. Returning to main menu." << endl;
    }
}

// view a ticket
void viewTicket(BookingManager &booker)
{
    string ticketId;
    cout << "\nEnter your Ticket ID (e.g., T-1001): ";
    getline(cin, ticketId);
    booker.viewTicket(ticketId);
}

// cancel a ticket
void cancelTicket(BookingManager &booker)
{
    string ticketId;
    cout << "\nEnter Ticket ID to cancel (e.g., T-1001): ";
    getline(cin, ticketId);
    booker.cancelTicket(ticketId);
}

int main()
{
    // --- 1. System Initialization ---
    cout << "Initializing Smart Bus System..." << endl;

    RouteGraph graph;
    BusScheduler scheduler(&graph);
    BookingManager booker("tickets.csv");
    // using the DataLoader to populate the system
    DataLoader::loadNetworkData(graph);
    DataLoader::loadBusSchedules(scheduler, graph);

    cout << "\n--- Welcome to the Smart Bus Route Planner ---" << endl;

    string choice;
    while (true)
    {
        printMenu();
        getline(cin, choice);

        if (choice == "1")
        {
            planAndBookJourney(graph, scheduler, booker);
        }
        else if (choice == "2")
        {
            viewTicket(booker);
        }
        else if (choice == "3")
        {
            cancelTicket(booker);
        }
        else if (choice == "4")
        {
            graph.printAllStops();
        }
        else if (choice == "5")
        {
            cout << "Thank you for using the Smart Bus Planner!" << endl;
            break;
        }
        else
        {
            cout << "Invalid choice. Please enter a number from 1 to 5." << endl;
        }
    }

    return 0;
}