;;; completion/ivy/packages.el

;; Show more information in ivy-switch-buffer; and only display
;; project/workgroup-relevant buffers.
(defun +ivy-get-buffers (&optional buffer-list)
  (let ((min-name 5)
        (min-mode 5)
        (proot (doom-project-root)))
    (mapcar
     (lambda (b) (format (format "%%-%ds %%-%ds %%s" min-name min-mode)
                    (nth 0 b)
                    (nth 1 b)
                    (or (nth 2 b) "")))
     (mapcar (lambda (b)
               (with-current-buffer b
                 (let ((buffer-name (buffer-name b))
                       (mode-name (symbol-name major-mode)))
                   (when (> (length buffer-name) min-name)
                     (setq min-name (+ (length buffer-name) 15)))
                   (when (> (length mode-name) min-mode)
                     (setq min-mode (+ (length mode-name) 3)))
                   (list
                    (concat
                     (propertize buffer-name
                                 'face (cond ((string-match-p "^ ?\\*" buffer-name)
                                              'font-lock-comment-face)
                                             ((not (string= proot (doom-project-root)))
                                              'font-lock-keyword-face)
                                             (buffer-read-only
                                              'error)))
                     (when (and buffer-file-name (buffer-modified-p))
                       (propertize "[+]" 'face 'doom-modeline-buffer-modified)))
                    (propertize mode-name 'face 'font-lock-constant-face)
                    (when buffer-file-name
                      (f-slash (abbreviate-file-name (f-dirname buffer-file-name))))))))
             (or buffer-list (doom/get-buffers t))))))

(defun +ivy--select-buffer-action (buffer)
  (ivy--switch-buffer-action
   (s-chop-suffix
    "[+]"
    (substring buffer 0 (s-index-of "   " buffer)))))

;;;###autoload
(defun +ivy/switch-project-buffer (&optional all-p)
  "Displays open buffers in current project and workspace. If ALL-P, then show
all open buffers."
  (interactive)
  (ivy-read (format "%s buffers: " (if all-p "All" "Project"))
            (+ivy-get-buffers (if all-p (buffer-list)))
            :matcher #'ivy--switch-buffer-matcher
            :action #'+ivy--select-buffer-action
            :keymap ivy-switch-buffer-map
            :caller '+ivy/switch-project-buffer))

;;;###autoload
(defun +ivy/switch-buffer ()
  "Displays all open buffers, across projects and workspaces."
  (interactive)
  (+ivy/switch-project-buffer t))

;;;###autoload
(defun +ivy/kill-ring ()
  (interactive)
  (ivy-read "Kill ring:" (--filter (not (or (< (length it) 3)
                                            (string-match-p "\\`[\n[:blank:]]+\\'" it)))
                                   (cl-remove-duplicates kill-ring :test 'equal))))

;;;###autoload
(defun +ivy/recentf ()
  "Find a file on `recentf-list'."
  (interactive)
  (while (doom-popup-p)
    (doom/popup-close))
  (call-interactively 'counsel-recentf))

;;;###autoload (autoload '+ivy:ag-search "completion/ivy/autoload" nil t)
;;;###autoload (autoload '+ivy:ag-search-cwd "completion/ivy/autoload" nil t)
;;;###autoload (autoload '+ivy:swiper "completion/ivy/autoload" nil t)
(@after evil
  (defvar doom-ivy-ag-last-search nil)

  (evil-define-operator +ivy:ag-search (beg end search regex-p &optional dir)
    "Preform a counsel search with SEARCH. If SEARCH is nil and in visual mode,
use the selection, otherwise activate live ag searching in helm.

If REGEX-P is non-nil, SEARCH will be treated as a regular expression.
DIR specifies the default-directory from which ag is run."
    :type inclusive :repeat nil
    (interactive "<r><a><!>")
    (let ((search (or search
                      (and (evil-visual-state-p)
                           (and beg end (rxt-quote-pcre (buffer-substring-no-properties beg end))))
                      doom-ivy-ag-last-search)))
      (setq doom-ivy-ag-last-search search)
      (counsel-ag search (or dir (f-slash (doom-project-root)))
                  (concat "--nocolor --nogroup" (if regex-p " -Q")))))

  (evil-define-operator +ivy:ag-search-cwd (beg end search regex-p)
    :type inclusive :repeat nil
    (interactive "<r><a><!>")
    (+ivy:ag-search beg end search regex-p default-directory))

  (evil-define-command +ivy:swiper (&optional search)
    "Invoke `swiper' with SEARCH, otherwise with the symbol at point."
    (interactive "<a>")
    (swiper (or search (thing-at-point 'symbol)))))

;;;###autoload
(defun +ivy/tasks ()
  (interactive)
  ;; TODO Something a little nicer
  (counsel-ag " (TODO|FIXME|NOTE) " (doom-project-root)))

;;;###autoload
(defun +ivy*counsel-ag-function (string base-cmd extra-ag-args)
  "Advice to get rid of the character limit from `counsel-ag-function', which
interferes with my custom :ag ex command `+ivy:ag-search'."
  (when (null extra-ag-args)
    (setq extra-ag-args ""))
  (if (< (length string) 1)
      (counsel-more-chars 1)
    (let ((default-directory counsel--git-grep-dir)
          (regex (counsel-unquote-regex-parens
                  (setq ivy--old-re
                        (ivy--regex string)))))
      (let ((ag-cmd (format base-cmd
                            (concat extra-ag-args
                                    " -- "
                                    (shell-quote-argument regex)))))
        (if (file-remote-p default-directory)
            (split-string (shell-command-to-string ag-cmd) "\n" t)
          (counsel--async-command ag-cmd)
          nil)))))

;;;###autoload
(defun +ivy/counsel-ag-occur ()
  "Invoke the search+replace wgrep buffer on the current ag search results."
  (interactive)
  (require 'wgrep)
  (call-interactively 'ivy-occur))

(provide 'completion/ivy/autoload)
;;; completion/ivy/autoload.el ends here