feat: custom index command

Author: williamfzc

action.yml

@@ -37,6 +37,10 @@ inputs:
     description: "specific lsif file input"
     default: ""
     required: false
+  index_cmd:
+    description: "custom index command"
+    default: ""
+    required: false
 runs:
   using: 'docker'
   image: 'Dockerfile'

debug.py

@@ -11,7 +11,7 @@ def debug_main():
     check_call(["git", "clone", "--depth=2", "https://github.com/gin-gonic/gin.git"])
     # check_call(["git", "clone", "--depth=2", "https://github.com/square/okhttp"])
 
-    gen_index("golang", "gin")
+    gen_index("golang", "gin", "")
     # gen_index("kotlin", "okhttp")
 
     os.chdir("gin")

index.py

@@ -5,10 +5,15 @@
 from utils import check_call
 
 
-def gen_index(lang: str, directory: str):
+def gen_index(lang: str, directory: str, index_command: str):
     current_directory = os.getcwd()
     try:
         os.chdir(directory)
+
+        if index_command:
+            logger.info(f"custom index command: {index_command}")
+            check_call(index_command.split(" "))
+
         if lang == "golang":
             gen_golang_index()
         elif lang == "python":

main.py

@@ -33,6 +33,7 @@ def main():
     openai_api_key = args[5]
     debug_mode = args[6]
     lsif_file = args[7]
+    index_command = args[8]
 
     if debug_mode:
         logger.warning("in debug mode, start testing")
@@ -57,7 +58,7 @@ def main():
     logger.info(f"files: {files}")
 
     if not lsif_file:
-        gen_index(lang, user_dir)
+        gen_index(lang, user_dir, index_command)
     gen_diff(before_sha, after_sha, lsif_file)
 
     with open(csv_result_file, encoding="utf-8") as f:

test_index.py

@@ -2,4 +2,4 @@
 
 
 def test_index_py():
-    gen_index("python", ".")
+    gen_index("python", ".", "")