1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
|
import dotenv from "dotenv";
import express from "express";
import { MongoClient } from "mongodb";
import path from "path";
import fs from "fs";
import winston from "winston";
import expressWinston from "express-winston";
dotenv.config();
import processTemplate, { Template } from "./templates";
import routes from "./routes";
import Metadata from "./metadata";
import loggerConfig from "./util/logger";
import { configOrError, handleShutdown } from "./util/config";
// Start-up configuration
const BIND_ADDRESS = process.env.BIND_ADDRESS ?? "localhost";
const PORT = Number(process.env.PORT ?? 3000);
const TARGET_URL = process.env.TARGET_URL ?? "http://localhost:3000";
const logger = winston.createLogger(loggerConfig("ROOT"));
const MONGO_URI = configOrError("MONGO_URI");
const MONGO_DB = process.env.MONGO_DB ?? "ao-coverage";
const HOST_DIR = configOrError("HOST_DIR");
fs.accessSync(HOST_DIR, fs.constants.R_OK | fs.constants.W_OK);
if (!path.isAbsolute(HOST_DIR)) {
logger.error("HOST_DIR must be an absolute path");
process.exit(1);
}
// prepare template files
const bashTemplate = {
inputFile: path.join(__dirname, "..", "public", "bash.template"),
outputFile: path.join(HOST_DIR, "bash"),
context: { TARGET_URL }
} as Template;
processTemplate(bashTemplate)
.then(template => {
logger.debug("Generated '%s' from template file", template.outputFile);
})
.catch(err => {
logger.error("Unable to process template file: %s", err);
// if the output file exists, then we are fine with continuing without
return fs.promises.access(bashTemplate.outputFile, fs.constants.R_OK);
})
.catch(err => {
logger.error("Cannot proceed: %s", err);
process.exit(1);
});
new MongoClient(MONGO_URI, { useUnifiedTopology: true }).connect(
(err, mongo) => {
if (err !== null) {
logger.error(err);
process.exit(1);
}
const app: express.Application = express();
const metadata = new Metadata(mongo.db(MONGO_DB));
app.use(
expressWinston.logger({
...loggerConfig("HTTP"),
colorize: true,
// filter out token query param from URL
msg:
'{{req.method}} {{req.url.replace(/token=[-\\w.~]*(&*)/, "token=$1")}} - {{res.statusCode}} {{res.responseTime}}ms'
})
);
// actual app routes
app.use(routes(metadata));
app.use(expressWinston.errorLogger(loggerConfig("_ERR")));
const server = app.listen(PORT, BIND_ADDRESS, () => {
logger.info("Express has started: http://%s:%d/", BIND_ADDRESS, PORT);
});
// application exit handling
const signalCodes: NodeJS.Signals[] = ["SIGTERM", "SIGHUP", "SIGINT"];
signalCodes.map((code: NodeJS.Signals) => {
process.on(code, handleShutdown(mongo, server));
});
}
);
|
