From d809e06791c6b6c8844b103a8323b4be7f955174 Mon Sep 17 00:00:00 2001
From: Seohyun Lee <studioseorii@gmail.com>
Date: Sat, 1 Jan 2022 17:33:23 +0900
Subject: [PATCH] feat: add support for golang & rust

---
 Dockerfile           |  7 ++++
 res/test.html        | 93 ++++++++++++++++----------------------------
 src/runner/golang.ts | 16 ++++++++
 src/runner/index.ts  |  4 +-
 4 files changed, 60 insertions(+), 60 deletions(-)
 create mode 100644 src/runner/golang.ts

diff --git a/Dockerfile b/Dockerfile
index 0e91e8c..5bd6cb5 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -36,6 +36,13 @@ RUN java -version
 RUN apk add rust
 RUN rustc --version
 
+# Install Go
+RUN apk add go
+RUN mkdir /tmp/gocache
+RUN chmod 777 /tmp/gocache
+ENV GOROOT /usr/lib/go
+ENV GOCACHE /tmp/gocache
+ENV PATH /go/bin:$PATH
 
 # Copy files & Install requirements
 RUN addgroup execute
diff --git a/res/test.html b/res/test.html
index aaa963d..5954fb1 100644
--- a/res/test.html
+++ b/res/test.html
@@ -137,6 +137,15 @@
                 CP: 'Compiling',
                 RUN: 'Running',
             }
+            const supportedLanguages = [
+                'C',
+                'CPP',
+                'PYTHON3',
+                'PYPY3',
+                'JAVA',
+                'RUST',
+                'GO',
+            ]
             const ext = {
                 PYTHON3: 'py',
                 PYPY3: 'py',
@@ -144,6 +153,7 @@
                 C: 'c',
                 JAVA: 'java',
                 RUST: 'rs',
+                GO: 'go',
             }
 
             const exampleCode = {
@@ -202,6 +212,17 @@
     };
 
     println!("{}", number_a + number_b);
+}`,
+                GO: `package main
+
+import (
+\t"fmt"
+)
+
+func main() {
+\tvar a, b int
+\tfmt.Scanf("%d %d", &a, &b)
+\tfmt.Println(a+b)
 }`,
             }
 
@@ -358,7 +379,7 @@ <h4>State</h4>
                             ],
                         },
                         timeLimit: 2000,
-                        memoryLimit: 100,
+                        memoryLimit: 1000,
                     }),
                 })
             }
@@ -373,6 +394,17 @@ <h4>State</h4>
             }
 
             document.addEventListener('DOMContentLoaded', () => {
+                for (let i of supportedLanguages) {
+                    document.getElementById('supportedLang').innerHTML += `<li
+                                class="mdc-list-item ripple"
+                                aria-selected="false"
+                                data-value="${i}"
+                                role="option"
+                            >
+                                <span class="mdc-list-item__ripple"></span>
+                                <span class="mdc-list-item__text">${i}</span>
+                            </li>`
+                }
                 setInterval(() => {
                     if (document.documentElement.scrollTop)
                         document
@@ -482,64 +514,7 @@ <h3>Problem</h3>
                     <div
                         class="mdc-select__menu mdc-menu mdc-menu-surface mdc-menu-surface--fullwidth"
                     >
-                        <ul class="mdc-list">
-                            <li
-                                class="mdc-list-item ripple"
-                                aria-selected="false"
-                                data-value="C"
-                                role="option"
-                            >
-                                <span class="mdc-list-item__ripple"></span>
-                                <span class="mdc-list-item__text"> C </span>
-                            </li>
-                            <li
-                                class="mdc-list-item ripple"
-                                aria-selected="false"
-                                data-value="CPP"
-                                role="option"
-                            >
-                                <span class="mdc-list-item__ripple"></span>
-                                <span class="mdc-list-item__text"> CPP </span>
-                            </li>
-                            <li
-                                class="mdc-list-item ripple"
-                                aria-selected="false"
-                                data-value="PYTHON3"
-                                role="option"
-                            >
-                                <span class="mdc-list-item__ripple"></span>
-                                <span class="mdc-list-item__text">
-                                    PYTHON3
-                                </span>
-                            </li>
-                            <li
-                                class="mdc-list-item ripple"
-                                aria-selected="false"
-                                data-value="PYPY3"
-                                role="option"
-                            >
-                                <span class="mdc-list-item__ripple"></span>
-                                <span class="mdc-list-item__text"> PYPY3 </span>
-                            </li>
-                            <li
-                                class="mdc-list-item ripple"
-                                aria-selected="false"
-                                data-value="JAVA"
-                                role="option"
-                            >
-                                <span class="mdc-list-item__ripple"></span>
-                                <span class="mdc-list-item__text"> JAVA </span>
-                            </li>
-                            <li
-                                class="mdc-list-item ripple"
-                                aria-selected="false"
-                                data-value="RUST"
-                                role="option"
-                            >
-                                <span class="mdc-list-item__ripple"></span>
-                                <span class="mdc-list-item__text"> RUST </span>
-                            </li>
-                        </ul>
+                        <ul class="mdc-list" id="supportedLang"></ul>
                     </div>
                 </div>
                 <label
diff --git a/src/runner/golang.ts b/src/runner/golang.ts
new file mode 100644
index 0000000..39710f2
--- /dev/null
+++ b/src/runner/golang.ts
@@ -0,0 +1,16 @@
+import { JudgeRequest } from '../types/request'
+import { execute, getLimitString } from './util'
+import commonJudge from './common'
+
+export default function (data: JudgeRequest) {
+    return commonJudge(
+        data,
+        (path) =>
+            execute(
+                `p-${data.uid}`,
+                getLimitString({ cpuLimit: 50 }, `go build ${path}/Main.go`),
+                { cwd: path }
+            ),
+        (path) => path + '/Main'
+    )
+}
diff --git a/src/runner/index.ts b/src/runner/index.ts
index a466072..44a2c4c 100644
--- a/src/runner/index.ts
+++ b/src/runner/index.ts
@@ -1,5 +1,4 @@
 import {
-    CommonDataSet,
     JudgeRequest,
     JudgeSourceType,
     JudgeType,
@@ -14,6 +13,7 @@ import judgePython3 from './python3'
 import judgePypy3 from './pypy3'
 import judgeJava from './java'
 import judgeRust from './rust'
+import judgeGo from './golang'
 
 export default function (data: JudgeRequest): Promise<JudgeResult> {
     switch (data.judgeType) {
@@ -39,6 +39,8 @@ export default function (data: JudgeRequest): Promise<JudgeResult> {
                     return judgeJava(data)
                 case JudgeSourceType.RUST:
                     return judgeRust(data)
+                case JudgeSourceType.GO:
+                    return judgeGo(data)
             }
             return Promise.resolve({
                 uid: data.uid,