;;; keybinds.el --- All keybindings -*- lexical-binding: t -*-

;; C-c to escape insert mode
(define-key evil-insert-state-map (kbd "C-c") 'evil-normal-state)

;;; Global keys
(global-set-key (kbd "M-x") 'execute-extended-command)
(global-set-key (kbd "C-x b") 'consult-buffer)
(global-set-key (kbd "C-c m s") 'magit-status)
(global-set-key (kbd "C-c m l") 'magit-log)

;;; Multiple cursors
(global-set-key (kbd "C-S-c C-S-c") 'mc/edit-lines)
(global-set-key (kbd "C->")         'mc/mark-next-like-this)
(global-set-key (kbd "C-<")         'mc/mark-previous-like-this)
(global-set-key (kbd "C-c C-<")     'mc/mark-all-like-this)
(global-set-key (kbd "C-\"")        'mc/skip-to-next-like-this)
(global-set-key (kbd "C-:")         'mc/skip-to-previous-like-this)

;;; Move text
(global-set-key (kbd "M-p") 'move-text-up)
(global-set-key (kbd "M-n") 'move-text-down)

;;; Dired evil bindings
(with-eval-after-load 'dired
  (evil-define-key 'normal dired-mode-map
    "." 'dired-create-empty-file
    "h" 'dired-up-directory
    "l" 'dired-find-file
    "n" 'evil-search-next
    "N" 'evil-search-previous))

;;; Org mode evil bindings
(defun my/org-heading-has-checkbox-p ()
  "Check if current heading has [ ] or [X] checkbox."
  (save-excursion
    (beginning-of-line)
    (looking-at "^\\*+\\s-+\\[[ X-]\\]")))

(defun my/org-update-parent-cookie ()
  "Update parent heading's [/] or [n/m] cookie by counting child heading checkboxes."
  (save-excursion
    (when (org-up-heading-safe)
      (let ((start (point))
            (end (save-excursion (org-end-of-subtree t) (point)))
            (level (org-current-level))
            (checked 0)
            (total 0))
        ;; Count direct child headings with checkboxes
        (save-excursion
          (while (re-search-forward (format "^\\*\\{%d\\}\\s-+\\[\\([ X-]\\)\\]" (1+ level)) end t)
            (setq total (1+ total))
            (when (string= (match-string 1) "X")
              (setq checked (1+ checked)))))
        ;; Update the cookie in parent heading
        (beginning-of-line)
        (when (re-search-forward "\\[\\([0-9]*/[0-9]*\\|/\\)\\]" (line-end-position) t)
          (replace-match (format "[%d/%d]" checked total) t t))))))

(defun my/org-toggle-heading-checkbox ()
  "Toggle [ ] <-> [X] in current heading and update parent cookie."
  (save-excursion
    (beginning-of-line)
    (when (re-search-forward "\\[\\([ X-]\\)\\]" (line-end-position) t)
      (replace-match (if (string= (match-string 1) " ") "[X]" "[ ]") t t)))
  (my/org-update-parent-cookie))

(defun my/org-dwim-at-point ()
  "Do-what-I-mean at point: toggle checkbox, follow link, or cycle TODO."
  (interactive)
  (cond
   ;; Heading with [ ] or [X]
   ((my/org-heading-has-checkbox-p)
    (my/org-toggle-heading-checkbox))
   ;; List checkbox
   ((org-at-item-checkbox-p)
    (org-toggle-checkbox))
   ;; Link
   ((org-in-regexp org-link-any-re)
    (org-open-at-point))
   ;; Regular TODO heading
   ((org-at-heading-p)
    (org-todo))
   (t (org-return))))

(with-eval-after-load 'org
  (evil-define-key 'normal org-mode-map
    (kbd "RET") 'my/org-dwim-at-point
    (kbd "<return>") 'my/org-dwim-at-point
    "t" 'org-todo))

;;; LSP keybindings
(with-eval-after-load 'eglot
  (evil-define-key 'normal eglot-mode-map
    "K" 'eldoc-box-help-at-point
    "gd" 'xref-find-definitions
    "gr" 'xref-find-references))

;;; Leader keybindings (SPC)
(evil-leader/set-key
  ;; files
  "ff" 'my/telescope-find-files
  "fF" 'my/fzf-find-file
  "fg" 'my/consult-ripgrep
  "fs" 'my/consult-ripgrep-symbol
  "fo" 'consult-recent-file
  "fl" 'consult-line
  "fi" 'my/find-emacs-config
  "fp" 'my/switch-project
  ;; buffers
  "bb" 'consult-buffer
  "fb" 'consult-buffer
  "bp" 'previous-buffer
  "bn" 'next-buffer
  "bd" 'kill-current-buffer
  "bm" 'ibuffer
  ;; custom
  "cr" 'my/set-base-dir
  "cd" (lambda () (interactive) (dired (file-name-directory (or buffer-file-name default-directory))))
  "cl" 'my/close-popup-window
  "cn" 'next-error
  "cp" 'previous-error
  "cc" 'compile
  "cm" 'recompile
  "cb" (lambda () (interactive) (compile "bear -- make"))
  ;; magit
  "ms" 'magit-status
  "ml" 'magit-log
  ;; ssh/servers
  ; "st" 'my/connect-tonydev
  ;; terminal
  "to" 'my/vterm-here
  ;; window
  "wv" 'split-window-right
  "ws" 'split-window-below
  "wd" 'delete-window
  "wh" 'evil-window-left
  "wj" 'evil-window-down
  "wk" 'evil-window-up
  "wl" 'evil-window-right
  ;; reformat
  "qq" 'my/reformat-parenthesized-content
  ;; ERC
  "ec" 'my/erc-connect-libera)