chore: use from_repo + clean-up from_repo (#462)

Author: christinewangcw

src/codegen/git/repo_operator/local_repo_operator.py

@@ -12,7 +12,9 @@
 from codegen.git.repo_operator.repo_operator import RepoOperator
 from codegen.git.schemas.enums import FetchResult
 from codegen.git.schemas.repo_config import RepoConfig
+from codegen.git.utils.clone_url import add_access_token_to_url
 from codegen.git.utils.file_utils import create_files
+from codegen.shared.configs.session_configs import config
 
 logger = logging.getLogger(__name__)
 
@@ -112,6 +114,9 @@ def create_from_repo(cls, repo_path: str, url: str, access_token: str | None = N
             url (str): Git URL of the repository
             access_token (str | None): Optional GitHub API key for operations that need GitHub access
         """
+        access_token = access_token or config.secrets.github_token
+        url = add_access_token_to_url(url=url, token=access_token)
+
         # Check if repo already exists
         if os.path.exists(repo_path):
             try:

src/codegen/sdk/core/codebase.py

@@ -1219,11 +1219,10 @@ def set_session_options(self, **kwargs: Unpack[SessionOptions]) -> None:
     @classmethod
     def from_repo(
         cls,
-        repo_name: str,
+        repo_full_name: str,
         *,
-        tmp_dir: str | None = None,
+        tmp_dir: str | None = "/tmp/codegen",
         commit: str | None = None,
-        shallow: bool = True,
         programming_language: ProgrammingLanguage | None = None,
         config: CodebaseConfig = DefaultConfig,
     ) -> "Codebase":
@@ -1240,23 +1239,21 @@ def from_repo(
         Returns:
             Codebase: A Codebase instance initialized with the cloned repository
         """
-        logger.info(f"Fetching codebase for {repo_name}")
+        logger.info(f"Fetching codebase for {repo_full_name}")
 
         # Parse repo name
-        if "/" not in repo_name:
+        if "/" not in repo_full_name:
             msg = "repo_name must be in format 'owner/repo'"
             raise ValueError(msg)
-        owner, repo = repo_name.split("/")
+        owner, repo = repo_full_name.split("/")
 
         # Setup temp directory
-        if tmp_dir is None:
-            tmp_dir = "/tmp/codegen"
         os.makedirs(tmp_dir, exist_ok=True)
         logger.info(f"Using directory: {tmp_dir}")
 
         # Setup repo path and URL
         repo_path = os.path.join(tmp_dir, repo)
-        repo_url = f"https://github.com/{repo_name}.git"
+        repo_url = f"https://github.com/{repo_full_name}.git"
         logger.info(f"Will clone {repo_url} to {repo_path}")
 
         try:

tests/integration/codegen/git/codebase/conftest.py

@@ -1,36 +1,13 @@
 import os
 
 import pytest
-from git import Repo as GitRepo
 
-from codegen.git.repo_operator.local_repo_operator import LocalRepoOperator
-from codegen.git.schemas.repo_config import RepoConfig
-from codegen.git.utils.clone_url import get_authenticated_clone_url_for_repo_config
-from codegen.sdk.codebase.config import ProjectConfig
 from codegen.sdk.core.codebase import Codebase
-from codegen.shared.configs.session_configs import config
+from codegen.shared.enums.programming_language import ProgrammingLanguage
 
 
 @pytest.fixture
-def repo_config(tmpdir):
-    repo_config = RepoConfig(
-        name="Kevin-s-Adventure-Game",
-        full_name="codegen-sh/Kevin-s-Adventure-Game",
-        base_dir=str(tmpdir),
-    )
-    yield repo_config
-
-
-@pytest.fixture
-def op(repo_config):
-    os.chdir(repo_config.base_dir)
-    GitRepo.clone_from(url=get_authenticated_clone_url_for_repo_config(repo_config, token=config.secrets.github_token), to_path=os.path.join(repo_config.base_dir, repo_config.name), depth=1)
-    op = LocalRepoOperator(repo_config=repo_config)
-    yield op
-
-
-@pytest.fixture
-def codebase(op: LocalRepoOperator):
-    project_config = ProjectConfig(repo_operator=op)
-    codebase = Codebase(projects=[project_config])
+def codebase(tmpdir):
+    os.chdir(tmpdir)
+    codebase = Codebase.from_repo(repo_full_name="codegen-sh/Kevin-s-Adventure-Game", tmp_dir=tmpdir, programming_language=ProgrammingLanguage.PYTHON)
     yield codebase