From 389b00ba154172efcfc27172053a50c546853fc3 Mon Sep 17 00:00:00 2001
From: 1computer1 <mineox100@gmail.com>
Date: Tue, 14 May 2019 19:50:21 -0400
Subject: [PATCH] Keep containers running instead of making new ones

---
 src/commands/exit.js            | 18 ++++++++
 src/listeners/messageInvalid.js |  2 +-
 src/struct/LanguageHandler.js   | 80 ++++++++++++++++++++++++---------
 3 files changed, 77 insertions(+), 23 deletions(-)
 create mode 100644 src/commands/exit.js

diff --git a/src/commands/exit.js b/src/commands/exit.js
new file mode 100644
index 0000000..d198327
--- /dev/null
+++ b/src/commands/exit.js
@@ -0,0 +1,18 @@
+const { Command } = require('discord-akairo');
+
+class ExitCommand extends Command {
+    constructor() {
+        super('exit', {
+            aliases: ['exit'],
+            ownerOnly: true,
+            clientPermissions: ['SEND_MESSAGES']
+        });
+    }
+
+    async exec() {
+        await this.client.languageHandler.cleanup();
+        process.exit();
+    }
+}
+
+module.exports = ExitCommand;
diff --git a/src/listeners/messageInvalid.js b/src/listeners/messageInvalid.js
index 9bb7b00..332c4cd 100644
--- a/src/listeners/messageInvalid.js
+++ b/src/listeners/messageInvalid.js
@@ -27,7 +27,7 @@ class MessageInvalidListener extends Listener {
         }
 
         let errored = false;
-        const result = await this.client.languageHandler.evalCode(message, parse)
+        const result = await this.client.languageHandler.evalCode(parse)
             .catch(e => {
                 errored = true;
                 return e.message;
diff --git a/src/struct/LanguageHandler.js b/src/struct/LanguageHandler.js
index c8d70a9..e1ed26a 100644
--- a/src/struct/LanguageHandler.js
+++ b/src/struct/LanguageHandler.js
@@ -23,6 +23,7 @@ class LanguageHandler extends AkairoHandler {
         });
 
         this.aliases = new Collection();
+        this.containers = new Collection();
     }
 
     register(language, filepath) {
@@ -59,30 +60,50 @@ class LanguageHandler extends AkairoHandler {
         }));
     }
 
-    evalCode(message, { language, code, options }) {
+    async setupContainer(id) {
+        if (this.containers.has(id)) {
+            return this.containers.get(id);
+        }
+
+        const name = `comp_iler-${id}-${Date.now()}`;
+        const proc = childProcess.spawn('docker', [
+            'run', '--rm', `--name=${name}`, '-u1000', '-w/tmp/', '-t', '-d',
+            '--net=none', `--cpus=${this.client.config.cpus}`, `-m=${this.client.config.memory}`,
+            `1computer1/comp_iler:${id}`
+        ]);
+
+        try {
+            await this.handleSpawn(proc);
+            this.containers.set(id, { name });
+            return this.containers.get(id);
+        } catch (err) {
+            throw err;
+        }
+    }
+
+    async evalCode({ language, code, options }) {
+        const { id = language.id, env = {} } = language.runWith(options);
+        const { name } = await this.setupContainer(id);
+        const proc = childProcess.spawn('docker', [
+            'exec',
+            ...Object.entries(env).map(([k, v]) => `-e${k}=${v}`),
+            name, '/bin/sh', '/var/run/run.sh', code
+        ]);
+
+        try {
+            const result = await this.handleSpawn(proc);
+            return result;
+        } catch (err) {
+            this.images.delete(id);
+            await this.kill(name);
+            throw err;
+        }
+    }
+
+    handleSpawn(proc) {
         return new Promise((resolve, reject) => {
-            const name = `comp_iler-${message.id}-${Date.now()}`;
-            const { id = language.id, env = {} } = language.runWith(options);
-            const proc = childProcess.spawn('docker', [
-                'run', '--rm', `--name=${name}`, '-u1000', '-w/tmp/',
-                '--net=none', `--cpus=${this.client.config.cpus}`, `-m=${this.client.config.memory}`,
-                ...Object.entries(env).map(([k, v]) => `-e${k}=${v}`),
-                `1computer1/comp_iler:${id}`,
-                '/bin/sh', '/var/run/run.sh', code
-            ]);
-
             setTimeout(() => {
-                try {
-                    if (process.platform === 'win32') {
-                        childProcess.execSync(`docker kill --signal=9 ${name} >nul 2>nul`);
-                    } else {
-                        childProcess.execSync(`docker kill --signal=9 ${name} >/dev/null 2>/dev/null`);
-                    }
-
-                    reject(new Error('Timed out'));
-                } catch (e) {
-                    reject(e);
-                }
+                reject(new Error('Timed out'));
             }, this.client.config.timeout);
 
             let data = '';
@@ -108,6 +129,21 @@ class LanguageHandler extends AkairoHandler {
             });
         });
     }
+
+    kill(name) {
+        let cmd;
+        if (process.platform === 'win32') {
+            cmd = `docker kill --signal=9 ${name} >nul 2>nul`;
+        } else {
+            cmd = `docker kill --signal=9 ${name} >/dev/null 2>/dev/null`;
+        }
+
+        return util.promisify(childProcess.exec)(cmd);
+    }
+
+    cleanup() {
+        return Promise.all(this.images.map(({ name }) => this.kill(name)));
+    }
 }
 
 module.exports = LanguageHandler;