addressing comments

Author: Kang Tu

ai-code-prompt-mode.el

@@ -415,21 +415,21 @@ that root, otherwise return the absolute path."
 (defun ai-code--insert-prompt (prompt-text)
   "Preprocess and insert PROMPT-TEXT into the AI prompt file.
 If PROMPT-TEXT is a command (starts with /), execute it directly instead."
-  (let* ((processed-prompt (if ai-code-prompt-preprocess-filepaths
-                               (ai-code--preprocess-prompt-text prompt-text)
-                             prompt-text))
-         (append-summary-p (and (derived-mode-p 'ai-code-prompt-mode)
-                                (org-at-heading-p)
-                                (y-or-n-p "Append result summary to current section? ")))
-         (final-prompt (if append-summary-p
-                           (concat processed-prompt
-                                   (format "\n\nAfter completing, append a concise result summary as a sub-heading at the end of the current section in file %s near line %d."
-                                           buffer-file-name (line-number-at-pos)))
-                         processed-prompt)))
-    (if (and (string-prefix-p "/" final-prompt)
-             (not (string-match-p " " final-prompt)))
-        (ai-code--execute-command final-prompt)
-      (ai-code--write-prompt-to-file-and-send final-prompt))))
+  (let ((processed-prompt (if ai-code-prompt-preprocess-filepaths
+                              (ai-code--preprocess-prompt-text prompt-text)
+                            prompt-text)))
+    (if (and (string-prefix-p "/" processed-prompt)
+             (not (string-match-p " " processed-prompt)))
+        (ai-code--execute-command processed-prompt)
+      (let* ((append-summary-p (and (derived-mode-p 'ai-code-prompt-mode)
+                                    (org-at-heading-p)
+                                    (y-or-n-p "Append result summary to current section? ")))
+             (final-prompt (if append-summary-p
+                               (concat processed-prompt
+                                       (format "\n\nAfter completing, append a concise result summary as a sub-heading at the end of the current section in file %s near line %d."
+                                               buffer-file-name (line-number-at-pos)))
+                             processed-prompt)))
+        (ai-code--write-prompt-to-file-and-send final-prompt)))))
 
 ;; Define the AI Prompt Mode (derived from org-mode)
 ;;;###autoload

test/test_ai-code-change.el

@@ -909,13 +909,12 @@ is between the function definition and its body."
         (should-not (string-match-p "summary" captured-prompt))))))
 
 (ert-deftest ai-code-test-insert-prompt-org-heading-append-summary-yes ()
-  "Test that `ai-code--insert-prompt' appends summary instruction on org heading when user confirms."
+  "Test that `ai-code--insert-prompt' appends summary instruction on prompt-mode heading when user confirms."
   (with-temp-buffer
-    (require 'org)
-    (setq buffer-file-name "/tmp/project/plan.org")
+    (setq buffer-file-name "/tmp/project/.ai.code.files/.ai.code.prompt.org")
     (insert "* TODO Build search feature\n")
     (insert "Design the API first.\n")
-    (org-mode)
+    (ai-code-prompt-mode)
     (goto-char (point-min))
 
     (let (captured-prompt
@@ -929,16 +928,15 @@ is between the function definition and its body."
         (should (stringp captured-prompt))
         (should (string-match-p "Test prompt text" captured-prompt))
         (should (string-match-p "summary" captured-prompt))
-        (should (string-match-p "plan\\.org" captured-prompt))))))
+        (should (string-match-p "\\.ai\\.code\\.prompt\\.org" captured-prompt))))))
 
 (ert-deftest ai-code-test-insert-prompt-org-heading-append-summary-no ()
   "Test that `ai-code--insert-prompt' does NOT append summary when user declines."
   (with-temp-buffer
-    (require 'org)
-    (setq buffer-file-name "/tmp/project/plan.org")
+    (setq buffer-file-name "/tmp/project/.ai.code.files/.ai.code.prompt.org")
     (insert "* TODO Build search feature\n")
     (insert "Design the API first.\n")
-    (org-mode)
+    (ai-code-prompt-mode)
     (goto-char (point-min))
 
     (let (captured-prompt
@@ -953,8 +951,8 @@ is between the function definition and its body."
         (should (string-match-p "Test prompt text" captured-prompt))
         (should-not (string-match-p "summary" captured-prompt))))))
 
-(ert-deftest ai-code-test-insert-prompt-non-org-no-summary-asked ()
-  "Test that `ai-code--insert-prompt' does NOT ask about summary in non-org buffer."
+(ert-deftest ai-code-test-insert-prompt-non-prompt-mode-no-summary-asked ()
+  "Test that `ai-code--insert-prompt' does NOT ask about summary in non-prompt-mode buffer."
   (with-temp-buffer
     (setq buffer-file-name "/tmp/project/test.el")
     (setq-local comment-start ";")
@@ -976,6 +974,30 @@ is between the function definition and its body."
         (should (string-match-p "Test prompt text" captured-prompt))
         (should-not (string-match-p "summary" captured-prompt))))))
 
+(ert-deftest ai-code-test-insert-prompt-slash-command-on-heading-executes ()
+  "Test that slash commands execute directly even when on a prompt-mode heading."
+  (with-temp-buffer
+    (setq buffer-file-name "/tmp/project/.ai.code.files/.ai.code.prompt.org")
+    (insert "* TODO Build search feature\n")
+    (ai-code-prompt-mode)
+    (goto-char (point-min))
+
+    (let (y-or-n-called command-executed
+          (ai-code-prompt-preprocess-filepaths nil))
+      (cl-letf (((symbol-function 'y-or-n-p)
+                 (lambda (_)
+                   (setq y-or-n-called t)
+                   t))
+                ((symbol-function 'ai-code--execute-command)
+                 (lambda (_cmd) (setq command-executed t)))
+                ((symbol-function 'ai-code--write-prompt-to-file-and-send)
+                 (lambda (_) (error "Should not reach write-prompt"))))
+
+        (ai-code--insert-prompt "/status")
+
+        (should command-executed)
+        (should-not y-or-n-called)))))
+
 (provide 'test_ai-code-change)
 
 ;;; test_ai-code-change.el ends here