feat: remove joern integration

Author: SunBK201

scubatrace/function.py

@@ -8,7 +8,6 @@
 from tree_sitter import Node
 
 from . import language as lang
-from .call import Call
 from .identifier import Identifier
 from .statement import BlockStatement, Statement
 
@@ -21,16 +20,10 @@ class Function(BlockStatement):
     A function in the source code.
     """
 
-    def __init__(
-        self, node: Node, file: File | BlockStatement, joern_id: str | None = None
-    ):
+    def __init__(self, node: Node, file: File | BlockStatement):
         super().__init__(node, file)
-        self.joern_id = joern_id
         self._is_build_cfg = False
 
-        self.callers_joern: list[Call] = []
-        self.callees_joern: list[Call] = []
-
     @staticmethod
     def create(node: Node, parent: File | BlockStatement):
         """
@@ -106,9 +99,6 @@ def __str__(self) -> str:
             f'"{self.name.replace("::", "--")} ({self.file.name}\\:{self.start_line})"'
         )
 
-    def set_joernid(self, joern_id: str):
-        self.joern_id = joern_id
-
     @property
     def signature(self) -> str:
         """

scubatrace/project.py

@@ -11,9 +11,7 @@
 from scubalspy.scubalspy_config import ScubalspyConfig
 from scubalspy.scubalspy_logger import ScubalspyLogger
 
-from . import joern
 from . import language as lang
-from .call import Call
 from .file import File
 from .function import Function, FunctionDeclaration
 from .parser import Parser
@@ -98,26 +96,9 @@ def __init__(
         path: str,
         language: type[lang.Language],
         enable_lsp: bool = True,
-        enable_joern: bool = False,
     ):
         self.path = path
         self.language = language
-        if enable_joern:
-            if language == lang.C:
-                joern_language = joern.Language.C
-            elif language == lang.JAVA:
-                joern_language = joern.Language.JAVA
-            elif language == lang.PYTHON:
-                joern_language = joern.Language.PYTHON
-            elif language == lang.JAVASCRIPT:
-                joern_language = joern.Language.JAVASCRIPT
-            else:
-                raise ValueError("Joern unsupported language")
-            self.joern = joern.Joern(
-                path,
-                joern_language,
-            )
-            self.joern.export_with_preprocess()
         if enable_lsp:
             self.start_lsp()
 
@@ -160,13 +141,6 @@ def start_lsp(self):
                 atexit.register(os.remove, self.conf_file)
         self.lsp.sync_start_server()
 
-    def close(self):
-        if "joern" in self.__dict__:
-            self.joern.close()
-
-    def __exit__(self, exc_type, exc_value, traceback):
-        self.close()
-
     @property
     def abspath(self) -> str:
         """
@@ -289,69 +263,6 @@ def callgraph(self) -> nx.MultiDiGraph:
         cg = self.__build_callgraph(entry)
         return cg
 
-    @cached_property
-    def callgraph_joern(self) -> nx.MultiDiGraph:
-        if self.joern is None:
-            raise ValueError("Joern is not enabled for this project.")
-        joern_cg = self.joern.callgraph
-        cg = nx.MultiDiGraph()
-        for node in joern_cg.nodes:
-            if joern_cg.nodes[node]["NODE_TYPE"] != "METHOD":
-                continue
-            if joern_cg.nodes[node]["IS_EXTERNAL"] == "true":
-                continue
-            func = self.search_function(
-                joern_cg.nodes[node]["FILENAME"],
-                int(joern_cg.nodes[node]["LINE_NUMBER"]),
-            )
-            if func is None:
-                continue
-            func.set_joernid(node)
-            cg.add_node(
-                func,
-                label=func.dot_text,
-            )
-        for u, v, data in joern_cg.edges(data=True):
-            if joern_cg.nodes[u]["NODE_TYPE"] != "METHOD":
-                continue
-            if joern_cg.nodes[v]["NODE_TYPE"] != "METHOD":
-                continue
-
-            # search by joern_id
-            src_func: Function | None = None
-            dst_func: Function | None = None
-            for node in cg.nodes:
-                if node.joern_id == u:
-                    src_func = node
-                if node.joern_id == v:
-                    dst_func = node
-            if src_func is None or dst_func is None:
-                continue
-            if src_func == dst_func:
-                continue
-            src_func.callees_joern.append(
-                Call(
-                    src_func,
-                    dst_func,
-                    int(data["LINE_NUMBER"]),
-                    int(data["COLUMN_NUMBER"]),
-                )
-            )
-            dst_func.callers_joern.append(
-                Call(
-                    src_func,
-                    dst_func,
-                    int(data["LINE_NUMBER"]),
-                    int(data["COLUMN_NUMBER"]),
-                )
-            )
-            cg.add_edge(
-                src_func,
-                dst_func,
-                **data,
-            )
-        return cg
-
     def export_callgraph(self, output_path: str):
         """
         Exports the call graph of the project to a DOT file.