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

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

8
const PORT_NUM = "5700";
9
const PORT_URL = ("http://localhost:" + PORT_NUM);
10

11
const DOCKER_IMAGE = "mohit_website";
12
const DOCKER_CONTAINER = "mohit_website_main_docker_container";
13
const DOCKER_START_ARGS = ("--env-file .env --env PORT=" + PORT_NUM + " --env HOST=0.0.0.0 -p " + PORT_NUM + ":" + PORT_NUM);
14

15
const DOCKER_BUILD_COMMAND = ("docker build -t " + DOCKER_IMAGE + " .");
16
const DOCKER_START_COMMAND = ("docker run " + DOCKER_START_ARGS + " --name " + DOCKER_CONTAINER + " " + DOCKER_IMAGE);
17
const DOCKER_STOP_COMMAND = ("docker stop " + DOCKER_CONTAINER);
18
const DOCKER_REMOVE_COMMAND = ("docker rm " + DOCKER_CONTAINER);
19

20
readline.emitKeypressEvents(process.stdin);
21
if(process.stdin.isTTY) { process.stdin.setRawMode(true); }
22

23
/**
24
 * This function is ran if an error occurs.
25
 * @param {import("node:child_process").ExecException} error The error itself. 
26
 */
27
function onError(error) {
28
    console.error(error);
29
    process.exit(1);
30
}
31

32
/** This function runs the main logic for this script. */
33
function main() {
34
    var dockerBuilt = false;
35
    var dockerBuildCommandComplete = false;
36

37
    // This figures out if the docker instance needs to be built.
38
    try {
39
        if(-1 != args.findIndex((item) => { return (item === "--force-build"); })) { throw new Error("FORCE_DOCKER_BUILD"); }
40
        dockerBuilt = (execSync(`docker images -q ${DOCKER_IMAGE}`, { encoding: 'utf8' }).trim().length > 0);
41
    } catch(err) {
42
        if(err.message !== "FORCE_DOCKER_BUILD") {
43
            console.error(err);
44
            console.log("\n\n");
45
        }
46
        dockerBuilt = false;
47
    }
48

49
    // This builds the docker instance if necessary.
50
    if(!dockerBuilt) {
51
        try {
52
            execSync(DOCKER_BUILD_COMMAND, { stdio: "inherit" });
53
        } catch(err) {
54
            onError(err);
55
        }
56
    }
57

58
    // This runs the docker instance.
59
    dockerBuildCommandComplete = true;
60
    console.log("Starting Docker Instance...");
61

62
    const platform = os.platform();
63
    const openBrowserCommand = (platform === "darwin" ? 'open' : (platform === "win32" ? 'start' : 'xdg-open'));
64

65
    const startProcess = exec(DOCKER_START_COMMAND);
66
    setTimeout(() => { execSync(openBrowserCommand + " " + PORT_URL); }, 1000);
67

68
    startProcess.stdout.pipe(process.stdout);
69
    startProcess.stderr.pipe(process.stderr);
70
    startProcess.on("error", (err) => { onError(err); });
71

72
    process.stdin.resume();
73
    process.on("SIGINT", () => {
74
        if(!dockerBuildCommandComplete) { return; }
75
        try {
76
            console.log("\n\nGracefully shutting down Docker container...");
77
            execSync(DOCKER_STOP_COMMAND, { stdio: 'inherit' });
78
            execSync(DOCKER_REMOVE_COMMAND, { stdio: 'inherit' });
79

80
            console.log("Docker Container Shut Down!");
81
            process.exit(0);
82
        } catch(err) {
83
            onError(err);
84
        }
85
    });
86

87
    process.stdin.on("keypress", (chunk = "", key) => {
88
        if(!dockerBuildCommandComplete) { return; }
89
        if((key.name !== "c" || !key.ctrl) && key.name !== "q") { return; }
90

91
        // Shuts down the docker instance.
92
        try {
93
            console.log("\n\nGracefully shutting down Docker container...");
94
            execSync(DOCKER_STOP_COMMAND, { stdio: 'inherit' });
95
            execSync(DOCKER_REMOVE_COMMAND, { stdio: 'inherit' });
96

97
            console.log("Docker Container Shut Down!");
98
            process.exit(0);
99
        } catch(err) {
100
            onError(err);
101
        }
102
    });
103
}
104

105
// Runs the main function.
106
main();