ai-cr 2.0.0.dev1__py3-none-any.whl → 2.0.1__py3-none-any.whl

gito/report_struct.py

@@ -1,132 +1,134 @@
-import json
-import logging
-from dataclasses import dataclass, field, asdict
-from datetime import datetime
-from enum import StrEnum
-from pathlib import Path
-
-import microcore as mc
-from colorama import Fore, Style, Back
-from microcore.utils import file_link
-import textwrap
-
-from .constants import JSON_REPORT_FILE_NAME
-from .project_config import ProjectConfig
-from .utils import syntax_hint, block_wrap_lr, max_line_len
-
-
-@dataclass
-class Issue:
-    @dataclass
-    class AffectedCode:
-        start_line: int = field()
-        end_line: int | None = field(default=None)
-        file: str = field(default="")
-        proposal: str = field(default="")
-        affected_code: str = field(default="")
-
-        @property
-        def syntax_hint(self) -> str:
-            return syntax_hint(self.file)
-
-    id: str = field()
-    title: str = field()
-    details: str = field(default="")
-    severity: int | None = field(default=None)
-    confidence: int | None = field(default=None)
-    tags: list[str] = field(default_factory=list)
-    file: str = field(default="")
-    affected_lines: list[AffectedCode] = field(default_factory=list)
-
-    def __post_init__(self):
-        self.affected_lines = [
-            Issue.AffectedCode(**dict(file=self.file) | i)
-            for i in self.affected_lines
-        ]
-
-    def github_code_link(self, github_env: dict) -> str:
-        url = (
-            f"https://github.com/{github_env['github_repo']}"
-            f"/blob/{github_env['github_pr_sha_or_branch']}"
-            f"/{self.file}"
-        )
-        if self.affected_lines:
-            url += f"#L{self.affected_lines[0].start_line}"
-            if self.affected_lines[0].end_line:
-                url += f"-L{self.affected_lines[0].end_line}"
-        return url
-
-
-@dataclass
-class Report:
-    class Format(StrEnum):
-        MARKDOWN = "md"
-        CLI = "cli"
-
-    issues: dict = field(default_factory=dict)
-    summary: str = field(default="")
-    number_of_processed_files: int = field(default=0)
-    total_issues: int = field(init=False)
-    created_at: str = field(default_factory=lambda: datetime.now().strftime("%Y-%m-%d %H:%M:%S"))
-    model: str = field(default_factory=lambda: mc.config().MODEL)
-
-    @property
-    def plain_issues(self):
-        return [
-            issue
-            for file, issues in self.issues.items()
-            for issue in issues
-        ]
-
-    def __post_init__(self):
-        issue_id: int = 0
-        for file in self.issues.keys():
-            self.issues[file] = [
-                Issue(
-                    **{
-                        "id": (issue_id := issue_id + 1),
-                        "file": file,
-                    } | issue
-                )
-                for issue in self.issues[file]
-            ]
-        self.total_issues = issue_id
-
-    def save(self, file_name: str = ""):
-        file_name = file_name or JSON_REPORT_FILE_NAME
-        with open(file_name, "w") as f:
-            json.dump(asdict(self), f, indent=4)
-        logging.info(f"Report saved to {mc.utils.file_link(file_name)}")
-
-    @staticmethod
-    def load(file_name: str | Path = ""):
-        with open(file_name or JSON_REPORT_FILE_NAME, "r") as f:
-            data = json.load(f)
-        data.pop("total_issues", None)
-        return Report(**data)
-
-    def render(
-        self,
-        config: ProjectConfig = None,
-        report_format: Format = Format.MARKDOWN,
-    ) -> str:
-        config = config or ProjectConfig.load()
-        template = getattr(config, f"report_template_{report_format}")
-        return mc.prompt(
-            template,
-            report=self,
-            ui=mc.ui,
-            Fore=Fore,
-            Style=Style,
-            Back=Back,
-            file_link=file_link,
-            textwrap=textwrap,
-            block_wrap_lr=block_wrap_lr,
-            max_line_len=max_line_len,
-            **config.prompt_vars
-        )
-
-    def to_cli(self, report_format=Format.CLI):
-        output = self.render(report_format=report_format)
-        print("")
-        print(output)
+import json
+import logging
+from dataclasses import dataclass, field, asdict
+from datetime import datetime
+from enum import StrEnum
+from pathlib import Path
+
+import microcore as mc
+from colorama import Fore, Style, Back
+from microcore.utils import file_link
+import textwrap
+
+from .constants import JSON_REPORT_FILE_NAME, HTML_TEXT_ICON
+from .project_config import ProjectConfig
+from .utils import syntax_hint, block_wrap_lr, max_line_len
+
+
+@dataclass
+class Issue:
+    @dataclass
+    class AffectedCode:
+        start_line: int = field()
+        end_line: int | None = field(default=None)
+        file: str = field(default="")
+        proposal: str = field(default="")
+        affected_code: str = field(default="")
+
+        @property
+        def syntax_hint(self) -> str:
+            return syntax_hint(self.file)
+
+    id: str = field()
+    title: str = field()
+    details: str = field(default="")
+    severity: int | None = field(default=None)
+    confidence: int | None = field(default=None)
+    tags: list[str] = field(default_factory=list)
+    file: str = field(default="")
+    affected_lines: list[AffectedCode] = field(default_factory=list)
+
+    def __post_init__(self):
+        self.affected_lines = [
+            Issue.AffectedCode(**dict(file=self.file) | i)
+            for i in self.affected_lines
+        ]
+
+    def github_code_link(self, github_env: dict) -> str:
+        url = (
+            f"https://github.com/{github_env['github_repo']}"
+            f"/blob/{github_env['github_pr_sha_or_branch']}"
+            f"/{self.file}"
+        )
+        if self.affected_lines:
+            url += f"#L{self.affected_lines[0].start_line}"
+            if self.affected_lines[0].end_line:
+                url += f"-L{self.affected_lines[0].end_line}"
+        return url
+
+
+@dataclass
+class Report:
+    class Format(StrEnum):
+        MARKDOWN = "md"
+        CLI = "cli"
+
+    issues: dict[str, list[Issue]] = field(default_factory=dict)
+    summary: str = field(default="")
+    number_of_processed_files: int = field(default=0)
+    total_issues: int = field(init=False)
+    created_at: str = field(default_factory=lambda: datetime.now().strftime("%Y-%m-%d %H:%M:%S"))
+    model: str = field(default_factory=lambda: mc.config().MODEL)
+    pipeline_out: dict = field(default_factory=dict)
+
+    @property
+    def plain_issues(self):
+        return [
+            issue
+            for file, issues in self.issues.items()
+            for issue in issues
+        ]
+
+    def __post_init__(self):
+        issue_id: int = 0
+        for file in self.issues.keys():
+            self.issues[file] = [
+                Issue(
+                    **{
+                        "id": (issue_id := issue_id + 1),
+                        "file": file,
+                    } | issue
+                )
+                for issue in self.issues[file]
+            ]
+        self.total_issues = issue_id
+
+    def save(self, file_name: str = ""):
+        file_name = file_name or JSON_REPORT_FILE_NAME
+        with open(file_name, "w") as f:
+            json.dump(asdict(self), f, indent=4)
+        logging.info(f"Report saved to {mc.utils.file_link(file_name)}")
+
+    @staticmethod
+    def load(file_name: str | Path = ""):
+        with open(file_name or JSON_REPORT_FILE_NAME, "r") as f:
+            data = json.load(f)
+        data.pop("total_issues", None)
+        return Report(**data)
+
+    def render(
+        self,
+        config: ProjectConfig = None,
+        report_format: Format = Format.MARKDOWN,
+    ) -> str:
+        config = config or ProjectConfig.load()
+        template = getattr(config, f"report_template_{report_format}")
+        return mc.prompt(
+            template,
+            report=self,
+            ui=mc.ui,
+            Fore=Fore,
+            Style=Style,
+            Back=Back,
+            file_link=file_link,
+            textwrap=textwrap,
+            block_wrap_lr=block_wrap_lr,
+            max_line_len=max_line_len,
+            HTML_TEXT_ICON=HTML_TEXT_ICON,
+            **config.prompt_vars
+        )
+
+    def to_cli(self, report_format=Format.CLI):
+        output = self.render(report_format=report_format)
+        print("")
+        print(output)

