290 lines
13 KiB
EmacsLisp
290 lines
13 KiB
EmacsLisp
|
;;; helm-adaptive.el --- Adaptive Sorting of Candidates. -*- lexical-binding: t -*-
|
|||
|
|
|
|||
|
|
;; Original Author: Tamas Patrovics
|
|||
|
|
|
|||
|
|
;; Copyright (C) 2007 Tamas Patrovics
|
|||
|
|
;; Copyright (C) 2012 ~ 2021 Thierry Volpiatto <thierry.volpiatto@gmail.com>
|
|||
|
|
|
|||
|
|
;; This program is free software; you can redistribute it and/or modify
|
|||
|
|
;; it under the terms of the GNU General Public License as published by
|
|||
|
|
;; the Free Software Foundation, either version 3 of the License, or
|
|||
|
|
;; (at your option) any later version.
|
|||
|
|
|
|||
|
|
;; This program is distributed in the hope that it will be useful,
|
|||
|
|
;; but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|||
|
|
;; MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|||
|
|
;; GNU General Public License for more details.
|
|||
|
|
|
|||
|
|
;; You should have received a copy of the GNU General Public License
|
|||
|
|
;; along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|||
|
|
|
|||
|
|
;;; Code:
|
|||
|
|
|
|||
|
|
(require 'cl-lib)
|
|||
|
|
(require 'helm)
|
|||
|
|
|
|||
|
|
|
|||
|
|
(defgroup helm-adapt nil
|
|||
|
|
"Adaptative sorting of candidates for Helm."
|
|||
|
|
:group 'helm)
|
|||
|
|
|
|||
|
|
(defcustom helm-adaptive-history-file
|
|||
|
|
(locate-user-emacs-file "helm-adaptive-history")
|
|||
|
|
"Path of file where history information is stored.
|
|||
|
|
When nil history is not saved nor restored after Emacs restart
|
|||
|
|
unless you save/restore `helm-adaptive-history' with something
|
|||
|
|
else like psession or desktop."
|
|||
|
|
:type 'string
|
|||
|
|
:group 'helm-adapt)
|
|||
|
|
|
|||
|
|
(defcustom helm-adaptive-history-length 50
|
|||
|
|
"Maximum number of candidates stored for a source."
|
|||
|
|
:type 'number
|
|||
|
|
:group 'helm-adapt)
|
|||
|
|
|
|||
|
|
(defcustom helm-adaptive-sort-by-frequent-recent-usage t
|
|||
|
|
"Try to sort on an average of frequent and recent usage when non-nil.
|
|||
|
|
|
|||
|
|
When nil sort on frequency usage only.
|
|||
|
|
|
|||
|
|
Only frequency:
|
|||
|
|
When candidate have low frequency, you have to hit on it many
|
|||
|
|
times to make it going up on top.
|
|||
|
|
|
|||
|
|
Frequency+recent:
|
|||
|
|
Even with a low frequency, candidate go up on top. If a candidate
|
|||
|
|
have a high frequency but it is not used since some time, it goes
|
|||
|
|
down slowly, but as soon you reuse it it go up on top quickly."
|
|||
|
|
:group 'helm-adapt
|
|||
|
|
:type 'boolean)
|
|||
|
|
|
|||
|
|
;; Internal
|
|||
|
|
(defvar helm-adaptive-done nil
|
|||
|
|
"nil if history information is not yet stored for the current
|
|||
|
|
selection.")
|
|||
|
|
|
|||
|
|
(defvar helm-adaptive-history nil
|
|||
|
|
"Contains the stored history information.
|
|||
|
|
Format: ((SOURCE-NAME
|
|||
|
|
(SELECTED-CANDIDATE (PATTERN . NUMBER-OF-USE) ...) ...) ...)")
|
|||
|
|
|
|||
|
|
(defconst helm-adaptive-freq-coefficient 5)
|
|||
|
|
(defconst helm-adaptive-recent-coefficient 2)
|
|||
|
|
|
|||
|
|
(defun helm-adaptive-done-reset ()
|
|||
|
|
(setq helm-adaptive-done nil))
|
|||
|
|
|
|||
|
|
;;;###autoload
|
|||
|
|
(define-minor-mode helm-adaptive-mode
|
|||
|
|
"Toggle adaptive sorting in all sources."
|
|||
|
|
:group 'helm-adapt
|
|||
|
|
:require 'helm-adaptive
|
|||
|
|
:global t
|
|||
|
|
(if helm-adaptive-mode
|
|||
|
|
(progn
|
|||
|
|
(unless helm-adaptive-history
|
|||
|
|
(helm-adaptive-maybe-load-history))
|
|||
|
|
(add-hook 'kill-emacs-hook 'helm-adaptive-save-history)
|
|||
|
|
;; Should run at beginning of `helm-initial-setup'.
|
|||
|
|
(add-hook 'helm-before-initialize-hook 'helm-adaptive-done-reset)
|
|||
|
|
;; Should run at beginning of `helm-exit-minibuffer'.
|
|||
|
|
(add-hook 'helm-before-action-hook 'helm-adaptive-store-selection)
|
|||
|
|
;; Should run at beginning of `helm-select-action'.
|
|||
|
|
(add-hook 'helm-select-action-hook 'helm-adaptive-store-selection))
|
|||
|
|
(helm-adaptive-save-history)
|
|||
|
|
(setq helm-adaptive-history nil)
|
|||
|
|
(remove-hook 'kill-emacs-hook 'helm-adaptive-save-history)
|
|||
|
|
(remove-hook 'helm-before-initialize-hook 'helm-adaptive-done-reset)
|
|||
|
|
(remove-hook 'helm-before-action-hook 'helm-adaptive-store-selection)
|
|||
|
|
(remove-hook 'helm-select-action-hook 'helm-adaptive-store-selection)))
|
|||
|
|
|
|||
|
|
(defun helm-adapt-use-adaptive-p (&optional source-name)
|
|||
|
|
"Return current source only if it use adaptive history, nil otherwise."
|
|||
|
|
(when helm-adaptive-mode
|
|||
|
|
(let* ((source (or source-name (helm-get-current-source)))
|
|||
|
|
(adapt-source (or (assoc-default 'filtered-candidate-transformer source)
|
|||
|
|
(assoc-default 'candidate-transformer source))))
|
|||
|
|
(if (listp adapt-source)
|
|||
|
|
(and (memq 'helm-adaptive-sort adapt-source) source)
|
|||
|
|
(and (eq adapt-source 'helm-adaptive-sort) source)))))
|
|||
|
|
|
|||
|
|
(defun helm-adaptive-store-selection ()
|
|||
|
|
"Store history information for the selected candidate."
|
|||
|
|
(unless helm-adaptive-done
|
|||
|
|
(setq helm-adaptive-done t)
|
|||
|
|
(let ((source (helm-adapt-use-adaptive-p)))
|
|||
|
|
(when source
|
|||
|
|
(let* ((source-name (assoc-default 'name source))
|
|||
|
|
(source-info (or (assoc source-name helm-adaptive-history)
|
|||
|
|
(progn
|
|||
|
|
(push (list source-name) helm-adaptive-history)
|
|||
|
|
(car helm-adaptive-history))))
|
|||
|
|
(selection (helm-get-selection nil t))
|
|||
|
|
(selection-info (progn
|
|||
|
|
(setcdr source-info
|
|||
|
|
(cons
|
|||
|
|
(let ((found (assoc selection (cdr source-info))))
|
|||
|
|
(if (not found)
|
|||
|
|
;; new entry
|
|||
|
|
(list selection)
|
|||
|
|
;; move entry to the beginning of the
|
|||
|
|
;; list, so that it doesn't get
|
|||
|
|
;; trimmed when the history is
|
|||
|
|
;; truncated
|
|||
|
|
(setcdr source-info
|
|||
|
|
(delete found (cdr source-info)))
|
|||
|
|
found))
|
|||
|
|
(cdr source-info)))
|
|||
|
|
(cadr source-info)))
|
|||
|
|
(pattern-info (progn
|
|||
|
|
(setcdr selection-info
|
|||
|
|
(cons
|
|||
|
|
(let ((found (assoc helm-pattern (cdr selection-info))))
|
|||
|
|
(if (not found)
|
|||
|
|
;; new entry
|
|||
|
|
(cons helm-pattern 0)
|
|||
|
|
;; move entry to the beginning of the
|
|||
|
|
;; list, so if two patterns used the
|
|||
|
|
;; same number of times then the one
|
|||
|
|
;; used last appears first in the list
|
|||
|
|
(setcdr selection-info
|
|||
|
|
(delete found (cdr selection-info)))
|
|||
|
|
found))
|
|||
|
|
(cdr selection-info)))
|
|||
|
|
(cadr selection-info)))
|
|||
|
|
(timestamp-info (helm-aif (assq 'timestamp (cdr selection-info))
|
|||
|
|
it
|
|||
|
|
(setcdr selection-info (cons (cons 'timestamp 0) (cdr selection-info)))
|
|||
|
|
(cadr selection-info))))
|
|||
|
|
;; Increase usage count.
|
|||
|
|
(setcdr pattern-info (1+ (cdr pattern-info)))
|
|||
|
|
;; Update timestamp.
|
|||
|
|
(setcdr timestamp-info (float-time))
|
|||
|
|
;; Truncate history if needed.
|
|||
|
|
(if (> (length (cdr selection-info)) helm-adaptive-history-length)
|
|||
|
|
(setcdr selection-info
|
|||
|
|
(cl-subseq (cdr selection-info) 0 helm-adaptive-history-length))))))))
|
|||
|
|
|
|||
|
|
(defun helm-adaptive-maybe-load-history ()
|
|||
|
|
"Load `helm-adaptive-history-file' which contain `helm-adaptive-history'.
|
|||
|
|
Returns nil if `helm-adaptive-history-file' doesn't exist."
|
|||
|
|
(when (and helm-adaptive-history-file
|
|||
|
|
(file-readable-p helm-adaptive-history-file))
|
|||
|
|
(load-file helm-adaptive-history-file)))
|
|||
|
|
|
|||
|
|
(defun helm-adaptive-save-history (&optional arg)
|
|||
|
|
"Save history information to the file given by `helm-adaptive-history-file'."
|
|||
|
|
(interactive "p")
|
|||
|
|
(when helm-adaptive-history-file
|
|||
|
|
(with-temp-buffer
|
|||
|
|
(insert
|
|||
|
|
";; -*- mode: emacs-lisp -*-\n"
|
|||
|
|
";; History entries used for helm adaptive display.\n")
|
|||
|
|
(let (print-length print-level)
|
|||
|
|
(prin1 `(setq helm-adaptive-history ',helm-adaptive-history)
|
|||
|
|
(current-buffer)))
|
|||
|
|
(insert ?\n)
|
|||
|
|
(write-region (point-min) (point-max) helm-adaptive-history-file nil
|
|||
|
|
(unless arg 'quiet)))))
|
|||
|
|
|
|||
|
|
(defun helm-adaptive-sort (candidates source)
|
|||
|
|
"Sort the CANDIDATES for SOURCE by usage frequency.
|
|||
|
|
This is a filtered candidate transformer you can use with the
|
|||
|
|
`filtered-candidate-transformer' attribute."
|
|||
|
|
(let* ((source-name (assoc-default 'name source))
|
|||
|
|
(source-info (assoc source-name helm-adaptive-history)))
|
|||
|
|
(if source-info
|
|||
|
|
(let ((usage
|
|||
|
|
;; Loop in the SOURCE entry of `helm-adaptive-history'
|
|||
|
|
;; and assemble a list containing the (CANDIDATE
|
|||
|
|
;; . USAGE-COUNT) pairs.
|
|||
|
|
(cl-loop with cf = (if helm-adaptive-sort-by-frequent-recent-usage
|
|||
|
|
helm-adaptive-freq-coefficient 1)
|
|||
|
|
with cr = helm-adaptive-recent-coefficient
|
|||
|
|
for (src-cand . infos) in (cdr source-info)
|
|||
|
|
for count-freq = 0
|
|||
|
|
for count-rec =
|
|||
|
|
(helm-aif (and helm-adaptive-sort-by-frequent-recent-usage
|
|||
|
|
(assq 'timestamp infos))
|
|||
|
|
(* cr (+ (float-time) (cdr it)))
|
|||
|
|
0)
|
|||
|
|
do (cl-loop for (pattern . score) in
|
|||
|
|
(remove (assq 'timestamp infos) infos)
|
|||
|
|
;; If current pattern is equal to
|
|||
|
|
;; the previously used one then
|
|||
|
|
;; this candidate has priority
|
|||
|
|
;; (that's why its count-freq is
|
|||
|
|
;; boosted by 10000) and it only
|
|||
|
|
;; has to compete with other
|
|||
|
|
;; candidates which were also
|
|||
|
|
;; selected with the same pattern.
|
|||
|
|
if (equal pattern helm-pattern)
|
|||
|
|
return (setq count-freq (+ 10000 score))
|
|||
|
|
else do (cl-incf count-freq score))
|
|||
|
|
and collect (cons src-cand (+ (* count-freq cf) count-rec))
|
|||
|
|
into results
|
|||
|
|
;; Sort the list in descending order, so
|
|||
|
|
;; candidates with highest priority come
|
|||
|
|
;; first.
|
|||
|
|
finally return
|
|||
|
|
(sort results (lambda (first second)
|
|||
|
|
(> (cdr first) (cdr second)))))))
|
|||
|
|
(if (consp usage)
|
|||
|
|
;; Put those candidates first which have the highest usage count.
|
|||
|
|
(cl-loop for (cand . _freq) in usage
|
|||
|
|
for info = (or (and (assq 'multiline source)
|
|||
|
|
(replace-regexp-in-string
|
|||
|
|
"\n\\'" "" cand))
|
|||
|
|
;; Some transformers like in
|
|||
|
|
;; bookmarks may add a leading
|
|||
|
|
;; space to provide additional
|
|||
|
|
;; infos like an icon as a
|
|||
|
|
;; display prop, strip out this
|
|||
|
|
;; leading space for
|
|||
|
|
;; comparison. Same for a
|
|||
|
|
;; trailing space (helm
|
|||
|
|
;; boookmark add bmk location as
|
|||
|
|
;; a display prop when
|
|||
|
|
;; displaying it).
|
|||
|
|
(helm-aand (replace-regexp-in-string "\\` " "" cand)
|
|||
|
|
(replace-regexp-in-string " \\'" "" it)))
|
|||
|
|
when (cl-member info candidates
|
|||
|
|
:test 'helm-adaptive-compare)
|
|||
|
|
collect (car it) into sorted
|
|||
|
|
and do (setq candidates
|
|||
|
|
(cl-remove info candidates
|
|||
|
|
:test 'helm-adaptive-compare))
|
|||
|
|
finally return (append sorted candidates))
|
|||
|
|
(message "Your `%s' is maybe corrupted or too old, \
|
|||
|
|
you should reinitialize it with `helm-reset-adaptive-history'"
|
|||
|
|
helm-adaptive-history-file)
|
|||
|
|
(sit-for 1)
|
|||
|
|
candidates))
|
|||
|
|
;; if there is no information stored for this source then do nothing
|
|||
|
|
candidates)))
|
|||
|
|
|
|||
|
|
;;;###autoload
|
|||
|
|
(defun helm-reset-adaptive-history ()
|
|||
|
|
"Delete all `helm-adaptive-history' and his file.
|
|||
|
|
Useful when you have a old or corrupted
|
|||
|
|
`helm-adaptive-history-file'."
|
|||
|
|
(interactive)
|
|||
|
|
(when (y-or-n-p "Really delete all your `helm-adaptive-history'? ")
|
|||
|
|
(setq helm-adaptive-history nil)
|
|||
|
|
(when (and helm-adaptive-history-file
|
|||
|
|
(file-exists-p helm-adaptive-history-file))
|
|||
|
|
(delete-file helm-adaptive-history-file))))
|
|||
|
|
|
|||
|
|
(defun helm-adaptive-compare (x y)
|
|||
|
|
"Compare display parts if some of candidates X and Y.
|
|||
|
|
|
|||
|
|
Arguments X and Y are cons cell in (DISPLAY . REAL) format or
|
|||
|
|
atoms."
|
|||
|
|
(equal (if (listp x) (car x) x)
|
|||
|
|
(if (listp y) (car y) y)))
|
|||
|
|
|
|||
|
|
|
|||
|
|
(provide 'helm-adaptive)
|
|||
|
|
|
|||
|
|
;;; helm-adaptive.el ends here
|