coriolis-web/server/api/LogosApi.ts

/*
Copyright (C) 2019  Cloudbase Solutions SRL
This program is free software: you can redistribute it and/or modify
it under the terms of the GNU Affero General Public License as
published by the Free Software Foundation, either version 3 of the
License, or (at your option) any later version.
This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
GNU Affero General Public License for more details.
You should have received a copy of the GNU Affero General Public License
along with this program.  If not, see <http://www.gnu.org/licenses/>.
*/

import express from "express";
import path from "path";
import fs from "fs";

const getModJsonProviders = (jsonPath: string) => {
  const jsonContent: any = fs.readFileSync(jsonPath);
  const json = JSON.parse(jsonContent);
  if (!json.providers) {
    throw new Error();
  }
  return json.providers;
};

const getOptimalLogoHeightKey = (
  availableHeightKeys: string[],
  requestedHeight: number,
  style?: string | null
): string => {
  let heightKeys = availableHeightKeys;
  if (style) {
    const styledKeys = heightKeys.filter((k) =>
      style ? k.indexOf(style) > -1 : false
    );
    if (styledKeys.length) {
      heightKeys = styledKeys;
    }
  }

  const optimal = heightKeys.reduce((prev, curr) => {
    let prevHeight: any = /d+/.exec(prev);
    let currHeight: any = /d+/.exec(curr);
    prevHeight = prevHeight ? Number(prevHeight[0]) : 0;
    currHeight = currHeight ? Number(currHeight[0]) : 0;
    return Math.abs(currHeight - requestedHeight) <
      Math.abs(prevHeight - requestedHeight)
      ? curr
      : prev;
  });
  return optimal;
};

export default (router: express.Router) => {
  router.get("/logos/:provider/:size/:style?", (req, res) => {
    const SIZES = [32, 42, 64, 128];
    const STYLES = ["white", "disabled"];
    const { provider, style } = req.params;
    const size = Number(req.params.size);

    if (SIZES.indexOf(size) === -1) {
      res
        .status(400)
        .json({ error: { message: `Valid sizes are: ${SIZES.join(", ")}` } });
      return;
    }
    if (style && STYLES.indexOf(style) === -1) {
      res
        .status(400)
        .json({ error: { message: `Valid styles are: ${STYLES.join(", ")}` } });
      return;
    }
    const logoBase = path.join(__dirname, "/resources/providerLogos");
    let logoPath = `${logoBase}/${provider}-${size}`;
    logoPath = style ? `${logoPath}-${style}.svg` : `${logoPath}.svg`;

    const modJsonPath: string | null | undefined = process.env.MOD_JSON;
    if (!modJsonPath) {
      res.sendFile(logoPath);
      return;
    }

    try {
      const providersJson = getModJsonProviders(modJsonPath);
      const providerJson = providersJson[provider];
      if (!providerJson) {
        res.sendFile(logoPath);
        return;
      }
      const providerLogosJson = providerJson.logos;
      if (!providerLogosJson) {
        console.log(
          `No logos specified in MOD_JSON file for '${provider}' provider`
        );
        res.sendFile(logoPath);
        return;
      }
      const providerLogosKeys = Object.keys(providerLogosJson);
      if (!providerLogosKeys.length) {
        console.log(
          `No logo heights specified in MOD_JSON file for '${provider}' provider`
        );
        res.sendFile(logoPath);
        return;
      }
      const optimalHeightKey = getOptimalLogoHeightKey(
        providerLogosKeys,
        size,
        style
      );
      const modLogoPath = providerLogosJson[optimalHeightKey].path;
      if (!modLogoPath) {
        console.log(
          `No logo path specified in MOD_JSON file for '${provider}' provider`
        );
        res.sendFile(logoPath);
        return;
      }
      res.sendFile(modLogoPath);
    } catch (err) {
      console.error(err);
      res.status(400).json({ error: { message: "Invalid Mod JSON file" } });
    }
  });
};