tvl-depot/users/tazjin/emacs/config/functions.el
Vincent Ambo 261ae2de95 feat(tazjin/emacs): make zoxide project opener work without git
Work projects are not in git, but I still want to fuzzy jump to them.

Change-Id: Iced78519da69b6992d253e2a0c9ed485f1347164
Reviewed-on: https://cl.tvl.fyi/c/depot/+/9005
Reviewed-by: tazjin <tazjin@tvl.su>
Tested-by: BuildkiteCI
Autosubmit: tazjin <tazjin@tvl.su>
2023-08-04 16:35:47 +00:00

374 lines
14 KiB
EmacsLisp

(require 'chart)
(require 'dash)
(require 'map)
(defun load-file-if-exists (filename)
(if (file-exists-p filename)
(load filename)))
(defun goto-line-with-feedback ()
"Show line numbers temporarily, while prompting for the line number input"
(interactive)
(unwind-protect
(progn
(setq-local display-line-numbers t)
(let ((target (read-number "Goto line: ")))
(avy-push-mark)
(goto-line target)))
(setq-local display-line-numbers nil)))
;; These come from the emacs starter kit
(defun esk-add-watchwords ()
(font-lock-add-keywords
nil '(("\\<\\(FIX\\(ME\\)?\\|TODO\\|DEBUG\\|HACK\\|REFACTOR\\|NOCOMMIT\\)"
1 font-lock-warning-face t))))
(defun esk-sudo-edit (&optional arg)
(interactive "p")
(if (or arg (not buffer-file-name))
(find-file (concat "/sudo:root@localhost:" (read-file-name "File: ")))
(find-alternate-file (concat "/sudo:root@localhost:" buffer-file-name))))
;; Open the NixOS man page
(defun nixos-man ()
(interactive)
(man "configuration.nix"))
;; Get the nix store path for a given derivation.
;; If the derivation has not been built before, this will trigger a build.
(defun nix-store-path (derivation)
(let ((expr (concat "with import <nixos> {}; " derivation)))
(s-chomp (shell-command-to-string (concat "nix-build -E '" expr "'")))))
(defun insert-nix-store-path ()
(interactive)
(let ((derivation (read-string "Derivation name (in <nixos>): ")))
(insert (nix-store-path derivation))))
(defun toggle-force-newline ()
"Buffer-local toggle for enforcing final newline on save."
(interactive)
(setq-local require-final-newline (not require-final-newline))
(message "require-final-newline in buffer %s is now %s"
(buffer-name)
require-final-newline))
(defun list-external-commands ()
"Creates a list of all external commands available on $PATH
while filtering NixOS wrappers."
(cl-loop
for dir in (split-string (getenv "PATH") path-separator)
when (and (file-exists-p dir) (file-accessible-directory-p dir))
for lsdir = (cl-loop for i in (directory-files dir t)
for bn = (file-name-nondirectory i)
when (and (not (s-contains? "-wrapped" i))
(not (member bn completions))
(not (file-directory-p i))
(file-executable-p i))
collect bn)
append lsdir into completions
finally return (sort completions 'string-lessp)))
(defvar external-command-flag-overrides
'(("google-chrome" . "--force-device-scale-factor=1.4"))
"This setting lets me add additional flags to specific commands
that are run interactively via `run-external-command'.")
(defun run-external-command--handler (cmd)
"Execute the specified command and notify the user when it
finishes."
(let* ((extra-flags (cdr (assoc cmd external-command-flag-overrides)))
(cmd (if extra-flags (s-join " " (list cmd extra-flags)) cmd)))
(message "Starting %s..." cmd)
(set-process-sentinel
(start-process-shell-command cmd nil cmd)
(lambda (process event)
(when (string= event "finished\n")
(message "%s process finished." process))))))
(defun run-external-command ()
"Prompts the user with a list of all installed applications and
lets them select one to launch."
(interactive)
(let ((external-commands-list (list-external-commands)))
(run-external-command--handler
(completing-read "Command: " external-commands-list
nil ;; predicate
t ;; require-match
nil ;; initial-input
;; hist
'external-commands-history))))
(defun password-store-lookup (&optional password-store-dir)
"Interactive password-store lookup function that actually uses
the GPG agent correctly."
(interactive)
(let* ((entry (completing-read "Copy password of entry: "
(password-store-list (or password-store-dir
(password-store-dir)))
nil ;; predicate
t ;; require-match
))
(password (auth-source-pass-get 'secret entry)))
(password-store-clear)
(kill-new password)
(setq password-store-kill-ring-pointer kill-ring-yank-pointer)
(message "Copied %s to the kill ring. Will clear in %s seconds."
entry (password-store-timeout))
(setq password-store-timeout-timer
(run-at-time (password-store-timeout)
nil 'password-store-clear))))
(defun browse-repositories ()
"Select a git repository and open its associated magit buffer."
(interactive)
(magit-status
(completing-read "Repository: " (magit-list-repos))))
(defun bottom-right-window-p ()
"Determines whether the last (i.e. bottom-right) window of the
active frame is showing the buffer in which this function is
executed."
(let* ((frame (selected-frame))
(right-windows (window-at-side-list frame 'right))
(bottom-windows (window-at-side-list frame 'bottom))
(last-window (car (seq-intersection right-windows bottom-windows))))
(eq (current-buffer) (window-buffer last-window))))
(defhydra mc/mark-more-hydra (:color pink)
("<up>" mc/mmlte--up "Mark previous like this")
("<down>" mc/mmlte--down "Mark next like this")
("<left>" mc/mmlte--left (if (eq mc/mark-more-like-this-extended-direction 'up)
"Skip past the cursor furthest up"
"Remove the cursor furthest down"))
("<right>" mc/mmlte--right (if (eq mc/mark-more-like-this-extended-direction 'up)
"Remove the cursor furthest up"
"Skip past the cursor furthest down"))
("f" nil "Finish selecting"))
;; Mute the message that mc/mmlte wants to print on its own
(advice-add 'mc/mmlte--message :around (lambda (&rest args) (ignore)))
(defun mc/mark-dwim (arg)
"Select multiple things, but do what I mean."
(interactive "p")
(if (not (region-active-p)) (mc/mark-next-lines arg)
(if (< 1 (count-lines (region-beginning)
(region-end)))
(mc/edit-lines arg)
;; The following is almost identical to `mc/mark-more-like-this-extended',
;; but uses a hydra (`mc/mark-more-hydra') instead of a transient key map.
(mc/mmlte--down)
(mc/mark-more-hydra/body))))
(setq mc/cmds-to-run-for-all '(kill-region paredit-newline))
(setq mc/cmds-to-run-once '(mc/mark-dwim
mc/mark-more-hydra/mc/mmlte--down
mc/mark-more-hydra/mc/mmlte--left
mc/mark-more-hydra/mc/mmlte--right
mc/mark-more-hydra/mc/mmlte--up
mc/mark-more-hydra/mmlte--up
mc/mark-more-hydra/nil))
(defun memespace-region ()
"Make a meme out of it."
(interactive)
(let* ((start (region-beginning))
(end (region-end))
(memed
(message
(s-trim-right
(apply #'string
(-flatten
(nreverse
(-reduce-from (lambda (acc x)
(cons (cons x (-repeat (+ 1 (length acc)) 32)) acc))
'()
(string-to-list (buffer-substring-no-properties start end))))))))))
(save-excursion (delete-region start end)
(goto-char start)
(insert memed))))
(defun insert-todo-comment (prefix todo)
"Insert a comment at point with something for me to do."
(interactive "P\nsWhat needs doing? ")
(save-excursion
(move-end-of-line nil)
(insert (format " %s TODO(%s): %s"
(s-trim-right comment-start)
(if prefix (read-string "Who needs to do this? ")
(getenv "USER"))
todo))))
;; Custom text scale adjustment functions that operate on the entire instance
(defun modify-text-scale (factor)
(set-face-attribute 'default nil
:height (+ (* factor 5) (face-attribute 'default :height))))
(defun increase-default-text-scale (prefix)
"Increase default text scale in all Emacs frames, or just the
current frame if PREFIX is set."
(interactive "P")
(if prefix (text-scale-increase 1)
(modify-text-scale 1)))
(defun decrease-default-text-scale (prefix)
"Increase default text scale in all Emacs frames, or just the
current frame if PREFIX is set."
(interactive "P")
(if prefix (text-scale-decrease 1)
(modify-text-scale -1)))
(defun set-default-text-scale (prefix &optional to)
"Set the default text scale to the specified value, or the
default. Restores current frame's text scale only, if PREFIX is
set."
(interactive "P")
(if prefix (text-scale-adjust 0)
(set-face-attribute 'default nil :height (or to 120))))
(defun screenshot-select (filename)
"Take a screenshot based on a mouse-selection and save it to
~/screenshots."
(interactive "sScreenshot filename: ")
(let* ((path (f-join "~/screenshots"
(format "%s-%d.png"
(if (string-empty-p filename) "shot" filename)
(time-convert nil 'integer)))))
(shell-command (format "maim --select %s" path))
(message "Wrote screenshot to %s" path)))
(defun graph-unread-mails ()
"Create a bar chart of unread mails based on notmuch tags.
Certain tags are excluded from the overview."
(interactive)
(let ((tag-counts
(-keep (-lambda ((name . search))
(let ((count
(string-to-number
(s-trim
(notmuch-command-to-string "count" search "and" "tag:unread")))))
(when (>= count 1) (cons name count))))
(notmuch-hello-generate-tag-alist '("unread" "signed" "attachment" "important")))))
(chart-bar-quickie
(if (< (length tag-counts) 6)
'vertical 'horizontal)
"Unread emails"
(-map #'car tag-counts) "Tag:"
(-map #'cdr tag-counts) "Count:")))
(defun notmuch-show-open-or-close-subthread (&optional prefix)
"Open or close the subthread from (and including) the message at point."
(interactive "P")
(save-excursion
(let ((current-depth (map-elt (notmuch-show-get-message-properties) :depth 0)))
(loop do (notmuch-show-message-visible (notmuch-show-get-message-properties) prefix)
until (or (not (notmuch-show-goto-message-next))
(= (map-elt (notmuch-show-get-message-properties) :depth) current-depth)))))
(force-window-update))
(defun vterm-send-ctrl-x ()
"Sends `C-x' to the libvterm."
(interactive)
(vterm-send-key "x" nil nil t))
(defun find-depot-project (dir)
"Function used in the `project-find-functions' hook list to
determine the current project root of a depot project."
(when (s-starts-with? "/depot" dir)
(if (f-exists-p (f-join dir "default.nix"))
(cons 'transient dir)
(find-depot-project (f-parent dir)))))
(add-to-list 'project-find-functions #'find-depot-project)
(defun find-cargo-project (dir)
"Attempt to find the current project in `project-find-functions'
by looking for a `Cargo.toml' file."
(when dir
(unless (equal "/" dir)
(if (f-exists-p (f-join dir "Cargo.toml"))
(cons 'transient dir)
(find-cargo-project (f-parent dir))))))
(add-to-list 'project-find-functions #'find-cargo-project)
(defun magit-find-file-worktree ()
(interactive)
"Find a file in the current (ma)git worktree."
(magit-find-file--internal "{worktree}"
(magit-read-file-from-rev "HEAD" "Find file")
#'pop-to-buffer-same-window))
(defun songwhip--handle-result (status &optional cbargs)
;; TODO(tazjin): Inspect status, which looks different in practice
;; than the manual claims.
(if-let* ((response (json-parse-string
(buffer-substring url-http-end-of-headers (point-max))))
(sw-path (ht-get* response "data" "path"))
(link (format "https://songwhip.com/%s" sw-path))
(select-enable-clipboard t))
(progn
(kill-new link)
(message "Copied Songwhip link (%s)" link))
(warn "Something went wrong while retrieving Songwhip link!")
;; For debug purposes, the buffer is persisted in this case.
(setq songwhip--debug-buffer (current-buffer))))
(defun songwhip-lookup-url (url)
"Look up URL on Songwhip and copy the resulting link to the clipboard."
(interactive "sEnter source URL: ")
(let ((songwhip-url "https://songwhip.com/api/")
(url-request-method "POST")
(url-request-extra-headers '(("Content-Type" . "application/json")))
(url-request-data
(json-serialize `((country . "GB")
(url . ,url)))))
(url-retrieve "https://songwhip.com/api/" #'songwhip--handle-result nil t t)
(message "Requesting Songwhip URL ... please hold the line.")))
(defun rg-in-project (&optional prefix)
"Interactively call ripgrep in the current project, or fall
back to ripgrep default behaviour if prefix is set."
(interactive "P")
(counsel-rg nil (unless prefix
(if-let ((pr (project-current)))
(project-root pr)))))
(defun zoxide-open-project ()
"Query Zoxide for paths, and open the result as appropriate (magit or dired)."
(interactive)
(zoxide-open-with
nil
(lambda (path)
(condition-case err (magit-status-setup-buffer path)
(magit-outside-git-repo (dired path))))))
(defun toggle-nix-test-and-exp ()
"Switch between the .nix and .exp file in a Tvix/Nix test."
(interactive)
(let* ((file (buffer-file-name))
(other (if (s-suffix? ".nix" file)
(s-replace-regexp ".nix$" ".exp" file)
(if (s-suffix? ".exp" file)
(s-replace-regexp ".exp$" ".nix" file)
(error "Not a .nix/.exp file!")))))
(find-file other)))
(provide 'functions)