"""Type definition for GitHub operation records in JSONL files."""

timestamp: str

function: str

args: list[Any]

"""Decorator to log function inputs in a blue panel."""

@wraps(func)

def wrapper(*args, **kwargs):

# Get function name and format it nicely

func_name = func.__name__.replace('_', ' ').title()

# Format parameters

params = []

for k, v in kwargs.items():

if isinstance(v, str) and len(v) > 50:

params.append(f"{k}='{v[:50]}...'")

else:

params.append(f"{k}='{v}'")

console.print(Panel(", ".join(params), title=f"[bold blue]{func_name}", border_style="blue"))

return func(*args, **kwargs)

method: str, endpoint: str, repo: str | None = None, data: dict | None = None, params: dict | None = None, should_raise: bool = False

) -> dict[str, Any] | str:

"""Make a GitHub API request with common error handling.

if repo is None:

repo = os.environ.get("GITHUB_REPOSITORY")

if not repo:

return "Error: GITHUB_REPOSITORY environment variable not found"

token = os.environ.get("GITHUB_TOKEN", "")

if not token:

return "Error: GITHUB_TOKEN environment variable not found"

url = f"https://api.github.com/repos/{repo}/{endpoint}"

headers = {

"Authorization": f"Bearer {token}",

"Accept": "application/vnd.github.v3+json",

}

try:

if method.upper() == "GET":

response = requests.get(url, headers=headers, params=params, timeout=30)

elif method.upper() == "POST":

response = requests.post(url, headers=headers, json=data, params=params, timeout=30)

else:

response = requests.request(method, url, headers=headers, json=data, params=params, timeout=30)

response.raise_for_status()

return response.json() # type: ignore[no-any-return]

except Exception as e:

if should_raise:

raise e

"""Decorator that checks if a write api should be called, or if the tool should record to JSONL."""

@wraps(func)

def wrapper(*args, **kwargs):

try:

if not _should_call_write_api():

# Record the tool request to JSONL file

record_entry: GitHubOperation = {

"timestamp": datetime.utcnow().isoformat() + "Z",

"function": func.__name__,

"args": args,

"kwargs": kwargs

}

os.makedirs(".artifact", exist_ok=True)

with open(".artifact/write_operations.jsonl", "a") as f:

f.write(json.dumps(record_entry) + "\n")

# Generate and return deferred message

params = dict(kwargs)

if args:

# Map positional args to parameter names from function signature

import inspect

sig = inspect.signature(func)

param_names = list(sig.parameters.keys())

for i, arg in enumerate(args):

if i < len(param_names):

params[param_names[i]] = arg

deferred_msg = _generate_deferred_message(func.__name__, params)

console.print(Panel(escape(deferred_msg), title="[bold yellow]Operation Deferred", border_style="yellow"))

return deferred_msg

except Exception as e:

error_msg = f"Error checking permissions: {e!s}"

console.print(Panel(escape(error_msg), title="[bold red]Error", border_style="red"))

return error_msg

return func(*args, **kwargs)

"""Generate a consistent deferred message for write operations.

if not params:

return f"Operation deferred: {operation_name}"

# Format parameters, truncating long values

param_strs = []

for key, value in params.items():

if isinstance(value, str) and len(value) > 50:

param_strs.append(f"{key}='{value[:50]}...'")

elif isinstance(value, str):

param_strs.append(f"{key}='{value}'")

else:

param_strs.append(f"{key}={value}")

"""Checks if GITHUB_WRITE environment variable is set to true.

"""Creates a new issue in the specified repository.

result = _github_request("POST", "issues", repo, {"title": title, "body": body})

if isinstance(result, str):

console.print(Panel(escape(result), title="[bold red]Error", border_style="red"))

return result

message = f"Issue created: #{result['number']} - {result['html_url']}"

console.print(Panel(escape(message), title="[bold green]Success", border_style="green"))

issue_number: int,

title: str | None = None,

body: str | None = None,

state: str | None = None,

repo: str | None = None,

) -> str:

"""Updates an issue's title, body, or state.

data = {}

if title is not None:

data["title"] = title

if body is not None:

data["body"] = body

if state is not None:

data["state"] = state

if not data:

error_msg = "Error: At least one field (title, body, or state) must be provided"

console.print(Panel(escape(error_msg), title="[bold red]Error", border_style="red"))

return error_msg

result = _github_request("PATCH", f"issues/{issue_number}", repo, data)

if isinstance(result, str):

console.print(Panel(escape(result), title="[bold red]Error", border_style="red"))

return result

message = f"Issue updated: #{result['number']} - {result['html_url']}"

console.print(Panel(escape(message), title="[bold green]Success", border_style="green"))

"""Adds a comment to an issue or pull request in the specified repository or GITHUB_REPOSITORY environment variable.

