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
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
|
import winston from "winston";
import { MongoClient, MongoError } from "mongodb";
import { Server } from "http";
import path from "path";
import fs from "fs";
import { v4 as uuid } from "uuid";
import loggerConfig from "./logger";
import processTemplate, { Template } from "../templates";
import { EnvConfig } from "../metadata";
const logger = winston.createLogger(loggerConfig("ROOT"));
export const initializeToken = (): string => {
//TODO check for token in hostDir/persist created token in hostDir so it's not regenerated on startup
const newToken = uuid();
logger.warn(
"TOKEN variable not provided, using this value instead: %s",
newToken
);
logger.warn(
"Use this provided token to push your coverage reports to the server."
);
return newToken;
};
export const configOrError = (varName: string): string => {
const value = process.env[varName];
if (value !== undefined) {
return value;
} else {
logger.error("%s must be defined", varName);
process.exit(1);
}
};
export const persistTemplate = async (input: Template): Promise<void> => {
try {
const template = await processTemplate(input);
logger.debug("Generated '%s' from template file", template.outputFile);
} catch (err1) {
try {
await fs.promises.access(input.outputFile, fs.constants.R_OK);
} catch (err2) {
logger.error(
"Error while generating '%s' from template file: %s",
input.outputFile,
err1
);
logger.error("Cannot proceed due to error: %s", err2);
process.exit(1);
}
// if the output file exists, then we are fine with continuing without
logger.warn(
"Could not generate '%s' from template file, but file already exists: %s",
input.outputFile,
err1
);
}
};
export const handleStartup = async (
mongoUri: string,
config: EnvConfig,
targetUrl: string
): Promise<MongoClient> => {
try {
const { hostDir, publicDir } = config;
await fs.promises.access(hostDir, fs.constants.R_OK | fs.constants.W_OK);
if (!path.isAbsolute(hostDir)) {
await Promise.reject("hostDir must be an absolute path");
}
const mongo = await MongoClient.connect(mongoUri).catch((err: MongoError) =>
Promise.reject(err.message ?? "Unable to connect to database")
);
await persistTemplate({
inputFile: path.join(publicDir, "templates", "sh.tmpl"),
outputFile: path.join(hostDir, "sh"),
context: { TARGET_URL: targetUrl },
} as Template);
await persistTemplate({
inputFile: path.join(publicDir, "templates", "index.html.tmpl"),
outputFile: path.join(hostDir, "index.html"),
context: {
TARGET_URL: targetUrl,
CURL_HTTPS: targetUrl.includes("https")
? "--proto '=https' --tlsv1.2 "
: "",
},
} as Template);
return mongo;
} catch (err) {
logger.error("Error occurred during startup: %s", err);
process.exit(1);
}
};
export const handleShutdown =
(mongo: MongoClient, server: Server) =>
(signal: NodeJS.Signals): Promise<void> => {
logger.warn("%s signal received. Closing shop.", signal);
return mongo
.close()
.then(() => {
logger.info("MongoDB client connection closed.");
return new Promise((res, rej) =>
server.close((err) => {
logger.info("Express down.");
(err ? rej : res)(err);
})
);
})
.then(() => process.exit(0))
.catch(() => process.exit(1));
};
|
