index.ts

import { LDBWSClient } from "ldbws-client";
import dotenv from "dotenv";
import chalk from "chalk";
import ora from "ora";

dotenv.config();

// Log in to NR api
const api = new LDBWSClient(process.env.TOKEN!!)

// Get the station code args
const crs = process.argv[2].toUpperCase();

// Get the number of services
if (process.argv.length == 4) {
    var num = process.argv[3]
} else num = 10

// Clear the terminal before printing
print("\x1Bc");
// Add loading message

const spinner = ora("Loading departure board for " + crs).start();

// Update the data every 30 seconds
async function update() {
    // Move the cursor to remove the new line
    setTimeout(async function () {
        // Get the departures from NR
        var resp = await api.GetDepartureBoard(crs, num);
        // Move the cursor to write over the previous table
        print("\x1B[5;0H");
        // Print the data
        printBody(resp);
        // Live updates
        update();
    }, 30000);
}

// Get the initial data & then begin updates
const get = async () => {
    // Get the departures from NR
    try {
        var resp = await api.GetDepartureBoard(crs, num);
    } catch(err) {
        // If an error occured
        print(chalk.red("\nError: " + err.message));
        return
    }
    // Stop spinning & clear the terminal 
    // to remove loading message
    spinner.stop();
    print("\x1Bc");
    // Print the title block
    title("BHM");
    // Print the table headings
    print(chalk.bold(formatString("Time", 5)));
    print(chalk.bold(formatString("Destination", 25)));
    print(chalk.bold(formatString("Plat", 4)));
    print(chalk.bold(formatString("Expected", 11)));
    print(chalk.bold(formatString("Operator", 30)) + "\n");
    // Print the data
    printBody(resp);
    // Live updates
    update();
}

get()

// Print the main body of data from the response
function printBody(resp) {
    resp.trainServices.service.forEach(departure => {
        // If there isn't a platform, set it to "-"
        if (departure.platform != null) {
            var platform = departure.platform
        } else platform = "-"
        // Calculate formatting for the expected time
        var expected
        if (departure.etd == "On time") expected = chalk.green(formatString("✓ " + departure.etd, 11))
        else if (departure.etd == "Cancelled") expected = chalk.red(formatString("✗ " + departure.etd, 11))
        else expected = chalk.yellow(formatString("! " + departure.etd, 11))
        // Print the row out
        print(chalk.gray(formatString(departure.std, 5)))
        print(chalk.white(formatString(departure.destination.location[0].locationName, 25)))
        print(chalk.grey(formatString(platform, 4)))
        print(expected)
        print(chalk.grey(formatString(departure.operator, 30)))
        print("\n") // New line for next row
    });
}

// Prin the title block
function title() {
    print(chalk.redBright("LIVE") + chalk.white(" Departures from " + crs + "\n")) // Station code
    print(chalk.gray("Powered by National Rail Enquiries\n")) // NR Attribution
    console.log() // New line for spacing
}

// Wrapper function for printing without new line
function print(string) { process.stdout.write(string); }

// Format a string by adding spaces to the end
function formatString(string, maxLength) {
    var spaces = maxLength - string.length;
    var result = string;
    for (var i = 0; i < spaces + 2; i++) {
        result += " "
    }
    return result
}