result = _github_request("POST", f"issues/{issue_number}/comments", repo, {"body": comment_text})

if isinstance(result, str):

console.print(Panel(escape(result), title="[bold red]Error", border_style="red"))

return result

message = f"Comment added successfully: {result['html_url']} (created: {result['created_at']})"

console.print(Panel(escape(message), title="[bold green]Success", border_style="green"))

"""Creates a new pull request, or optionally comments on the fallback_issue_id for a link to create a pull request.

try:

result = _github_request(

"POST",

"pulls",

repo,

{"title": title, "head": head, "base": base, "body": body},

should_raise=True

)

if isinstance(result, str):

console.print(Panel(escape(result), title="[bold red]Error", border_style="red"))

return result

message = f"Pull request created: #{result['number']} - {result['html_url']}"

console.print(Panel(escape(message), title="[bold green]Success", border_style="green"))

return message

except Exception as e:

if fallback_issue_id is not None:

agent_message = "Failed to create pull request, commenting on issue instead."

console.print(Panel(escape(agent_message), title="[bold yellow]Fallback", border_style="yellow"))

repo_name = repo or os.environ.get("GITHUB_REPOSITORY", "")

query_params = urlencode({

'quick_pull': '1',

'title': title,

'body': body

}, quote_via=quote)

pr_link = f"https://github.com/{repo_name}/compare/{base}...{head}?{query_params}"

fallback_comment = f"Unable to create pull request via API. You can create it manually by clicking [here]({pr_link})."

add_issue_comment(fallback_issue_id, fallback_comment, repo)

return f"Unable to create pull request via API - posted a manual creation link as a comment on issue #{fallback_issue_id}"

else:

error_msg = f"Error: {e!s}"

console.print(Panel(escape(error_msg), title="[bold red]Error", border_style="red"))

pr_number: int,

title: str | None = None,

body: str | None = None,

base: str | None = None,

repo: str | None = None,

) -> str:

"""Updates a pull request's title, body, or base branch.

data = {}

if title is not None:

data["title"] = title

if body is not None:

data["body"] = body

if base is not None:

data["base"] = base

if not data:

error_msg = "Error: At least one field (title, body, or base) must be provided"

console.print(Panel(escape(error_msg), title="[bold red]Error", border_style="red"))

return error_msg

result = _github_request("PATCH", f"pulls/{pr_number}", repo, data)

if isinstance(result, str):

console.print(Panel(escape(result), title="[bold red]Error", border_style="red"))

return result

message = f"Pull request updated: #{result['number']} - {result['html_url']}"

console.print(Panel(escape(message), title="[bold green]Success", border_style="green"))

"""Replies to a pull request review comment.

result = _github_request("POST", f"pulls/{pr_number}/comments/{comment_id}/replies", repo, {"body": reply_text})

if isinstance(result, str):

console.print(Panel(escape(result), title="[bold red]Error", border_style="red"))

return result

message = f"Reply added to review comment: {result['html_url']}"

reply_details = f"Reply: {reply_text}\nURL: {result['html_url']}"

console.print(Panel(escape(reply_details), title="[bold green]✅ Reply Added", border_style="green"))

"""Gets details of a specific issue.

result = _github_request("GET", f"issues/{issue_number}", repo)

if isinstance(result, str):

console.print(Panel(escape(result), title="[bold red]Error", border_style="red"))

return result

details = (

f"#{result['number']} - {result['title']}\n"

f"State: {result['state']}\n"

f"Author: {result['user']['login']}\n"

f"URL: {result['html_url']}\n\n{result['body']}"

)

console.print(

Panel(

escape(details),

title=f"[bold green]📋 Issue #{result['number']}",

border_style="blue",

)

)

