book_flight.php

<?php
session_start();
header('Content-Type: application/json');
require_once 'db_config.php';

// Check if user is logged in
if (!isset($_SESSION['logged_in']) || $_SESSION['logged_in'] !== true) {
    echo json_encode(['success' => false, 'message' => 'Please login to book flights.']);
    exit;
}

if ($_SERVER['REQUEST_METHOD'] === 'POST') {
    $bookingData = json_decode(file_get_contents('php://input'), true);
    
    if (!$bookingData) {
        echo json_encode(['success' => false, 'message' => 'Invalid booking data.']);
        exit;
    }
    
    $flightId = $bookingData['flightId'] ?? '';
    $passengers = $bookingData['passengers'] ?? [];
    $isRoundTrip = isset($bookingData['returnFlightId']);
    $returnFlightId = $bookingData['returnFlightId'] ?? '';
    
    if (empty($flightId) || empty($passengers)) {
        echo json_encode(['success' => false, 'message' => 'Missing required booking information.']);
        exit;
    }
    
    try {
        $conn->beginTransaction();
        
        // Get flight information
        $stmt = $conn->prepare("SELECT * FROM flights WHERE flight_id = ?");
        $stmt->execute([$flightId]);
        $flight = $stmt->fetch(PDO::FETCH_ASSOC);
        
        if (!$flight) {
            throw new Exception("Flight not found.");
        }
        
        // Check available seats
        $seatsNeeded = count($passengers);
        if ($flight['available_seats'] < $seatsNeeded) {
            throw new Exception("Not enough available seats.");
        }
        
        $basePrice = (float)$flight['price'];
        $totalPrice = 0;
        $bookingResults = [];
        
        // Create one flight booking for all passengers
        $flightBookingId = 'FB-' . time() . '-' . rand(1000, 9999);
        
        // Process each passenger
        foreach ($passengers as $passenger) {
            $ssn = trim($passenger['ssn'] ?? '');
            $firstName = trim($passenger['firstName'] ?? '');
            $lastName = trim($passenger['lastName'] ?? '');
            $dob = trim($passenger['dob'] ?? '');
            $category = trim($passenger['category'] ?? 'adults');
            
            if (empty($ssn) || empty($firstName) || empty($lastName) || empty($dob)) {
                throw new Exception("Missing passenger information.");
            }
            
            // Calculate ticket price based on category
            $ticketPrice = $basePrice;
            if ($category === 'children') {
                $ticketPrice = $basePrice * 0.7;
            } elseif ($category === 'infants') {
                $ticketPrice = $basePrice * 0.1;
            }
            
            $totalPrice += $ticketPrice;
            
            // Insert or update passenger
            $stmt = $conn->prepare("
                INSERT INTO passenger (ssn, first_name, last_name, date_of_birth, category)
                VALUES (?, ?, ?, ?, ?)
                ON DUPLICATE KEY UPDATE
                    first_name = VALUES(first_name),
                    last_name = VALUES(last_name),
                    date_of_birth = VALUES(date_of_birth),
                    category = VALUES(category)
            ");
            $stmt->execute([$ssn, $firstName, $lastName, $dob, $category]);
            
            // Create ticket
            $ticketId = 'T-' . time() . '-' . rand(1000, 9999) . '-' . rand(100, 999);
            $stmt = $conn->prepare("
                INSERT INTO ticket (ticket_id, flight_booking_id, ssn, price)
                VALUES (?, ?, ?, ?)
            ");
            $stmt->execute([$ticketId, $flightBookingId, $ssn, $ticketPrice]);
            
            $bookingResults[] = [
                'flightBookingId' => $flightBookingId,
                'ticketId' => $ticketId,
                'ssn' => $ssn,
                'firstName' => $firstName,
                'lastName' => $lastName,
                'dob' => $dob,
                'category' => $category,
                'price' => $ticketPrice
            ];
        }
        
        // Create flight booking with total price
        $stmt = $conn->prepare("
            INSERT INTO flight_booking (flight_booking_id, flight_id, total_price)
            VALUES (?, ?, ?)
        ");
        $stmt->execute([$flightBookingId, $flightId, $totalPrice]);
        
        // Update available seats (one per passenger)
        $stmt = $conn->prepare("
            UPDATE flights SET available_seats = available_seats - ? WHERE flight_id = ?
        ");
        $stmt->execute([count($passengers), $flightId]);
        
        // Handle return flight if round trip
        $returnBookingResults = [];
        if ($isRoundTrip && !empty($returnFlightId)) {
            $stmt = $conn->prepare("SELECT * FROM flights WHERE flight_id = ?");
            $stmt->execute([$returnFlightId]);
            $returnFlight = $stmt->fetch(PDO::FETCH_ASSOC);
            
            if (!$returnFlight) {
                throw new Exception("Return flight not found.");
            }
            
            if ($returnFlight['available_seats'] < $seatsNeeded) {
                throw new Exception("Not enough available seats for return flight.");
            }
            
            $returnBasePrice = (float)$returnFlight['price'];
            $returnTotalPrice = 0;
            
            // Create one return flight booking for all passengers
            $returnFlightBookingId = 'FB-' . (time() + 1) . '-' . rand(1000, 9999);
            
            foreach ($passengers as $passenger) {
                $ssn = trim($passenger['ssn'] ?? '');
                $category = trim($passenger['category'] ?? 'adults');
                
                $ticketPrice = $returnBasePrice;
                if ($category === 'children') {
                    $ticketPrice = $returnBasePrice * 0.7;
                } elseif ($category === 'infants') {
                    $ticketPrice = $returnBasePrice * 0.1;
                }
                
                $returnTotalPrice += $ticketPrice;
                
                // Create return ticket
                $returnTicketId = 'T-' . (time() + 1) . '-' . rand(1000, 9999) . '-' . rand(100, 999);
                $stmt = $conn->prepare("
                    INSERT INTO ticket (ticket_id, flight_booking_id, ssn, price)
                    VALUES (?, ?, ?, ?)
                ");
                $stmt->execute([$returnTicketId, $returnFlightBookingId, $ssn, $ticketPrice]);
                
                $returnBookingResults[] = [
                    'flightBookingId' => $returnFlightBookingId,
                    'ticketId' => $returnTicketId,
                    'ssn' => $ssn,
                    'firstName' => $passenger['firstName'],
                    'lastName' => $passenger['lastName'],
                    'dob' => $passenger['dob'],
                    'category' => $category,
                    'price' => $ticketPrice
                ];
            }
            
            // Create return flight booking with total price
            $stmt = $conn->prepare("
                INSERT INTO flight_booking (flight_booking_id, flight_id, total_price)
                VALUES (?, ?, ?)
            ");
            $stmt->execute([$returnFlightBookingId, $returnFlightId, $returnTotalPrice]);
            
            // Update return flight seats (one per passenger)
            $stmt = $conn->prepare("
                UPDATE flights SET available_seats = available_seats - ? WHERE flight_id = ?
            ");
            $stmt->execute([count($passengers), $returnFlightId]);
        }
        
        $conn->commit();
        
        echo json_encode([
            'success' => true,
            'message' => 'Flight(s) booked successfully!',
            'outbound' => [
                'flight' => $flight,
                'bookingId' => $bookingResults[0]['flightBookingId'] ?? '',
                'totalPrice' => $totalPrice,
                'tickets' => $bookingResults
            ],
            'return' => $isRoundTrip && !empty($returnBookingResults) ? [
                'flight' => $returnFlight ?? null,
                'bookingId' => $returnBookingResults[0]['flightBookingId'] ?? '',
                'totalPrice' => $returnTotalPrice ?? 0,
                'tickets' => $returnBookingResults
            ] : null
        ]);
        
    } catch (Exception $e) {
        $conn->rollBack();
        echo json_encode(['success' => false, 'message' => $e->getMessage()]);
    }
} else {
    echo json_encode(['success' => false, 'message' => 'Invalid request method.']);
}
?>