FlightDataLayer.jsx

import { useState } from 'react';
import { useError } from '../components/errorDialog/ErrorProvider.jsx';

//Defines the initial flight object.
const initialFlight = {
    airlineID: 0,
    codeShares: [],
    description: '',
    departTime: new Date(new Date().setHours(0, 0, 0, 0)),
    destination: '',
    flightNumber: '',
    gateID: 0,
    gateName: '',
    name: '',
    sortDestinationID: 0,
    sortDestinationName: '',
};
export default initialFlight;

//The function returns a hook to interact with the server for the flights.
export function useFlightDataLayer() {
    const { showError } = useError();
    const [flights, setFlights] = useState([]);
    const [addFlightSuccess, setAddFlightSuccess] = useState(false);
    const [addFlightValidationProblemDetails, setAddFlightValidationProblemDetails] = useState(null);
    const [deleteFlightSuccess, setDeleteFlightSuccess] = useState(false);
    const [updateFlightSuccess, setUpdateFlightSuccess] = useState(false);
    const [updateFlightValidationProblemDetails, setUpdateFlightValidationProblemDetails] = useState(null);

    //Constants for the status codes returned by the server.
    const BadRequestCode = 400;
    const NotFoundCode = 404;
    const ConflictCode = 409;
    const InternalServerError = 500;

    //The function adds a flight to the server.
    //@param {object} flight The flight to add.
    const addFlight = (flight) => {
        clearStates();

        let prepFlight = prepFlightObject(flight);

        fetch('api/Flight', {
            method: 'POST',
            body: JSON.stringify(prepFlight),
            headers: {
                "Content-Type": "application/json",
            },
        })
            .then(response => {
                if (response.ok) {
                    setAddFlightSuccess(true);
                }
                else if (response.status === BadRequestCode) {
                    response.json().then(validationProblemDetails => setAddFlightValidationProblemDetails(validationProblemDetails));
                }
                else if (response.status === InternalServerError) {
                    response.json().then(problemDetails => showError(problemDetails.detail));
                }
                else {
                    showError('Failed to create the flight because of an error on the server.');
                }
            })
            .catch(error => showError('Failed to communicate with the server.'));
    };

    //The function clears the states before an operation.
    const clearStates = () => {
        setAddFlightSuccess(false);
        setAddFlightValidationProblemDetails(null);
        setDeleteFlightSuccess(false);
        setUpdateFlightSuccess(false);
        setUpdateFlightValidationProblemDetails(null);
    };

    //The function deletes a flight from the server.
    //@param {object} flight The flight to delete.
    const deleteFlight = (flight) => {
        clearStates();

        fetch('/api/Flight/' + flight.integer64ID, {
            method: 'DELETE'
        })
            .then(response => {
                if (response.ok) {
                    setDeleteFlightSuccess(true);
                }
                else if (response.status === NotFoundCode || response.status === InternalServerError) {
                    response.json().then(problemDetails => showError(problemDetails.detail));
                }
                else {
                    showError('Failed to delete the flight because of an error on the server.');
                }
            })
            .catch(error => showError('Failed to communicate with the server.'));
    };

    //The function retrieves the flights from the server.
    const getFlights = () => {
        fetch('api/Flight/All')
            .then(response => response.json())
            .then(json => setFlights(json))
            .catch(error => showError('Failed to communicate with the server.'));
    };

    //The function prepares the flight object before being sent to the server.
    const prepFlightObject = (flight) => {
        const options = { hour12: false, hour: '2-digit', minute: '2-digit', second: '2-digit' };
        return {
            ...flight,
            //Only send the time to the server because the property on the server side is a C# TimeSpan.
            departTime: flight.departTime.toLocaleTimeString('en-US', options)
        };
    };

    //The function updates a flight to the server.
    //@param {object} flight The flight to update.
    const updateFlight = (flight) => {
        clearStates();

        let prepFlight = prepFlightObject(flight);

        fetch('api/Flight', {
            method: 'PUT',
            body: JSON.stringify(prepFlight),
            headers: {
                "Content-Type": "application/json",
            },
        })
            .then(response => {
                if (response.ok) {
                    setUpdateFlightSuccess(true);
                }
                else if (response.status === BadRequestCode) {
                    response.json().then(validationProblemDetails => setUpdateFlightValidationProblemDetails(validationProblemDetails));
                }
                else if (response.status === NotFoundCode || response.status == ConflictCode || response.status === InternalServerError) {
                    response.json().then(problemDetails => showError(problemDetails.detail));
                }
                else {
                    showError('Failed to update the flight because of an error on the server.');
                }
            })
            .catch(error => showError('Failed to communicate with the server.'));
    };

    return {
        flights,
        getFlights,
        addFlight,
        addFlightValidationProblemDetails,
        addFlightSuccess,
        deleteFlight,
        deleteFlightSuccess,
        updateFlight,
        updateFlightValidationProblemDetails,
        updateFlightSuccess
    };
};