(require 'magit)

(require 'ai-code-input)

(require 'ai-code-prompt-mode)

(require 'which-func nil t)

(declare-function ai-code-read-string "ai-code-input")

(declare-function ai-code--insert-prompt "ai-code-prompt-mode" (prompt-text))

(declare-function magit-git-string "magit-git" (&rest args))

(declare-function difftastic-magit-diff "difftastic" ())

(declare-function ai-code--validate-git-repository "ai-code-git" ())

(declare-function ai-code--require-current-branch "ai-code-git" ())

(declare-function ai-code--default-pr-target-branch "ai-code-git" (current-branch))

(declare-function ai-code--magit-generate-feature-branch-diff-file "ai-code-git" ())

(declare-function ai-code--generate-staged-diff "ai-code-git" (diff-file))

(declare-function ai-code--generate-branch-or-commit-diff "ai-code-git" (diff-params diff-file))

(declare-function ai-code--open-diff-file "ai-code-git" (diff-file))

(declare-function ai-code--explain-code-change "ai-code-discussion" (&optional review-source))

(declare-function ai-code--get-context-files-string "ai-code-utils" ())

(declare-function ai-code--format-repo-context-info "ai-code-utils" ())

(declare-function ai-code--current-line-scope-context "ai-code-utils" ())

(declare-function ai-code--scope-context-for-region "ai-code-utils" (beg end))

(declare-function ai-code--format-scope-context "ai-code-utils" (context))

(declare-function ai-code--get-region-location-info "ai-code-discussion" (region-beginning region-end))

(defcustom ai-code-default-review-source nil

"Default review source for pull request and issue analysis.

:type '(choice (const :tag "Prompt user" nil)

(const :tag "GitHub MCP server" github-mcp)

(const :tag "gh CLI tool" gh-cli))

:group 'ai-code)

(defvar ai-code-pr-title-history nil

"Minibuffer history for optional PR titles.")

(defvar ai-code--grill-me-workflow)

(defconst ai-code-github--analysis-only-note

"No need to make code change. Provide analysis only."

"Prompt note marking a GitHub workflow as analysis without code changes.")

(defconst ai-code-github--resolution-analysis-only-note

"Do not make code changes. Provide analysis and resolution suggestions only."

"Prompt note marking a resolution workflow as analysis without code changes.")

(defconst ai-code-github--pr-creation-no-code-change-note

"No need to make code change. Only create the pull request and report its details."

"Prompt note marking PR creation as a workflow without code changes.")

(defconst ai-code-github--analysis-only-notes

(list ai-code-github--analysis-only-note

ai-code-github--resolution-analysis-only-note

ai-code-github--pr-creation-no-code-change-note)

"Generated prompt notes that mark a GitHub workflow as analysis only.

editing the note out of a prompt restores normal code-change routing.")

(defun ai-code--extract-url-from-region ()

"Return the first URL found in the active region, or nil."

(when (use-region-p)

(let ((text (buffer-substring-no-properties (region-beginning) (region-end))))

(when (string-match "https?://[^ \t\n]+" text)

(match-string 0 text)))))

(defun ai-code--pull-or-review-action-choice ()

"Prompt user for action in `ai-code-pull-or-review-diff-file'.

(or ai-code-default-review-source

(let* ((action-alist '(("Use GitHub MCP server" . github-mcp)

("Use gh CLI tool" . gh-cli)))

(choice (completing-read "Select review source: "

action-alist

nil t nil nil "Use GitHub MCP server")))

(alist-get choice action-alist nil nil #'string=))))

(defun ai-code--message-review-source-config-hint ()

"Message a minibuffer hint about configuring `ai-code-default-review-source'."

(message

"Tip: set ai-code-default-review-source to github-mcp or gh-cli to skip this review-source prompt in future C-c a v runs."))

(defun ai-code--pull-or-review-source-instruction (review-source &optional review-mode)

"Return source instruction string for REVIEW-SOURCE and REVIEW-MODE."

(pcase review-mode

('investigate-issue

(pcase review-source

('github-mcp

"Use GitHub MCP server to inspect the GitHub issue and relevant repository context.")

('gh-cli

"Use gh CLI tool to inspect the GitHub issue and relevant repository context.")

(_ "Investigate this GitHub issue using the repository as context.")))

('review-ci-checks

(pcase review-source

('github-mcp

"Use GitHub MCP server to fetch pull request details and CI checks.")

('gh-cli

"Use gh CLI tool to fetch pull request details and CI checks.")

(_ "Review the GitHub CI checks for this pull request.")))

('resolve-merge-conflict

(pcase review-source

('github-mcp

"Use GitHub MCP server to fetch pull request branch details and merge status.")

('gh-cli

"Use gh CLI tool to fetch pull request branch details and merge status.")

(_ "Resolve merge conflicts for this pull request.")))

('explain-code-change

(pcase review-source

('github-mcp

"Use GitHub MCP server to inspect the pull request diff, changed files, commits, and relevant metadata.")

('gh-cli

"Use gh CLI tool to inspect the pull request diff, changed files, commits, and relevant metadata.")

(_ "Inspect the pull request diff and relevant metadata to understand the change.")))

(_

(pcase review-source

('github-mcp

"Use GitHub MCP server to fetch pull request details and review comments.")

('gh-cli

"Use gh CLI tool to fetch pull request details and review comments.")

(_ "Review this pull request.")))))

(defun ai-code--build-pr-review-init-prompt (review-source pr-url)

"Build PR review initial prompt for REVIEW-SOURCE with PR-URL."

(let ((source-instruction

(ai-code--pull-or-review-source-instruction review-source)))

(format "Review pull request: %s

pr-url source-instruction

ai-code-github--analysis-only-note)))

(defun ai-code--build-send-current-branch-pr-init-prompt

(review-source current-branch target-branch &optional pr-title)

"Build a PR creation prompt.

;; DONE: it should ask user if this is a draft PR or not (y/n). If yes, the PR should be created as a draft PR. If no, the PR should be created as a normal PR.

(let* ((source-instruction

(ai-code--send-current-branch-pr-source-instruction review-source))

(draft-pr-p

(y-or-n-p "Create this PR as a draft? "))

(pr-type-instruction

(if draft-pr-p

"Create a draft pull request. If you use GitHub CLI, pass `--draft`.\n"

"Create a normal pull request, not a draft.\n"))

(title-instruction

(if (string-empty-p (or pr-title ""))

"2. Generate a concise PR title based on the code change.\n"

(format "2. Use this PR title exactly: %s\n" pr-title))))

(format "Create a pull request from branch %s into %s.

current-branch

target-branch

source-instruction

target-branch

pr-type-instruction

title-instruction

ai-code-github--pr-creation-no-code-change-note)))

(defun ai-code--build-pr-feedback-check-init-prompt (review-source pr-url)

"Build unresolved feedback check prompt for REVIEW-SOURCE with PR-URL."

(let ((source-instruction

(ai-code--pull-or-review-source-instruction review-source)))

(format "Check unresolved feedback for pull request: %s

pr-url source-instruction

ai-code-github--analysis-only-note)))

(defun ai-code--build-issue-investigation-init-prompt (review-source issue-url &optional include-context)

"Build issue investigation prompt for REVIEW-SOURCE with ISSUE-URL.

(let* ((source-instruction

(ai-code--pull-or-review-source-instruction review-source 'investigate-issue))

(prompt

(format "Investigate issue: %s

issue-url source-instruction

ai-code-github--analysis-only-note)))

(if (and include-context (or buffer-file-name (use-region-p)))

(let* ((region-text (when (use-region-p)

(buffer-substring-no-properties (region-beginning) (region-end))))

(region-location-info (when region-text

(ai-code--get-region-location-info

(region-beginning)

(region-end))))

(semantic-scope

(ai-code--format-scope-context

(if region-text

(ai-code--scope-context-for-region

(region-beginning) (region-end))

(ai-code--current-line-scope-context))))

(files-context-string (ai-code--get-context-files-string))

(repo-context-string (ai-code--format-repo-context-info))

(context-blocks nil))

(push (if buffer-file-name

(format "Current file: %s" buffer-file-name)

(format "Current buffer: %s" (buffer-name)))

context-blocks)

(unless (string-empty-p semantic-scope)

(push semantic-scope context-blocks))

(when region-text

(push (concat "Selected region:\n"

(when region-location-info

(concat region-location-info "\n"))

region-text)

context-blocks))

(when (and files-context-string (not (string-empty-p files-context-string)))

(push files-context-string context-blocks))

(when (and repo-context-string (not (string-empty-p repo-context-string)))

(push repo-context-string context-blocks))

(concat prompt "\n\nLocal Context:\n"

(mapconcat #'identity (nreverse context-blocks) "\n\n")))

prompt)))

(defun ai-code--build-pr-description-init-prompt (review-source pr-url)

"Build PR description prompt for REVIEW-SOURCE with PR-URL."

(let ((source-instruction

(ai-code--pull-or-review-source-instruction review-source)))

(format "Prepare a pull request description for: %s

pr-url source-instruction

ai-code-github--analysis-only-note)))

(defun ai-code--build-pr-ci-check-init-prompt (review-source pr-url)

"Build a CI check review prompt for REVIEW-SOURCE with PR-URL."

(let ((source-instruction

(ai-code--pull-or-review-source-instruction review-source 'review-ci-checks)))

(format "Review GitHub CI checks for pull request: %s

pr-url source-instruction

ai-code-github--analysis-only-note)))

(defun ai-code--build-resolve-merge-conflict-init-prompt (review-source pr-url)

"Build merge conflict resolution prompt for REVIEW-SOURCE with PR-URL."

(let ((source-instruction

(ai-code--pull-or-review-source-instruction review-source 'resolve-merge-conflict)))

(format "Resolve merge conflict for pull request: %s

pr-url source-instruction

ai-code-github--resolution-analysis-only-note)))

(defun ai-code--pull-or-review-url-prompt (review-mode)

"Return the URL prompt string for REVIEW-MODE."

(if (eq review-mode 'investigate-issue)

"GitHub issue URL: "

"Pull request URL: "))

(defconst ai-code-github--pull-or-review-pr-mode-alist

'(("Review the PR" . review-pr)

("Check unresolved feedback" . check-feedback)

("Prepare PR description" . prepare-pr-description)

("Send out PR for current branch" . send-current-branch-pr)

("Review current branch with difftastic"

. review-current-branch-with-difftastic)

("Investigate issue" . investigate-issue)

("Review GitHub CI checks" . review-ci-checks)

("Explain code change" . explain-code-change)

("Resolve merge conflict" . resolve-merge-conflict)

("Generate diff file" . generate-diff-file))

"Analysis modes offered for a pull request or issue.

`ai-code-github--analysis-only-notes' unless it really requests code changes.")

(defun ai-code--pull-or-review-pr-mode-choice ()

"Prompt user to choose analysis mode for a pull request or issue."

(let* ((review-mode-alist ai-code-github--pull-or-review-pr-mode-alist)

(review-mode (completing-read "Select analysis mode (PR or issue): "

review-mode-alist

nil t nil nil "Review the PR")))

(or (alist-get review-mode review-mode-alist nil nil #'string=)

'review-pr)))

(defun ai-code--review-current-branch-with-difftastic ()

"Review the current branch with `difftastic-magit-diff'.

(require 'difftastic nil t)

(unless (fboundp 'difftastic-magit-diff)

(user-error

(concat

"The package difftastic is not installed. "

"Install it from MELPA or https://github.com/pkryger/difftastic.el")))

(call-interactively #'difftastic-magit-diff))

(defun ai-code--send-current-branch-pr-source-instruction (review-source)

"Return PR creation instructions for REVIEW-SOURCE."

(pcase review-source

('gh-cli

(concat

"Use GitHub CLI to create the pull request. "

"Run `gh pr create` with the current branch as the head branch, "

"target the requested base branch, and include the final title and body."))

('github-mcp

(concat

"Use GitHub MCP tools to create the pull request directly. "

"Do not fetch review comments before the PR exists; "

"create the PR first, then return the resulting PR URL."))

(_

(concat

"Create the pull request using the backend's PR creation capability. "

"Do not treat this as a PR review flow before the PR exists."))))

(defun ai-code--build-pr-init-prompt (review-source target-url review-mode &optional include-context)

"Build initial prompt for REVIEW-SOURCE, TARGET-URL and REVIEW-MODE.

(pcase review-mode

('investigate-issue

(ai-code--build-issue-investigation-init-prompt review-source target-url include-context))

('check-feedback

(ai-code--build-pr-feedback-check-init-prompt review-source target-url))

('review-ci-checks

(ai-code--build-pr-ci-check-init-prompt review-source target-url))

('prepare-pr-description

(ai-code--build-pr-description-init-prompt review-source target-url))

('resolve-merge-conflict

(ai-code--build-resolve-merge-conflict-init-prompt review-source target-url))

(_

(ai-code--build-pr-review-init-prompt review-source target-url))))

(defun ai-code--pull-or-review-pr-with-source (review-source &optional arg)

"Prompt for a mode and send a prompt for REVIEW-SOURCE to AI.

(require 'ai-code-git nil t)

(let ((review-mode (ai-code--pull-or-review-pr-mode-choice)))

(cond

((eq review-mode 'generate-diff-file)

(ai-code--magit-generate-feature-branch-diff-file))

((eq review-mode 'review-current-branch-with-difftastic)

(ai-code--review-current-branch-with-difftastic))

((eq review-mode 'explain-code-change)

(unless (fboundp 'ai-code--explain-code-change)

(require 'ai-code-discussion nil t))

(unless (fboundp 'ai-code--explain-code-change)

(user-error "Code change explanation support is not available"))

(ai-code--explain-code-change review-source))

(t

(let* ((init-prompt

(if (eq review-mode 'send-current-branch-pr)

(progn

(unless (magit-toplevel)

(user-error "Not inside a Git repository"))

(let* ((current-branch (ai-code--require-current-branch))

(default-target-branch

(ai-code--default-pr-target-branch current-branch))

(target-branch

(ai-code-read-string "Target branch to merge into: "

default-target-branch))

(pr-title

(read-string

"PR title (optional, leave empty for AI to generate): "

nil

'ai-code-pr-title-history)))

(ai-code--build-send-current-branch-pr-init-prompt

review-source current-branch target-branch pr-title)))

(let* ((url-prompt (ai-code--pull-or-review-url-prompt review-mode))

(region-url (ai-code--extract-url-from-region))

(target-url (ai-code-read-string url-prompt region-url))

(has-context (or buffer-file-name (use-region-p)))

(include-context (and (eq review-mode 'investigate-issue)

has-context

(or arg

(y-or-n-p "Include active buffer and editor context? ")))))

(ai-code--build-pr-init-prompt review-source target-url review-mode include-context))))

(prompt-label (if (eq review-mode 'send-current-branch-pr)

"Enter PR creation prompt: "

"Enter review prompt: ")))

(let ((ai-code--grill-me-workflow review-mode))

(ai-code--confirm-and-send prompt-label init-prompt)))))))

(defun ai-code--get-git-web-repo-url ()

"Get Git repository web URL from git remote.

(let* ((remote-url (magit-git-string "config" "--get" "remote.origin.url")))

(when (and remote-url (stringp remote-url) (not (string-empty-p remote-url)))

(cond

;; HTTPS URL: https://host.com/user/repo.git or https://host.com/user/repo

((string-match "https://\\([^/]+\\)/\\(.+\\)" remote-url)

(let ((host (match-string 1 remote-url))

(path (match-string 2 remote-url)))

(format "https://%s/%s" host (replace-regexp-in-string "\\.git$" "" path))))

;; SSH URL: git@host.com:user/repo.git or git@host.com:user/repo

((string-match "git@\\([^:]+\\):\\(.+\\)" remote-url)

(let ((host (match-string 1 remote-url))

(path (match-string 2 remote-url)))

(format "https://%s/%s" host (replace-regexp-in-string "\\.git$" "" path))))

(t nil)))))

(defun ai-code--open-git-web-compare (start end)

"Open Git web compare page for START..END in browser.

(let ((repo-url (ai-code--get-git-web-repo-url)))

(if repo-url

(let ((compare-url (format "%s/compare/%s...%s" repo-url start end)))

(browse-url compare-url)

(message "Opened web compare: %s" compare-url))

(message "Unable to determine repository web URL"))))

(defun ai-code--open-git-web-commit (commit)

"Open Git web commit page for COMMIT in browser.

(let ((repo-url (ai-code--get-git-web-repo-url)))

(if repo-url

(let ((commit-url (format "%s/commit/%s" repo-url commit)))

(browse-url commit-url)

(message "Opened web commit: %s" commit-url))

(message "Unable to determine repository web URL"))))

(defun ai-code-pull-or-review-diff-file (&optional arg)

"Review a diff file with AI Code or choose a PR review workflow.

(interactive "P")

(if (and buffer-file-name (string-match-p "\\.diff$" buffer-file-name))

(let* ((file-name (file-name-nondirectory buffer-file-name))

(init-prompt (format "Code review for %s. Use relevant file in repository as context.

**Requirement**: " file-name

ai-code-github--analysis-only-note)))

(ai-code--confirm-and-send "Enter review prompt (type requirement at end): " init-prompt))

;; For non-diff files, let user choose PR review via MCP/gh CLI.

(unless ai-code-default-review-source

(ai-code--message-review-source-config-hint))

(let ((review-source (ai-code--pull-or-review-action-choice)))

(ai-code--pull-or-review-pr-with-source review-source arg))))

(provide 'ai-code-github)