toki/emacs-winum
1
0
Fork 0
Code Issues Pull requests Activity

feat: define convenience macro for auto-assign

Browse source
This commit is contained in:
Kiana Sheibani 2024-11-14 00:27:30 -05:00
parent aca0f74f3a
commit efa0aedb7c
Signed by: toki
GPG key ID: 6CB106C25E86A9F7
1 changed files with 33 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

40
winum.el
View file

@ -128,9 +128,11 @@ and *NeoTree* the number 0:
This function is called after `winum-assign-functions' to automatically assign This function is called after `winum-assign-functions' to automatically assign
indices to the remaining windows. It must take in a list of windows and call indices to the remaining windows. It must take in a list of windows and call
the function `winum--assign' to assign an index to each, while avoiding the function `winum--assign' to assign an index to each.
assigning any indices already taken (stored in `winum--assigned-indices').
Members of this list should be tested for using `equal'. Auto-assign functions must not assign any indices that are `equal' to an index
already used (stored in `winum--assigned-indices'). The convenience macro
`winum--assign-unique' will handle this automatically.
The default auto-assign function is `winum--auto-assign', which assigns strictly The default auto-assign function is `winum--auto-assign', which assigns strictly
increasing integers to each window." increasing integers to each window."
@ -505,16 +507,40 @@ POSITION: position in the mode-line."
(when windows (when windows
(funcall winum-auto-assign-function windows)))) (funcall winum-auto-assign-function windows))))
(defmacro winum--assign-unique (window var &optional after update pred)
"Ensure that the index in VAR is unique, then assign it to WINDOW.
If AFTER is non-nil, also ensure that VAR is unique after the assignment.
UPDATE and PRED are quoted function symbols or unquoted sexps. If they
are functions, then the value of VAR is passed as input.
UPDATE must return a new index to be assigned to VAR, with the guarantee
that repeatedly executing it will never return the same index twice. The
default updating function is `1+'.
PRED must be non-nil if VAR must be updated. By default, PRED checks if
the variable's value is in `winum--assigned-indices'."
(unless pred
(setq pred `(member ,var winum--assigned-indices)))
(unless update
(setq update '#'1+))
(and (= (safe-length pred) 2)
(memq (car pred) '(quote function))
(setq pred `(funcall ,pred ,var)))
(and (= (safe-length update) 2)
(memq (car update) '(quote function))
(setq update `(funcall ,update ,var)))
`(progn
(while ,pred (setq ,var ,update))
(winum--assign ,window ,var)
,@(and after `((setq ,var ,update)))))
(defun winum--auto-assign (windows) (defun winum--auto-assign (windows)
"The default auto-assign function for assigning indices to windows. "The default auto-assign function for assigning indices to windows.
Takes in the list of windows WINDOWS and uses `winum--assign' to assign Takes in the list of windows WINDOWS and uses `winum--assign' to assign
increasing integers to it." increasing integers to it."
(let ((index 1)) (let ((index 1))
(dolist (w windows) (dolist (w windows)
(while (member index winum--assigned-indices) (winum--assign-unique w index t))))
(setq index (1+ index)))
(winum--assign w index)
(setq index (1+ index)))))
(defun winum--try-to-find-custom-index (window) (defun winum--try-to-find-custom-index (window)
"Try to find and assign a custom index for WINDOW. "Try to find and assign a custom index for WINDOW.