core: frontends: expose source API to all languages (#1893)

Signed-off-by: David Korczynski <[email protected]>

Author: DavidKorczynski

src/fuzz_introspector/frontends/core.py

@@ -15,8 +15,7 @@
 
 import logging
 
-from fuzz_introspector.frontends import frontend_c
-from fuzz_introspector.frontends import frontend_cpp
+from fuzz_introspector.frontends import (frontend_c, frontend_cpp, frontend_go)
 
 logger = logging.getLogger(name=__name__)
 
@@ -35,7 +34,7 @@ def analyse_source_file(code: bytes, language: str):
     elif language == 'cpp':
         return frontend_cpp.analyse_source_code(code)
     elif language == 'go':
-        logger.info('Go is not implement yet.')
+        return frontend_go.analyse_source_code(code)
     else:
         logger.info('Language %s not supported', language)
     return None

src/fuzz_introspector/frontends/frontend_c.py

@@ -602,13 +602,12 @@ def load_treesitter_trees(source_files, log_harnesses=True):
     return results
 
 
-def analyse_source_code(source_content: str) -> Project:
-    """Returns a Project based on a single source string."""
+def analyse_source_code(source_content: str) -> SourceCodeFile:
+    """Returns a source abstraction based on a single source string."""
     source_code = SourceCodeFile(source_file='in-memory string',
                                  language='c',
                                  source_content=source_content.encode())
-    project = Project([source_code])
-    return project
+    return source_code
 
 
 def analyse_folder(folder_path: str, language: str = 'c') -> Project:

src/fuzz_introspector/frontends/frontend_cpp.py

@@ -379,10 +379,9 @@ def load_treesitter_trees(source_files, log_harnesses=True):
     return results
 
 
-def analyse_source_code(source_content: str) -> Project:
-    """Returns a Project based on a single source string."""
+def analyse_source_code(source_content: str) -> SourceCodeFile:
+    """Returns a source abstraction based on a single source string."""
     source_code = SourceCodeFile(source_file='in-memory string',
                                  language='cpp',
                                  source_content=source_content.encode())
-    project = Project([source_code])
-    return project
+    return source_code

src/fuzz_introspector/frontends/frontend_go.py

@@ -604,3 +604,10 @@ def load_treesitter_trees(source_files: list[str],
         results.append(source_cls)
 
     return results
+
+
+def analyse_source_code(source_content: str) -> SourceCodeFile:
+    """Returns a source abstraction based on a single source string."""
+    source_code = SourceCodeFile(source_file='in-memory string',
+                                 source_content=source_content.encode())
+    return source_code