SmartMirror-Projektarbeit/modules/MMM-PublicTransportHafas/convenience/query_stations.js

"use strict";

const createClient = require("hafas-client");
const readline = require("readline");
const arrayUnique = require("array-unique");

let profileName = "";
let profile = "";

const productMap = {
  bus: "Bus",
  ferry: "Fähre",
  national: "Fernverkehr",
  nationalExpress: "Fernverkehr",
  regional: "Regionalverkehr",
  suburban: "S-Bahn",
  subway: "U-Bahn",
  taxi: "Taxi",
  tram: "Tram"
};

if (process.argv.length == 3) {
  profileName = process.argv[2];
  console.log("Using hafas-client profile: " + profileName);
} else {
  console.log("Using default hafas-client profile: 'db'");
  profileName = "db";
}

try {
  profile = require("hafas-client/p/" + profileName);
}
catch(err) {
  console.error("\n" + err.message + "\n Did you choose the right profile name? \n");
}

if (profile !== "") {
  const client = createClient(profile, 'MMM-PublicTransportHafas');
  const rl = readline.createInterface({
    input: process.stdin,
    output: process.stdout
  });

  rl.question("Geben Sie eine Adresse oder einen Stationsnamen ein: ", (answer) => {
    rl.close();

    const opt = {
      results: 10,
      stations: true,
      adresses: false,
      poi: false
    };

    client.locations(answer, opt).then((response) => {
      console.log("\nGefundene Haltestellen für \"" + answer + "\":\n");

      response.forEach((element) => {
        printStationInfo(element);
      });

      process.exit(0);
    }).catch(console.error);
  });
}


function printStationInfo(element) {
  let id = element.id;
  let name = element.name;
  let products = element.products;

  if (id && name) {
    console.log("> Haltestelle: \"" + name + "\"\n  ID: " + id + "\n  " + refineProducts(products) + "\n");
  }
}


function refineProducts(products) {
  let result = "Verkehrsmittel: ";

  if (!products) {
    return result + "keine";
  }

  let availableProducts = Object.keys(products).filter(key => products[key]);
  let availableProductsReadable = arrayUnique(availableProducts.map(product => productMap[product]));

  return result + availableProductsReadable.join(", ");
}