random-flight-generator/lib/FlightGenerator.js

A tool for generating random flights.

"use strict";
var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
    if (k2 === undefined) k2 = k;
    Object.defineProperty(o, k2, { enumerable: true, get: function() { return m[k]; } });
}) : (function(o, m, k, k2) {
    if (k2 === undefined) k2 = k;
    o[k2] = m[k];
}));
var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
    Object.defineProperty(o, "default", { enumerable: true, value: v });
}) : function(o, v) {
    o["default"] = v;
});
var __importStar = (this && this.__importStar) || function (mod) {
    if (mod && mod.__esModule) return mod;
    var result = {};
    if (mod != null) for (var k in mod) if (k !== "default" && Object.prototype.hasOwnProperty.call(mod, k)) __createBinding(result, mod, k);
    __setModuleDefault(result, mod);
    return result;
};
var __importDefault = (this && this.__importDefault) || function (mod) {
    return (mod && mod.__esModule) ? mod : { "default": mod };
};
Object.defineProperty(exports, "__esModule", { value: true });
exports.FlightGenerator = void 0;
var Airport_1 = require("./Airport");
var AirportUtils_1 = require("./AirportUtils");
var AirportsJson = __importStar(require("./data/airports.json"));
var Flight_1 = require("./Flight");
var lodash_1 = __importDefault(require("lodash"));
var FlightGenerator = /** @class */ (function () {
    function FlightGenerator(options) {
        this.tries = 0;
        this.maxTries = 10;
        this.airports = [];
        this.options = options || {};
        this.setupAirports();
    }
    FlightGenerator.prototype.setupAirports = function () {
        var _this = this;
        this.airports = lodash_1.default.map(AirportsJson, function (airport) { return new Airport_1.Airport(airport); }).filter(function (airport) {
            return airport.satisfiesFlightGeneratorOptions(_this.options);
        });
    };
    FlightGenerator.prototype.generateFlight = function () {
        var departure;
        var arrival;
        if (this.options.departure && this.options.arrival) {
            // Both airports specified
            departure = AirportUtils_1.AirportUtils.getAirport(this.options.departure);
            arrival = AirportUtils_1.AirportUtils.getAirport(this.options.arrival);
        }
        else if (this.options.departure) {
            // Departure specified
            departure = AirportUtils_1.AirportUtils.getAirport(this.options.departure);
            arrival = this.findMatchingAirport(departure);
        }
        else if (this.options.arrival) {
            // Arrival specified
            arrival = AirportUtils_1.AirportUtils.getAirport(this.options.arrival);
            departure = this.findMatchingAirport(arrival);
        }
        else {
            // None specified
            departure = AirportUtils_1.AirportUtils.randomAirport(this);
            arrival = this.findMatchingAirport(departure);
        }
        if (departure === arrival) {
            // If departure and arrival are the same, no valid flights were found
            this.tries++;
            if (this.tries < this.maxTries) {
                return this.generateFlight();
            }
        }
        return new Flight_1.Flight(departure, arrival);
    };
    /**
     * Finds an airport that satisfies the FlightGenerator options.
     * Returns the same airport if no match is found.
     */
    FlightGenerator.prototype.findMatchingAirport = function (inputAirport) {
        if (!inputAirport) {
            inputAirport = AirportUtils_1.AirportUtils.randomAirport(this);
        }
        var possibleAirports = AirportUtils_1.AirportUtils.getAllPossibleAirports(inputAirport, this);
        if (possibleAirports.length > 0) {
            var outputAirport = possibleAirports[Math.floor(Math.random() * possibleAirports.length)];
            outputAirport.randomlyGenerated = true;
            return outputAirport;
        }
        else {
            return inputAirport;
        }
    };
    return FlightGenerator;
}());
exports.FlightGenerator = FlightGenerator;