Adds auto-transient menu display for compose buffer

Author: xenodium

chatgpt-shell-prompt-compose.el

@@ -32,6 +32,7 @@
 (require 'ring)
 (require 'flymake)
 (require 'shell-maker)
+(require 'transient)
 
 (declare-function chatgpt-shell-previous-source-block "chatgpt-shell")
 (declare-function chatgpt-shell-next-source-block "chatgpt-shell")
@@ -50,6 +51,12 @@
 (declare-function chatgpt-shell--pretty-smerge-insert "chatgpt-shell")
 (declare-function chatgpt-shell-markdown-block-at-point "chatgpt-shell")
 (declare-function chatgpt-shell-view-block-at-point "chatgpt-shell")
+(declare-function chatgpt-shell-copy-block-at-point "chatgpt-shell")
+
+(defcustom chatgpt-shell-compose-auto-transient t
+  "When non-nil automatically display transient menu post compose submission."
+  :type 'boolean
+  :group 'chatgpt-shell)
 
 (defvar-local chatgpt-shell-prompt-compose--exit-on-submit nil
   "Whether or not compose buffer should close after submission.
@@ -109,6 +116,27 @@ t if invoked from a transient frame (quitting closes the frame).")
     map)
   "Keymap for `chatgpt-shell-prompt-compose-view-mode'.")
 
+(transient-define-prefix chatgpt-shell-prompt-compose-transient ()
+  "ChatGPT Shell Compose Transient."
+  [["Navigation"
+    ("n" "Next item" chatgpt-shell-prompt-compose-next-item :transient t)
+    ("p" "Previous item" chatgpt-shell-prompt-compose-previous-item :transient t)
+    ("f" "Next interaction" chatgpt-shell-prompt-compose-next-interaction :transient t)
+    ("b" "Previous interaction" chatgpt-shell-prompt-compose-previous-interaction :transient t)
+    ("o" "View other shell buffer" chatgpt-shell-prompt-compose-other-buffer)]
+   ["Prompts"
+    ("g" "Retry" chatgpt-shell-prompt-compose-retry)
+    ("r" "Reply with followup" chatgpt-shell-prompt-compose-reply)
+    ("m" "Request more" chatgpt-shell-prompt-compose-request-more)
+    ("e" "Request entire snippet" chatgpt-shell-prompt-compose-request-entire-snippet)]
+   ["Blocks"
+    ("v" "View block at point" chatgpt-shell-view-block-at-point)
+    ("i" "Insert block at point" chatgpt-shell-prompt-compose-insert-block-at-point)
+    ("w" "Copy block at point" chatgpt-shell-copy-block-at-point)
+    ("C-c C-c" "Execute block at point" chatgpt-shell-execute-block-action-at-point)]
+   ["Quit"
+    ("q" "Quit shell compose buffer" chatgpt-shell-prompt-compose-quit-and-close-frame)]])
+
 (define-minor-mode chatgpt-shell-prompt-compose-view-mode
   "Like `view-mode`, but extended for ChatGPT Compose."
   :lighter " ChatGPT view"
@@ -366,7 +394,9 @@ Optionally set its PROMPT."
                                         (lambda (_input _output _success)
                                           (with-current-buffer (chatgpt-shell-prompt-compose-buffer)
                                             (let ((inhibit-read-only t))
-                                              (chatgpt-shell--put-source-block-overlays))))
+                                              (chatgpt-shell--put-source-block-overlays)))
+                                          (when chatgpt-shell-compose-auto-transient
+                                            (chatgpt-shell-prompt-compose-transient)))
                                         'inline))
         ;; Point should go to beginning of response after submission.
         (goto-char (point-min))

chatgpt-shell.el

@@ -2911,6 +2911,17 @@ For example \"elisp\" -> \"emacs-lisp\"."
                                 :on-finished #'identity)
     (user-error "No block at point")))
 
+(defun chatgpt-shell-copy-block-at-point ()
+  "Copy code block at point to the kill ring."
+  (interactive)
+  (if-let ((block (chatgpt-shell-markdown-block-at-point)))
+      (let ((code (buffer-substring-no-properties
+                   (map-elt block 'start)
+                   (map-elt block 'end))))
+        (kill-new code)
+        (message "Copied block to kill ring"))
+    (user-error "No block at point")))
+
 (cl-defun chatgpt-shell--view-code (&key edit language code on-finished)
   "Open a temporary buffer for editing CODE in LANGUAGE major mode.
 When done, invoke the FINISHED function with the resulting code.