AirPort.cpp

#include <iostream>
#include <vector>
#include <string>
#include <iomanip>
#include <algorithm>
#include <fstream>
using namespace std;

// Class: Date
class Date {
private:
    int day, month, year, hour, minute;

public:
    void setDay(int d) { day = d; }
    int getDay() const { return day; }

    void setMonth(int m) { month = m; }
    int getMonth() const { return month; }

    void setYear(int y) { year = y; }
    int getYear() const { return year; }

    void setHour(int h) { hour = h; }
    int getHour() const { return hour; }

    void setMinute(int m) { minute = m; }
    int getMinute() const { return minute; }

    friend ostream& operator<<(ostream& out, const Date& date) {
        out << setw(2) << setfill('0') << date.day << "/"
            << setw(2) << setfill('0') << date.month << "/"
            << date.year << " "
            << setw(2) << setfill('0') << date.hour << ":"
            << setw(2) << setfill('0') << date.minute;
        return out;
    }

    friend istream& operator>>(istream& in, Date& date) {
        cout << "Enter date (DD MM YYYY HH MM): ";
        in >> date.day >> date.month >> date.year >> date.hour >> date.minute;
        return in;
    }

    bool operator<(const Date& other) const {
        if (year != other.year) return year < other.year;
        if (month != other.month) return month < other.month;
        if (day != other.day) return day < other.day;
        if (hour != other.hour) return hour < other.hour;
        return minute < other.minute;
    }
};

// Class: Food
class Food {
private:
    string type; // Economic or Business
    string meal;

public:
    void setType(const string& t) { type = t; }
    string getType() const { return type; }

    void setMeal(const string& m) { meal = m; }
    string getMeal() const { return meal; }

    friend ostream& operator<<(ostream& out, const Food& food) {
        out << "Type: " << food.type << ", Meal: " << food.meal;
        return out;
    }

    friend istream& operator>>(istream& in, Food& food) {
        cout << "Enter food type (Economic/Business): ";
        in >> food.type;
        cout << "Enter meal: ";
        in >> ws;
        getline(in, food.meal);
        return in;
    }
};

// Class: Airport
class Airport {
protected:
    vector<string> domesticFlights;
    vector<string> foreignFlights;

public:
    virtual void addFlight(const string& flight) = 0;
    virtual void removeFlight(const string& flight) = 0;
    virtual void displayFlights() const = 0;

    virtual ~Airport() = default;
};

// Class: DomesticFlights
class DomesticFlights : public Airport {
private:
    struct Flight {
        string originCity;
        string destinationCity;
        Date flightDate;
        string planeName;
        bool isCancelled;
    };

    vector<Flight> flights;

public:
    void addFlight(const string& flight) override {
        cout << "This method is not directly used in this class. Use the extended version.\n";
    }

    void addFlight(const string& origin, const string& destination, const Date& date, const string& plane, bool cancelled) {
        if (flights.size() >= 10) {
            cout << "Cannot add more than 10 flights." << endl;
            return;
        }
        flights.push_back({origin, destination, date, plane, cancelled});
        cout << "Flight added successfully." << endl;
    }

    void removeFlight(const string& flight) override {
        cout << "This method is not directly used in this class. Use the extended version.\n";
    }

    void removeFlight(const string& origin, const string& destination) {
        auto it = find_if(flights.begin(), flights.end(), [&](const Flight& f) {
            return f.originCity == origin && f.destinationCity == destination;
        });
        if (it != flights.end()) {
            flights.erase(it);
            cout << "Flight removed successfully." << endl;
        } else {
            cout << "Flight not found." << endl;
        }
    }

    void displayFlights() const override {
        vector<Flight> sortedFlights = flights;
        sort(sortedFlights.begin(), sortedFlights.end(), [](const Flight& a, const Flight& b) {
            return a.flightDate < b.flightDate;
        });
        ofstream outFile("DomesticFlights.txt");
        if (outFile.is_open()) {
            for (const auto& f : sortedFlights) {
                outFile << "Origin: " << f.originCity << " | Destination: " << f.destinationCity
                        << " | Date: " << f.flightDate << " | Plane: " << f.planeName
                        << " | Cancelled: " << (f.isCancelled ? "Yes" : "No") << endl;
                cout << "Origin: " << f.originCity << " | Destination: " << f.destinationCity
                     << " | Date: " << f.flightDate << " | Plane: " << f.planeName
                     << " | Cancelled: " << (f.isCancelled ? "Yes" : "No") << endl;
            }
            outFile.close();
            cout << "Flight details saved to DomesticFlights.txt." << endl;
        } else {
            cout << "Error opening file to save flights." << endl;
        }
    }
};

// Class: ForeignFlights
class ForeignFlights : public Airport {
private:
    struct Flight {
        string originCity;
        string destinationCity;
        Date flightDate;
        string planeName;
        bool isCancelled;
    };