gito/utils.py

@@ -1,132 +1,226 @@
-import sys
-import os
-from pathlib import Path
-import typer
-
-
-_EXT_TO_HINT: dict[str, str] = {
-    # scripting & languages
-    ".py": "python",
-    ".js": "javascript",
-    ".ts": "typescript",
-    ".java": "java",
-    ".c": "c",
-    ".cpp": "cpp",
-    ".cc": "cpp",
-    ".cxx": "cpp",
-    ".h": "cpp",
-    ".hpp": "cpp",
-    ".cs": "csharp",
-    ".rb": "ruby",
-    ".go": "go",
-    ".rs": "rust",
-    ".swift": "swift",
-    ".kt": "kotlin",
-    ".scala": "scala",
-    ".dart": "dart",
-    ".php": "php",
-    ".pl": "perl",
-    ".pm": "perl",
-    ".lua": "lua",
-    # web & markup
-    ".html": "html",
-    ".htm": "html",
-    ".css": "css",
-    ".scss": "scss",
-    ".less": "less",
-    ".json": "json",
-    ".xml": "xml",
-    ".yaml": "yaml",
-    ".yml": "yaml",
-    ".toml": "toml",
-    ".ini": "ini",
-    ".csv": "csv",
-    ".md": "markdown",
-    ".rst": "rest",
-    # shell & config
-    ".sh": "bash",
-    ".bash": "bash",
-    ".zsh": "bash",
-    ".fish": "bash",
-    ".ps1": "powershell",
-    ".dockerfile": "dockerfile",
-    # build & CI
-    ".makefile": "makefile",
-    ".mk": "makefile",
-    "CMakeLists.txt": "cmake",
-    "Dockerfile": "dockerfile",
-    ".gradle": "groovy",
-    ".travis.yml": "yaml",
-    # data & queries
-    ".sql": "sql",
-    ".graphql": "graphql",
-    ".proto": "protobuf",
-    ".yara": "yara",
-}
-
-
-def syntax_hint(file_path: str | Path) -> str:
-    """
-    Returns a syntax highlighting hint based on the file's extension or name.
-
-    This can be used to annotate code blocks for rendering with syntax highlighting,
-    e.g., using Markdown-style code blocks: ```<syntax_hint>\n<code>\n```.
-
-    Args:
-      file_path (str | Path): Path to the file.
-
-    Returns:
-      str: A syntax identifier suitable for code highlighting (e.g., 'python', 'json').
-    """
-    p = Path(file_path)
-    ext = p.suffix.lower()
-    if not ext:
-        name = p.name.lower()
-        if name == "dockerfile":
-            return "dockerfile"
-        return ""
-    return _EXT_TO_HINT.get(ext, ext.lstrip("."))
-
-
-def is_running_in_github_action():
-    return os.getenv("GITHUB_ACTIONS") == "true"
-
-
-def no_subcommand(app: typer.Typer) -> bool:
-    """
-    Checks if the current script is being invoked as a command in a target Typer application.
-    """
-    return not (
-        (first_arg := next((a for a in sys.argv[1:] if not a.startswith('-')), None))
-        and first_arg in (
-            cmd.name or cmd.callback.__name__.replace('_', '-')
-            for cmd in app.registered_commands
-        )
-        or '--help' in sys.argv
-    )
-
-
-def parse_refs_pair(refs: str) -> tuple[str | None, str | None]:
-    SEPARATOR = '..'
-    if not refs:
-        return None, None
-    if SEPARATOR not in refs:
-        return refs, None
-    what, against = refs.split(SEPARATOR, 1)
-    return what or None, against or None
-
-
-def max_line_len(text: str) -> int:
-    return max((len(line) for line in text.splitlines()), default=0)
-
-
-def block_wrap_lr(text: str, left: str = "", right: str = "", max_rwrap: int = 60) -> str:
-    ml = max_line_len(text)
-    lines = text.splitlines()
-    wrapped_lines = []
-    for line in lines:
-        ln = left+line
-        if ml <= max_rwrap:
-            ln += ' ' * (ml - len(line)) + right
-        wrapped_lines.append(ln)
-    return "\n".join(wrapped_lines)
+import re
+import sys
+import os
+from pathlib import Path
+
+import typer
+import git
+from git import Repo
+from microcore import ui
+
+
+_EXT_TO_HINT: dict[str, str] = {
+    # scripting & languages
+    ".py": "python",
+    ".js": "javascript",
+    ".ts": "typescript",
+    ".java": "java",
+    ".c": "c",
+    ".cpp": "cpp",
+    ".cc": "cpp",
+    ".cxx": "cpp",
+    ".h": "cpp",
+    ".hpp": "cpp",
+    ".cs": "csharp",
+    ".rb": "ruby",
+    ".go": "go",
+    ".rs": "rust",
+    ".swift": "swift",
+    ".kt": "kotlin",
+    ".scala": "scala",
+    ".dart": "dart",
+    ".php": "php",
+    ".pl": "perl",
+    ".pm": "perl",
+    ".lua": "lua",
+    # web & markup
+    ".html": "html",
+    ".htm": "html",
+    ".css": "css",
+    ".scss": "scss",
+    ".less": "less",
+    ".json": "json",
+    ".xml": "xml",
+    ".yaml": "yaml",
+    ".yml": "yaml",
+    ".toml": "toml",
+    ".ini": "ini",
+    ".csv": "csv",
+    ".md": "markdown",
+    ".rst": "rest",
+    # shell & config
+    ".sh": "bash",
+    ".bash": "bash",
+    ".zsh": "bash",
+    ".fish": "bash",
+    ".ps1": "powershell",
+    ".dockerfile": "dockerfile",
+    # build & CI
+    ".makefile": "makefile",
+    ".mk": "makefile",
+    "CMakeLists.txt": "cmake",
+    "Dockerfile": "dockerfile",
+    ".gradle": "groovy",
+    ".travis.yml": "yaml",
+    # data & queries
+    ".sql": "sql",
+    ".graphql": "graphql",
+    ".proto": "protobuf",
+    ".yara": "yara",
+}
+
+
+def syntax_hint(file_path: str | Path) -> str:
+    """
+    Returns a syntax highlighting hint based on the file's extension or name.
+
+    This can be used to annotate code blocks for rendering with syntax highlighting,
+    e.g., using Markdown-style code blocks: ```<syntax_hint>\n<code>\n```.
+
+    Args:
+      file_path (str | Path): Path to the file.
+
+    Returns:
+      str: A syntax identifier suitable for code highlighting (e.g., 'python', 'json').
+    """
+    p = Path(file_path)
+    ext = p.suffix.lower()
+    if not ext:
+        name = p.name.lower()
+        if name == "dockerfile":
+            return "dockerfile"
+        return ""
+    return _EXT_TO_HINT.get(ext, ext.lstrip("."))
+
+
+def is_running_in_github_action():
+    return os.getenv("GITHUB_ACTIONS") == "true"
+
+
+def no_subcommand(app: typer.Typer) -> bool:
+    """
+    Checks if the current script is being invoked as a command in a target Typer application.
+    """
+    return not (
+        (first_arg := next((a for a in sys.argv[1:] if not a.startswith('-')), None))
+        and first_arg in (
+            cmd.name or cmd.callback.__name__.replace('_', '-')
+            for cmd in app.registered_commands
+        )
+        or '--help' in sys.argv
+    )
+
+
+def parse_refs_pair(refs: str) -> tuple[str | None, str | None]:
+    SEPARATOR = '..'
+    if not refs:
+        return None, None
+    if SEPARATOR not in refs:
+        return refs, None
+    what, against = refs.split(SEPARATOR, 1)
+    return what or None, against or None
+
+
+def max_line_len(text: str) -> int:
+    return max((len(line) for line in text.splitlines()), default=0)
+
+
+def block_wrap_lr(
+    text: str,
+    left: str = "",
+    right: str = "",
+    max_rwrap: int = 60,
+    min_wrap: int = 0,
+) -> str:
+    ml = max(max_line_len(text), min_wrap)
+    lines = text.splitlines()
+    wrapped_lines = []
+    for line in lines:
+        ln = left+line
+        if ml <= max_rwrap:
+            ln += ' ' * (ml - len(line)) + right
+        wrapped_lines.append(ln)
+    return "\n".join(wrapped_lines)
+
+
+def extract_gh_owner_repo(repo: git.Repo) -> tuple[str, str]:
+    """
+    Extracts the GitHub owner and repository name.
+
+    Returns:
+        tuple[str, str]: A tuple containing the owner and repository name.
+    """
+    remote_url = repo.remotes.origin.url
+    if remote_url.startswith('git@github.com:'):
+        # SSH format: git@github.com:owner/repo.git
+        repo_path = remote_url.split(':')[1].replace('.git', '')
+    elif remote_url.startswith('https://github.com/'):
+        # HTTPS format: https://github.com/owner/repo.git
+        repo_path = remote_url.replace('https://github.com/', '').replace('.git', '')
+    else:
+        raise ValueError("Unsupported remote URL format")
+    owner, repo_name = repo_path.split('/')
+    return owner, repo_name
+
+
+def detect_github_env() -> dict:
+    """
+    Try to detect GitHub repository/PR info from environment variables (for GitHub Actions).
+    Returns a dict with github_repo, github_pr_sha, github_pr_number, github_ref, etc.
+    """
+    repo = os.environ.get("GITHUB_REPOSITORY", "")
+    pr_sha = os.environ.get("GITHUB_SHA", "")
+    pr_number = os.environ.get("GITHUB_REF", "")
+    branch = ""
+    ref = os.environ.get("GITHUB_REF", "")
+    # Try to resolve PR head SHA if available.
+    # On PRs, GITHUB_HEAD_REF/BASE_REF contain branch names.
+    if "GITHUB_HEAD_REF" in os.environ:
+        branch = os.environ["GITHUB_HEAD_REF"]
+    elif ref.startswith("refs/heads/"):
+        branch = ref[len("refs/heads/"):]
+    elif ref.startswith("refs/pull/"):
+        # for pull_request events
+        branch = ref
+
+    d = {
+        "github_repo": repo,
+        "github_pr_sha": pr_sha,
+        "github_pr_number": pr_number,
+        "github_branch": branch,
+        "github_ref": ref,
+    }
+    # Fallback for local usage: try to get from git
+    if not repo:
+        git_repo = None
+        try:
+            git_repo = Repo(".", search_parent_directories=True)
+            origin = git_repo.remotes.origin.url
+            # e.g. git@github.com:Nayjest/ai-code-review.git -> Nayjest/ai-code-review
+            match = re.search(r"[:/]([\w\-]+)/([\w\-\.]+?)(\.git)?$", origin)
+            if match:
+                d["github_repo"] = f"{match.group(1)}/{match.group(2)}"
+            d["github_pr_sha"] = git_repo.head.commit.hexsha
+            d["github_branch"] = (
+                git_repo.active_branch.name if hasattr(git_repo, "active_branch") else ""
+            )
+        except Exception:
+            pass
+        finally:
+            if git_repo:
+                try:
+                    git_repo.close()
+                except Exception:
+                    pass
+    # If branch is not a commit SHA, prefer branch for links
+    if d["github_branch"]:
+        d["github_pr_sha_or_branch"] = d["github_branch"]
+    elif d["github_pr_sha"]:
+        d["github_pr_sha_or_branch"] = d["github_pr_sha"]
+    else:
+        d["github_pr_sha_or_branch"] = "main"
+    return d
+
+
+def stream_to_cli(text):
+    print(ui.blue(text), end='')

