Add existing

PythonCoderAS · PythonCoderAS · commit 76bc9509b28f · 2022-06-04T20:57:42.000-04:00
diff --git a/handlers/abc.py b/handlers/abc.py
@@ -3,7 +3,7 @@
 from os.path import abspath, expanduser, join
 
 
-class BaseHandler(metaclass=abc.ABCMeta):
+class BaseHandler(abc.ABC):
     """Base Handler class that defines abstract methods"""
 
     @property
diff --git a/handlers/c.py b/handlers/c.py
@@ -7,20 +7,20 @@
 class CHandler(BaseHandler):
     """Compile and run C programs."""
 
-    def run(self, code_file):
+    def run(self, code_file, directory):
         try:
             check_call(["bash", "-c", 'which -s clang'])
         except Exception:
             compiler = "gcc"
         else:
             compiler = "clang"
-        comp = run([compiler, code_file, "-o", "script"])
+        comp = run([compiler, code_file, "-o", f"{directory}/script"])
         if comp.returncode != 0:
             exit(comp.returncode)
         else:
-            exit(run(["bash", "-c", "./script"]).returncode)
+            exit(run(["bash", "-c", f"{directory}/script"]).returncode)
 
     def build(self, code, directory=None):
         with self.file("c", directory=directory) as f:
             f.write(code)
-        self.run(f.name)
+        self.run(f.name, directory=directory if directory else self.tempdir)
diff --git a/handlers/cpp.py b/handlers/cpp.py
@@ -0,0 +1,26 @@
+from subprocess import check_call, run
+from sys import exit
+
+from .abc import BaseHandler
+
+
+class CPPHandler(BaseHandler):
+    """Compile and run C++ programs."""
+
+    def run(self, code_file, directory):
+        try:
+            check_call(["bash", "-c", 'which -s clang++'])
+        except Exception:
+            compiler = "g++"
+        else:
+            compiler = "clang++"
+        comp = run([compiler, code_file, "-o", f"{directory}/script"])
+        if comp.returncode != 0:
+            exit(comp.returncode)
+        else:
+            exit(run(["bash", "-c", f"{directory}/script"]).returncode)
+
+    def build(self, code, directory=None):
+        with self.file("cpp", directory=directory) as f:
+            f.write(code)
+        self.run(f.name, directory=directory if directory else self.tempdir)
diff --git a/handlers/java.py b/handlers/java.py
@@ -0,0 +1,27 @@
+import re
+from os.path import abspath, expanduser, join, dirname, basename
+from os import chdir
+from subprocess import CalledProcessError, check_call, run
+from sys import exit
+
+from .abc import BaseHandler
+
+
+class JavaHandler(BaseHandler):
+    """Run a Java program."""
+
+    def run(self, code_file):
+        try:
+            check_call(["javac", code_file])
+        except CalledProcessError as e:
+            exit(e.returncode)
+        chdir(dirname(code_file))
+        code_file = basename(code_file)
+        exit(run(["java", code_file.rpartition(".java")[0]]).returncode)
+
+    def build(self, code, directory=None):
+        classname = re.search(r"public class ([A-Za-z0-9_]+)", code).group(1)
+        directory = self.tempdir if directory is None else abspath(expanduser(directory))
+        with open(join(directory, classname + ".java"), "w") as file:
+            file.write(code)
+        self.run(file.name)
diff --git a/main.py b/main.py
@@ -4,19 +4,23 @@
 import sys
 
 from handlers.c import CHandler
-from handlers.python import PythonHandler, Python2Handler
+from handlers.cpp import CPPHandler
+from handlers.java import JavaHandler
+from handlers.python import Python2Handler, PythonHandler
 from handlers.shell import BashShellHandler, ShellHandler, ZshShellHandler
+from util import reverse_dict
 
-handlers = {
-    "--bash": BashShellHandler,
-    "--c": CHandler,
-    "--python": PythonHandler,
-    "--py": PythonHandler,
-    "--py2": Python2Handler,
-    "--python2": Python2Handler,
-    "--sh": ShellHandler,
-    "--zsh": ZshShellHandler,
+handlers_unwrapped = {
+    BashShellHandler: ["--bash"],
+    CHandler: ["--c"],
+    CPPHandler: ["--cpp", "--c++", "--cplusplus"],
+    JavaHandler: ["--java"],
+    PythonHandler: ["--python", "--python3", "--py", "--py3"],
+    Python2Handler: ["--python2", "--py2"],
+    ShellHandler: ["--sh"],
+    ZshShellHandler: ["--zsh"]
 }
+handlers = reverse_dict(handlers_unwrapped)
 
 
 def main(args=None):
@@ -35,13 +39,13 @@ def main(args=None):
         help="Give the stored file the speified name (does not do anything without --keep)",
         default="script",
     )
-    for argname, handler in handlers.items():
+    for handler, argnames in handlers_unwrapped.items():
         parser.add_argument(
-            argname, action="store_true", default=None, help=handler.__doc__
+            *argnames, action="store_true", default=None, help=handler.__doc__
         )
     ns = parser.parse_args(args)
     parsed_dict = {
-        argname.split("-")[-1]: getattr(ns, argname.split("-")[-1])
+        argname.split("-")[-1]: getattr(ns, argname.split("-")[-1], None)
         for argname in handlers.keys()
     }
     parse_count = [bool(val) for name, val in parsed_dict.items()].count(True)
diff --git a/util.py b/util.py
@@ -0,0 +1,6 @@
+def reverse_dict(dictionary):
+    d = {}
+    for key, values in dictionary.items():
+        for value in values:
+            d[value] = key
+    return d