    vector<Flight> flights;

public:
    void addFlight(const string& flight) override {
        cout << "This method is not directly used in this class. Use the extended version.\n";
    }

    void addFlight(const string& origin, const string& destination, const Date& date, const string& plane, bool cancelled) {
        if (flights.size() >= 10) {
            cout << "Cannot add more than 10 flights." << endl;
            return;
        }
        flights.push_back({origin, destination, date, plane, cancelled});
        cout << "Flight added successfully." << endl;
    }

    void removeFlight(const string& flight) override {
        cout << "This method is not directly used in this class. Use the extended version.\n";
    }

    void removeFlight(const string& origin, const string& destination) {
        auto it = find_if(flights.begin(), flights.end(), [&](const Flight& f) {
            return f.originCity == origin && f.destinationCity == destination;
        });
        if (it != flights.end()) {
            flights.erase(it);
            cout << "Flight removed successfully." << endl;
        } else {
            cout << "Flight not found." << endl;
        }
    }

    void displayFlights() const override {
        vector<Flight> sortedFlights = flights;
        sort(sortedFlights.begin(), sortedFlights.end(), [](const Flight& a, const Flight& b) {
            return a.flightDate < b.flightDate;
        });
        ofstream outFile("ForeignFlights.txt");
        if (outFile.is_open()) {
            for (const auto& f : sortedFlights) {
                outFile << "Origin: " << f.originCity << " | Destination: " << f.destinationCity
                        << " | Date: " << f.flightDate << " | Plane: " << f.planeName
                        << " | Cancelled: " << (f.isCancelled ? "Yes" : "No") << endl;
                cout << "Origin: " << f.originCity << " | Destination: " << f.destinationCity
                     << " | Date: " << f.flightDate << " | Plane: " << f.planeName
                     << " | Cancelled: " << (f.isCancelled ? "Yes" : "No") << endl;
            }
            outFile.close();
            cout << "Flight details saved to ForeignFlights.txt." << endl;
        } else {
            cout << "Error opening file to save flights." << endl;
        }
    }
};

// Class: Tax
class Tax {
private:
    const double percentage;

public:
    Tax(double p) : percentage(p) {}

    double calculateTax(double price) const {
        return price + (price * (percentage / 100));
    }
};

// Class: Customer
class Customer {
private:
    string name;
    double balance;
    vector<string> purchasedFlights;

public:
    void setName(const string& n) { name = n; }
    string getName() const { return name; }

    void setBalance(double b) { balance = b; }
    double getBalance() const { return balance; }

    void addPurchasedFlight(const string& flight) {
        purchasedFlights.push_back(flight);
        cout << "Flight purchased: " << flight << endl;
        ofstream outFile("PurchasedFlights.txt", ios::app);
        if (outFile.is_open()) {
            outFile << flight << endl;
            outFile.close();
        } else {
            cout << "Error saving purchased flight to file." << endl;
        }
    }

    void refundTicket(const string& flight, double penalty) {
        auto it = find(purchasedFlights.begin(), purchasedFlights.end(), flight);
        if (it != purchasedFlights.end()) {
            purchasedFlights.erase(it);
            balance -= penalty;
            cout << "Flight refunded: " << flight << " | Penalty: " << penalty << endl;
        } else {
            cout << "Ticket not found." << endl;
        }
    }

    void displayPurchasedFlights() const {
        cout << "Purchased Flights:" << endl;
        for (const auto& flight : purchasedFlights) {
            cout << "- " << flight << endl;
        }
    }

    void increaseBalance(double amount) {
        balance += amount;
        cout << "Balance increased by " << amount << ". New balance: " << balance << endl;
    }
};

// Class: Admin
class Admin {
private:
    string adminName;
    string adminPassword;

public:
    void setAdminName(const string& name) { adminName = name; }
    string getAdminName() const { return adminName; }

    void setAdminPassword(const string& password) { adminPassword = password; }
    string getAdminPassword() const { return adminPassword; }

    bool authenticate(const string& name, const string& password) const {
        return (name == adminName && password == adminPassword);
    }
};

