ai-cr 2.0.2__py3-none-any.whl → 3.0.0__py3-none-any.whl

gito/gh_api.py

@@ -1,6 +1,12 @@
+import os
 import logging
 
 import requests
+from fastcore.basics import AttrDict  # objects returned by ghapi
+
+
+def resolve_gh_token(token_or_none: str | None = None) -> str | None:
+    return token_or_none or os.getenv("GITHUB_TOKEN", None) or os.getenv("GH_TOKEN", None)
 
 
 def post_gh_comment(
@@ -33,3 +39,47 @@ def post_gh_comment(
     else:
         logging.error(f"Failed to post comment: {resp.status_code} {resp.reason}\n{resp.text}")
         return False
+
+
+def hide_gh_comment(
+    comment: dict | str,
+    token: str = None,
+    reason: str = "OUTDATED"
+) -> bool:
+    """
+    Hide a GitHub comment using GraphQL API with specified reason.
+    Args:
+        comment (dict | str):
+            The comment to hide,
+            either as a object returned from ghapi or a string node ID.
+            note: comment.id is not the same as node_id.
+        token (str): GitHub personal access token with permissions to minimize comments.
+        reason (str): The reason for hiding the comment, e.g., "OUTDATED".
+    """
+    comment_node_id = comment.node_id if isinstance(comment, AttrDict) else comment
+    token = resolve_gh_token(token)
+    mutation = """
+    mutation($commentId: ID!, $reason: ReportedContentClassifiers!) {
+        minimizeComment(input: {subjectId: $commentId, classifier: $reason}) {
+            minimizedComment { isMinimized }
+        }
+    }"""
+
+    response = requests.post(
+        "https://api.github.com/graphql",
+        headers={"Authorization": f"Bearer {token}"},
+        json={
+            "query": mutation,
+            "variables": {"commentId": comment_node_id, "reason": reason}
+        }
+    )
+    success = (
+        response.status_code == 200
+        and response.json().get("data", {}).get("minimizeComment") is not None
+    )
+    if not success:
+        logging.error(
+            f"Failed to hide comment {comment_node_id}: "
+            f"{response.status_code} {response.reason}\n{response.text}"
+        )
+    return success

gito/issue_trackers.py

@@ -15,9 +15,10 @@ class IssueTrackerIssue:
 
 
 def extract_issue_key(branch_name: str, min_len=2, max_len=10) -> str | None:
-    pattern = fr"\b[A-Z][A-Z0-9]{{{min_len - 1},{max_len - 1}}}-\d+\b"
+    boundary = r'\b|_|-|/|\\'
+    pattern = fr"(?:{boundary})([A-Z][A-Z0-9]{{{min_len - 1},{max_len - 1}}}-\d+)(?:{boundary})"
     match = re.search(pattern, branch_name)
-    return match.group(0) if match else None
+    return match.group(1) if match else None
 
 
 def get_branch(repo: git.Repo):

gito/pipeline.py

@@ -2,10 +2,12 @@ import logging
 from enum import StrEnum
 from dataclasses import dataclass, field
 
-from gito.utils import is_running_in_github_action
 from microcore import ui
 from microcore.utils import resolve_callable
 
+from .context import Context
+from .utils import is_running_in_github_action
+
 
 class PipelineEnv(StrEnum):
     LOCAL = "local"
@@ -42,7 +44,7 @@ class PipelineStep:
 
 @dataclass
 class Pipeline:
-    ctx: dict = field(default_factory=dict)
+    ctx: Context = field()
     steps: dict[str, PipelineStep] = field(default_factory=dict)
     verbose: bool = False
 
@@ -55,15 +57,14 @@ class Pipeline:
     def run(self, *args, **kwargs):
         cur_env = PipelineEnv.current()
         logging.info("Running pipeline... [env: %s]", ui.yellow(cur_env))
-        self.ctx["pipeline_out"] = self.ctx.get("pipeline_out", {})
         for step_name, step in self.enabled_steps.items():
             if cur_env in step.envs:
                 logging.info(f"Running pipeline step: {step_name}")
                 try:
-                    step_output = step.run(*args, **kwargs, **self.ctx)
+                    step_output = step.run(*args, **kwargs, **vars(self.ctx))
                     if isinstance(step_output, dict):
-                        self.ctx["pipeline_out"].update(step_output)
-                    self.ctx["pipeline_out"][step_name] = step_output
+                        self.ctx.pipeline_out.update(step_output)
+                    self.ctx.pipeline_out[step_name] = step_output
                     if self.verbose and step_output:
                         logging.info(
                             f"Pipeline step {step_name} output: {repr(step_output)}"
@@ -79,4 +80,4 @@ class Pipeline:
                     f"Skipping pipeline step: {step_name}"
                     f" [env: {ui.yellow(cur_env)} not in {step.envs}]"
                 )
-        return self.ctx["pipeline_out"]
+        return self.ctx.pipeline_out

gito/pipeline_steps/jira.py

@@ -2,7 +2,7 @@ import logging
 import os
 
 import git
-from jira import JIRA
+from jira import JIRA, JIRAError
 
 from gito.issue_trackers import IssueTrackerIssue, resolve_issue_key
 
@@ -16,6 +16,11 @@ def fetch_issue(issue_key, jira_url, username, api_token) -> IssueTrackerIssue |
             description=issue.fields.description or "",
             url=f"{jira_url.rstrip('/')}/browse/{issue_key}"
         )
+    except JIRAError as e:
+        logging.error(
+            f"Failed to fetch Jira issue {issue_key}: code {e.status_code} :: {e.text}"
+        )
+        return None
     except Exception as e:
         logging.error(f"Failed to fetch Jira issue {issue_key}: {e}")
         return None

gito/project_config.py

@@ -33,6 +33,11 @@ class ProjectConfig:
     when referenced in code review comments.
     """
     pipeline_steps: dict[str, dict | PipelineStep] = field(default_factory=dict)
+    collapse_previous_code_review_comments: bool = field(default=True)
+    """
+    If True, previously added code review comments in the pull request
+    will be collapsed automatically when a new comment is added.
+    """
 
     def __post_init__(self):
         self.pipeline_steps = {
@@ -61,10 +66,16 @@ class ProjectConfig:
             logging.info(
                 f"Loading project-specific configuration from {mc.utils.file_link(config_path)}...")
             default_prompt_vars = config["prompt_vars"]
+            default_pipeline_steps = config["pipeline_steps"]
             with open(config_path, "rb") as f:
                 config.update(tomllib.load(f))
             # overriding prompt_vars config section will not empty default values
             config["prompt_vars"] = default_prompt_vars | config["prompt_vars"]
+            # merge individual pipeline steps
+            for k, v in config["pipeline_steps"].items():
+                config["pipeline_steps"][k] = default_pipeline_steps.get(k, {}) | v
+            # merge pipeline steps dict
+            config["pipeline_steps"] = default_pipeline_steps | config["pipeline_steps"]
         else:
             logging.info(
                 f"No project config found at {ui.blue(config_path)}, using defaults"

gito/report_struct.py

@@ -10,9 +10,9 @@ 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 .constants import JSON_REPORT_FILE_NAME, HTML_TEXT_ICON, HTML_CR_COMMENT_MARKER
 from .project_config import ProjectConfig
-from .utils import syntax_hint, block_wrap_lr, max_line_len
+from .utils import syntax_hint, block_wrap_lr, max_line_len, remove_html_comments
 
 
 @dataclass
@@ -125,6 +125,8 @@ class Report:
             block_wrap_lr=block_wrap_lr,
             max_line_len=max_line_len,
             HTML_TEXT_ICON=HTML_TEXT_ICON,
+            HTML_CR_COMMENT_MARKER=HTML_CR_COMMENT_MARKER,
+            remove_html_comments=remove_html_comments,
             **config.prompt_vars
         )
 

gito/tpl/github_workflows/components/env-vars.j2

@@ -0,0 +1,10 @@
+
+        LLM_API_TYPE: {{ api_type }}
+        LLM_API_KEY: {{ "${{ secrets." + secret_name + " }}" }}
+        MODEL: {{ model }}
+        {% raw -%}
+        JIRA_TOKEN: ${{ secrets.JIRA_TOKEN }}
+        JIRA_URL: ${{ secrets.JIRA_URL }}
+        JIRA_USER: ${{ secrets.JIRA_USER }}
+        LINEAR_API_KEY: ${{ secrets.LINEAR_API_KEY }}
+        {%- endraw %}

gito/tpl/github_workflows/components/installs.j2

@@ -0,0 +1,23 @@
+
+
+    - name: Set up Python
+      uses: actions/setup-python@v5
+      with: { python-version: "3.13" }
+
+    - name: Fetch Latest Gito Version
+      id: gito-version
+      run: pip index versions gito.bot 2>/dev/null | head -1 | sed -n 's/.* (\([^)]*\)).*/version=\1/p' >> $GITHUB_OUTPUT
+
+    {% raw -%}
+    - uses: actions/cache@v4
+      id: cache
+      with:
+        path: |
+          ${{ env.pythonLocation }}/lib/python3.13/site-packages
+          ${{ env.pythonLocation }}/bin
+        key: gito_v${{ steps.gito-version.outputs.version }}
+    {%- endraw %}
+
+    - name: Install Gito
+      if: steps.cache.outputs.cache-hit != 'true'
+      run: pip install gito.bot~={{ major }}.{{ minor }}

gito/tpl/github_workflows/gito-code-review.yml.j2

@@ -0,0 +1,33 @@
+name: "Gito: AI Code Reviewer"
+on:
+  pull_request:
+    types: [opened, synchronize, reopened]
+  workflow_dispatch:
+    inputs:
+      pr_number:
+        description: "Pull Request number"
+        required: true
+jobs:
+  review:
+    runs-on: ubuntu-latest
+    permissions: { contents: read, pull-requests: write } # 'write' for leaving the summary comment
+    steps:
+    - uses: actions/checkout@v4
+      with: { fetch-depth: 0 }
+
+    {%- include("github_workflows/components/installs.j2") %}
+
+    - name: Run AI code review
+      env:
+        {%- include("github_workflows/components/env-vars.j2") %}
+        PR_NUMBER_FROM_WORKFLOW_DISPATCH: {% raw %}${{ github.event.inputs.pr_number }}{% endraw %}
+      run: |{% raw %}
+        gito --verbose review ${{ github.event_name == 'pull_request' && github.event.pull_request.base.ref && format(' --against="origin/{0}"', github.event.pull_request.base.ref) || '' }}
+        gito github-comment --token ${{ secrets.GITHUB_TOKEN }}{% endraw %}
+
+    - uses: actions/upload-artifact@v4
+      with:
+        name: gito-code-review-results
+        path: |
+          code-review-report.md
+          code-review-report.json

gito/tpl/github_workflows/gito-react-to-comments.yml.j2

@@ -0,0 +1,70 @@
+name: "Gito: React to GitHub comment"
+
+on:
+  issue_comment:
+    types: [created]
+
+permissions:
+  contents: write  # to make PR
+  issues: write
+  pull-requests: write
+  # read: to download the code review artifact
+  # write: to trigger other actions
+  actions: write
+
+jobs:
+  process-comment:
+    if: |
+      github.event.issue.pull_request &&
+      (
+        github.event.comment.author_association == 'OWNER' ||
+        github.event.comment.author_association == 'MEMBER' ||
+        github.event.comment.author_association == 'COLLABORATOR'
+      ) &&
+      (
+        startsWith(github.event.comment.body, '/') ||
+        startsWith(github.event.comment.body, 'gito') ||
+        startsWith(github.event.comment.body, 'ai') ||
+        startsWith(github.event.comment.body, 'bot') ||
+        contains(github.event.comment.body, '@gito') ||
+        contains(github.event.comment.body, '@ai') ||
+        contains(github.event.comment.body, '@bot')
+      )
+    runs-on: ubuntu-latest
+
+    steps:
+    - name: Get PR details
+      id: pr
+      uses: actions/github-script@v7
+      with:
+        script: |
+          const pr = await github.rest.pulls.get({
+            owner: context.repo.owner,
+            repo: context.repo.repo,
+            pull_number: context.issue.number
+          });
+          return {
+            head_ref: pr.data.head.ref,
+            head_sha: pr.data.head.sha,
+            base_ref: pr.data.base.ref
+          };
+
+    - name: Checkout repository
+      uses: actions/checkout@v4
+      with:
+        {% raw -%}
+        repository: ${{ github.repository }}
+        token: ${{ secrets.GITHUB_TOKEN }}
+        ref: ${{ fromJson(steps.pr.outputs.result).head_ref }}
+        fetch-depth: 0
+        {%- endraw %}
+
+      {%- include("github_workflows/components/installs.j2") %}
+
+    - name: Run Gito react
+      env:
+        # LLM config is needed only if answer_github_comments = true in .gito/config.toml
+        # Otherwise, use LLM_API_TYPE: none
+        {%- include("github_workflows/components/env-vars.j2") %}
+      run: |
+        {% raw %}gito react-to-comment ${{ github.event.comment.id }} --token ${{ secrets.GITHUB_TOKEN }}{%- endraw %}

gito/tpl/release_notes.j2

@@ -0,0 +1,24 @@
+{{ self_id }}
+----TASK----
+Write release notes for public documentation.
+Summarize the following changes, focusing on what is new, improved, or fixed for the end user.
+Do not include internal or technical details.
+Structure release notes using clear sections: Added, Changed, Fixed.
+Avoid internal technical jargon or developer-specific details.
+
+
+----RELATED CODEBASE CHANGES----
+{% for part in diff %}{{ part }}\n{% endfor %}
+
+----FULL FILE CONTENT AFTER APPLYING CHANGES----
+{% for file, file_lines in all_file_lines.items() %}
+--FILE: {{ file }}--
+{{ file_lines }}
+{% endfor %}
+
+{%- if pipeline_out.associated_issue and pipeline_out.associated_issue.title %}
+----ASSOCIATED ISSUE----
+# {{ pipeline_out.associated_issue.title }}
+{{ pipeline_out.associated_issue.description }}
+URL: {{ pipeline_out.associated_issue.url }}
+{%- endif -%}{{ '\n' }}

gito/utils.py

@@ -2,11 +2,12 @@ import re
 import sys
 import os
 from pathlib import Path
+import importlib.metadata
+from typing import Optional
 
 import typer
 import git
 from git import Repo
-from microcore import ui
 
 
 _EXT_TO_HINT: dict[str, str] = {
@@ -222,5 +223,20 @@ def detect_github_env() -> dict:
     return d
 
 
-def stream_to_cli(text):
-    print(ui.blue(text), end='')
+def make_streaming_function(handler: Optional[callable] = None) -> callable:
+    def stream(text):
+        if handler:
+            text = handler(text)
+        print(text, end='', flush=True)
+    return stream
+
+
+def version() -> str:
+    return importlib.metadata.version("gito.bot")
+
+
+def remove_html_comments(text):
+    """
+    Removes all HTML comments (<!-- ... -->) from the input text.
+    """
+    return re.sub(r'<!--.*?-->\s*', '', text, flags=re.DOTALL)

ai_cr-2.0.2.dist-info/RECORD

@@ -1,26 +0,0 @@
-gito/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-gito/__main__.py,sha256=MSmt_5Xg84uHqzTN38JwgseJK8rsJn_11A8WD99VtEo,61
-gito/bootstrap.py,sha256=ceeoSCD8RX4xKxWri_n0xx00aTFS6xPUHhAs3rKVP5o,2337
-gito/cli.py,sha256=qFY1ObA0-Ep19bNSEcwjn6KIu2OOSWDj6QGe_o7Ujoc,6555
-gito/commands/__init__.py,sha256=B2uUQsLMEsHfNT1N3lWYm38WSuQIHFmjiGs2tdBuDBA,55
-gito/commands/fix.py,sha256=OS829xVrLMLQqlol9AoxJMZ9qaAWRCSbUkVNszK96ws,5305
-gito/commands/gh_post_review_comment.py,sha256=oZh3kSRBx3cSMozI2llDEOVZ2WeT_5msXMLLCiLina4,1984
-gito/commands/gh_react_to_comment.py,sha256=4uUb39FnM45DO-wESQq6pUlO5AFvh0FSluCCTxkl8_g,6442
-gito/commands/repl.py,sha256=wDHWl2VyRK32hLZHImrhqpQNkWV3u3358fwDiA6D3iA,521
-gito/config.toml,sha256=iK0CeP3mynTnWIEsCi_jA9Gtb_UdAqwLC5nWHSTc-zc,17367
-gito/constants.py,sha256=9G4sZbqLBg13FYieJqkq_hzek4-smmL4KKYF64tovQQ,698
-gito/core.py,sha256=tp-eB_FIsZ2wffTiMrRLc_9MaFaGIky_ExWHWTf1DTk,9341
-gito/gh_api.py,sha256=syhM8sbs_lLNG53bfRHQpusO818nwXCl4jDxVY__IK0,1204
-gito/issue_trackers.py,sha256=nfib6zhvmL_zjRDCdZ0d6rMT4ZFJ3PxO5UORaJzb6gk,1495
-gito/pipeline.py,sha256=Nq5VUVrXVDXVVYX6nVSMX1CyhoWsNtSeTOQyNgQAsdE,2672
-gito/pipeline_steps/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-gito/pipeline_steps/jira.py,sha256=cgRSHoU6j0lcK6hQfhg950PEmIu3uWLGMKZbW3-H3mg,1683
-gito/pipeline_steps/linear.py,sha256=6UDc8nGKGpwHruPq8VItE2QBWshWxaTapoMhu_qjN_g,2445
-gito/project_config.py,sha256=XdpXsRDFWBNNpktbZDp9p-nbijyKjG_oD6UWzGZSLjM,2580
-gito/report_struct.py,sha256=tLhdmCPse3Jbo56y752vuGlndWY2f2g_mNFQ-BmJZGw,4160
-gito/utils.py,sha256=HpterYtjceVhklJe0w1n7PIQJk8MJFUYi4FiG-CzMqg,6418
-ai_cr-2.0.2.dist-info/LICENSE,sha256=VbdF_GbbDK24JvdTfnsxa2M6jmhsxmRSFeHCx-lICGE,1075
-ai_cr-2.0.2.dist-info/METADATA,sha256=K8YJPk8Mv3GwYKbGe3gpE4qJgoWfv2wQBemoXOe-awc,7989
-ai_cr-2.0.2.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-ai_cr-2.0.2.dist-info/entry_points.txt,sha256=Ua1DxkhJJ8TZuLgnH-IlWCkrre_0S0dq_GtYRaYupWk,38
-ai_cr-2.0.2.dist-info/RECORD,,