Add codegen/utils module

Author: caroljung-cg

src/codegen/cli/codemod/convert.py

@@ -7,7 +7,7 @@ def convert_to_cli(input: str, language: str, name: str) -> str:
 # from app.codemod.compilation.models.context import CodemodContext
 #from app.codemod.compilation.models.pr_options import PROptions
 
-from graph_sitter import {codebase_type}
+from codegen.sdk import {codebase_type}
 
 context: Any
 

src/codegen/cli/utils/count_functions.py

@@ -5,7 +5,7 @@
 
 # from app.codemod.compilation.models.context import CodemodContext
 # from app.codemod.compilation.models.pr_options import PROptions
-# from graph_sitter import PyCodebaseType
+# from codegen.sdk import PyCodebaseType
 
 # context: CodemodContext
 

src/codegen/git/__init__.py

@@ -3,3 +3,5 @@
 CODEGEN_BOT_NAME = "codegen-bot"
 CODEGEN_BOT_EMAIL = "team+codegenbot@codegen.sh"
 CODEOWNERS_FILEPATHS = [".github/CODEOWNERS", "CODEOWNERS", "docs/CODEOWNERS"]
+HIGHSIDE_REMOTE_NAME = "highside"
+LOWSIDE_REMOTE_NAME = "lowside"

src/codegen/git/configs/constants.py

@@ -0,0 +1,35 @@
+from __future__ import annotations
+
+import logging
+from typing import Any
+
+from pydantic import BaseModel, Field
+
+from codegen.git.models.pull_request_context import PullRequestContext
+
+logger = logging.getLogger(__name__)
+
+
+class CodemodContext(BaseModel):
+    CODEMOD_ID: int | None = None
+    CODEMOD_LINK: str | None = None
+    CODEMOD_AUTHOR: str | None = None
+    TEMPLATE_ARGS: dict[str, Any] = Field(default_factory=dict)
+
+    # TODO: add fields for version
+    # CODEMOD_VERSION_ID: int | None = None
+    # CODEMOD_VERSION_AUTHOR: str | None = None
+
+    PULL_REQUEST: PullRequestContext | None = None
+
+    @classmethod
+    def _render_template(cls, template_schema: dict[str, str], template_values: dict[str, Any]) -> dict[str, Any]:
+        template_data: dict[str, Any] = {}
+        for var_name, var_value in template_values.items():
+            var_type = template_schema.get(var_name)
+
+            if var_type == "list":
+                template_data[var_name] = [str(v).strip() for v in var_value.split(",")]
+            else:
+                template_data[var_name] = str(var_value)
+        return template_data

src/codegen/git/models/codemod_context.py

@@ -0,0 +1,12 @@
+from pydantic import BaseModel
+
+
+class GithubNamedUserContext(BaseModel):
+    """Represents a GitHub user parsed from a webhook payload"""
+
+    login: str
+    email: str | None = None
+
+    @classmethod
+    def from_payload(cls, payload: dict) -> "GithubNamedUserContext":
+        return cls(login=payload.get("login"), email=payload.get("email"))

src/codegen/git/models/github_named_user_context.py

@@ -0,0 +1,13 @@
+from pydantic import BaseModel
+
+from codegen.utils.decorators.docs import apidoc
+
+
+@apidoc
+class PROptions(BaseModel):
+    """Options for generating a PR."""
+
+    title: str | None = None
+    body: str | None = None
+    labels: list[str] | None = None  # TODO: not used until we add labels to GithubPullRequestModel
+    force_push_head_branch: bool | None = None

src/codegen/git/models/pr_options.py

@@ -0,0 +1,12 @@
+from pydantic import BaseModel
+
+
+class PRPartContext(BaseModel):
+    """Represents a GitHub pull request part parsed from a webhook payload"""
+
+    ref: str
+    sha: str
+
+    @classmethod
+    def from_payload(cls, payload: dict) -> "PRPartContext":
+        return cls(ref=payload.get("ref"), sha=payload.get("sha"))

src/codegen/git/models/pr_part_context.py

@@ -0,0 +1,52 @@
+from pydantic import BaseModel
+
+from codegen.git.models.github_named_user_context import GithubNamedUserContext
+from codegen.git.models.pr_part_context import PRPartContext
+from codegen.git.schemas.github import GithubType
+
+
+class PullRequestContext(BaseModel):
+    """Represents a GitHub pull request"""
+
+    id: int
+    url: str
+    html_url: str
+    number: int
+    state: str
+    title: str
+    user: GithubNamedUserContext
+    body: str
+    draft: bool
+    head: PRPartContext
+    base: PRPartContext
+    merged: bool | None
+    merged_by: dict | None
+    additions: int | None
+    deletions: int | None
+    changed_files: int | None
+    github_type: GithubType | None = None
+    webhook_data: dict | None = None
+
+    @classmethod
+    def from_payload(cls, webhook_payload: dict) -> "PullRequestContext":
+        webhook_data = webhook_payload.get("pull_request", {})
+        return cls(
+            id=webhook_data.get("id"),
+            url=webhook_data.get("url"),
+            html_url=webhook_data.get("html_url"),
+            number=webhook_data.get("number"),
+            state=webhook_data.get("state"),
+            title=webhook_data.get("title"),
+            user=GithubNamedUserContext.from_payload(webhook_data.get("user", {})),
+            body=webhook_data.get("body"),
+            draft=webhook_data.get("draft"),
+            head=PRPartContext.from_payload(webhook_data.get("head", {})),
+            base=PRPartContext.from_payload(webhook_data.get("base", {})),
+            merged=webhook_data.get("merged"),
+            merged_by=webhook_data.get("merged_by", {}),
+            additions=webhook_data.get("additions"),
+            deletions=webhook_data.get("deletions"),
+            changed_files=webhook_data.get("changed_files"),
+            github_type=GithubType.from_url(webhook_data.get("html_url")),
+            webhook_data=webhook_data,
+        )

src/codegen/git/models/pull_request_context.py

@@ -12,10 +12,10 @@
 import plotly
 """.strip()
 CODEGEN_IMPORTS = """
-from app.codemod.compilation.models.context import CodemodContext
-from app.codemod.compilation.models.github_named_user_context import GithubNamedUserContext
-from app.codemod.compilation.models.pr_part_context import PRPartContext
-from app.codemod.compilation.models.pull_request_context import PullRequestContext
+from codegen.git.models.codemod_context import CodemodContext
+from codegen.git.models.github_named_user_context import GithubNamedUserContext
+from codegen.git.models.pr_part_context import PRPartContext
+from codegen.git.models.pull_request_context import PullRequestContext
 """
 # TODO: these should also be made public (i.e. included in the docs site)
 GS_PRIVATE_IMPORTS = """