chore: debug mode for test

Author: williamfzc

.github/workflows/main.yml

@@ -1,4 +1,4 @@
-on: [push, pull_request]
+on: [ push, pull_request ]
 
 permissions:
   pull-requests: write
@@ -13,17 +13,8 @@ jobs:
         with:
           fetch-depth: 0
 
-      - name: Python Diff Context
+      - name: Test Action
         uses: ./
         with:
           lang: "python"
-
-      - name: Golang Diff Context
-        uses: ./
-        with:
-          lang: "golang"
-
-      - name: Java Diff Context
-        uses: ./
-        with:
-          lang: "java"
+          debug_mode: "true"

Dockerfile

@@ -15,7 +15,8 @@ RUN curl -fLo scip-linux-amd64.tar.gz https://github.com/sourcegraph/scip/releas
   && chmod +x ./scip
 
 # java/kotlin/scala scip/lsif
-RUN apk add openjdk17 \
+RUN apk add openjdk11 gradle maven \
+  && echo $JAVA_HOME \
   && curl -fLo coursier https://git.io/coursier-cli \
   && chmod +x coursier \
   && ./coursier bootstrap --standalone -o scip-java com.sourcegraph:scip-java_2.13:0.8.18 --main com.sourcegraph.scip_java.ScipJava \

action.yml

@@ -29,6 +29,10 @@ inputs:
     description: "api key for openai"
     default: ""
     required: false
+  debug_mode:
+    description: "enable debug mode if not empty"
+    default: ""
+    required: false
 runs:
   using: 'docker'
   image: 'Dockerfile'
@@ -39,3 +43,4 @@ runs:
     - ${{ inputs.repo_token }}
     - ${{ inputs.issue_number }}
     - ${{ inputs.openai_api_key }}
+    - ${{ inputs.debug_mode }}

debug.py

@@ -0,0 +1,11 @@
+from index import gen_index
+from utils import check_call
+
+
+def debug_main():
+    # pull some repos for test
+    check_call(["git", "clone", "--depth=1", "https://github.com/gin-gonic/gin.git"])
+    check_call(["git", "clone", "--depth=1", "https://github.com/alibaba/jvm-sandbox"])
+
+    gen_index("golang", "gin")
+    gen_index("java", "jvm-sandbox")

index.py

@@ -2,23 +2,26 @@
 
 from loguru import logger
 
-from diff import set_safe_git_dir
-from config import user_dir
 from utils import check_call
 
 
-def gen_index(lang: str):
-    set_safe_git_dir()
-    files = os.listdir(user_dir)
-    logger.info(f"files: {files}")
-
-    if lang == "golang":
-        gen_golang_index()
-    elif lang == "python":
-        gen_py_index()
-    else:
-        logger.error("no index mapping")
-        return
+def gen_index(lang: str, directory: str):
+    current_directory = os.getcwd()
+    try:
+        os.chdir(directory)
+        if lang == "golang":
+            gen_golang_index()
+        elif lang == "python":
+            gen_py_index()
+        elif lang == "java":
+            gen_java_and_kotlin_index()
+        elif lang == "kotlin":
+            gen_java_and_kotlin_index()
+        else:
+            logger.error("no index mapping")
+            return
+    finally:
+        os.chdir(current_directory)
 
 
 def gen_golang_index():

main.py

@@ -10,8 +10,15 @@
 
 from ai import process_with_ai
 from comment import send_comment
-from config import support_langs, csv_result_file, json_result_file, dot_result_file
-from diff import gen_diff
+from config import (
+    support_langs,
+    csv_result_file,
+    json_result_file,
+    dot_result_file,
+    user_dir,
+)
+from debug import debug_main
+from diff import gen_diff, set_safe_git_dir
 from index import gen_index
 from object import FileList
 
@@ -24,6 +31,12 @@ def main():
     repo_token = args[3]
     issue_number = args[4]
     openai_api_key = args[5]
+    debug_mode = args[6]
+
+    if debug_mode:
+        logger.warning("in debug mode, start testing")
+        debug_main()
+        logger.warning("debug mode end")
 
     # check
     if lang not in support_langs:
@@ -38,7 +51,11 @@ def main():
         return
 
     # data prepare
-    gen_index(lang)
+    set_safe_git_dir()
+    files = os.listdir(user_dir)
+    logger.info(f"files: {files}")
+
+    gen_index(lang, user_dir)
     gen_diff(before_sha, after_sha)
 
     with open(csv_result_file, encoding="utf-8") as f:

test_index.py

@@ -0,0 +1,5 @@
+from index import gen_index
+
+
+def test_index_py():
+    gen_index("python", ".")

test_main.py

@@ -1,7 +1,5 @@
 import subprocess
 
-from config import user_dir
-
 
 def check_call(commands: list):
-    subprocess.check_call(commands, cwd=user_dir)
+    subprocess.check_call(commands)