pygitcode 0.1.0__py3-none-any.whl

gitcode_cli/commands/pr.py

@@ -0,0 +1,497 @@
+from __future__ import annotations
+
+import json
+import subprocess
+
+import click
+
+from ..cli_compat import (
+    get_body_from_options,
+    get_default_base_branch,
+    get_fill_info,
+    normalize_multi_values,
+    resolve_pr_identifier_or_current_branch,
+)
+from ..formatters import format_pr_detail, format_pr_list, output_result
+from ..repo import resolve_repo
+from ..services import PullRequestService
+from ..utils import get_current_git_branch, open_in_browser, prompt_if_missing, resolve_pr_arg
+
+
+@click.group("pr")
+def pr_group() -> None:
+    pass
+
+
+@pr_group.command("list")
+@click.option("-R", "--repo", "repo_name", help="Select another repository using the [HOST/]OWNER/REPO format.")
+@click.option("-s", "--state")
+@click.option("-A", "--author")
+@click.option("-B", "--base")
+@click.option("--assignee")
+@click.option("--draft", is_flag=True, default=None)
+@click.option("--head")
+@click.option("-l", "--label", "labels", multiple=True)
+@click.option("-S", "--search")
+@click.option("-L", "--limit", type=int, default=30, show_default=True, help="Maximum number of items to fetch.")
+@click.option("-w", "--web", is_flag=True, help="Open the pull requests list in the web browser.")
+@click.option("--json", "json_fields", help="Output JSON. Optionally specify comma-separated fields.")
+@click.option("-q", "--jq", "jq_query", help="Filter JSON output using a jq expression.")
+@click.option("-t", "--template", help="Format output using a Go template string.")
+@click.pass_context
+def pr_list(
+    ctx: click.Context,
+    repo_name: str | None,
+    state: str | None,
+    author: str | None,
+    base: str | None,
+    assignee: str | None,
+    draft: bool | None,
+    head: str | None,
+    labels: tuple[str, ...] | None,
+    search: str | None,
+    limit: int | None,
+    web: bool,
+    json_fields: str | None,
+    jq_query: str | None,
+    template: str | None,
+) -> None:
+    app = ctx.obj["app"]
+    owner, repo = resolve_repo(repo_name or app.repo)
+    if web:
+        open_in_browser(f"https://gitcode.com/{owner}/{repo}/pulls")
+        return
+    service = PullRequestService(app.client())
+    labels_str = normalize_multi_values(labels)
+    items = service.list(
+        owner,
+        repo,
+        state=state,
+        author=author,
+        base=base,
+        assignee=assignee,
+        draft=draft,
+        head=head,
+        labels=labels_str,
+        search=search,
+    )
+    if limit is not None:
+        items = items[:limit]
+
+    def default_formatter(data):
+        output = format_pr_list(data)
+        if output:
+            click.echo(output)
+
+    output_result(
+        items,
+        json_fields,
+        jq_query,
+        template,
+        default_formatter=default_formatter,
+    )
+
+
+@pr_group.command("view")
+@click.option("-R", "--repo", "repo_name", help="Select another repository using the [HOST/]OWNER/REPO format.")
+@click.argument("identifier", required=False)
+@click.option("-w", "--web", is_flag=True, help="Open the pull request in the web browser.")
+@click.option("--json", "json_fields", help="Output JSON. Optionally specify comma-separated fields.")
+@click.option("-q", "--jq", "jq_query", help="Filter JSON output using a jq expression.")
+@click.option("-t", "--template", help="Format output using a Go template string.")
+@click.pass_context
+def pr_view(
+    ctx: click.Context,
+    repo_name: str | None,
+    identifier: str | None,
+    web: bool,
+    json_fields: str | None,
+    jq_query: str | None,
+    template: str | None,
+) -> None:
+    app = ctx.obj["app"]
+    owner, repo = resolve_repo(repo_name or app.repo)
+    service = PullRequestService(app.client())
+    resolved_identifier = resolve_pr_identifier_or_current_branch(identifier)
+    owner, repo, number = resolve_pr_arg(resolved_identifier, owner, repo, service)
+    number = int(number)
+    item = service.get(owner, repo, number)
+    if web:
+        open_in_browser(item["html_url"])
+        return
+    output_result(
+        item,
+        json_fields,
+        jq_query,
+        template,
+        default_formatter=lambda data: click.echo(format_pr_detail(data)),
+    )
+
+
+@pr_group.command("create")
+@click.option("-R", "--repo", "repo_name", help="Select another repository using the [HOST/]OWNER/REPO format.")
+@click.option("-t", "--title")
+@click.option("-b", "--body")
+@click.option("-F", "--body-file")
+@click.option("--editor", is_flag=True)
+@click.option("--fill", is_flag=True, help="Use commit info for title and body.")
+@click.option("--fill-first", is_flag=True, help="Use first commit info for title and body.")
+@click.option("--fill-verbose", is_flag=True, help="Use all commits for body description.")
+@click.option("--dry-run", is_flag=True)
+@click.option("-B", "--base")
+@click.option("-H", "--head")
+@click.option("-d", "--draft", is_flag=True)
+@click.option("--milestone")
+@click.option("-l", "--label", "labels", multiple=True)
+@click.option("-r", "--reviewer", "reviewers", multiple=True)
+@click.option("-a", "--assignee", "assignees", multiple=True)
+@click.option("-w", "--web", is_flag=True, help="Open the pull request in the web browser.")
+@click.option("--json", "json_fields", help="Output JSON. Optionally specify comma-separated fields.")
+@click.option("-q", "--jq", "jq_query", help="Filter JSON output using a jq expression.")
+@click.option("--template", help="Format output using a Go template string.")
+@click.pass_context
+def pr_create(
+    ctx: click.Context,
+    repo_name: str | None,
+    title: str | None,
+    body: str | None,
+    body_file: str | None,
+    editor: bool,
+    fill: bool,
+    fill_first: bool,
+    fill_verbose: bool,
+    dry_run: bool,
+    base: str | None,
+    head: str | None,
+    draft: bool,
+    milestone: str | None,
+    labels: tuple[str, ...] | None,
+    reviewers: tuple[str, ...] | None,
+    assignees: tuple[str, ...] | None,
+    web: bool,
+    json_fields: str | None,
+    jq_query: str | None,
+    template: str | None,
+) -> None:
+    app = ctx.obj["app"]
+    owner, repo = resolve_repo(repo_name or app.repo)
+    if web:
+        open_in_browser(f"https://gitcode.com/{owner}/{repo}/pulls/new")
+        return
+    if not head:
+        head = get_current_git_branch()
+        if not head:
+            raise click.ClickException("Unable to detect current branch. Use --head.")
+    if not base:
+        base = get_default_base_branch()
+
+    fill_mode = None
+    if fill_verbose:
+        fill_mode = "verbose"
+    elif fill_first:
+        fill_mode = "first"
+    elif fill:
+        fill_mode = "last"
+
+    if fill_mode:
+        fill_title, fill_body = get_fill_info(fill_mode)
+        if title is None:
+            title = fill_title
+        if body is None:
+            body = fill_body
+
+    title = prompt_if_missing(title, "Title")
+    body = get_body_from_options(body=body, body_file=body_file, editor=editor)
+    payload = {
+        "title": title,
+        "body": body,
+        "base": base,
+        "head": head,
+        "draft": draft,
+        "labels": normalize_multi_values(labels),
+        "assignees": normalize_multi_values(assignees),
+        "reviewers": normalize_multi_values(reviewers),
+        "milestone": milestone,
+    }
+    if dry_run:
+        click.echo(json.dumps({k: v for k, v in payload.items() if v is not None}, indent=2, sort_keys=True))
+        return
+    service = PullRequestService(app.client())
+    item = service.create(owner, repo, **payload)
+    output_result(
+        item,
+        json_fields,
+        jq_query,
+        template,
+        default_formatter=lambda data: click.echo(data["html_url"]),
+    )
+
+
+@pr_group.command("close")
+@click.option("-R", "--repo", "repo_name", help="Select another repository using the [HOST/]OWNER/REPO format.")
+@click.argument("identifier", required=False)
+@click.option("-c", "--comment")
+@click.option("-d", "--delete-branch", is_flag=True, help="Delete the remote branch after closing.")
+@click.pass_context
+def pr_close(
+    ctx: click.Context, repo_name: str | None, identifier: str | None, comment: str | None, delete_branch: bool
+) -> None:
+    app = ctx.obj["app"]
+    owner, repo = resolve_repo(repo_name or app.repo)
+    service = PullRequestService(app.client())
+    resolved_identifier = resolve_pr_identifier_or_current_branch(identifier)
+    owner, repo, number = resolve_pr_arg(resolved_identifier, owner, repo, service)
+    number = int(number)
+    if comment:
+        service.comment(owner, repo, number, body=comment)
+    item = service.update(owner, repo, number, state="closed")
+    if delete_branch:
+        try:
+            branch = item.get("head", {}).get("ref")
+            if branch:
+                subprocess.run(["git", "push", "origin", "--delete", branch], check=True)
+                click.echo(f"Deleted remote branch {branch}")
+            else:
+                click.echo("Warning: could not determine branch to delete.", err=True)
+        except Exception as exc:
+            click.echo(f"Warning: could not delete remote branch: {exc}", err=True)
+    click.echo(f"Closed pull request #{item['number']}")
+
+
+@pr_group.command("merge")
+@click.option("-R", "--repo", "repo_name", help="Select another repository using the [HOST/]OWNER/REPO format.")
+@click.argument("identifier", required=False)
+@click.option("-m", "--merge", "merge_mode", flag_value="merge")
+@click.option("-s", "--squash", "merge_mode", flag_value="squash")
+@click.option("-r", "--rebase", "merge_mode", flag_value="rebase")
+@click.pass_context
+def pr_merge(ctx: click.Context, repo_name: str | None, identifier: str | None, merge_mode: str | None) -> None:
+    app = ctx.obj["app"]
+    owner, repo = resolve_repo(repo_name or app.repo)
+    service = PullRequestService(app.client())
+    resolved_identifier = resolve_pr_identifier_or_current_branch(identifier)
+    owner, repo, number = resolve_pr_arg(resolved_identifier, owner, repo, service)
+    number = int(number)
+    item = service.merge(owner, repo, number, merge_method=merge_mode or "merge")
+    click.echo(item["message"])
+
+
+@pr_group.command("comment")
+@click.option("-R", "--repo", "repo_name", help="Select another repository using the [HOST/]OWNER/REPO format.")
+@click.argument("identifier", required=False)
+@click.option("-b", "--body")
+@click.option("--path")
+@click.option("--position", type=int)
+@click.pass_context
+def pr_comment(
+    ctx: click.Context,
+    repo_name: str | None,
+    identifier: str | None,
+    body: str | None,
+    path: str | None,
+    position: int | None,
+) -> None:
+    app = ctx.obj["app"]
+    owner, repo = resolve_repo(repo_name or app.repo)
+    service = PullRequestService(app.client())
+    resolved_identifier = resolve_pr_identifier_or_current_branch(identifier)
+    owner, repo, number = resolve_pr_arg(resolved_identifier, owner, repo, service)
+    number = int(number)
+    body = prompt_if_missing(body, "Body")
+    item = service.comment(owner, repo, number, body=body, path=path, position=position)
+    click.echo(str(item["id"]))
+
+
+@pr_group.command("review")
+@click.option("-R", "--repo", "repo_name", help="Select another repository using the [HOST/]OWNER/REPO format.")
+@click.argument("identifier", required=False)
+@click.option("-a", "--approve", is_flag=True, help="Approve the pull request. GitCode maps this to its review API.")
+@click.option("--body")
+@click.option("--comment", is_flag=True, help="Leave a review comment. Downgrades to a PR comment on GitCode.")
+@click.option("--request-changes", is_flag=True, help="Request changes. Downgrades to a PR comment on GitCode.")
+@click.option("--force", is_flag=True, help="Force review handling when supported by GitCode.")
+@click.pass_context
+def pr_review(
+    ctx: click.Context,
+    repo_name: str | None,
+    identifier: str | None,
+    approve: bool,
+    body: str | None,
+    comment: bool,
+    request_changes: bool,
+    force: bool,
+) -> None:
+    app = ctx.obj["app"]
+    owner, repo = resolve_repo(repo_name or app.repo)
+    service = PullRequestService(app.client())
+    resolved_identifier = resolve_pr_identifier_or_current_branch(identifier)
+    owner, repo, number = resolve_pr_arg(resolved_identifier, owner, repo, service)
+    number = int(number)
+
+    selected_modes = [approve, comment, request_changes]
+    if sum(1 for selected in selected_modes if selected) != 1:
+        raise click.ClickException("Specify exactly one of --approve, --comment, or --request-changes.")
+
+    if comment or request_changes:
+        if not body:
+            raise click.ClickException("Body is required when using --comment or --request-changes.")
+        item = service.comment(owner, repo, number, body=body)
+        if comment:
+            click.echo("GitCode review API does not support comment reviews; posted a pull request comment instead.")
+        else:
+            click.echo(
+                "GitCode review API does not support request-changes reviews; posted a pull request comment instead."
+            )
+        click.echo(f"Posted pull request comment {item['id']}")
+        return
+
+    item = service.review(owner, repo, number, body=body, force=force)
+    click.echo(f"Reviewed pull request #{number}")
+
+
+@pr_group.command("reopen")
+@click.option("-R", "--repo", "repo_name", help="Select another repository using the [HOST/]OWNER/REPO format.")
+@click.argument("identifier", required=False)
+@click.pass_context
+def pr_reopen(ctx: click.Context, repo_name: str | None, identifier: str | None) -> None:
+    app = ctx.obj["app"]
+    owner, repo = resolve_repo(repo_name or app.repo)
+    service = PullRequestService(app.client())
+    resolved_identifier = resolve_pr_identifier_or_current_branch(identifier)
+    owner, repo, number = resolve_pr_arg(resolved_identifier, owner, repo, service)
+    number = int(number)
+    item = service.update(owner, repo, number, state="open")
+    click.echo(f"Reopened pull request #{item['number']}")
+
+
+@pr_group.command("edit")
+@click.option("-R", "--repo", "repo_name", help="Select another repository using the [HOST/]OWNER/REPO format.")
+@click.argument("identifier", required=False)
+@click.option("-t", "--title")
+@click.option("-b", "--body")
+@click.option("-B", "--base")
+@click.option("-a", "--add-assignee")
+@click.option("-l", "--add-label")
+@click.option("-r", "--add-reviewer")
+@click.option("--remove-assignee")
+@click.option("--remove-label")
+@click.option("--remove-reviewer")
+@click.pass_context
+def pr_edit(
+    ctx: click.Context,
+    repo_name: str | None,
+    identifier: str | None,
+    title: str | None,
+    body: str | None,
+    base: str | None,
+    add_assignee: str | None,
+    add_label: str | None,
+    add_reviewer: str | None,
+    remove_assignee: str | None,
+    remove_label: str | None,
+    remove_reviewer: str | None,
+) -> None:
+    app = ctx.obj["app"]
+    owner, repo = resolve_repo(repo_name or app.repo)
+    service = PullRequestService(app.client())
+    resolved_identifier = resolve_pr_identifier_or_current_branch(identifier)
+    owner, repo, number = resolve_pr_arg(resolved_identifier, owner, repo, service)
+    number = int(number)
+    data = {
+        k: v
+        for k, v in {
+            "title": title,
+            "body": body,
+            "base": base,
+            "assignee": add_assignee,
+            "labels": add_label,
+            "reviewer": add_reviewer,
+            "unassignee": remove_assignee,
+            "unset_labels": remove_label,
+            "unset_reviewer": remove_reviewer,
+        }.items()
+        if v is not None
+    }
+    if not data:
+        raise click.UsageError("must specify at least one field to edit")
+    item = service.update(owner, repo, number, **data)
+    click.echo(f"Edited pull request #{item['number']}")
+
+
+@pr_group.command("diff")
+@click.option("-R", "--repo", "repo_name", help="Select another repository using the [HOST/]OWNER/REPO format.")
+@click.argument("identifier", required=False)
+@click.pass_context
+def pr_diff(ctx: click.Context, repo_name: str | None, identifier: str | None) -> None:
+    app = ctx.obj["app"]
+    owner, repo = resolve_repo(repo_name or app.repo)
+    service = PullRequestService(app.client())
+    resolved_identifier = resolve_pr_identifier_or_current_branch(identifier)
+    owner, repo, number = resolve_pr_arg(resolved_identifier, owner, repo, service)
+    diff_text = service.diff(owner, repo, int(number))
+    click.echo(diff_text)
+
+
+@pr_group.command("checkout")
+@click.option("-R", "--repo", "repo_name", help="Select another repository using the [HOST/]OWNER/REPO format.")
+@click.argument("identifier", required=False)
+@click.option("-b", "--branch", help="Local branch name to checkout into.")
+@click.pass_context
+def pr_checkout(ctx: click.Context, repo_name: str | None, identifier: str | None, branch: str | None) -> None:
+    app = ctx.obj["app"]
+    owner, repo = resolve_repo(repo_name or app.repo)
+    service = PullRequestService(app.client())
+    resolved_identifier = resolve_pr_identifier_or_current_branch(identifier)
+    owner, repo, number = resolve_pr_arg(resolved_identifier, owner, repo, service)
+    item = service.get(owner, repo, int(number))
+    head_ref = item.get("head", {}).get("ref")
+    if not head_ref:
+        raise click.ClickException("Unable to determine PR branch.")
+    local_branch = branch or head_ref
+    try:
+        subprocess.run(["git", "fetch", "origin", head_ref], check=True)
+        subprocess.run(["git", "checkout", "-b", local_branch, f"origin/{head_ref}"], check=True)
+        click.echo(f"Checked out branch {local_branch}")
+    except subprocess.CalledProcessError as exc:
+        raise click.ClickException(f"Git checkout failed: {exc}") from exc
+
+
+@pr_group.command("ready")
+@click.option("-R", "--repo", "repo_name", help="Select another repository using the [HOST/]OWNER/REPO format.")
+@click.argument("identifier", required=False)
+@click.option("--undo", is_flag=True, help="Convert a pull request to draft.")
+@click.pass_context
+def pr_ready(ctx: click.Context, repo_name: str | None, identifier: str | None, undo: bool) -> None:
+    app = ctx.obj["app"]
+    owner, repo = resolve_repo(repo_name or app.repo)
+    service = PullRequestService(app.client())
+    resolved_identifier = resolve_pr_identifier_or_current_branch(identifier)
+    owner, repo, number = resolve_pr_arg(resolved_identifier, owner, repo, service)
+    item = service.update(owner, repo, int(number), draft=undo)
+    if undo:
+        click.echo(f"Converted pull request #{item['number']} to draft")
+    else:
+        click.echo(f"Marked pull request #{item['number']} as ready for review")
+
+
+@pr_group.command("status")
+@click.option("-R", "--repo", "repo_name", help="Select another repository using the [HOST/]OWNER/REPO format.")
+@click.pass_context
+def pr_status(ctx: click.Context, repo_name: str | None) -> None:
+    app = ctx.obj["app"]
+    owner, repo = resolve_repo(repo_name or app.repo)
+    service = PullRequestService(app.client())
+    items = service.list(owner, repo, state="open")
+    click.echo(
+        f"Open pull requests in {owner}/{repo}  "
+        "(GitCode API approximation -- user-specific filtering is not available)"
+    )
+    if items:
+        for item in items:
+            click.echo(f"  #{item['number']}\t{item['state']}\t{item['title']}")
+    else:
+        click.echo("  No open pull requests")
+
+
+pr_group.add_command(pr_list, name="ls")
+pr_group.add_command(pr_create, name="new")

