OrderManagementSystem.java

import java.util.Scanner;
import java.time.LocalDate;

/**
 * Main controller class for the Dough Jamaica Order Management System (DJOMS).
 * 
 * Responsibilities:
 * - Initialize managers and load data from files
 * - Display menus and handle user navigation
 * - Coordinate between modules (Order, Customer, Payment managers)
 * - Trigger alerts for deadlines (startup + on demand)
 * - Save data back to CSV files via DataStore
 */
public class OrderManagementSystem {

    private final OrderManager orderManager;
    private final CustomerManager customerManager;
    private final PaymentManager paymentManager;
    private final DataStore datastore;
    private final Scanner scanner;

    /**
     * Constructor: loads all data and checks for alerts.
     */
    public OrderManagementSystem() {
        datastore = new DataStore();

        // Load stored CSV data into managers
        orderManager = datastore.loadOrders();
        customerManager = datastore.loadCustomers();
        paymentManager = datastore.loadPayments();

        scanner = new Scanner(System.in);

        System.out.println("\n=== Checking for Upcoming Order Deadlines... ===");
        orderManager.showUpcomingAlerts();    // startup alerts
    }

    /**
     * Main program loop. This displays menus until the user exits.
     */
    public void run() {
        boolean running = true;

        while (running) {
            printMainMenu();
            int choice = readInt("Enter your choice: ");

            switch (choice) {
                case 1 -> manageOrders();
                case 2 -> manageCustomers();
                case 3 -> managePayments();
                case 4 -> showAlerts();   // manual alert check
                case 5 -> saveAll();
                case 6 -> {
                    saveAll();
                    System.out.println("Goodbye! Exiting system...");
                    running = false;
                }
                default -> System.out.println("Invalid choice. Please try again.");
            }
        }
    }

    // --------------------------------------------------------------
    //                     MAIN MENU
    // --------------------------------------------------------------

    private void printMainMenu() {
        System.out.println("\n====== DOUGH JAMAICA ORDER MANAGEMENT SYSTEM ======");
        System.out.println("1. Order Management");
        System.out.println("2. Customer Management");
        System.out.println("3. Payment Management");
        System.out.println("4. Upcoming Alerts");
        System.out.println("5. Save All Changes");
        System.out.println("6. Exit System");
        System.out.println("====================================================");
    }

    // --------------------------------------------------------------
    //                     ORDER MENU
    // --------------------------------------------------------------

    private void manageOrders() {
        System.out.println("\n--- Order Menu ---");
        System.out.println("1. Add New Order");
        System.out.println("2. View All Orders");
        System.out.println("3. Update Order Status");
        System.out.println("4. Archive Completed Orders");
        System.out.println("5. Edit Customer Order");
        System.out.println("6. Delete Customer Order");
        System.out.println("7. Return to Main Menu");

        int choice = readInt("Enter your choice: ");

        switch (choice) {
            case 1 -> addOrderFlow();
            case 2 -> orderManager.viewOrders();
            case 3 -> updateOrderStatus();
            case 4 -> orderManager.archiveCompleted();
            case 5 -> editOrderFlow();
            case 6 -> deleteOrderFlow();
            default -> System.out.println("Invalid choice.");
        }
    }

    /**
     * Flow for adding an order (simplified; can be expanded).
     */
    private void addOrderFlow() {
        System.out.println("\n--- Add New Order ---");

        System.out.print("Enter Customer ID: ");
        String id = scanner.nextLine().trim();

        Customer c = customerManager.findCustomer(id);

        if (c == null) {
            System.out.println("Customer not found. Add customer first.");
            return;
        }

        System.out.print("Enter delivery date (YYYY-MM-DD): ");
        String date = scanner.nextLine();

        System.out.print("Enter order type: ");
        String type = scanner.nextLine();

        System.out.print("Enter order method (Please type: \"Pickup\" or \"Delivery\"): ");
        String order_method = scanner.nextLine();
        order_method.trim().toLowerCase();

        String address = "12 Fairway Avenue"; //Company Address

        if (order_method == "delivery"){
            System.out.print("Enter delivery/pickup address: ");
            address = scanner.nextLine();
        }

    
        
        
        //System.out.print("Enter delivery/pickup address: ");
        //String address = scanner.nextLine();

        System.out.print("Enter payment method (POS/Transfer): ");
        String method = scanner.nextLine();

        System.out.print("Enter total amount: ");
        double amt = readDouble("");

        Order o = new Order(orderManager.getNextOrderID(), c, 
                            java.time.LocalDate.parse(date), type, 
                            address, method, amt);

        orderManager.addOrder(o);

        System.out.println("Order added successfully!");
    }


