Refactor routes into separate file

1 files changed, 3 insertions, 129 deletions

diff --git a/src/index.ts b/src/index.ts
index 53592da..7bf2354 100644
--- a/src/index.ts
+++ b/src/index.ts
@@ -1,7 +1,5 @@
 import dotenv from "dotenv";
 import express from "express";
-import { JSDOM } from "jsdom";
-import { badgen } from "badgen";
 import { MongoClient } from "mongodb";
 import path from "path";
 import fs from "fs";
@@ -11,7 +9,7 @@ import expressWinston from "express-winston";
 
 dotenv.config();
 
-import formats, { Format } from "./formats";
+import routes from "./routes";
 import Metadata from "./metadata";
 import logger_config from "./util/logger";
 import { config_or_error, handle_shutdown } from "./util/config";
@@ -19,8 +17,6 @@ import { config_or_error, handle_shutdown } from "./util/config";
 // Start-up configuration
 const BIND_ADDRESS = process.env.BIND_ADDRESS || "localhost";
 const PORT = Number(process.env.PORT || 3000);
-const TOKEN = process.env.TOKEN || "";
-const UPLOAD_LIMIT = Number(process.env.UPLOAD_LIMIT || 4194304);
 
 const logger = winston.createLogger(logger_config("ROOT"));
 
@@ -53,130 +49,8 @@ new MongoClient(MONGO_URI, { useUnifiedTopology: true }).connect(
       })
     );
 
-    // Upload HTML file
-    app.post("/v1/:org/:repo/:branch/:commit.html", (req, res) => {
-      const { org, repo, branch, commit } = req.params;
-
-      const { token, format } = req.query;
-      //TODO @Metadata token should come from metadata
-      if (token != TOKEN) {
-        return res.status(401).send("Invalid token");
-      }
-
-      if (!formats.list_formats().includes(format)) {
-        return res.status(406).send("Report format unknown");
-      }
-
-      var contents = "";
-      req.on("data", raw => {
-        if (contents.length + raw.length > UPLOAD_LIMIT) {
-          res.status(413).send("Uploaded file is too large");
-        } else {
-          contents += raw;
-        }
-      });
-      req.on("end", () => {
-        let formatter: Format, coverage: number;
-        try {
-          const doc = new JSDOM(contents).window.document;
-          formatter = formats.get_format(format);
-          coverage = formatter.parse_coverage(doc);
-        } catch {
-          return res.status(400).send("Invalid report document");
-        }
-
-        const badge = badgen({
-          label: "coverage",
-          status: Math.floor(coverage).toString() + "%",
-          //TODO @Metadata stage values should come from metadata
-          color: formatter.match_color(coverage, 95, 80)
-        });
-
-        const report_path = path.join(HOST_DIR, org, repo, branch, commit);
-
-        fs.promises
-          .mkdir(report_path, { recursive: true })
-          .then(() =>
-            fs.promises.writeFile(path.join(report_path, "badge.svg"), badge)
-          )
-          .then(() =>
-            fs.promises.writeFile(
-              path.join(report_path, "index.html"),
-              contents
-            )
-          )
-          .then(() =>
-            metadata.updateBranch({ org, repo, name: branch, head: commit })
-          )
-          .then(result =>
-            result
-              ? res.status(200).send()
-              : res.status(500).send("Unknown error occurred")
-          );
-      });
-    });
-
-    app.get("/v1/:org/:repo/:branch.svg", (req, res) => {
-      const { org, repo, branch } = req.params;
-
-      metadata.getHeadCommit(org, repo, branch).then(
-        commit => {
-          logger.debug(
-            "Found commit %s for ORB %s/%s/%s",
-            commit,
-            org,
-            repo,
-            branch
-          );
-
-          res.sendFile(
-            path.join(HOST_DIR, org, repo, branch, commit, "badge.svg")
-          );
-        },
-        () => {
-          res.status(500).send("Unknown error occurred");
-        }
-      );
-    });
-
-    app.get("/v1/:org/:repo/:branch.html", (req, res) => {
-      const { org, repo, branch } = req.params;
-
-      metadata.getHeadCommit(org, repo, branch).then(
-        commit => {
-          logger.debug(
-            "Found commit %s for ORB %s/%s/%s",
-            commit,
-            org,
-            repo,
-            branch
-          );
-
-          res.sendFile(
-            path.join(HOST_DIR, org, repo, branch, commit, "index.html")
-          );
-        },
-        () => {
-          res.status(500).send("Unknown error occurred");
-        }
-      );
-    });
-
-    // provide hard link for commit
-    app.get("/v1/:org/:repo/:branch/:commit.svg", (req, res) => {
-      const { org, repo, branch, commit } = req.params;
-
-      res.sendFile(path.join(HOST_DIR, org, repo, branch, commit, "badge.svg"));
-    });
-
-    // provide hard link for commit
-    app.get("/v1/:org/:repo/:branch/:commit.html", (req, res) => {
-      const { org, repo, branch, commit } = req.params;
-
-      res.sendFile(
-        path.join(HOST_DIR, org, repo, branch, commit, "index.html")
-      );
-    });
+    // actual app routes
+    app.use(routes(metadata));
 
     app.use(expressWinston.errorLogger(logger_config("_ERR")));