Python API Add (#2)

* Base Python Code

* Python Compiler

* Update Dockerfile

* Update Dockerfile

Author: codewithvk

Dockerfile

@@ -6,7 +6,8 @@ WORKDIR /usr/src/app
 ENV DEBIAN_FRONTEND noninteractive
 RUN apt-get update && \
     apt-get -y install gcc mono-mcs && \
-    rm -rf /var/lib/apt/lists/*
+    rm -rf /var/lib/apt/lists/* && \
+    apt-get -y install python3
 
 COPY package*.json ./
 

api/pythonApi.js

@@ -0,0 +1,25 @@
+const express = require('express');
+const {  CppCompile } = require('../compiler/cpp');
+const { PythonCompile } = require('../compiler/python');
+const router = express.Router();
+
+
+router.post('/', async (req, res) => {
+    const InputCode = Buffer.from(req.body.code, 'base64').toString('binary')
+    const DeCode = Buffer.from(req.body.input, 'base64').toString('binary')
+    let response = await PythonCompile(InputCode, DeCode);
+    console.log({response})
+    if (response.statusMes === "Compiler Error") {
+        res.status(202).json(response)
+    } else if (response.statusMes === "Run Time Error") {
+        res.status(201).json(response)
+    } else {
+        res.status(200).json(response)
+    }
+
+});
+
+
+
+
+module.exports = router;

compiler/python.js

@@ -0,0 +1,143 @@
+const { v4: uuid } = require('uuid');
+const { exec } = require('child_process');
+const fs = require('fs');
+const os = require('os');
+const path = require('path');
+const saveFile = (file, data) => {
+    return new Promise((resolve, reject) => {
+        fs.writeFile(file, data, function (err) {
+            if (err) {
+                reject(err);
+            } else {
+                resolve([file]);
+            }
+        });
+    });
+};
+
+async function deleteFiles(pyPath, inputPath, exePath) {
+    if (fs.existsSync(pyPath)) {
+        await fs.unlinkSync(pyPath);
+    }
+
+    if (fs.existsSync(inputPath)) {
+        await fs.unlinkSync(inputPath);
+    }
+
+    if (fs.existsSync(exePath)) {
+        await fs.unlinkSync(exePath);
+    }
+}
+
+function getRunCommand(executable, input) {
+    return `python ${executable} < ${input}`;
+}
+
+function getExecutablePath(fileName) {
+    // console.log(os.platform());
+    if (os.platform() === 'win32') {
+        return `${path.join(__dirname, '..', 'upload', fileName)}.exe`;
+    }
+    if (os.platform() === 'linux') {
+        return `${path.join(__dirname, '..', 'upload', fileName)}`;
+    }
+}
+
+function getpyPath(fileName) {
+    return `${path.join(__dirname, '..', 'upload', fileName)}.py`;
+}
+
+function getInputPath(fileName) {
+    return `${path.join(__dirname, '..', 'upload', fileName)}-input.txt`;
+}
+
+
+
+function runProgram(exePath, inputPath) {
+    return new Promise((resolve, reject) => {
+
+        exec(getRunCommand(exePath, inputPath), (error, stdout, stderr) => {
+            if (error) {
+                reject({ error, stdout, stderr });
+            } else {
+                resolve({ stdout, stderr });
+            }
+        });
+    });
+}
+
+
+function runProgramNoIP(exePath) {
+    return new Promise((resolve, reject) => {
+
+        exec(`python ${exePath}`, (error, stdout, stderr) => {
+            if (error) {
+              reject({ error, stdout, stderr });
+            } else {
+                resolve({ stdout, stderr });
+            }
+        });
+    });
+}
+const PythonCompile = async (code, input) => {
+    let state = {
+        stdout: null,
+        stderr: null,
+        statusMes: "",
+    }
+    if (input.length > 0) {
+        let uniqueFileName = uuid();
+        let executePath = getExecutablePath(uniqueFileName)
+        let pyPath = getpyPath(uniqueFileName)
+        let ipPath = getInputPath(uniqueFileName)
+
+        await saveFile(pyPath, code);
+        await saveFile(ipPath, input);
+
+        try {
+            let { stdout, stderr } = await runProgram(pyPath, ipPath);
+            state.stdout = stdout;
+            state.stderr = stderr;
+        } catch (err) {
+            state.stderr = err.stderr;
+            state.statusMes = "Compiler Error";
+            deleteFiles(pyPath, ipPath, executePath);
+            return state;
+        }
+        if (state.stderr === '') {
+            state.stderr = null;
+        }
+        state.statusMes = "Successfully Compiled";
+        await deleteFiles(pyPath, ipPath, executePath);
+        return state;
+    } else {
+        let uniqueFileName = uuid();
+        let executePath = getExecutablePath(uniqueFileName)
+        let pyPath = getpyPath(uniqueFileName)
+        await saveFile(pyPath, code);
+        try {
+            let { stdout, stderr } = await runProgramNoIP(pyPath);
+            state.stdout = stdout;
+            state.stderr = stderr;
+        } catch (err) {
+            state.stderr = err.stderr;
+            state.statusMes = "Compiler Error";
+            deleteFiles(pyPath, executePath);
+            return state;
+        }
+        if (state.stderr === '') {
+            state.stderr = null;
+        }
+        state.statusMes = "Successfully Compiled";
+        await deleteFiles(pyPath, executePath);
+        return state;
+
+    }
+    return state;
+
+}
+
+
+
+
+module.exports = { PythonCompile };

server.js

@@ -11,6 +11,8 @@ app.get('/', (req, res) => {
     res.send("API IS RUNNING")
 })
 app.use('/api', require('./api/cppApi'));
+app.use('/api/python', require('./api/pythonApi'));
+
 
 app.listen(PORT, () => {
     console.log(`Listening at port ${PORT}`);