e865526c75
This allows you to control what search engines for project-search commands (bound to SPC / p) to try, and in what order. If you didn't want to use ripgrep, for instance, remove 'rg from these variables, or move it to the end of the list.
364 lines
15 KiB
EmacsLisp
364 lines
15 KiB
EmacsLisp
;;; completion/ivy/autoload/ivy.el -*- lexical-binding: t; -*-
|
|
|
|
(defun +ivy--is-workspace-or-other-buffer-p (buffer)
|
|
(let ((buffer (car buffer)))
|
|
(when (stringp buffer)
|
|
(setq buffer (get-buffer buffer)))
|
|
(and (not (eq buffer (current-buffer)))
|
|
(+workspace-contains-buffer-p buffer))))
|
|
|
|
(defun +ivy*rich-switch-buffer-buffer-name (str)
|
|
(propertize
|
|
(ivy-rich-pad str ivy-rich-switch-buffer-name-max-length)
|
|
'face (cond ((string-match-p "^ *\\*" str)
|
|
'font-lock-comment-face)
|
|
((and buffer-file-truename
|
|
(not (file-in-directory-p buffer-file-truename (doom-project-root))))
|
|
'font-lock-doc-face)
|
|
(t nil))))
|
|
(advice-add 'ivy-rich-switch-buffer-buffer-name :override #'+ivy*rich-switch-buffer-buffer-name)
|
|
|
|
|
|
;;
|
|
;; Library
|
|
;;
|
|
|
|
;;;###autoload
|
|
(defun +ivy-projectile-find-file-transformer (str)
|
|
"Highlight entries that have been visited. This is the opposite of
|
|
`counsel-projectile-find-file'."
|
|
(cond ((get-file-buffer (projectile-expand-root str))
|
|
(propertize str 'face '(:weight ultra-bold :slant italic)))
|
|
(t str)))
|
|
|
|
;;;###autoload
|
|
(defun +ivy-recentf-transformer (str)
|
|
"Dim recentf entries that are not in the current project of the buffer you
|
|
started `counsel-recentf' from. Also uses `abbreviate-file-name'."
|
|
(let ((str (abbreviate-file-name str)))
|
|
(if (file-in-directory-p str (doom-project-root))
|
|
str
|
|
(propertize str 'face 'ivy-virtual))))
|
|
|
|
;;;###autoload
|
|
(defun +ivy-buffer-transformer (str)
|
|
"Dim special buffers, buffers whose file aren't in the current buffer, and
|
|
virtual buffers. Uses `ivy-rich' under the hood."
|
|
(let ((buf (get-buffer str)))
|
|
(require 'ivy-rich)
|
|
(cond (buf (ivy-rich-switch-buffer-transformer str))
|
|
((and (eq ivy-virtual-abbreviate 'full)
|
|
ivy-rich-switch-buffer-align-virtual-buffer)
|
|
(ivy-rich-switch-buffer-virtual-buffer str))
|
|
((eq ivy-virtual-abbreviate 'full)
|
|
(propertize (abbreviate-file-name str) 'str 'ivy-virtual))
|
|
(t (propertize str 'face 'ivy-virtual)))))
|
|
|
|
;;;###autoload
|
|
(defun +ivy/switch-workspace-buffer (&optional arg)
|
|
"Switch to another buffer within the current workspace.
|
|
|
|
If ARG (universal argument), open selection in other-window."
|
|
(interactive "P")
|
|
(ivy-read "Switch to workspace buffer: "
|
|
'internal-complete-buffer
|
|
:predicate #'+ivy--is-workspace-or-other-buffer-p
|
|
:action (if arg
|
|
#'ivy--switch-buffer-other-window-action
|
|
#'ivy--switch-buffer-action)
|
|
:matcher #'ivy--switch-buffer-matcher
|
|
:keymap ivy-switch-buffer-map
|
|
:caller #'+ivy/switch-workspace-buffer))
|
|
|
|
(defun +ivy--tasks-candidates (tasks)
|
|
"Generate a list of task tags (specified by `+ivy-task-tags') for
|
|
`+ivy/tasks'."
|
|
(let* ((max-type-width
|
|
(cl-loop for task in +ivy-task-tags maximize (length (car task))))
|
|
(max-desc-width
|
|
(cl-loop for task in tasks maximize (length (cl-cdadr task))))
|
|
(max-width (max (- (frame-width) (1+ max-type-width) max-desc-width)
|
|
25)))
|
|
(cl-loop
|
|
with fmt = (format "%%-%ds %%-%ds%%s%%s:%%s" max-type-width max-width)
|
|
for alist in tasks
|
|
collect
|
|
(let-alist alist
|
|
(format fmt
|
|
(propertize .type 'face (cdr (assoc .type +ivy-task-tags)))
|
|
(substring .desc 0 (min max-desc-width (length .desc)))
|
|
(propertize " | " 'face 'font-lock-comment-face)
|
|
(propertize (abbreviate-file-name .file) 'face 'font-lock-keyword-face)
|
|
(propertize .line 'face 'font-lock-constant-face))))))
|
|
|
|
(defun +ivy--tasks (target)
|
|
(let* (case-fold-search
|
|
(task-tags (mapcar #'car +ivy-task-tags))
|
|
(cmd
|
|
(format "%s -H -S --no-heading -- %s %s"
|
|
(or (when-let* ((bin (executable-find "rg")))
|
|
(concat bin " --line-number"))
|
|
(when-let* ((bin (executable-find "ag")))
|
|
(concat bin " --numbers"))
|
|
(error "ripgrep & the_silver_searcher are unavailable"))
|
|
(shell-quote-argument
|
|
(concat "\\s("
|
|
(string-join task-tags "|")
|
|
")([\\s:]|\\([^)]+\\):?)"))
|
|
target)))
|
|
(save-match-data
|
|
(cl-loop with out = (shell-command-to-string cmd)
|
|
for x in (and out (split-string out "\n" t))
|
|
when (condition-case-unless-debug ex
|
|
(string-match
|
|
(concat "^\\([^:]+\\):\\([0-9]+\\):.+\\("
|
|
(string-join task-tags "\\|")
|
|
"\\):?\\s-*\\(.+\\)")
|
|
x)
|
|
(error
|
|
(print! (red "Error matching task in file: (%s) %s"
|
|
(error-message-string ex)
|
|
(car (split-string x ":"))))
|
|
nil))
|
|
collect `((type . ,(match-string 3 x))
|
|
(desc . ,(match-string 4 x))
|
|
(file . ,(match-string 1 x))
|
|
(line . ,(match-string 2 x)))))))
|
|
|
|
(defun +ivy--tasks-open-action (x)
|
|
"Jump to the file and line of the current task."
|
|
(let ((location (cadr (split-string x " | ")))
|
|
(type (car (split-string x " "))))
|
|
(cl-destructuring-bind (file line) (split-string location ":")
|
|
(with-ivy-window
|
|
(find-file (expand-file-name file (doom-project-root)))
|
|
(goto-char (point-min))
|
|
(forward-line (1- (string-to-number line)))
|
|
(search-forward type (line-end-position) t)
|
|
(backward-char (length type))
|
|
(recenter)))))
|
|
|
|
;;;###autoload
|
|
(defun +ivy/tasks (&optional arg)
|
|
"Search through all TODO/FIXME tags in the current project. If ARG, only
|
|
search current file. See `+ivy-task-tags' to customize what this searches for."
|
|
(interactive "P")
|
|
(ivy-read (format "Tasks (%s): "
|
|
(if arg
|
|
(concat "in: " (file-relative-name buffer-file-name))
|
|
"project"))
|
|
(+ivy--tasks-candidates
|
|
(+ivy--tasks (if arg buffer-file-name (doom-project-root))))
|
|
:action #'+ivy--tasks-open-action
|
|
:caller '+ivy/tasks))
|
|
|
|
;;;###autoload
|
|
(defun +ivy/wgrep-occur ()
|
|
"Invoke the search+replace wgrep buffer on the current ag/rg search results."
|
|
(interactive)
|
|
(unless (window-minibuffer-p)
|
|
(user-error "No completion session is active"))
|
|
(require 'wgrep)
|
|
(let* ((caller (ivy-state-caller ivy-last))
|
|
(occur-fn (plist-get ivy--occurs-list caller))
|
|
(buffer
|
|
(generate-new-buffer
|
|
(format "*ivy-occur%s \"%s\"*"
|
|
(if caller (concat " " (prin1-to-string caller)) "")
|
|
ivy-text))))
|
|
(with-current-buffer buffer
|
|
(let ((inhibit-read-only t))
|
|
(erase-buffer)
|
|
(funcall occur-fn))
|
|
(setf (ivy-state-text ivy-last) ivy-text)
|
|
(setq ivy-occur-last ivy-last)
|
|
(setq-local ivy--directory ivy--directory))
|
|
(ivy-exit-with-action
|
|
`(lambda (_)
|
|
(pop-to-buffer ,buffer)
|
|
(ivy-wgrep-change-to-wgrep-mode)))))
|
|
|
|
;;;###autoload
|
|
(defun +ivy-yas-prompt (prompt choices &optional display-fn)
|
|
(yas-completing-prompt prompt choices display-fn #'ivy-completing-read))
|
|
|
|
;;;###autoload
|
|
(defun +ivy-git-grep-other-window-action (x)
|
|
"Opens the current candidate in another window."
|
|
(when (string-match "\\`\\(.*?\\):\\([0-9]+\\):\\(.*\\)\\'" x)
|
|
(select-window
|
|
(with-ivy-window
|
|
(let ((file-name (match-string-no-properties 1 x))
|
|
(line-number (match-string-no-properties 2 x)))
|
|
(find-file-other-window (expand-file-name file-name (ivy-state-directory ivy-last)))
|
|
(goto-char (point-min))
|
|
(forward-line (1- (string-to-number line-number)))
|
|
(re-search-forward (ivy--regex ivy-text t) (line-end-position) t)
|
|
(run-hooks 'counsel-grep-post-action-hook)
|
|
(selected-window))))))
|
|
|
|
;;;###autoload
|
|
(defun +ivy-confirm-delete-file (x)
|
|
(dired-delete-file x 'confirm-each-subdirectory))
|
|
|
|
|
|
;;
|
|
;; File searching
|
|
;;
|
|
|
|
;;;###autoload
|
|
(cl-defun +ivy-file-search (engine &key query in all-files (recursive t))
|
|
"Conduct a file search using ENGINE, which can be any of: rg, ag, pt, and
|
|
grep. If omitted, ENGINE will default to the first one it detects, in that
|
|
order.
|
|
|
|
:query STRING
|
|
Determines the initial input to search for.
|
|
:in PATH
|
|
Sets what directory to base the search out of. Defaults to the current
|
|
project's root.
|
|
:recursive BOOL
|
|
Whether or not to search files recursively from the base directory."
|
|
(declare (indent defun))
|
|
(let* ((project-root (doom-project-root))
|
|
(directory (or in project-root))
|
|
(default-directory directory)
|
|
(engine (or engine
|
|
(cl-loop for tool in +ivy-project-search-engines
|
|
if (executable-find (symbol-name tool))
|
|
return tool)
|
|
(and (or (executable-find "grep")
|
|
(executable-find "git"))
|
|
'grep)
|
|
(error "No search engine specified (is ag, rg, pt or git installed?)")))
|
|
(query
|
|
(or query
|
|
(when (use-region-p)
|
|
(let ((beg (or (bound-and-true-p evil-visual-beginning) (region-beginning)))
|
|
(end (or (bound-and-true-p evil-visual-end) (region-end))))
|
|
(when (> (abs (- end beg)) 1)
|
|
(rxt-quote-pcre (buffer-substring-no-properties beg end)))))))
|
|
(prompt
|
|
(format "%s%%s %s"
|
|
(symbol-name engine)
|
|
(cond ((equal directory default-directory)
|
|
"./")
|
|
((equal directory project-root)
|
|
(projectile-project-name))
|
|
(t
|
|
(file-relative-name directory project-root))))))
|
|
(require 'counsel)
|
|
(let ((counsel-more-chars-alist
|
|
(if query '((t . 1)) counsel-more-chars-alist)))
|
|
(pcase engine
|
|
('grep
|
|
(let ((args (if recursive " -R"))
|
|
(counsel-projectile-grep-initial-input query))
|
|
(if all-files
|
|
(cl-letf (((symbol-function #'projectile-ignored-directories-rel)
|
|
(symbol-function #'ignore))
|
|
((symbol-function #'projectile-ignored-files-rel)
|
|
(symbol-function #'ignore)))
|
|
(counsel-projectile-grep args))
|
|
(counsel-projectile-grep args))))
|
|
('ag
|
|
(let ((args (concat (if all-files " -a")
|
|
(unless recursive " --depth 1"))))
|
|
(counsel-ag query directory args (format prompt args))))
|
|
('rg
|
|
(let ((args (concat (if all-files " -uu")
|
|
(unless recursive " --maxdepth 1"))))
|
|
(counsel-rg query directory args (format prompt args))))
|
|
('pt
|
|
(let ((counsel-pt-base-command
|
|
(concat counsel-pt-base-command
|
|
(if all-files " -U")
|
|
(unless recursive " --depth=1")))
|
|
(default-directory directory))
|
|
(counsel-pt query)))
|
|
(_ (error "No search engine specified"))))))
|
|
|
|
;;;###autoload
|
|
(defun +ivy/project-search (arg)
|
|
"Performs a project search using the first available search backend from a
|
|
list of: ripgrep, ag, pt, git-grep and grep. If ARG (universal argument),
|
|
preform search from current directory."
|
|
(interactive "P")
|
|
(call-interactively
|
|
(or (cl-loop for tool in (cl-remove-duplicates +ivy-project-search-engines)
|
|
if (executable-find (symbol-name tool))
|
|
return (intern (format "+ivy/%s%s" tool (if arg "-from-cwd" ""))))
|
|
(if arg
|
|
#'+ivy/grep-from-cwd
|
|
#'+ivy/grep))))
|
|
|
|
;;;###autoload
|
|
(defun +ivy/rg (all-files-p &optional query directory)
|
|
"Perform a project file search using ripgrep. QUERY is a regexp. If omitted,
|
|
the current selection is used. If no selection is active, the last known search
|
|
is used.
|
|
|
|
If ALL-FILES-P, don't respect .gitignore files and search everything.
|
|
|
|
NOTE: ripgrep doesn't support multiline searches (yet)."
|
|
(interactive "P")
|
|
(+ivy-file-search 'rg :query query :in directory :all-files all-files-p))
|
|
|
|
;;;###autoload
|
|
(defun +ivy/ag (all-files-p &optional query directory)
|
|
"Perform a project file search using the silver searcher. QUERY is a pcre
|
|
regexp. If omitted, the current selection is used. If no selection is active,
|
|
the last known search is used.
|
|
|
|
If ALL-FILES-P, don't respect .gitignore files and search everything."
|
|
(interactive "P")
|
|
(+ivy-file-search 'ag :query query :in directory :all-files all-files-p))
|
|
|
|
;;;###autoload
|
|
(defun +ivy/pt (all-files-p &optional query directory)
|
|
"Perform a project file search using the platinum searcher. QUERY is a grep
|
|
regexp. If omitted, the current selection is used. If no selection is active,
|
|
the last known search is used.
|
|
|
|
If ALL-FILES-P, don't respect .gitignore files and search everything."
|
|
(interactive "P")
|
|
(+ivy-file-search 'pt :query query :in directory :all-files all-files-p))
|
|
|
|
;;;###autoload
|
|
(defun +ivy/grep (all-files-p &optional query directory)
|
|
"Perform a project file search using grep (or git-grep in git repos). QUERY is
|
|
a grep regexp. If omitted, the current selection is used. If no selection is
|
|
active, the last known search is used.
|
|
|
|
If ALL-FILES-P, don't respect .gitignore files and search everything."
|
|
(interactive "P")
|
|
(+ivy-file-search 'grep :query query :in directory :all-files all-files-p))
|
|
|
|
;; Relative to current directory
|
|
;;;###autoload
|
|
(defun +ivy/rg-from-cwd (recursive-p &optional query)
|
|
"Like `+ivy/rg', but from the current directory (recursively if RECURSIVE-P is
|
|
non-nil)."
|
|
(interactive "P")
|
|
(+ivy-file-search 'rg :query query :in default-directory :recursive recursive-p))
|
|
|
|
;;;###autoload
|
|
(defun +ivy/ag-from-cwd (recursive-p &optional query)
|
|
"Like `+ivy/ag', but from the current directory (recursively if RECURSIVE-P is
|
|
non-nil)."
|
|
(interactive "P")
|
|
(+ivy-file-search 'ag :query query :in default-directory :recursive recursive-p))
|
|
|
|
;;;###autoload
|
|
(defun +ivy/pt-from-cwd (recursive-p &optional query)
|
|
"Like `+ivy/pt', but from the current directory (recursively if RECURSIVE-P is
|
|
non-nil)."
|
|
(interactive "P")
|
|
(+ivy-file-search 'pt :query query :in default-directory :recursive recursive-p))
|
|
|
|
;;;###autoload
|
|
(defun +ivy/grep-from-cwd (recursive-p &optional query)
|
|
"Like `+ivy/grep', but from the current directory (recursively if RECURSIVE-P is
|
|
non-nil)."
|
|
(interactive "P")
|
|
(+ivy-file-search 'grep :query query :in default-directory :recursive recursive-p))
|