1
import { exec, execSync } from "node:child_process";
2
import readline from "node:readline";
3
import os from "node:os";
4

5
/** This is the OS being used as a string. */
6
const platform = os.platform();
7

8
/** This array contains the arguments that can be passed in for this script. */
9
const args = process.argv.slice(2);
10

11
var dockerBuilt = false;
12
var dockerBuildCommandComplete = false;
13
var dockerScriptClosing = false;
14

15
const PORT_NUM = "5700";
16
const PORT_URL = ("http://localhost:" + PORT_NUM);
17

18
const DOCKER_IMAGE = "mohit_website";
19
const DOCKER_CONTAINER = "mohit_website_main_docker_container";
20
const DOCKER_START_ENV_ARGS = (`--env PORT=${PORT_NUM} --env HOST=0.0.0.0 --env HOST_PLATFORM=${platform} --env DOCKER_SCRIPT=true`);
21
const DOCKER_START_ARGS = (`--env-file .env ${DOCKER_START_ENV_ARGS} -p ${PORT_NUM}:${PORT_NUM} -v "${process.cwd()}:/app" -v /app/node_modules`);
22

23
const DOCKER_BUILD_COMMAND = ("docker build -t " + DOCKER_IMAGE + " .");
24
const DOCKER_START_COMMAND = ("docker run " + DOCKER_START_ARGS + " --name " + DOCKER_CONTAINER + " " + DOCKER_IMAGE);
25
const DOCKER_STOP_COMMAND = ("docker stop " + DOCKER_CONTAINER);
26
const DOCKER_REMOVE_COMMAND = ("docker rm " + DOCKER_CONTAINER);
27

28
readline.emitKeypressEvents(process.stdin);
29
if(process.stdin.isTTY) { process.stdin.setRawMode(true); }
30

31
/**
32
 * This function is ran if an error occurs.
33
 * @param {import("node:child_process").ExecException} error The error itself. 
34
 */
35
function onError(error) {
36
    console.error(error);
37
    process.exit(1);
38
}
39

40
/** This function shuts down the docker instance. */
41
function shutdownDockerInstance() {
42
    if(!dockerBuildCommandComplete || dockerScriptClosing) { return; }
43
    dockerScriptClosing = true;
44

45
    try {
46
        console.log("\n\nGracefully shutting down Docker container...");
47
        execSync(DOCKER_STOP_COMMAND, { stdio: 'inherit' });
48
        execSync(DOCKER_REMOVE_COMMAND, { stdio: 'inherit' });
49

50
        console.log("Docker Container Shut Down!");
51
        process.exit(0);
52
    } catch(err) {
53
        onError(err);
54
    }
55
}
56

57
/** This function runs the main logic for this script. */
58
function main() {
59
    // This figures out if the docker instance needs to be built.
60
    try {
61
        if(-1 != args.findIndex((item) => { return (item === "--force-build"); })) { throw new Error("FORCE_DOCKER_BUILD"); }
62
        dockerBuilt = (execSync(`docker images -q ${DOCKER_IMAGE}`, { encoding: 'utf8' }).trim().length > 0);
63
    } catch(err) {
64
        if(err.message !== "FORCE_DOCKER_BUILD") {
65
            console.error(err);
66
            console.log("\n\n");
67
        }
68
        dockerBuilt = false;
69
    }
70

71
    // This builds the docker instance if necessary.
72
    if(!dockerBuilt) {
73
        try {
74
            execSync(DOCKER_BUILD_COMMAND, { stdio: "inherit" });
75
        } catch(err) {
76
            onError(err);
77
        }
78
    }
79

80
    // This runs the docker instance.
81
    dockerBuildCommandComplete = true;
82
    console.log("Starting Docker Instance...");
83

84
    const openBrowserCommand = (platform === "darwin" ? 'open' : (platform === "win32" ? 'start' : 'xdg-open'));
85
    const startProcess = exec(DOCKER_START_COMMAND);
86
    setTimeout(() => { execSync(openBrowserCommand + " " + PORT_URL); }, 1000);
87

88
    startProcess.stdout.pipe(process.stdout);
89
    startProcess.stderr.pipe(process.stderr);
90
    startProcess.on("error", (err) => { onError(err); });
91

92
    process.stdin.resume();
93
    process.on("SIGINT", () => { shutdownDockerInstance(); });
94
    process.on("SIGTERM", () => { shutdownDockerInstance(); });
95

96
    process.stdin.on("keypress", (chunk = "", key) => {
97
        if((key.name !== "c" || !key.ctrl) && key.name !== "q") { return; }
98
        shutdownDockerInstance();
99
    });
100
}
101

102
// Runs the main function.
103
main();