"""Lists issues from the specified GitHub repository or GITHUB_REPOSITORY environment variable.

result = _github_request("GET", "issues", repo, params={"state": state})

if isinstance(result, str):

console.print(Panel(escape(result), title="[bold red]Error", border_style="red"))

return result

# Filter out pull requests from issues list

issues = [issue for issue in result if "pull_request" not in issue]

if not issues:

message = f"No {state} issues found in {repo or os.environ.get('GITHUB_REPOSITORY')}"

console.print(Panel(escape(message), title="[bold yellow]Info", border_style="yellow"))

return message

table = Table(title=f"🐛 Issues ({state})", box=box.DOUBLE)

table.add_column("Issue #", style="cyan")

table.add_column("Title", style="white")

table.add_column("Author", style="green")

table.add_column("URL", style="blue")

for issue in issues:

table.add_row(

f"#{issue['number']}", # type: ignore[index]

issue["title"], # type: ignore[index]

issue["user"]["login"], # type: ignore[index]

issue["html_url"], # type: ignore[index]

)

console.print(table)

output = f"Issues ({state}) in {repo or os.environ.get('GITHUB_REPOSITORY')}:\n"

for issue in issues:

output += f"#{issue['number']} - {issue['title']} by {issue['user']['login']} - {issue['html_url']}\n" # type: ignore[index]

"""Gets all comments for a specific issue.

params = {"since": since} if since else None

result = _github_request("GET", f"issues/{issue_number}/comments", repo, params=params)

if isinstance(result, str):

console.print(Panel(escape(result), title="[bold red]Error", border_style="red"))

return result

if not result:

message = f"No comments found for issue #{issue_number}" + (f" updated after {since}" if since else "")

console.print(Panel(escape(message), title="[bold yellow]Info", border_style="yellow"))

return message

output = f"Comments for issue #{issue_number}:\n"

for comment in result:

output += f"{comment['user']['login']} - updated: {comment['updated_at']}\n{comment['body']}\n\n" # type: ignore[index]

console.print(Panel(escape(output), title=f"[bold green]💬 Issue #{issue_number} Comments", border_style="blue"))

"""Gets details of a specific pull request.

result = _github_request("GET", f"pulls/{pr_number}", repo)

if isinstance(result, str):

console.print(Panel(escape(result), title="[bold red]Error", border_style="red"))

return result

details = (

f"#{result['number']} - {result['title']}\n"

f"State: {result['state']}\n"

f"Author: {result['user']['login']}\n"

f"Head: {result['head']['ref']} -> Base: {result['base']['ref']}\n"

f"URL: {result['html_url']}\n\n{result['body']}"

)

console.print(

Panel(

escape(details),

title=f"[bold green]🔀 PR #{result['number']}",

border_style="blue",

)

)

"""Lists pull requests from the specified GitHub repository or GITHUB_REPOSITORY environment variable.

result = _github_request("GET", "pulls", repo, params={"state": state})

if isinstance(result, str):

console.print(Panel(escape(result), title="[bold red]Error", border_style="red"))

return result

if not result:

message = f"No {state} pull requests found in {repo or os.environ.get('GITHUB_REPOSITORY')}"

console.print(Panel(escape(message), title="[bold yellow]Info", border_style="yellow"))

return message

table = Table(title=f"🔀 Pull Requests ({state})", box=box.DOUBLE)

table.add_column("PR #", style="cyan")

table.add_column("Title", style="white")

table.add_column("Author", style="green")

table.add_column("URL", style="blue")

for pr in result:

table.add_row(f"#{pr['number']}", pr["title"], pr["user"]["login"], pr["html_url"]) # type: ignore[index]

console.print(table)

output = f"Pull Requests ({state}) in {repo or os.environ.get('GITHUB_REPOSITORY')}:\n"

for pr in result:

output += f"#{pr['number']} - {pr['title']} by {pr['user']['login']} - {pr['html_url']}\n" # type: ignore[index]

"""Gets all review threads and comments for a PR.

if repo is None:

repo = os.environ.get("GITHUB_REPOSITORY")

if not repo:

return "Error: GITHUB_REPOSITORY environment variable not found"

token = os.environ.get("GITHUB_TOKEN", "")

if not token:

return "Error: GITHUB_TOKEN environment variable not found"

owner, repo_name = repo.split("/")

query = """

variables = {"owner": owner, "name": repo_name, "number": pr_number}

try:

response = requests.post(

"https://api.github.com/graphql",

headers={"Authorization": f"Bearer {token}"},

json={"query": query, "variables": variables},

timeout=30

)

response.raise_for_status()

data = response.json()

if "errors" in data:

return f"GraphQL Error: {data['errors']}"

pr_data = data["data"]["repository"]["pullRequest"]

# Filter by since if provided

if since:

cutoff = datetime.fromisoformat(since.replace('Z', '+00:00'))