// Admin Panel
void adminPanel(Admin& admin, DomesticFlights& domestic, ForeignFlights& foreign) {
    string name, password;
    cout << "Enter Admin Name: ";
    cin >> name;
    cout << "Enter Admin Password: ";
    cin >> password;

    if (admin.authenticate(name, password)) {
        int choice;
        do {
            cout << "\n--- Admin Panel ---\n";
            cout << "1. Add Domestic Flight\n";
            cout << "2. Add Foreign Flight\n";
            cout << "3. Remove Domestic Flight\n";
            cout << "4. Remove Foreign Flight\n";
            cout << "5. Display Domestic Flights\n";
            cout << "6. Display Foreign Flights\n";
            cout << "7. Logout\n";
            cout << "Enter your choice: ";
            cin >> choice;

            switch (choice) {
                case 1: {
                    string origin, destination, plane;
                    Date date;
                    bool cancelled;
                    cout << "Enter origin city: ";
                    cin >> origin;
                    cout << "Enter destination city: ";
                    cin >> destination;
                    cout << "Enter flight date: \n";
                    cin >> date;
                    cout << "Enter plane name: ";
                    cin >> plane;
                    cout << "Is flight cancelled (1 for Yes, 0 for No): ";
                    cin >> cancelled;
                    domestic.addFlight(origin, destination, date, plane, cancelled);
                    break;
                }
                case 2: {
                    string origin, destination, plane;
                    Date date;
                    bool cancelled;
                    cout << "Enter origin city: ";
                    cin >> origin;
                    cout << "Enter destination city: ";
                    cin >> destination;
                    cout << "Enter flight date: \n";
                    cin >> date;
                    cout << "Enter plane name: ";
                    cin >> plane;
                    cout << "Is flight cancelled (1 for Yes, 0 for No): ";
                    cin >> cancelled;
                    foreign.addFlight(origin, destination, date, plane, cancelled);
                    break;
                }
                case 3: {
                    string origin, destination;
                    cout << "Enter origin city: ";
                    cin >> origin;
                    cout << "Enter destination city: ";
                    cin >> destination;
                    domestic.removeFlight(origin, destination);
                    break;
                }
                case 4: {
                    string origin, destination;
                    cout << "Enter origin city: ";
                    cin >> origin;
                    cout << "Enter destination city: ";
                    cin >> destination;
                    foreign.removeFlight(origin, destination);
                    break;
                }
                case 5:
                    domestic.displayFlights();
                    break;
                case 6:
                    foreign.displayFlights();
                    break;
                case 7:
                    cout << "Logging out...\n";
                    break;
                default:
                    cout << "Invalid choice. Try again.\n";
            }
        } while (choice != 7);
    } else {
        cout << "Authentication failed.\n";
    }
}

// Customer Panel
void customerPanel(Customer& customer, DomesticFlights& domestic, ForeignFlights& foreign, Tax& tax) {
    int choice;
    do {
        cout << "\n--- Customer Panel ---\n";
        cout << "1. View Domestic Flights\n";
        cout << "2. View Foreign Flights\n";
        cout << "3. Purchase Flight\n";
        cout << "4. Refund Flight\n";
        cout << "5. View Purchased Flights\n";
        cout << "6. Increase Balance\n";
        cout << "7. Logout\n";
        cout << "Enter your choice: ";
        cin >> choice;

        switch (choice) {
            case 1:
                domestic.displayFlights();
                break;
            case 2:
                foreign.displayFlights();
                break;
            case 3: {
                string origin, destination;
                double price;
                cout << "Enter origin city: ";
                cin >> origin;
                cout << "Enter destination city: ";
                cin >> destination;
                cout << "Enter ticket price: ";
                cin >> price;

                double totalPrice = tax.calculateTax(price);
                if (customer.getBalance() >= totalPrice) {
                    customer.setBalance(customer.getBalance() - totalPrice);
                    customer.addPurchasedFlight(origin + "-" + destination);
                    cout << "Ticket purchased successfully. Total cost: " << totalPrice << endl;
                } else {
                    cout << "Insufficient balance." << endl;
                }
                break;
            }
            case 4: {
                string flight;
                double penalty;
                cout << "Enter flight to refund: ";
                cin.ignore();
                getline(cin, flight);
                cout << "Enter refund penalty: ";
                cin >> penalty;
                customer.refundTicket(flight, penalty);
                break;
            }
            case 5:
                customer.displayPurchasedFlights();
                break;
            case 6: {
                double amount;
                cout << "Enter amount to increase balance: ";
                cin >> amount;
                customer.increaseBalance(amount);
                break;
            }
            case 7:
                cout << "Logging out...\n";
                break;
            default:
                cout << "Invalid choice. Try again.\n";
        }
    } while (choice != 7);
}

int main() {
    DomesticFlights domestic;
    ForeignFlights foreign;
    Customer customer;
    Admin admin;
    Tax tax(10.0); // 10% tax

    string adminName = "admin";
    string adminPassword = "1234";
    admin.setAdminName(adminName);
    admin.setAdminPassword(adminPassword);

    int mainChoice;
    do {
        cout << "\n--- Main Menu ---\n";
        cout << "1. Admin Panel\n";
        cout << "2. Customer Panel\n";
        cout << "3. Exit\n";
        cout << "Enter your choice: ";
        cin >> mainChoice;

        switch (mainChoice) {
            case 1:
                adminPanel(admin, domestic, foreign);
                break;
            case 2:
                customerPanel(customer, domestic, foreign, tax);
                break;
            case 3:
                cout << "Exiting the system. Goodbye!\n";
                break;
            default:
                cout << "Invalid choice. Try again.\n";
        }
    } while (mainChoice != 3);

    return 0;
}