(require 'consult)
(require 'beframe)
(require 'calendar)
(require 'org-roam-node)
(require 'denote)

(defvar bard/consult--source-notes
  `(:name     "Note Buffers"
    :narrow   ?n
    :category buffer
    :face     consult-buffer
    :history  buffer-name-history
    :items    ,(lambda ()
                 (mapcar #'buffer-name
                         (seq-filter
                          (lambda (buf)
                            (string-prefix-p "[Note]" (buffer-name buf)))
                          (beframe-buffer-list))))
    :action   ,#'switch-to-buffer
    :state    ,#'consult--buffer-state)
  "Consult source for note buffers (limited to beframe buffers).")


(defun bard/find-notes-file ()
  (interactive)
  (consult-find "~/Notes/denote"))

(defun bard/search-notes-directory ()
  (interactive)
  (consult-grep "~/Notes/denote"))

(defun bard/consult-buffer-notes ()
  "Show `consult-buffer` limited to buffers starting with [Note]."
  (interactive)
  (consult-buffer '(bard/consult--source-notes)))

(defun bard/ibuffer-notes ()
  "Open `ibuffer` limited to buffers starting with [Note]."
  (interactive)
  (ibuffer nil "*Ibuffer-Notes*"
           '((name . "^\\[Note\\]"))))

(defun bard/denote-insert-id-at-top ()
  "Insert or replace a top-level :ID: property at the very top of the current file."
  (interactive)
  (org-with-wide-buffer
   (goto-char (point-min))
   ;; If file already starts with a :PROPERTIES: drawer, remove it
   (when (looking-at ":PROPERTIES:")
     (let ((end (save-excursion
                  (re-search-forward ":END:" nil t))))
       (when end
         (delete-region (point-min) (min (point-max) (1+ end)))))))
  ;; Insert fresh ID drawer at very top
  (goto-char (point-min))
  (let ((id (org-id-new)))
    (insert (format ":PROPERTIES:\n:ID:       %s\n:END:\n" id)))
  (save-buffer))

(defun bard/denote-maybe-insert-id ()
  "Insert a top-level :ID: unless this is a denote journal file."
  (interactive)
  (let ((file (buffer-file-name)))
    (unless (and file
                 (boundp 'denote-journal-directory)
                 (string-prefix-p (expand-file-name denote-journal-directory)
                                  (expand-file-name file)))
      (bard/denote-insert-id-at-top))))

(defun bard/insert-or-create-node ()
  "Insert an Org-roam link if the note exists; otherwise create via Denote."
  (interactive)
  ;; Prompt for an existing node, but do not auto-create
  (let* ((node (org-roam-node-read))
         (title (org-roam-node-title node)))
    (if node
        ;; Node exists: insert Org-roam link at point
        (let ((id (org-roam-node-id node)))
          (if id
              (insert (org-link-make-string
                           (concat "id:" id)
                           title))
            ;; Node does not exist (no id): create via Denote
            (bard/denote-insert-node title))))))

(defun bard/find-or-create-node ()
  "Find an Org-roam node by title; if missing, create via Denote."
  (interactive)
  (let* ((node (org-roam-node-read nil nil))  ;; no auto-create; just selection
         (title (org-roam-node-title node)))
    (if (org-roam-node-id node)
        ;; Node exists → visit
        (org-roam-node-visit node)
      ;; Node missing → create via Denote
      (let ((keyword (denote-keywords-prompt)))
        (denote title keyword)))))

(defun bard/denote-insert-node (title)
  "Denote analogy for `org-roam-insert-node', takes TITLE as node title."
  (interactive)
  (let* ((keywords (denote-keywords-prompt))
         (file (denote title keywords nil nil nil nil nil nil)))

    (when-let ((buf (find-buffer-visiting file)))
      (with-current-buffer buf
        (save-buffer)
        (kill-buffer buf)))

    ;; extract uuid from file that is autogenerated by `bard/denote-insert-id-at-top'
    (let ((uuid
           (with-temp-buffer
             (insert-file-contents file)
             (goto-char (point-min))
             (when (re-search-forward "^:ID:\\s-+\\([[:alnum:]-]+\\)" nil t)
               (match-string 1)))))
      (unless uuid
        (error "No :ID: found in Denote file %s" file))

      ;; Insert link at point
      (insert (format "[[id:%s][%s]]" uuid title)))))

(defun denote-sequence-region ()
  "Call `denote-sequence' and insert therein the text of the active region.

Note that, currently, `denote-save-buffers' and
`denote-kill-buffers' are NOT respected.  The buffer is not
saved or killed at the end of `denote-sequence-region'."
  (declare (interactive-only t))
  (interactive)
  (if-let* (((region-active-p))
            ;; We capture the text early, otherwise it will be empty
            ;; the moment `insert' is called.
            (text (buffer-substring-no-properties (region-beginning) (region-end))))
      (progn
        (let ((denote-ignore-region-in-denote-command t)
              ;; FIXME: Find a way to insert the region before the buffer is
              ;; saved/killed by the creation command.
              (denote-save-buffers nil)
              (denote-kill-buffers nil))
          (call-interactively 'denote-sequence))
        (push-mark (point))
        (insert text)
        (run-hook-with-args 'denote-region-after-new-note-functions (mark) (point)))
    (call-interactively 'denote-sequence)))

(defvar bard/class-dirs
  '(("ANTH 204" . "~/Documents/dox/Uni/FALL2025-ANTH 204/")
    ("CHEM 201" . "~/Documents/dox/Uni/FALL2025-CHEM 201/")
    ("CHEM 207" . "~/Documents/dox/Uni/FALL2025-CHEM 207/")
    ("ENGL 105" . "~/Documents/dox/Uni/FALL2025-ENGL 105/")
    ("ENGR 101" . "~/Documents/dox/Uni/FALL2025-ENGR 101/")
    ("ENGR 110" . "~/Documents/dox/Uni/FALL2025-ENGR 110/"))
  "Mapping of class names to their document directories.")

(defvar bard/uni-notes-file "~/Notes/denote/uni.org"
  "Path to the main university org file.")

(defun bard/jump-to-class (class)
  "Jump to CLASS heading in `bard/uni-notes-file` and open its dir in dired."
  (interactive
   (list (completing-read "Class: " (mapcar #'car bard/class-dirs))))
  (let* ((dir (cdr (assoc class bard/class-dirs))))
    ;; split windows
    (delete-other-windows)
    (let ((notes-window (selected-window))
          (dired-window (split-window-right)))
      ;; open notes file and jump to heading
      (with-selected-window notes-window
        (find-file bard/uni-notes-file)
        (widen)
        (goto-char (point-min))
        (message class)
        (search-forward class nil nil))
      ;; open dired in right window
      (with-selected-window dired-window
        (dired dir)))))

;; (defun bard/jump-to-class-new-frame (class)
;;   "Open CLASS notes and dir in a new frame titled after CLASS."
;;   (interactive
;;    (list (completing-read "Class: " (mapcar #'car bard/class-dirs))))
;;   (let* ((dir (cdr (assoc class bard/class-dirs)))
;;          ;; make a new frame with title
;;          (frame (make-frame `((name . ,class)))))
;;     (select-frame-set-input-focus frame)
;;     (delete-other-windows)
;;     (let ((notes-window (selected-window))
;;           (dired-window (split-window-right)))
;;       ;; open notes file and jump to heading
;;       (with-selected-window notes-window
;;         (find-file bard/uni-notes-file)
;;         (widen)
;;         (goto-char (point-min))
;;         (search-forward class nil nil))
;;       ;; open dired in right window
;;       (with-selected-window dired-window
;;         (dired dir)))))

(defun bard/jump-to-class-new-frame (class)
  "Open CLASS notes and dir in a new frame titled after CLASS, even with beframe."
  (interactive
   (list (completing-read "Class: " (mapcar #'car bard/class-dirs))))
  (let* ((dir (cdr (assoc class bard/class-dirs)))
         (frame (make-frame `((frame-title-format . ,class)))))
    (select-frame-set-input-focus frame)
    (delete-other-windows)
    (let ((notes-window (selected-window))
          (dired-window (split-window-right)))
      (with-selected-window notes-window
        (find-file bard/uni-notes-file)
        (widen)
        (goto-char (point-min))
        (search-forward class nil nil))
      (with-selected-window dired-window
        (dired dir)
        (beframe-rename-current-frame)))))

;; Optional: bind to a key
(global-set-key (kbd "C-c u") #'bard/jump-to-class)
(global-set-key (kbd "C-c U") #'bard/jump-to-class-new-frame)

(defun bard/denote-todo-template ()
  "Return string for daily tasks heading in `denote-journal' entries"
  (format "* Tasks for %s\n\n* Notes for today"
          (format-time-string "%Y-%m-%d (%a)")))

(provide 'bard-writing)