Rename CodebaseGraph to CodebaseContext (#405)

`CodebaseGraph` is misleading and has been a deprecated term since the
Codebase Initialization Refactor.

Author: EdwardJXLi

src/codegen/runner/diff/get_raw_diff.py

@@ -27,7 +27,7 @@ def patch_to_limited_diff_string(patch, codebase: Codebase, max_lines=10000):
 
     # Add flags that are not in the diff
     filenames = [patched_file.path for patched_file in patch]
-    flags_not_in_diff = list(filter(lambda flag: flag.symbol.filepath not in filenames, codebase.G.flags._flags))
+    flags_not_in_diff = list(filter(lambda flag: flag.symbol.filepath not in filenames, codebase.ctx.flags._flags))
 
     for flag in flags_not_in_diff:
         filename = flag.symbol.filepath
@@ -39,7 +39,7 @@ def patch_to_limited_diff_string(patch, codebase: Codebase, max_lines=10000):
         patch.append(patched_file)
 
     for patched_file in patch:
-        filtered_flags = filter(lambda flag: flag.symbol.filepath == patched_file.path, codebase.G.flags._flags)
+        filtered_flags = filter(lambda flag: flag.symbol.filepath == patched_file.path, codebase.ctx.flags._flags)
         sorted_flags = list(map(lambda flag: flag.symbol.start_point.row + 1, filtered_flags))
         sorted_flags.sort()
 

src/codegen/runner/sandbox/executor.py

@@ -36,8 +36,8 @@ async def find_flags(self, execute_func: Callable) -> list[CodeFlag]:
         """Runs the execute_func in find_mode to find flags"""
         self.codebase.set_find_mode(True)
         await self._execute_with_try_catch(execute_func, commit=False)
-        code_flags = self.codebase.G.flags._flags
-        logger.info(f"> Found {len(self.codebase.G.flags._flags)} CodeFlags")
+        code_flags = self.codebase.ctx.flags._flags
+        logger.info(f"> Found {len(self.codebase.ctx.flags._flags)} CodeFlags")
         return code_flags
 
     async def find_flag_groups(self, code_flags: list[CodeFlag], grouping_config: GroupingConfig) -> list[Group]:
@@ -77,7 +77,7 @@ async def execute_flag_groups(self, commit_msg: str, execute_func: Callable, fla
             run_results.append(run_result)
             head_branches.append(created_branch)
 
-        self.codebase.G.flags._flags.clear()
+        self.codebase.ctx.flags._flags.clear()
         return run_results, head_branches
 
     async def execute(self, execute_func: Callable, group: Group | None = None, session_options: SessionOptions = SessionOptions()) -> CodemodRunResult:
@@ -155,7 +155,7 @@ async def _get_structured_run_output(self, result: CodemodRunResult) -> CodemodR
                 "messageType": str(flag.message_type),
                 "messageRecipient": flag.message_recipient,
             }
-            for flag in self.codebase.G.flags._flags
+            for flag in self.codebase.ctx.flags._flags
         ]
         result.flags = flags
         if result.observation_meta is None:

src/codegen/runner/sandbox/middlewares.py

@@ -69,6 +69,6 @@ async def cleanup_after_codemod(self, is_exception: bool = False):
         if is_exception:
             # TODO: instead of committing transactions, we should just rollback
             logger.info("Committing pending transactions due to exception")
-            self.runner.codebase.G.commit_transactions(sync_graph=False)
+            self.runner.codebase.ctx.commit_transactions(sync_graph=False)
         self.runner.reset_runner()
         self.server_info.is_running_codemod = False

src/codegen/runner/sandbox/runner.py

@@ -112,5 +112,5 @@ async def create_branch(self, request: CreateBranchRequest) -> CreateBranchRespo
         response.results = run_results
         response.branches = branches
 
-        self.codebase.G.flags._flags.clear()
+        self.codebase.ctx.flags._flags.clear()
         return response

src/codegen/sdk/code_generation/doc_utils/generate_docs_json.py

@@ -8,7 +8,7 @@
 from codegen.sdk.core.placeholder.placeholder_type import TypePlaceholder
 
 ATTRIBUTES_TO_IGNORE = [
-    "G",
+    "ctx",
     "node_id",
     "angular",
     "model_config",
@@ -19,7 +19,6 @@
     "node_type",
     "ts_node",
     "file_node_id",
-    "G",
     "statement_type",
     "assignment_types",
 ]

src/codegen/sdk/codebase/codebase_analysis.py

@@ -9,7 +9,7 @@
 
 
 def get_codebase_summary(codebase: Codebase) -> str:
-    node_summary = f"""Contains {len(codebase.G.get_nodes())} nodes
+    node_summary = f"""Contains {len(codebase.ctx.get_nodes())} nodes
 - {len(list(codebase.files))} files
 - {len(list(codebase.imports))} imports
 - {len(list(codebase.external_modules))} external_modules