    private void editOrderFlow(){
        System.out.println("\n--- Edit Order ---");
        System.out.print("Enter Order ID to edit: ");
        int orderID = Integer.parseInt(scanner.nextLine());

        Order ordertoEdit = null;
        for (Order o: orderManager.getOrdersList()){
            if (o.getOrderID() == orderID){
                ordertoEdit = o;
                break;
            }
        }

        if (ordertoEdit == null){
            System.out.print("Order ID not found");
            return;
        }

        System.out.print("Enter new delivery date: ");
        String newDate = scanner.nextLine();

        if(!newDate.isEmpty()){
            try{
                ordertoEdit.deliveryDate = LocalDate.parse(newDate);
            } catch(Exception e){
                System.out.print("Invalid Date Entered");
            }
        }

        System.out.print("Enter new order type: ");
        String newType = scanner.nextLine();

        if(!newType.isEmpty()){
            ordertoEdit.orderType = newType;
        }

        System.out.print("Enter new order method: ");
        String orderMethod = scanner.nextLine();

        if(!orderMethod.isEmpty()){
            ordertoEdit.method = orderMethod;
        }

        if(ordertoEdit.method.equalsIgnoreCase("Delivery")){
            System.out.print("Enter new delivery address: ");
            String newAddress = scanner.nextLine();
            if (!newAddress.isEmpty()){
                ordertoEdit.address = newAddress;
            }
        } else{
            ordertoEdit.address = "12 Fairway Avenue";
        }

        System.out.print("Enter new status method: ");
        String newStatus = scanner.nextLine();

        if(!orderMethod.isEmpty()){
            ordertoEdit.setStatus(newStatus);
        } else{
            System.out.print("Invalid Status");
        }
    
        System.out.print("Order edited sucessfully");
    }

    private void deleteOrderFlow(){
        System.out.print("Enter order ID to delete: ");
        int orderID = Integer.parseInt(scanner.nextLine());

        Order orderToDelete = null;
    for (Order o : orderManager.getOrdersList()) {
        if (o.getOrderID() == orderID) {
            orderToDelete = o;
            break;
        }
    }

    if (orderToDelete == null) {
        System.out.println("Error: Order not found.");
        return;
    }
    orderManager.getOrdersList().remove(orderToDelete);
}


    


    private void updateOrderStatus() {
        int id = readInt("Enter Order ID to update: ");
        orderManager.updateStatus(id, scanner);
    }



    // --------------------------------------------------------------
    //                 CUSTOMER MENU
    // --------------------------------------------------------------

    private void manageCustomers() {
        System.out.println("\n--- Customer Menu ---");
        System.out.println("1. Add Customer");
        System.out.println("2. View Customers");
        System.out.println("3. Edit Customers");
        System.out.println("4. Delete Customers");
        System.out.println("5. Return to Main Menu");

        int choice = readInt("Enter choice: ");

        switch (choice) {
            case 1 -> addCustomerFlow();
            case 2 -> customerManager.viewCustomers();
            case 3 -> editCustomerFlow();
            case 4 -> deleteCustomerFlow();
            default -> System.out.println("Invalid choice.");
        }
    }

