Feat: Add note taking functions, useful during discussing with AI (#46)

* add function for notes taking

* Add .ai.code.notes.org to config and gitignore entries

* Bump version to 0.52 and add HISTORY

* Update ai-code-discussion.el

Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com>

* Refactor note file name customization and gitignore entries for AI
code notes

* Update ai-code-discussion.el

Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com>

* Update ai-code-discussion.el

Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com>

---------

Co-authored-by: Kang Tu <kang_tu@apple.com>
Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com>

Author: 3 people

HISTORY.org

@@ -3,6 +3,7 @@
 
 ** Main branch change
 
+- Feat: Add note taking functions, useful during discussing with AI
 - Fix function detection for TODO comments preceding method definitions 
   - Addressing [[https://github.com/tninja/ai-code-interface.el/issues/40][Wrong function detection]], suggested by @Silex
 - Refactor: error investigation prompt, run command with comint buffer

ai-code-discussion.el

@@ -17,6 +17,7 @@
 (declare-function ai-code-read-string "ai-code-input")
 (declare-function ai-code--insert-prompt "ai-code-prompt-mode")
 (declare-function ai-code--get-clipboard-text "ai-code-interface")
+(declare-function ai-code-call-gptel-sync "ai-code-prompt-mode")
 
 ;;;###autoload
 (defun ai-code-ask-question (arg)
@@ -324,7 +325,6 @@ Explain what this function does, its parameters, return value, algorithm, and it
       (when final-prompt
         (ai-code--insert-prompt final-prompt)))))
 
-
 (defun ai-code--explain-file ()
   "Explain the current file."
   (let ((file-name (or buffer-file-name "current buffer")))
@@ -334,6 +334,85 @@ Explain what this function does, its parameters, return value, algorithm, and it
       (when final-prompt
         (ai-code--insert-prompt final-prompt)))))
 
+;;;###autoload
+(defcustom ai-code-notes-file-name ".ai.code.notes.org"
+  "Default note file name relative to the project root.
+This value is used by `ai-code-take-notes' when suggesting where to store notes."
+  :type 'string
+  :group 'ai-code)
+
+;;;###autoload
+(defcustom ai-code-notes-use-gptel-headline nil
+  "Whether to use GPTel to generate headline for notes.
+If non-nil, call `ai-code-call-gptel-sync` to generate a smart default
+headline based on the selected content. Otherwise, prompt with empty default."
+  :type 'boolean
+  :group 'ai-code)
+
+;;;###autoload
+(defun ai-code-take-notes ()
+  "Take notes from selected region and save to a note file.
+When there is a selected region, ask for note file path (default is
+.ai.code.notes.org in the git root) and section title.  Add the section
+title as a headline at the end of the note file, and put the selected
+region as content of that section."
+  (interactive)
+  (unless (region-active-p)
+    (user-error "No region selected. Please select the text you want to save as notes"))
+  (let* ((region-text (buffer-substring-no-properties (region-beginning) (region-end)))
+         (git-root (condition-case nil
+                       (magit-toplevel)
+                     (error nil)))
+         (default-note-file (if git-root
+                                (expand-file-name ai-code-notes-file-name git-root)
+                              (expand-file-name ai-code-notes-file-name default-directory)))
+         (note-file (read-file-name
+                     "Note file: "
+                     (file-name-directory default-note-file)
+                     default-note-file
+                     nil
+                     (file-name-nondirectory default-note-file)))
+         (default-title (when ai-code-notes-use-gptel-headline
+                          (condition-case err
+                              (string-trim
+                               (ai-code-call-gptel-sync
+                                (format "Generate a concise headline (max 10 words) for this note content. Only return the headline text without quotes or extra formatting:\n\n%s"
+                                        (if (> (length region-text) 500)
+                                            (substring region-text 0 500)
+                                          region-text))))
+                            (error
+                             (message "GPTel headline generation failed: %s" (error-message-string err))
+                             ""))))
+         (section-title (ai-code-read-string "Section title: " (or default-title ""))))
+    (when (string-empty-p section-title)
+      (user-error "Section title cannot be empty"))
+    ;; Create note file directory if it doesn't exist
+    (let ((note-dir (file-name-directory note-file)))
+      (unless (file-exists-p note-dir)
+        (make-directory note-dir t)))
+    ;; Append section to note file
+    (with-current-buffer (find-file-noselect note-file)
+      (save-excursion
+        (goto-char (point-max))
+        ;; Add newline before new section if file is not empty
+        (unless (bobp)
+          (insert "\n\n"))
+        ;; Insert headline
+        (insert "* " section-title "\n")
+        ;; Insert timestamp
+        (org-insert-time-stamp (current-time) t nil)
+        (insert "\n\n")
+        ;; Insert region content
+        (insert region-text)
+        (insert "\n"))
+      (save-buffer))
+    ;; Open note file in other window and scroll to bottom
+    (let ((note-buffer (find-file-other-window note-file)))
+      (with-selected-window (get-buffer-window note-buffer)
+        (goto-char (point-max))
+        (recenter -1)))
+    (message "Notes added to %s under section: %s" note-file section-title)))
+
 (provide 'ai-code-discussion)
 
 ;;; ai-code-discussion.el ends here

ai-code-git.el

@@ -545,7 +545,12 @@ If not inside a Git repository, do nothing."
     (if (not git-root)
         (message "ai-code-update-git-ignore: not in a git repository, skipped")
       (let* ((gitignore-path (expand-file-name ".gitignore" git-root))
-             (required-entries '(".ai.code.prompt.org" ".projectile" "GTAGS" "GRTAGS" "GPATH"))
+             (required-entries (list ai-code-prompt-file-name
+                                     ai-code-notes-file-name
+                                     ".projectile"
+                                     "GTAGS"
+                                     "GRTAGS"
+                                     "GPATH"))
              (gitignore-content (when (file-exists-p gitignore-path)
                                   (with-temp-buffer
                                     (insert-file-contents gitignore-path)

ai-code-interface.el

@@ -1,7 +1,7 @@
 ;;; ai-code-interface.el --- AI code interface for editing AI prompt files -*- lexical-binding: t; -*-
 
 ;; Author: Kang Tu <tninja@gmail.com>
-;; Version: 0.51
+;; Version: 0.52
 ;; Package-Requires: ((emacs "26.1") (transient "0.8.0") (magit "2.1.0"))
 
 ;; SPDX-License-Identifier: Apache-2.0
@@ -174,6 +174,7 @@ Shows the current backend label to the right."
     ("k" "Copy Cur File Name (C-u: full)" ai-code-copy-buffer-file-name-to-clipboard)
     ("o" "Open Clipboard file dir" ai-code-open-clipboard-file-path-as-dired)
     ("m" "Debug python MCP server" ai-code-debug-mcp)
+    ("n" "Take notes from AI session region" ai-code-take-notes)
     ]
    ])