gitcode_cli/config.py

@@ -0,0 +1,39 @@
+from __future__ import annotations
+
+import json
+import os
+from pathlib import Path
+
+from .errors import ConfigError
+
+CONFIG_DIR = Path.home() / ".config" / "gc"
+CONFIG_PATH = CONFIG_DIR / "config.json"
+TOKEN_ENV_VARS = ("GC_TOKEN",)
+
+
+def load_config() -> dict:
+    if not CONFIG_PATH.exists():
+        return {}
+    try:
+        return json.loads(CONFIG_PATH.read_text())
+    except json.JSONDecodeError as exc:
+        raise ConfigError(f"Invalid config file: {CONFIG_PATH}") from exc
+
+
+def save_config(data: dict) -> None:
+    CONFIG_DIR.mkdir(parents=True, exist_ok=True)
+    CONFIG_PATH.write_text(json.dumps(data, indent=2) + "\n")
+
+
+def get_token(explicit_token: str | None = None) -> str:
+    if explicit_token:
+        return explicit_token
+    for env_name in TOKEN_ENV_VARS:
+        value = os.getenv(env_name)
+        if value:
+            return value
+    config = load_config()
+    token = config.get("token")
+    if token:
+        return token
+    raise ConfigError("No token found. Set GC_TOKEN or run `gc auth login`.")

gitcode_cli/context.py

@@ -0,0 +1,14 @@
+from __future__ import annotations
+
+from dataclasses import dataclass
+
+from .client import GitCodeClient
+
+
+@dataclass
+class AppContext:
+    token: str
+    repo: str | None
+
+    def client(self) -> GitCodeClient:
+        return GitCodeClient(token=self.token)

gitcode_cli/errors.py

@@ -0,0 +1,23 @@
+from typing import Optional
+
+
+class GCError(Exception):
+    pass
+
+
+class ConfigError(GCError):
+    pass
+
+
+class AuthError(GCError):
+    pass
+
+
+class RepoResolutionError(GCError):
+    pass
+
+
+class APIError(GCError):
+    def __init__(self, message: str, status_code: Optional[int] = None):
+        super().__init__(message)
+        self.status_code = status_code