Initial commit

src/struct/LanguageHandler.js

@@ -0,0 +1,107 @@
+const { AkairoHandler } = require('discord-akairo');
+const { Collection } = require('discord.js');
+const Language = require('./Language');
+const childProcess = require('child_process');
+
+class LanguageHandler extends AkairoHandler {
+    constructor(client, {
+        directory,
+        classToHandle = Language,
+        extensions = ['.js', '.ts'],
+        automateCategories,
+        loadFilter
+    }) {
+        super(client, {
+            directory,
+            classToHandle,
+            extensions,
+            automateCategories,
+            loadFilter
+        });
+
+        this.aliases = new Collection();
+    }
+
+    register(language, filepath) {
+        super.register(language, filepath);
+
+        for (let alias of language.aliases) {
+            const conflict = this.aliases.get(alias.toLowerCase());
+            if (conflict) throw new TypeError(`Alias conflict of ${alias} between ${language.id} and ${conflict}`);
+
+            alias = alias.toLowerCase();
+            this.aliases.set(alias, language.id);
+        }
+    }
+
+    deregister(language) {
+        for (let alias of language.aliases) {
+            alias = alias.toLowerCase();
+            this.aliases.delete(alias);
+        }
+
+        super.deregister(language);
+    }
+
+    findLanguage(alias) {
+        return this.modules.get(this.aliases.get(alias.toLowerCase()));
+    }
+
+    buildDocker() {
+        for (const [, { id }] of this.modules) {
+            childProcess.execSync(`docker build -t "1computer1/comp_iler:${id}" ./docker/${id}`);
+        }
+    }
+
+    evalCode(message, { language, code, options }) {
+        return new Promise((resolve, reject) => {
+            const name = `comp_iler-${message.id}-${Date.now()}`;
+            const { id, env } = language.runWith(options);
+            const proc = childProcess.spawn('docker', [
+                'run', '--rm', `--name=${name}`,
+                '--net=none', '--cpus=0.5', '-m=256m',
+                ...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/nul`);
+                    }
+
+                    reject(new Error('Evaluation timed out'));
+                } catch (e) {
+                    reject(e);
+                }
+            }, 10000);
+
+            let data = '';
+            proc.stdout.on('data', chunk => {
+                data += chunk;
+            });
+
+            proc.stderr.on('data', chunk => {
+                data += chunk;
+            });
+
+            proc.on('error', error => {
+                error.data = data;
+                reject(error);
+            });
+
+            proc.on('exit', status => {
+                if (status !== 0) {
+                    reject(new Error(data));
+                } else {
+                    resolve(data);
+                }
+            });
+        });
+    }
+}
+
+module.exports = LanguageHandler;