ai_cr-2.0.0.dev1.dist-info/RECORD

@@ -1,18 +0,0 @@
-gito/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-gito/__main__.py,sha256=EClCwCzb6h6YBpt0hrnG4h0mlNhNePyg_xBNNSVm1os,65
-gito/bootstrap.py,sha256=ETKioiDc2Npc7znd8HJxA5-twd7sZMPCufIGwXFQSbY,2403
-gito/cli.py,sha256=JASFo1NGUqw9k6-ZnU5KXFuexxUc6EKJNOYs6uPPyjs,7943
-gito/commands/__init__.py,sha256=NKUUDskR6taZGbHk5qR9msDS22aHpdzAZlUDxzRdMYA,56
-gito/commands/fix.py,sha256=THEu7vdsK2--hOkz6rha4zESVfRVIDc88mHNNlYfM7Q,4421
-gito/commands/repl.py,sha256=waN7FJBl98gWmDwZWMa8x157iWbPHIDPJEKmTdzWQ70,396
-gito/config.toml,sha256=zd6s_UJs7Du1W7ulR2iGu7Grcs4l4qpq--VAS67pj4w,15003
-gito/constants.py,sha256=DNqh3hOxeLM_KscUH4uFdAc9295o1hiXtwoEw2KxLiU,392
-gito/core.py,sha256=a98HJyZMPswwbmfzJBJvnPZMWYJU7njs3O8B4Z_UGmc,7848
-gito/project_config.py,sha256=tbN1mf2FqpUQ9y9hGsGfM2CRX7AmD-ZT0QkXVUdHi4U,4279
-gito/report_struct.py,sha256=vC9Rs6OZvUspn0Ho4s9gVf0doHFt_dNbdqjdubqoIrY,4162
-gito/utils.py,sha256=Oh1hyg-YpUZgOlWNJdRTDLzZV-YyjMkgLsC5MiN1uow,3431
-ai_cr-2.0.0.dev1.dist-info/entry_points.txt,sha256=Ua1DxkhJJ8TZuLgnH-IlWCkrre_0S0dq_GtYRaYupWk,38
-ai_cr-2.0.0.dev1.dist-info/LICENSE,sha256=XATf3zv-CppUSJqI18KLhwnPEomUXEl5WbBzFyb9OSU,1096
-ai_cr-2.0.0.dev1.dist-info/METADATA,sha256=JTgfajLAdhqrWqNw40BfNHh-B3jQRbgJeoiorkuTAaw,7676
-ai_cr-2.0.0.dev1.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
-ai_cr-2.0.0.dev1.dist-info/RECORD,,