    private void addCustomerFlow() {
        System.out.println("\n--- Add Customer ---");

        String id;
        while (true) { // loop until a unique ID is entered
        System.out.print("Enter Customer ID: ");
        id = scanner.nextLine().trim();

        if (customerManager.findCustomer(id) != null) {
            System.out.println("Customer ID already exists. Please enter a unique ID.");
        } else {
            break; // unique ID, exit loop
        }
    }

        //System.out.print("Enter Customer ID: ");
        //String id = scanner.nextLine().trim();

        System.out.print("Enter name: ");
        String name = scanner.nextLine();

        System.out.print("Enter phone number: ");
        String phone = scanner.nextLine();

        System.out.print("Enter email: ");
        String email = scanner.nextLine();

        System.out.print("Enter address: ");
        String address = scanner.nextLine();

        Customer c = new Customer(id, name, phone, email, address);
        customerManager.addCustomer(c);

    }


    private void editCustomerFlow() {
    System.out.print("Enter Customer ID to edit: ");
    String id = scanner.nextLine();

    Customer c = customerManager.findCustomer(id);
    if (c == null) {
        System.out.println("Customer not found.");
        return;
    }

    System.out.print("Enter new name: ");
    String name = scanner.nextLine();
    System.out.print("Enter new phone: ");
    String phone = scanner.nextLine();
    System.out.print("Enter new email: ");
    String email = scanner.nextLine();
    System.out.print("Enter new address: ");
    String address = scanner.nextLine();

    customerManager.editCustomer(id, name, phone, email, address);
}


private void deleteCustomerFlow() {
    System.out.print("Enter Customer ID to delete: ");
    String id = scanner.nextLine().trim();

    Customer c = customerManager.findCustomer(id);
    if (c == null){
        System.out.print("Customer ID not found");
        return;
    }
    customerManager.deleteCustomer(id);
    
}



    // --------------------------------------------------------------
    //                 PAYMENT MENU
    // --------------------------------------------------------------

    private void managePayments() {
        System.out.println("\n--- Payment Menu ---");
        System.out.println("1. Record Payment");
        System.out.println("2. View Payments");
        System.out.println("3. Update Payment Status");
        System.out.println("4. Return to Main Menu");

        int choice = readInt("Enter choice: ");

        switch (choice) {
            case 1 -> recordPaymentFlow();
            case 2 -> paymentManager.viewPayments();
            case 3 -> updatePaymentStatus();
            case 4 -> {}
            default -> System.out.println("Invalid choice.");
        }
    }

    private void recordPaymentFlow() {
        System.out.println("\n--- Record Payment ---");

        int id = readInt("Enter Order ID: ");

        System.out.print("Enter payment method: ");
        String method = scanner.nextLine();

        double amount = readDouble("Enter amount: ");

        Payment p = new Payment(paymentManager.getNextPaymentID(), id, method, amount);
        paymentManager.addPayment(p);

        System.out.println("Payment recorded!");
    }

    private void updatePaymentStatus() {
        int id = readInt("Payment ID to update: ");
        paymentManager.updateStatus(id, scanner);
    }

    // --------------------------------------------------------------
    //                  ALERTS + SAVE
    // --------------------------------------------------------------

    private void showAlerts() {
        System.out.println("\n--- Alerts ---");
        orderManager.showUpcomingAlerts();
    }

    private void saveAll() {
        System.out.println("\nSaving data...");
        datastore.saveOrders(orderManager);
        datastore.saveCustomers(customerManager);
        datastore.savePayments(paymentManager);
        System.out.println("All data saved successfully.");
    }

    // --------------------------------------------------------------
    //                  INPUT HELPERS
    // --------------------------------------------------------------

    private int readInt(String msg) {
        while (true) {
            try {
                System.out.print(msg);
                String s = scanner.nextLine();
                return Integer.parseInt(s.trim());
            } catch (NumberFormatException e) {
                System.out.println("Please enter a valid number.");
            }
        }
    }

    private double readDouble(String msg) {
        while (true) {
            try {
                System.out.print(msg);
                String s = scanner.nextLine();
                return Double.parseDouble(s.trim());
            } catch (Exception e) {
                System.out.println("Enter a valid number.");
            }
        }
    }
}