@@ -19,10 +19,10 @@ def get_codebase_summary(codebase: Codebase) -> str:
 \t- {len(list(codebase.global_vars))} global_vars
 \t- {len(list(codebase.interfaces))} interfaces
 """
-    edge_summary = f"""Contains {len(codebase.G.edges)} edges
-- {len([x for x in codebase.G.edges if x[2].type == EdgeType.SYMBOL_USAGE])} symbol -> used symbol
-- {len([x for x in codebase.G.edges if x[2].type == EdgeType.IMPORT_SYMBOL_RESOLUTION])} import -> used symbol
-- {len([x for x in codebase.G.edges if x[2].type == EdgeType.EXPORT])} export -> exported symbol
+    edge_summary = f"""Contains {len(codebase.ctx.edges)} edges
+- {len([x for x in codebase.ctx.edges if x[2].type == EdgeType.SYMBOL_USAGE])} symbol -> used symbol
+- {len([x for x in codebase.ctx.edges if x[2].type == EdgeType.IMPORT_SYMBOL_RESOLUTION])} import -> used symbol
+- {len([x for x in codebase.ctx.edges if x[2].type == EdgeType.EXPORT])} export -> exported symbol
     """
 
     return f"{node_summary}\n{edge_summary}"

src/codegen/sdk/codebase/codebase_graph.py renamed to src/codegen/sdk/codebase/codebase_context.py

@@ -76,7 +76,7 @@ def get_node_classes(programming_language: ProgrammingLanguage) -> NodeClasses:
         raise ValueError(msg)
 
 
-class CodebaseGraph:
+class CodebaseContext:
     """MultiDiGraph Wrapper with TransactionManager"""
 
     # =====[ __init__ attributes ]=====

src/codegen/sdk/codebase/config_parser.py

@@ -4,21 +4,21 @@
 from codegen.sdk.enums import ProgrammingLanguage
 
 if TYPE_CHECKING:
-    from codegen.sdk.codebase.codebase_graph import CodebaseGraph
+    from codegen.sdk.codebase.codebase_context import CodebaseContext
 
 
 class ConfigParser(ABC):
     def __init__(self):
         pass
 
     @abstractmethod
-    def parse_configs(self, codebase_graph: "CodebaseGraph"): ...
+    def parse_configs(self, codebase_context: "CodebaseContext"): ...
 
 
-def get_config_parser_for_language(language: ProgrammingLanguage, codebase_graph: "CodebaseGraph") -> ConfigParser | None:
+def get_config_parser_for_language(language: ProgrammingLanguage, codebase_context: "CodebaseContext") -> ConfigParser | None:
     from codegen.sdk.typescript.config_parser import TSConfigParser
 
     if language == ProgrammingLanguage.TYPESCRIPT:
-        return TSConfigParser(codebase_graph)
+        return TSConfigParser(codebase_context)
 
     return None

src/codegen/sdk/codebase/factory/get_session.py

@@ -6,7 +6,7 @@
 
 from codegen.git.repo_operator.local_repo_operator import LocalRepoOperator
 from codegen.git.schemas.repo_config import BaseRepoConfig
-from codegen.sdk.codebase.codebase_graph import CodebaseGraph
+from codegen.sdk.codebase.codebase_context import CodebaseContext
 from codegen.sdk.codebase.config import CodebaseConfig, GSFeatureFlags, ProjectConfig, SessionOptions, TestFlags
 from codegen.sdk.codebase.factory.codebase_factory import CodebaseFactory
 from codegen.sdk.core.codebase import Codebase, PyCodebaseType, TSCodebaseType
@@ -108,12 +108,12 @@ def get_codebase_graph_session(
     files: dict[str, str] = {},
     sync_graph: bool = True,
     session_options: SessionOptions = SessionOptions(),
-) -> Generator[CodebaseGraph, None, None]:
+) -> Generator[CodebaseContext, None, None]:
     """Gives you a Codebase2 operating on the files you provided as a dict"""
     op = LocalRepoOperator.create_from_files(repo_path=tmpdir, files=files)
     config = CodebaseConfig(feature_flags=TestFlags)
     projects = [ProjectConfig(repo_operator=op, programming_language=programming_language)]
-    graph = CodebaseGraph(projects=projects, config=config)
+    graph = CodebaseContext(projects=projects, config=config)
     with graph.session(sync_graph=sync_graph, session_options=session_options):
         try:
             yield graph

src/codegen/sdk/codebase/node_classes/py_node_classes.py

@@ -31,10 +31,10 @@
 from codegen.sdk.python.statements.import_statement import PyImportStatement
 
 
-def parse_subscript(node: TSNode, file_node_id, G, parent):
+def parse_subscript(node: TSNode, file_node_id, ctx, parent):
     if (node.prev_named_sibling and node.prev_named_sibling.text.decode("utf-8") == "TypeAlias") or isinstance(parent, Type):
-        return PyGenericType(node, file_node_id, G, parent)
-    return SubscriptExpression(node, file_node_id, G, parent)
+        return PyGenericType(node, file_node_id, ctx, parent)
+    return SubscriptExpression(node, file_node_id, ctx, parent)
 
 
 PyExpressionMap = {