# Filter review threads - if any comment in thread is newer, include entire thread

filtered_threads = []

for thread in pr_data["reviewThreads"]["nodes"]:

has_newer_comment = any(datetime.fromisoformat(c['updatedAt'].replace('Z', '+00:00')) > cutoff

for c in thread["comments"]["nodes"])

if has_newer_comment:

filtered_threads.append(thread)

pr_data["reviewThreads"]["nodes"] = filtered_threads

# Filter general comments

pr_data["comments"]["nodes"] = [c for c in pr_data["comments"]["nodes"]

if datetime.fromisoformat(c['updatedAt'].replace('Z', '+00:00')) > cutoff]

output = f"Review threads and comments for PR #{pr_number}:\n\n"

# Group review threads by review ID

review_threads = {}

for thread in pr_data["reviewThreads"]["nodes"]:

if not show_resolved and thread["isResolved"]:

continue

if thread["comments"]["nodes"]:

first_comment = thread["comments"]["nodes"][0]

review_id = first_comment.get("pullRequestReview", {}).get("id", "N/A")

if review_id not in review_threads:

review_threads[review_id] = {

"review_data": first_comment.get("pullRequestReview", {}),

"threads": []

}

review_threads[review_id]["threads"].append(thread)

# Display grouped review threads

for review_id, review_info in review_threads.items():

review_data = review_info['review_data']

output += f"📝 Review [Review ID: {review_id}]\n"

# Always show review author and timestamps

if review_data.get('author'):

output += f" 👤 Review by {review_data['author']['login']} (updated: {review_data['updatedAt']})\n"

# Show top-level review comment if it exists

if review_data.get('body'):

output += f" 📋 Review Comment:\n"

output += f" {review_data['body']}\n"

output += "\n"

# Show all threads for this review

for thread in review_info["threads"]:

first_comment = thread["comments"]["nodes"][0]

line_info = f":{first_comment['line']}" if first_comment.get('line') else " (Comment on file)"

status = "✅ RESOLVED" if thread["isResolved"] else "🔄 OPEN"

output += f" 📍 Thread ({status}): {first_comment['path']}{line_info}\n"

# Show code context right after thread header

if first_comment.get('diffHunk') and first_comment.get('line'):

diff_lines = first_comment['diffHunk'].split('\n')

current_new_line = 0

target_line = first_comment['line']

start_line = first_comment.get('startLine') or target_line

output += f" Code context (lines {start_line}-{target_line}):\n"

for diff_line in diff_lines:

if diff_line.startswith('@@'):

parts = diff_line.split(' ')

if len(parts) >= 3:

new_start = parts[2].split(',')[0][1:]

current_new_line = int(new_start) - 1

elif diff_line.startswith('+'):

current_new_line += 1

if start_line <= current_new_line <= target_line:

output += f" +{current_new_line}: {diff_line[1:]}\n"

elif diff_line.startswith('-'):

pass

elif diff_line.startswith(' '):

current_new_line += 1

if start_line <= current_new_line <= target_line:

output += f" {current_new_line}: {diff_line[1:]}\n"

output += "\n"

# Group comments by reply relationships

comments = thread["comments"]["nodes"]

root_comments = [c for c in comments if not c.get('replyTo')]

for root_comment in root_comments:

output += f" 💬 {root_comment['author']['login']} (updated: {root_comment['updatedAt']}) [Comment ID: {root_comment['fullDatabaseId']}]:\n"

output += f" {root_comment['body']}\n"

# Find and show replies to this comment

replies = [c for c in comments if c.get('replyTo') and c['replyTo'].get('id') == root_comment['id']]

if replies:

for reply in replies:

output += f" ↳ {reply['author']['login']} (updated: {reply['updatedAt']}):\n"

output += f" {reply['body']}\n"

output += "\n"

output += "\n"

# General comments

if pr_data["comments"]["nodes"]:

for comment in pr_data["comments"]["nodes"]:

output += f"💬 Comment\n"

output += f" 👤 Comment by {comment['author']['login']} (updated: {comment['updatedAt']})\n"

output += f" 📝 Comment:\n"

output += f" {comment['body']}\n\n"

console.print(Panel(escape(output), title=f"[bold green]PR #{pr_number} Review Data", border_style="blue"))

return output

except Exception as e:

error_msg = f"Error: {e!s}\n\nStack trace:\n{traceback.format_exc()}"

console.print(Panel(escape(error_msg), title="[bold red]Error", border_style="red"))