From d218580feb8c67ebf918e2ee87d5658f1fbae436 Mon Sep 17 00:00:00 2001
From: B1nj0y <idegorepl@gmail.com>
Date: Wed, 22 Dec 2021 12:29:41 +0800
Subject: [PATCH] Support write_mode

---
 gitfx/run_funcs.py | 30 +++++++++++++++++++-----------
 1 file changed, 19 insertions(+), 11 deletions(-)

diff --git a/gitfx/run_funcs.py b/gitfx/run_funcs.py
index a5e3d80..30b2821 100644
--- a/gitfx/run_funcs.py
+++ b/gitfx/run_funcs.py
@@ -5,6 +5,7 @@
 """The main program to run functions."""
 
 import os
+import sys
 import subprocess
 
 from gitfx import parse_funcs
@@ -12,6 +13,7 @@
 
 
 ROOT_DIR = os.getenv('GITHUB_WORKSPACE', os.getcwd())
+WRITE_MODE = os.getenv('GITFX_WRITE_MODE', 'overwrite')
 
 SUPPORTED_LANGS = ['ruby',
                    'python',
@@ -87,16 +89,19 @@ def run_fun(func_path, func):
     # decide language version
     version = lang_version.get_version(func_path, func_lang) or 'latest'
 
-    if func_lang == 'bash':
-        cmd = ["bash", os.path.join(func_path, func_file_name)]
-    else:
-        cmd = ['docker', 'run', '--rm', '--workdir', '/github/workspace',
-               '-v', ROOT_DIR + ':/github/workspace',
-               docker_image(func_lang) + ':' + version, 'sh', '-c',
-               "cd " + os.path.relpath(func_path, ROOT_DIR) + " && " +        # noqa
-               deps_install_cmd(func_lang, func_path) + " && " +                    # noqa
-               run_before_script + " && " +                                   # noqa
-               RUN_CMDS[func_lang] + " " + func_file_name]
+    try:
+        if func_lang == 'bash':
+            cmd = ["bash", os.path.join(func_path, func_file_name)]
+        else:
+            cmd = ['docker', 'run', '--rm', '--workdir', '/github/workspace',
+                   '-v', ROOT_DIR + ':/github/workspace',
+                   docker_image(func_lang) + ':' + version, 'sh', '-c',
+                   "cd " + os.path.relpath(func_path, ROOT_DIR) + " && " +        # noqa
+                   deps_install_cmd(func_lang, func_path) + " && " +                    # noqa
+                   run_before_script + " && " +                                   # noqa
+                   RUN_CMDS[func_lang] + " " + func_file_name]
+    except:
+        sys.exit(1)
 
     output = subprocess.Popen(cmd, stdout=subprocess.PIPE).communicate()[0]
     return output.decode("utf8")
@@ -111,7 +116,10 @@ def write_to_route(result, func_route):
         os.makedirs(dst_dir)
 
     file_path = os.path.join(ROOT_DIR, func_route)
-    f = open(file_path, 'w')
+    if os.path.isfile(file_path) and WRITE_MODE == 'append':
+        f = open(file_path, 'a')
+    else:
+        f = open(file_path, 'w')
     f.write(result)
     f.close()