(require 'org)

(require 'which-func)

(require 'magit)

(require 'transient)

(require 'seq)

(defgroup ai-code nil

"Unified interface for multiple AI coding CLIs."

:group 'tools)

(defcustom ai-code-menu-layout 'default

"Layout used by `ai-code-menu`.

:type '(choice (const :tag "Default multi-column menu" default)

(const :tag "Narrower two-column menu" two-columns))

:group 'ai-code)

(require 'ai-code-backends)

(require 'ai-code-backends-infra)

(require 'ai-code-input)

(require 'ai-code-prompt-mode)

(require 'ai-code-agile)

(require 'ai-code-git)

(require 'ai-code-change)

(require 'ai-code-discussion)

(require 'ai-code-codex-cli)

(require 'ai-code-aider-cli)

(require 'ai-code-github-copilot-cli)

(require 'ai-code-opencode)

(require 'ai-code-grok-cli)

(require 'ai-code-codebuddy-cli)

(require 'ai-code-file)

(require 'ai-code-harness)

(require 'ai-code-ai)

(require 'ai-code-mcp-server)

(require 'ai-code-notifications)

(require 'ai-code-onboarding)

(declare-function ai-code-cli-start "ai-code-backends")

(declare-function ai-code-cli-resume "ai-code-backends")

(declare-function ai-code-cli-switch-to-buffer "ai-code-backends")

(declare-function ai-code-cli-send-command "ai-code-backends" (command))

(declare-function ai-code-current-backend-label "ai-code-backends")

(declare-function ai-code-set-backend "ai-code-backends")

(declare-function ai-code-select-backend "ai-code-backends")

(declare-function ai-code-open-backend-config "ai-code-backends")

(declare-function ai-code-open-backend-agent-file "ai-code-backends")

(declare-function ai-code-upgrade-backend "ai-code-backends")

(defvar ai-code-mcp-agent-enabled-backends)

(declare-function ai-code-install-backend-skills "ai-code-backends")

(declare-function ai-code-backends-infra--session-buffer-p "ai-code-backends-infra" (buffer))

(declare-function ai-code--process-word-for-filepath "ai-code-prompt-mode" (word git-root-truename))

(declare-function ai-code-call-gptel-sync "ai-code-prompt-mode" (question))

(defcustom ai-code-use-gptel-headline nil

"Whether to use GPTel to generate headlines for prompt sections.

:type 'boolean

:group 'ai-code)

(defcustom ai-code-prompt-suffix nil

"Suffix text to append to prompts after a new line.

:type '(choice (const nil) string)

:group 'ai-code)

(defcustom ai-code-use-prompt-suffix t

"When non-nil, append `ai-code-prompt-suffix` where supported."

:type 'boolean

:group 'ai-code)

(defvar ai-code-auto-test-suffix ai-code-test-after-code-change-suffix

"Default prompt suffix to request running tests after code changes.")

(defvar ai-code-auto-test-type nil

"Forward declaration for `ai-code-auto-test-type'.

See the later `defcustom' for user-facing documentation and default.")

(defvar ai-code-discussion-auto-follow-up-suffix nil

"Send-time prompt suffix that requests numbered next-step suggestions.")

(defvar ai-code-discussion-auto-follow-up-enabled nil

"Forward declaration for `ai-code-discussion-auto-follow-up-enabled'.

See the later `defcustom' for user-facing documentation and default.")

(defconst ai-code--auto-test-type-ask-choices

'(("Run tests after code change" . test-after-change)

("TDD Red + Green (write failing test, then make it pass)" . tdd)

("TDD Red + Green + Blue (refactor after Green)" . tdd-with-refactoring)

("Do not write or run tests" . no-test))

"Resolve auto test suffix choices for `ask-me` mode.")

(defconst ai-code--auto-test-type-persistent-choices

'(("Ask every time" . ask-me)

("Off" . nil))

"Persistent choices for `ai-code-auto-test-type`.")

(defconst ai-code--auto-follow-up-type-ask-choices

'(("Suggest next steps" . t)

("No next-step suggestions" . nil))

"Resolve next-step suggestion choices for `ask-me` mode.")

(defconst ai-code--auto-test-type-legacy-persistent-modes

'(test-after-change tdd tdd-with-refactoring)

"Legacy persistent values still honored for backward compatibility.")

(defun ai-code--read-auto-test-type-choice ()

"Read and return one prompt test type for this send action."

(let* ((choice (completing-read "Choose test prompt type for this send: "

(mapcar #'car ai-code--auto-test-type-ask-choices)

nil t nil nil

(caar ai-code--auto-test-type-ask-choices)))

(choice-cell (assoc choice ai-code--auto-test-type-ask-choices)))

(if choice-cell

(cdr choice-cell)

'test-after-change)))

(defun ai-code--read-auto-follow-up-choice ()

"Read whether to request numbered next-step suggestions for this send action."

(let* ((choice (completing-read "Discussion follow-up suggestions: "

(mapcar #'car ai-code--auto-follow-up-type-ask-choices)

nil t nil nil

(caar ai-code--auto-follow-up-type-ask-choices)))

(choice-cell (assoc choice ai-code--auto-follow-up-type-ask-choices)))

(and choice-cell

(cdr choice-cell))))

(defcustom ai-code-use-gptel-classify-prompt nil

"Whether to use GPTel to classify prompts before send-time suffix routing.

:type 'boolean

:group 'ai-code)

(defcustom ai-code-next-step-suggestion-suffix

(concat

"At the end of your response, provide 3-4 numbered candidate next\n"

"steps. Keep each option to one sentence. At least 2 candidates must\n"

"be AI-actionable items as follow up: either a code change or tool usage. Mark the\n"

"single best option with \"(Recommended)\". If the user replies with\n"

"only a number such as 1, 2, 3, or 4, treat that as selecting the\n"

"corresponding next step from your previous answer. The user may also\n"

"ignore these options and send a different follow-up request instead.")

"Prompt suffix for numbered next-step suggestions in discussion prompts."

:type '(choice (const nil) string)

:group 'ai-code)

(defun ai-code--downcase-strings (strings)

"Return STRINGS converted to lowercase."

(mapcar #'downcase strings))

(defconst ai-code--code-change-prompt-markers

(ai-code--downcase-strings

(list ai-code-change--selected-region-note

ai-code-change--generic-note

ai-code-change--selected-files-note))

"Prompt markers that clearly indicate a code-change request.")

(defconst ai-code--non-code-change-prompt-markers

(append

(ai-code--downcase-strings

(list ai-code-discussion--question-only-note

ai-code-discussion--selected-region-note))

(ai-code--downcase-strings

ai-code-discussion--explain-prompt-prefixes))

"Prompt markers that clearly indicate a non-code-change request.")

(defun ai-code--prompt-contains-any-marker-p (text markers)

"Return non-nil when any string in MARKERS appears in TEXT."

(seq-some (lambda (marker)

(string-match-p (regexp-quote marker) text))

markers))

(defun ai-code--simple-classify-prompt-code-change (prompt-text)

"Classify PROMPT-TEXT with cheap string matching before GPTel.

(let ((text (downcase (or prompt-text ""))))

(cond

((ai-code--prompt-contains-any-marker-p text

ai-code--code-change-prompt-markers)

'code-change)

((ai-code--prompt-contains-any-marker-p text

ai-code--non-code-change-prompt-markers)

'non-code-change)

(t 'unknown))))

(defun ai-code--classify-prompt-code-change (prompt-text)

"Classify whether PROMPT-TEXT requests a code change.

(let ((classification

(ai-code--simple-classify-prompt-code-change prompt-text)))

(if (eq classification 'unknown)

(ai-code--gptel-classify-prompt-code-change prompt-text)

classification)))

(defun ai-code--gptel-classify-prompt-code-change (prompt-text)

"Classify whether PROMPT-TEXT requests a code change using GPTel.

(let ((classification

(condition-case err

(if (require 'gptel nil t)

(let* ((raw-answer (ai-code-call-gptel-sync

(concat "Classify whether this user prompt requests program code changes in a repository.\n"

"Reply with exactly one token: CODE_CHANGE or NOT_CODE_CHANGE.\n"

"Treat edit/refactor/implement/fix/add/remove/update/tests as CODE_CHANGE.\n"

"Treat explain/summarize/discuss/review without editing as NOT_CODE_CHANGE.\n\n"

"Prompt:\n" prompt-text)))

(answer (upcase (string-trim (or raw-answer "")))))

(cond

((string-match-p "\\`CODE_CHANGE\\b" answer) 'code-change)

((string-match-p "\\`NOT_CODE_CHANGE\\b" answer) 'non-code-change)

(t 'unknown)))

'unknown)

(error

(message "GPTel prompt classification failed: %s" (error-message-string err))

'unknown))))

(message "GPTel prompt classification result: %s" classification)

classification))

(defun ai-code--resolve-auto-test-type-for-send (&optional prompt-text classification)

"Resolve the concrete auto test type for current send action for PROMPT-TEXT.

(if (eq ai-code-auto-test-type 'ask-me)

(ai-code--resolve-ask-auto-test-type-for-send prompt-text classification)

(and (memq ai-code-auto-test-type

ai-code--auto-test-type-legacy-persistent-modes)

ai-code-auto-test-type)))

(defun ai-code--resolve-ask-auto-test-type-for-send (&optional prompt-text classification)

"Resolve the send-time auto test type for ask-me mode with PROMPT-TEXT.

(if ai-code-use-gptel-classify-prompt

(pcase (or classification

(ai-code--classify-prompt-code-change prompt-text))

('code-change (ai-code--read-auto-test-type-choice))

('non-code-change nil)

(_ (ai-code--read-auto-test-type-choice)))

(ai-code--read-auto-test-type-choice)))

(defun ai-code--resolve-auto-follow-up-suffix-for-send (&optional prompt-text classification)

"Resolve next-step suggestion suffix for current send action for PROMPT-TEXT.

(when (and ai-code-discussion-auto-follow-up-enabled

ai-code-next-step-suggestion-suffix)

(let ((classification (or classification

(and ai-code-use-gptel-classify-prompt

(ai-code--classify-prompt-code-change prompt-text)))))

(unless (eq classification 'code-change)

(and (ai-code--read-auto-follow-up-choice)

ai-code-next-step-suggestion-suffix)))))

(defun ai-code--resolve-auto-test-suffix-for-send (&optional prompt-text classification)

"Resolve auto test suffix for current send action for PROMPT-TEXT.

(ai-code--auto-test-suffix-for-type

(ai-code--resolve-auto-test-type-for-send prompt-text classification)))

(defun ai-code--classify-prompt-for-send (&optional prompt-text)

"Return prompt classification for PROMPT-TEXT when needed.

(when (and ai-code-use-gptel-classify-prompt

(or ai-code-auto-test-type

ai-code-discussion-auto-follow-up-enabled))

(ai-code--classify-prompt-code-change prompt-text)))

(defun ai-code--with-auto-test-suffix-for-send (orig-fun prompt-text)

"Resolve and bind send-time suffixes before calling ORIG-FUN with PROMPT-TEXT."

(let* ((classification (ai-code--classify-prompt-for-send prompt-text))

(ai-code-auto-test-suffix

(ai-code--resolve-auto-test-suffix-for-send

prompt-text classification))

(ai-code-discussion-auto-follow-up-suffix

(ai-code--resolve-auto-follow-up-suffix-for-send

prompt-text classification)))

(funcall orig-fun prompt-text)))

(unless (advice-member-p #'ai-code--with-auto-test-suffix-for-send

'ai-code--write-prompt-to-file-and-send)

(advice-add 'ai-code--write-prompt-to-file-and-send

:around

#'ai-code--with-auto-test-suffix-for-send))

(defun ai-code--test-after-code-change--set (symbol value)

"Set SYMBOL to VALUE and update related suffix behavior."

(set-default symbol value)

(set symbol value)

(setq ai-code-auto-test-suffix

(ai-code--auto-test-suffix-for-type value)))

(defun ai-code--apply-auto-test-type (value)

"Set `ai-code-auto-test-type` to VALUE and refresh related suffix."

(setq ai-code-auto-test-type value)

(ai-code--test-after-code-change--set 'ai-code-auto-test-type value)

value)

(defun ai-code--apply-discussion-auto-follow-up-enabled (value)

"Set `ai-code-discussion-auto-follow-up-enabled` to VALUE."

(setq ai-code-discussion-auto-follow-up-enabled value)

value)

(defcustom ai-code-auto-test-type nil

"Select how prompts request tests after code changes."

:type '(choice (const :tag "Ask every time" ask-me)

(const :tag "Off" nil))

:set #'ai-code--test-after-code-change--set

:group 'ai-code)

(defcustom ai-code-discussion-auto-follow-up-enabled nil

"When non-nil, prompts may request numbered next-step suggestions.

:type 'boolean

:set (lambda (symbol value)

(set-default symbol value)

(set symbol value))

:group 'ai-code)

(defcustom ai-code-cli "claude"

"The command-line AI tool to use for `ai-code-apply-prompt-on-current-file`."

:type 'string

:group 'ai-code)

(defun ai-code--get-clipboard-text ()

"Return the current clipboard contents as a plain string, or nil if unavailable."

(let* ((selection (when (fboundp 'gui-get-selection)

(or (let ((text (gui-get-selection 'CLIPBOARD 'UTF8_STRING)))

(and (stringp text) (not (string-empty-p text)) text))

(let ((text (gui-get-selection 'CLIPBOARD 'STRING)))

(and (stringp text) (not (string-empty-p text)) text)))))

(kill-text (condition-case nil

(current-kill 0 t)

(error nil))))

(let ((text (or selection kill-text)))

(when (stringp text)

(substring-no-properties text)))))

(defun ai-code-send-command (arg)

"Read a prompt from the user and send it to the AI service.

;; Prefix levels control whether files/repo and clipboard context are included,

;; and the prompt label reflects the selected context.

(interactive "P")

(let* ((initial-input (when (use-region-p)

(string-trim-right

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

(region-end))

"\n")))

(prefix-value (when arg (prefix-numeric-value arg)))

(include-files-and-repo (and arg (>= prefix-value 4)))

(include-clipboard (and arg (>= prefix-value 16)))

(files-context-string (when include-files-and-repo

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

(repo-context-string (when include-files-and-repo

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

(clipboard-context (when include-clipboard

(ai-code--get-clipboard-text)))

(prompt-label

(cond

(include-clipboard "Send to AI (files/repo/clipboard context): ")

(include-files-and-repo "Send to AI (files/repo context): ")

(t "Send to AI: "))))

(when-let* ((prompt (ai-code-read-string prompt-label initial-input)))

(let ((final-prompt

(concat prompt

(or files-context-string "")

(or repo-context-string "")

(when (and clipboard-context

(string-match-p "\\S-" clipboard-context))

(concat "\n\nClipboard context:\n"

clipboard-context)))))

(ai-code--insert-prompt final-prompt)))))

(defun ai-code-cli-switch-to-buffer-or-hide ()

"Hide the current buffer when its name both begins and ends with '*'.

(interactive)

(if (and current-prefix-arg

(ai-code-backends-infra--session-buffer-p (current-buffer)))

(quit-window)

;; Try with argument first; fall back to no-arg call if function doesn't accept it

(condition-case nil

(ai-code-cli-switch-to-buffer t)

(wrong-number-of-arguments ;; will be triggered during calling corresponding function in external backends such as claude-code-ide.el, claude-code.el, since the corresponding function doesn't have parameter

(ai-code-cli-switch-to-buffer)))))

(defclass ai-code--use-prompt-suffix-type (transient-lisp-variable)

((variable :initform 'ai-code-use-prompt-suffix)

(format :initform "%k %d %v")

(reader :initform #'transient-lisp-variable--read-value))

"Toggle helper for `ai-code-use-prompt-suffix`.")

(transient-define-infix ai-code--infix-toggle-suffix ()

"Toggle `ai-code-use-prompt-suffix`."

:class 'ai-code--use-prompt-suffix-type

:key "^"

:description "Use prompt suffix:"

:reader (lambda (_prompt _initial-input _history)

(not ai-code-use-prompt-suffix)))

(defclass ai-code--code-change-auto-test-type (transient-lisp-variable)

((variable :initform 'ai-code-auto-test-type)

(format :initform "%k %d %v")

(reader :initform #'transient-lisp-variable--read-value))

"Selection helper for `ai-code-auto-test-type`.")

(transient-define-infix ai-code--infix-select-code-change-auto-test ()

"Select `ai-code-auto-test-type` mode."

:class 'ai-code--code-change-auto-test-type

:key "T"

:description "Auto test type:"

:reader (lambda (_prompt _initial-input _history)

(let* ((choices ai-code--auto-test-type-persistent-choices)

(choice (completing-read "Test after code change: "

(mapcar #'car choices)

nil t nil nil

(caar choices))))

(let ((value (cdr (assoc choice choices))))

(ai-code--apply-auto-test-type value)

(message "Auto test type set to %s; prompt suffix is now %s"

(or value "off")

(if (eq value 'ask-me)

"ask each send"

(or ai-code-auto-test-suffix "cleared")))

value))))

(defclass ai-code--discussion-auto-follow-up-enabled-type (transient-lisp-variable)

((variable :initform 'ai-code-discussion-auto-follow-up-enabled)

(format :initform "%k %d %v")

(reader :initform #'transient-lisp-variable--read-value))

"Selection helper for `ai-code-discussion-auto-follow-up-enabled`.")

(transient-define-infix ai-code--infix-toggle-auto-follow-up ()

"Toggle `ai-code-discussion-auto-follow-up-enabled`."

:class 'ai-code--discussion-auto-follow-up-enabled-type

:key "F"

:description "Discussion follow-up:"

:reader (lambda (_prompt _initial-input _history)

(ai-code--apply-discussion-auto-follow-up-enabled

(not ai-code-discussion-auto-follow-up-enabled))))

(defun ai-code--select-backend-description (&rest _)

"Dynamic description for the Select Backend menu item.

(format "Select Backend (%s)" (ai-code-current-backend-label)))

(defconst ai-code--terminal-backend-choices

'(("vterm" . vterm)

("eat" . eat)

("ghostel" . ghostel))

"Display choices for `ai-code-backends-infra-terminal-backend'.")

(defun ai-code--ordered-terminal-backend-choices ()

"Return terminal backend choices with the current backend first."

(let ((current-label

(car (seq-find

(lambda (it)

(eq (cdr it) ai-code-backends-infra-terminal-backend))

ai-code--terminal-backend-choices))))

(if current-label

(let ((current (assoc current-label ai-code--terminal-backend-choices)))

(cons current

(seq-remove (lambda (it)

(equal (car it) current-label))

ai-code--terminal-backend-choices)))

ai-code--terminal-backend-choices)))

(defun ai-code-select-terminal ()

"Interactively select the terminal backend for AI sessions."

(interactive)

(let* ((ordered-choices (ai-code--ordered-terminal-backend-choices))

(current-label (caar ordered-choices))

(choice (completing-read "Select terminal: "

(mapcar #'car ordered-choices)

nil t nil nil current-label))

(backend (cdr (assoc choice ordered-choices))))

(setq ai-code-backends-infra-terminal-backend backend)

(ai-code-backends-infra--sync-reflow-filter-advice)

(message "AI Code terminal backend switched to: %s" choice)))

(defun ai-code--select-terminal-description (&rest _)

"Dynamic description for the Select Terminal menu item."

(format "Select Terminal (%s)"

(symbol-name ai-code-backends-infra-terminal-backend)))

(transient-define-group ai-code--menu-ai-cli-session

("a" "Start AI CLI (C-u: args)" ai-code-cli-start)

("R" "Resume AI CLI (C-u: args)" ai-code-cli-resume)

("z" "Switch to AI CLI (C-u: hide)" ai-code-cli-switch-to-buffer-or-hide)

("s" ai-code-select-backend :description ai-code--select-backend-description)

;; DONE: similar to ai-code-select-backend, add ai-code-select-terminal, it will use ai-code-backends-infra-terminal-backend to select between different terminal emulators for AI sessions, such as vterm, eat, and ghostel.

("u" "Install / Upgrade AI CLI" ai-code-upgrade-backend)

("S" "(Un)Install skills for backend" ai-code-install-backend-skills)

("g" "Open backend config (eg. add mcp)" ai-code-open-backend-config)

("G" "Open backend repo agent file" ai-code-open-backend-agent-file)

("l" ai-code-select-terminal :description ai-code--select-terminal-description)

("|" "Apply prompt on file" ai-code-apply-prompt-on-current-file))

(transient-define-group ai-code--menu-actions-with-context

(ai-code--infix-toggle-suffix)

("c" "Code change (C-u: clipboard)" ai-code-code-change)

("i" "Implement TODO (C-u: clipboard)" ai-code-implement-todo)

("q" "Ask question (C-u: clipboard)" ai-code-ask-question)

("x" "Explain code in scope" ai-code-explain)

("<SPC>" "Send command (C-u: context)" ai-code-send-command)

("@" "Context (add/show/clear)" ai-code-context-action)

("C" "Create file or dir with AI" ai-code-create-file-or-dir)

("w" "New worktree branch (C-u: status)" ai-code-git-worktree-action))

(transient-define-group ai-code--menu-agile-development

(ai-code--infix-select-code-change-auto-test)

("r" "Refactor Code" ai-code-refactor-book-method)

("t" "Test Driven Development" ai-code-tdd-cycle)

("v" "GitHub PR AI Action" ai-code-pull-or-review-diff-file)

("!" "Run Current File or Command" ai-code-run-current-file-or-shell-cmd)

("b" "Build/Test/Lint (AI follow-up)" ai-code-build-or-test-project)

("K" "Create/Open task file (C-u: Search)" ai-code-create-or-open-task-file)

("n" "Take notes from AI session region" ai-code-take-notes)

(":" "Speech to text input" ai-code-speech-to-text-input))

(transient-define-group ai-code--menu-other-tools

(ai-code--infix-toggle-auto-follow-up)

("." "Init projectile and gtags" ai-code-init-project)

("e" "Debug exception (C-u: clipboard)" ai-code-investigate-exception)

("f" "Fix Flycheck errors in scope" ai-code-flycheck-fix-errors-in-scope)

("k" "Copy Cur File Name (C-u: full)" ai-code-copy-buffer-file-name-to-clipboard)

;; ("o" "Open recent file (C-u: insert)" ai-code-git-repo-recent-modified-files)

("p" "Open prompt history file" ai-code-open-prompt-file)

("m" "Debug python MCP server" ai-code-debug-mcp)

("N" "Toggle notifications" ai-code-notifications-toggle)

("h" "Help / Quick Start" ai-code-onboarding-open-quickstart))

(transient-define-prefix ai-code-menu-default ()

"Default transient menu for AI Code Interface interactive functions."

["AI Code Commands"

["AI CLI session" ai-code--menu-ai-cli-session]

["AI Code Actions With Context" ai-code--menu-actions-with-context]

["AI Agile Development With Harness" ai-code--menu-agile-development]

["Other Tools" ai-code--menu-other-tools]])

(transient-define-prefix ai-code-menu-2-columns ()

"Narrower two-column transient menu for AI Code Interface interactive functions."

["AI Code Commands"

["AI CLI session" ai-code--menu-ai-cli-session]

["AI Code Actions With Context" ai-code--menu-actions-with-context]]

[["AI Agile Development With Harness" ai-code--menu-agile-development]

["Other Tools" ai-code--menu-other-tools]])

(defun ai-code--menu-prefix-command ()

"Return the transient prefix command selected by `ai-code-menu-layout`."

(pcase ai-code-menu-layout

('two-columns #'ai-code-menu-2-columns)

('default #'ai-code-menu-default)

(_ #'ai-code-menu-default)))

(defun ai-code-menu ()

"Show the AI Code transient menu selected by `ai-code-menu-layout`."

(interactive)

(ai-code-onboarding-maybe-show-quickstart)

(call-interactively (ai-code--menu-prefix-command)))

(provide 'ai-code)