]> code.delx.au - gnu-emacs-elpa/blob - packages/el-search/el-search.el
simplify el-search-hl-post-command-fun
[gnu-emacs-elpa] / packages / el-search / el-search.el
1 ;;; el-search.el --- Expression based incremental search for emacs-lisp-mode -*- lexical-binding: t -*-
2
3 ;; Copyright (C) 2015 Free Software Foundation, Inc
4
5 ;; Author: Michael Heerdegen <michael_heerdegen@web.de>
6 ;; Maintainer: Michael Heerdegen <michael_heerdegen@web.de>
7 ;; Created: 29 Jul 2015
8 ;; Keywords: lisp
9 ;; Compatibility: GNU Emacs 25
10 ;; Version: 0.0.3
11 ;; Package-Requires: ((emacs "25"))
12
13
14 ;; This file is not part of GNU Emacs.
15
16 ;; GNU Emacs is free software: you can redistribute it and/or modify
17 ;; it under the terms of the GNU General Public License as published by
18 ;; the Free Software Foundation, either version 3 of the License, or
19 ;; (at your option) any later version.
20
21 ;; GNU Emacs is distributed in the hope that it will be useful,
22 ;; but WITHOUT ANY WARRANTY; without even the implied warranty of
23 ;; MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
24 ;; GNU General Public License for more details.
25
26 ;; You should have received a copy of the GNU General Public License
27 ;; along with GNU Emacs. If not, see <http://www.gnu.org/licenses/>.
28
29
30 ;;; Commentary:
31
32 ;; Introduction
33 ;; ============
34 ;;
35 ;;
36 ;; The main user entry point is `el-search-pattern'. This command
37 ;; prompts for a `pcase' pattern and searches the current buffer for
38 ;; matching expressions by iteratively `read'ing buffer contents. For
39 ;; any match, point is put at the beginning of the expression found
40 ;; (unlike isearch which puts point at the end of matches).
41 ;;
42 ;; It doesn't matter how the code is actually formatted. Comments are
43 ;; ignored, and strings are treated as atomic objects, their contents
44 ;; are not being searched.
45 ;;
46 ;; Example 1: if you enter
47 ;;
48 ;; 97
49 ;;
50 ;; at the prompt, this will find any occurrence of the number 97 in
51 ;; the code, but not 977 or (+ 90 7) or "My string containing 97".
52 ;; But it will find anything `eq' to 97 after reading, e.g. #x61 or
53 ;; ?a.
54 ;;
55 ;;
56 ;; Example 2: If you enter the pattern
57 ;;
58 ;; `(defvar ,_)
59 ;;
60 ;; you search for all defvar forms that don't specify an init value.
61 ;;
62 ;; The following will search for defvar forms with a docstring whose
63 ;; first line is longer than 70 characters:
64 ;;
65 ;; `(defvar ,_ ,_
66 ;; ,(and s (guard (< 70 (length (car (split-string s "\n")))))))
67 ;;
68 ;;
69 ;; When a search pattern is processed, the searched buffer is current
70 ;; with point at the beginning of the currently tested expression.
71 ;;
72 ;;
73 ;; Example 3:
74 ;;
75 ;; I can be useful to use (guard EXP) patterns for side effects.
76 ;;
77 ;; The following pattern will search for symbols defined in any
78 ;; library whose name starts with "cl". As a side effect, it prints
79 ;; the current line number, whether we have a macro or a function, and
80 ;; the defining file in the echo area for each match:
81 ;;
82 ;; (and (pred symbolp)
83 ;; (let file (symbol-file exp))
84 ;; (guard file)
85 ;; (let lib-name (file-name-sans-extension
86 ;; (file-name-nondirectory file)))
87 ;; (guard (string-match-p "^cl" lib-name))
88 ;; (or (and (pred macrop) (let type "macro "))
89 ;; (and (pred functionp) (let type "function "))
90 ;; (let type ""))
91 ;; (guard (message "Line %d: %s`%S' (from \"%s\")"
92 ;; (line-number-at-pos)
93 ;; type
94 ;; exp
95 ;; lib-name)))
96 ;;
97 ;; `message' never returns nil, so the last `guard' always "matches".
98 ;;
99 ;;
100 ;; Convenience
101 ;; ===========
102 ;;
103 ;; For pattern input, the minibuffer is put into `emacs-lisp-mode'.
104 ;;
105 ;; Any input PATTERN is silently transformed into (and exp PATTERN)
106 ;; so that you can always refer to the whole currently tested
107 ;; expression via the variable `exp'.
108 ;;
109 ;; Example 4:
110 ;;
111 ;; If you want to search a buffer for symbols that are defined in
112 ;; "cl-lib", you can use this pattern
113 ;;
114 ;; (guard (and (symbolp exp)
115 ;; (when-let ((file (symbol-file exp)))
116 ;; (string-match-p "cl-lib\\.elc?$" file))))
117 ;;
118 ;;
119 ;; ,----------------------------------------------------------------------
120 ;; | Q: "But I hate `pcase'! Can't we just do without?" |
121 ;; | |
122 ;; | A: Respect that you kept up until here! Just use (guard CODE), where|
123 ;; | CODE is any normal Elisp expression that returns non-nil when and |
124 ;; | only when you have a match. Use the variable `exp' to refer to |
125 ;; | the currently tested expression. Just like in the last example! |
126 ;; `----------------------------------------------------------------------
127 ;;
128 ;;
129 ;; It's cumbersome to write out the same complicated pattern
130 ;; constructs in the minibuffer again and again. You can define your
131 ;; own pcase pattern types for the purpose of el-search with
132 ;; `el-search-defpattern'. It is just like `pcase-defmacro', but the
133 ;; effect is limited to this package. See C-h f `el-search-pattern'
134 ;; for a list of predefined additional pattern forms.
135 ;;
136 ;;
137 ;; Replacing
138 ;; =========
139 ;;
140 ;; You can replace expressions with command `el-search-query-replace'.
141 ;; You are queried for a (pcase) pattern and a replacement expression.
142 ;; For each match of the pattern, the replacement expression is
143 ;; evaluated with the bindings created by the pcase matching in
144 ;; effect, and printed to produce the replacement string.
145 ;;
146 ;; Example: In some buffer you want to swap the two expressions at the
147 ;; places of the first two arguments in all calls of function `foo',
148 ;; so that e.g.
149 ;;
150 ;; (foo 'a (* 2 (+ 3 4)) t)
151 ;;
152 ;; becomes
153 ;;
154 ;; (foo (* 2 (+ 3 4)) 'a t).
155 ;;
156 ;; This will do it:
157 ;;
158 ;; M-x el-search-query-replace RET
159 ;; `(foo ,a ,b . ,rest) RET
160 ;; `(foo ,b ,a . ,rest) RET
161 ;;
162 ;; Type y to replace a match and go to the next one, r to replace
163 ;; without moving, SPC to go to the next match and ! to replace all
164 ;; remaining matches automatically. q quits. n is like SPC, so that
165 ;; y and n work like in isearch (meaning "yes" and "no") if you are
166 ;; used to that.
167 ;;
168 ;;
169 ;; Suggested key bindings
170 ;; ======================
171 ;;
172 ;; (define-key emacs-lisp-mode-map [(control ?S)] #'el-search-pattern)
173 ;; (define-key emacs-lisp-mode-map [(control ?%)] #'el-search-query-replace)
174 ;;
175 ;; (define-key isearch-mode-map [(control ?S)] #'el-search-search-from-isearch)
176 ;; (define-key isearch-mode-map [(control ?%)] #'el-search-replace-from-isearch)
177 ;;
178 ;; The bindings in `isearch-mode-map' let you conveniently switch to
179 ;; elisp searching from isearch.
180 ;;
181 ;;
182 ;; Bugs, Known Limitations
183 ;; =======================
184 ;;
185 ;; - Replacing: in some cases the reader syntax of forms
186 ;; is changing due to reading+printing. "Some" because we can treat
187 ;; that problem in most cases.
188 ;;
189 ;; - Similarly: Comments are normally preserved (where it makes
190 ;; sense). But when replacing like `(foo ,a ,b) -> `(foo ,b ,a)
191 ;;
192 ;; in a content like
193 ;;
194 ;; (foo
195 ;; a
196 ;; ;;a comment
197 ;; b)
198 ;;
199 ;; the comment will be lost.
200 ;;
201 ;;
202 ;; Acknowledgments
203 ;; ===============
204 ;;
205 ;; Thanks to Stefan Monnier for corrections and advice.
206 ;;
207 ;;
208 ;; TODO:
209 ;;
210 ;; - implement backward searching
211 ;;
212 ;; - improve docstrings
213 ;;
214 ;; - handle more reader syntaxes, e.g. #n, #n#
215 ;;
216 ;; - Implement sessions; add multi-file support based on iterators. A
217 ;; file list is read in (or the user can specify an iterator as a
218 ;; variable). The state in the current buffer is just (buffer
219 ;; . marker). Or should this be abstracted into an own lib? Could be
220 ;; named "files-session" or so.
221
222
223
224 ;;; Code:
225
226 ;;;; Requirements
227
228 (eval-when-compile
229 (require 'subr-x))
230
231 (require 'cl-lib)
232 (require 'elisp-mode)
233 (require 'thingatpt)
234
235
236 ;;;; Configuration stuff
237
238 (defgroup el-search nil
239 "Expression based search and replace for `emacs-lisp-mode'."
240 :group 'lisp)
241
242 (defcustom el-search-this-expression-identifier 'exp
243 "Name of the identifier referring to the current expression.
244 The default value is `exp'. You can use this name in the search
245 prompt to refer to the value of the currently tested expression."
246 :type 'symbol)
247
248 (defface el-search-match '((((background dark)) (:background "#0000A0"))
249 (t (:background "DarkSlateGray1")))
250 "Face for highlighting the current match.")
251
252
253 ;;;; Helpers
254
255 (defun el-search--print (expr)
256 (let ((print-quoted t)
257 (print-length nil)
258 (print-level nil))
259 (prin1-to-string expr)))
260
261 (defvar el-search-read-expression-map
262 (let ((map (make-sparse-keymap)))
263 (set-keymap-parent map read-expression-map)
264 (define-key map [(control ?g)] #'abort-recursive-edit)
265 (define-key map [up] nil)
266 (define-key map [down] nil)
267 (define-key map [(control meta backspace)] #'backward-kill-sexp)
268 (define-key map [(control ?S)] #'exit-minibuffer)
269 map)
270 "Map for reading input with `el-search-read-expression'.")
271
272 ;; $$$$$FIXME: this should be in Emacs! There is only a helper `read--expression'.
273 (defun el-search-read-expression (prompt &optional initial-contents hist default read)
274 "Read expression for `my-eval-expression'."
275 (minibuffer-with-setup-hook
276 (lambda ()
277 (emacs-lisp-mode)
278 (use-local-map el-search-read-expression-map)
279 (setq font-lock-mode t)
280 (funcall font-lock-function 1)
281 (backward-sexp)
282 (indent-sexp)
283 (goto-char (point-max)))
284 (read-from-minibuffer prompt initial-contents el-search-read-expression-map read
285 (or hist 'read-expression-history) default)))
286
287 (defun el-search--read-pattern (prompt &optional default read)
288 (let ((this-sexp (sexp-at-point)))
289 (minibuffer-with-setup-hook
290 (lambda ()
291 (when this-sexp
292 (let ((more-defaults (list (concat "'" (el-search--print this-sexp)))))
293 (setq-local minibuffer-default-add-function
294 (lambda () (if (listp minibuffer-default)
295 (append minibuffer-default more-defaults)
296 (cons minibuffer-default more-defaults)))))))
297 (el-search-read-expression
298 prompt el-search--initial-mb-contents 'el-search-history default read))))
299
300 (defun el-search--end-of-sexp ()
301 ;;Point must be at sexp beginning
302 (or (scan-sexps (point) 1) (point-max)))
303
304 (defun el-search--ensure-sexp-start ()
305 "Move point to the beginning of the next sexp if necessary.
306 Don't move if already at beginning of a sexp.
307 Point must not be inside a string or comment."
308 (let ((not-done t) res)
309 (while not-done
310 (let ((stop-here nil)
311 (looking-at-from-back (lambda (regexp n)
312 (save-excursion
313 (backward-char n)
314 (looking-at regexp)))))
315 (while (not stop-here)
316 (cond
317 ((eobp) (signal 'end-of-buffer nil))
318 ((looking-at (rx (and (* space) ";"))) (forward-line))
319 ((looking-at (rx (+ (or space "\n")))) (goto-char (match-end 0)))
320
321 ;; FIXME: can the rest be done more generically?
322 ((and (looking-at (rx (or (syntax symbol) (syntax word))))
323 (not (looking-at "\\_<"))
324 (not (funcall looking-at-from-back ",@" 2)))
325 (forward-symbol 1))
326 ((or (and (looking-at "'") (funcall looking-at-from-back "#" 1))
327 (and (looking-at "@") (funcall looking-at-from-back "," 1)))
328 (forward-char))
329 (t (setq stop-here t)))))
330 (condition-case nil
331 (progn
332 (setq res (save-excursion (read (current-buffer))))
333 (setq not-done nil))
334 (error (forward-char))))
335 res))
336
337 (defvar el-search--pcase-macros '()
338 "List of additional \"el-search\" pcase macros.")
339
340 (defun el-search--make-docstring ()
341 ;; code mainly from `pcase--make-docstring'
342 (let* ((main (documentation (symbol-function 'el-search-pattern) 'raw))
343 (ud (help-split-fundoc main 'pcase)))
344 (require 'help-fns)
345 (with-temp-buffer
346 (insert (or (cdr ud) main))
347 (mapc
348 (pcase-lambda (`(,symbol . ,fun))
349 (when-let ((doc (documentation fun)))
350 (insert "\n\n-- ")
351 (setq doc (help-fns--signature symbol doc nil fun nil))
352 (insert "\n" (or doc "Not documented."))))
353 (reverse el-search--pcase-macros))
354 (let ((combined-doc (buffer-string)))
355 (if ud (help-add-fundoc-usage combined-doc (car ud)) combined-doc)))))
356
357 (put 'el-search-pattern 'function-documentation '(el-search--make-docstring))
358
359 (defmacro el-search-defpattern (name args &rest body)
360 "Like `pcase-defmacro', but limited to el-search patterns.
361 The semantics is exactly that of `pcase-defmacro', but the scope
362 of the definitions is limited to \"el-search\"."
363 (declare (indent 2) (debug defun))
364 `(setf (alist-get ',name el-search--pcase-macros)
365 (lambda ,args ,@body)))
366
367 (el-search-defpattern string (&rest regexps)
368 "Matches any string that is matched by all REGEXPS."
369 (let ((string (make-symbol "string"))
370 (regexp (make-symbol "regexp")))
371 `(and (pred stringp)
372 (pred (lambda (,string)
373 (cl-every
374 (lambda (,regexp) (string-match-p ,regexp ,string))
375 (list ,@regexps)))))))
376
377 (el-search-defpattern symbol (&rest regexps)
378 "Matches any symbol whose name is matched by all REGEXPS."
379 `(and (pred symbolp)
380 (app symbol-name (string ,@regexps))))
381
382 (defun el-search--match-symbol-file (regexp symbol)
383 (when-let ((symbol-file (and (symbolp symbol)
384 (symbol-file symbol))))
385 (string-match-p
386 (if (symbolp regexp) (concat "\\`" (symbol-name regexp) "\\'") regexp)
387 (file-name-sans-extension (file-name-nondirectory symbol-file)))))
388
389 (el-search-defpattern source (regexp)
390 "Matches any symbol whose `symbol-file' is matched by REGEXP.
391
392 This pattern matches when the object is a symbol for that
393 `symbol-file' returns a (non-nil) FILE-NAME that fulfills
394 (string-match-p REGEXP (file-name-sans-extension
395 (file-name-nondirectory FILENAME)))
396
397 REGEXP can also be a symbol, in which case
398
399 (concat \"^\" (symbol-name regexp) \"$\")
400
401 is used as regular expression."
402 `(pred (el-search--match-symbol-file ,regexp)))
403
404 (defun el-search--match-key-sequence (keys expr)
405 (when-let ((expr-keys (pcase expr
406 ((or (pred stringp) (pred vectorp)) expr)
407 (`(kbd ,(and (pred stringp) string)) (ignore-errors (kbd string))))))
408 (apply #'equal
409 (mapcar (lambda (keys) (ignore-errors (key-description keys)))
410 (list keys expr-keys)))))
411
412 (el-search-defpattern keys (key-sequence)
413 "Matches any description of the KEY-SEQUENCE.
414 KEY-SEQUENCE is a key description in a format that Emacs
415 understands.
416
417 This pattern matches any description of the same key sequence.
418
419 Example: the pattern
420
421 (keys (kbd \"C-s\"))
422
423 matches any of these expressions:
424
425 (kbd \"C-s\")
426 [(control ?s)]
427 \"\\C-s\"
428
429 Any of these could be used as equivalent KEY-SEQUENCE in terms of
430 this pattern type."
431 `(pred (el-search--match-key-sequence ,key-sequence)))
432
433 (defmacro el-search--with-additional-pcase-macros (&rest body)
434 `(cl-letf ,(mapcar (pcase-lambda (`(,symbol . ,fun))
435 `((get ',symbol 'pcase-macroexpander) #',fun))
436 el-search--pcase-macros)
437 ,@body))
438
439 (defun el-search--matcher (pattern &rest body)
440 (let ((warning-suppress-log-types '((bytecomp))))
441 (el-search--with-additional-pcase-macros
442 (byte-compile
443 `(lambda (expression)
444 (pcase expression
445 (,pattern ,@(or body (list t)))
446 (_ nil)))))))
447
448 (defun el-search--match-p (matcher expression)
449 (funcall matcher expression))
450
451 (defun el-search--wrap-pattern (pattern)
452 `(and ,el-search-this-expression-identifier ,pattern))
453
454 (defun el-search--skip-expression (expression &optional read)
455 ;; Move forward at least one character. Don't move into a string or
456 ;; comment. Don't move further than the beginning of the next sexp.
457 ;; Try to move as far as possible. Point must be at the beginning
458 ;; of an expression.
459 ;; If there are positions where `read' would succeed, but that do
460 ;; not represent a valid sexp start, move past them (e.g. when
461 ;; before "#'" move past both characters).
462 ;;
463 ;; EXPRESSION must be the (read) expression at point, but when READ
464 ;; is non-nil, ignore the first argument and read the expression at
465 ;; point instead.
466 (when read (setq expression (save-excursion (read (current-buffer)))))
467 (cond
468 ((or (null expression)
469 (equal [] expression)
470 (not (or (listp expression) (vectorp expression))))
471 (goto-char (el-search--end-of-sexp)))
472 ((looking-at (rx (or ",@" "," "#'" "'")))
473 (goto-char (match-end 0)))
474 (t (forward-char))))
475
476 (defun el-search--search-pattern (pattern &optional noerror)
477 "Search elisp buffer with `pcase' PATTERN.
478 Set point to the beginning of the occurrence found and return
479 point. Optional second argument, if non-nil, means if fail just
480 return nil (no error)."
481
482 (let ((matcher (el-search--matcher pattern)) (match-beg nil) (opoint (point)) current-expr)
483
484 ;; when inside a string or comment, move past it
485 (let ((syntax-here (syntax-ppss)))
486 (when (nth 3 syntax-here) ;inside a string
487 (goto-char (nth 8 syntax-here))
488 (forward-sexp))
489 (when (nth 4 syntax-here) ;inside a comment
490 (forward-line 1)
491 (while (and (not (eobp)) (looking-at (rx (and (* space) ";"))))
492 (forward-line 1))))
493
494 (if (catch 'no-match
495 (while (not match-beg)
496 (condition-case nil
497 (setq current-expr (el-search--ensure-sexp-start))
498 (end-of-buffer
499 (goto-char opoint)
500 (throw 'no-match t)))
501 (if (el-search--match-p matcher current-expr)
502 (setq match-beg (point)
503 opoint (point))
504 (el-search--skip-expression current-expr))))
505 (if noerror nil (signal 'end-of-buffer nil)))
506 match-beg))
507
508 (defun el-search--do-subsexps (pos do-fun &optional ret-fun bound)
509 ;; In current buffer, for any expression start between POS and BOUND
510 ;; or (point-max), in order, call two argument function DO-FUN with
511 ;; the current sexp string and the ending position of the current
512 ;; sexp. When done, with RET-FUN given, call it with no args and
513 ;; return the result; else, return nil.
514 (save-excursion
515 (goto-char pos)
516 (condition-case nil
517 (while (< (point) (or bound (point-max)))
518 (let* ((this-sexp-end (save-excursion (thing-at-point--end-of-sexp) (point)))
519 (this-sexp-string (buffer-substring-no-properties (point) this-sexp-end)))
520 (funcall do-fun this-sexp-string this-sexp-end)
521 (el-search--skip-expression (read this-sexp-string))
522 (el-search--ensure-sexp-start)))
523 (end-of-buffer))
524 (when ret-fun (funcall ret-fun))))
525
526 (defun el-search--create-read-map (&optional pos)
527 (let ((mapping '()))
528 (el-search--do-subsexps
529 (or pos (point))
530 (lambda (sexp _) (push (cons (read sexp) sexp) mapping))
531 (lambda () (nreverse mapping))
532 (save-excursion (thing-at-point--end-of-sexp) (point)))))
533
534 (defun el-search--repair-replacement-layout (printed mapping)
535 (with-temp-buffer
536 (insert printed)
537 (el-search--do-subsexps
538 (point-min)
539 (lambda (sexp sexp-end)
540 (when-let ((old (cdr (assoc (read sexp) mapping))))
541 (delete-region (point) sexp-end)
542 (when (string-match-p "\n" old)
543 (unless (looking-back "^[[:space:]]*" (line-beginning-position))
544 (insert "\n"))
545 (unless (looking-at "[[:space:]\)]*$")
546 (insert "\n")
547 (backward-char)))
548 (save-excursion (insert old))))
549 (lambda () (buffer-substring (point-min) (point-max))))))
550
551
552 ;;;; Highlighting
553
554 (defvar-local el-search-hl-overlay nil)
555
556 (defvar el-search-keep-hl nil)
557
558 (defun el-search-hl-sexp-at-point ()
559 (let ((bounds (list (point) (el-search--end-of-sexp))))
560 (if (overlayp el-search-hl-overlay)
561 (apply #'move-overlay el-search-hl-overlay bounds)
562 (overlay-put (setq el-search-hl-overlay (apply #'make-overlay bounds))
563 'face 'el-search-match)))
564 (add-hook 'post-command-hook #'el-search-hl-post-command-fun t t))
565
566 (defun el-search-hl-remove ()
567 (when (overlayp el-search-hl-overlay)
568 (delete-overlay el-search-hl-overlay)))
569
570 (defun el-search-hl-post-command-fun ()
571 (unless (or el-search-keep-hl
572 (eq this-command 'el-search-query-replace)
573 (eq this-command 'el-search-pattern))
574 (el-search-hl-remove)
575 (remove-hook 'post-command-hook 'el-search-hl-post-command-fun t)))
576
577
578 ;;;; Core functions
579
580 (defvar el-search-history '()
581 "List of input strings.")
582
583 (defvar el-search-success nil)
584 (defvar el-search-current-pattern nil)
585
586 ;;;###autoload
587 (defun el-search-pattern (pattern)
588 "Start new or resume last elisp search.
589
590 Search current buffer for expressions that are matched by `pcase'
591 PATTERN. Use `read' to transform buffer contents into
592 expressions.
593
594
595 Additional `pcase' pattern types to be used with this command can
596 be defined with `el-search-defpattern'.
597
598 The following additional pattern types are currently defined:\n"
599 (interactive (list (if (and (eq this-command last-command)
600 el-search-success)
601 el-search-current-pattern
602 (let ((pattern
603 (el-search--read-pattern "Find pcase pattern: "
604 (car el-search-history)
605 t)))
606 ;; A very common mistake: input "foo" instead of "'foo"
607 (when (and (symbolp pattern)
608 (not (eq pattern '_))
609 (or (not (boundp pattern))
610 (not (eq (symbol-value pattern) pattern))))
611 (error "Please don't forget the quote when searching for a symbol"))
612 (el-search--wrap-pattern pattern)))))
613 (setq this-command 'el-search-pattern) ;in case we come from isearch
614 (setq el-search-current-pattern pattern)
615 (let ((opoint (point)))
616 (when (and (eq this-command last-command) el-search-success)
617 (el-search--skip-expression nil t))
618 (setq el-search-success nil)
619 (message "%s" (substitute-command-keys "Type \\[el-search-pattern] to repeat"))
620 (when (condition-case nil
621 (el-search--search-pattern pattern)
622 (end-of-buffer (message "No match")
623 (goto-char opoint)
624 (el-search-hl-remove)
625 (ding)
626 nil))
627 (setq el-search-success t)
628 (el-search-hl-sexp-at-point))))
629
630 (defun el-search-search-and-replace-pattern (pattern replacement &optional mapping)
631 (let ((replace-all nil) (nbr-replaced 0) (nbr-skipped 0) (done nil)
632 (el-search-keep-hl t) (opoint (point))
633 (get-replacement (el-search--matcher pattern replacement)))
634 (unwind-protect
635 (while (and (not done) (el-search--search-pattern pattern t))
636 (setq opoint (point))
637 (unless replace-all (el-search-hl-sexp-at-point))
638 (let* ((read-mapping (el-search--create-read-map))
639 (region (list (point) (el-search--end-of-sexp)))
640 (substring (apply #'buffer-substring-no-properties region))
641 (expr (read substring))
642 (replaced-this nil)
643 (new-expr (funcall get-replacement expr))
644 (to-insert (el-search--repair-replacement-layout
645 (el-search--print new-expr) (append mapping read-mapping)))
646 (do-replace (lambda ()
647 (atomic-change-group
648 (apply #'delete-region region)
649 (let ((inhibit-message t)
650 (opoint (point)))
651 (insert to-insert)
652 (indent-region opoint (point))
653 (goto-char opoint)
654 (el-search-hl-sexp-at-point)))
655 (cl-incf nbr-replaced)
656 (setq replaced-this t))))
657 (if replace-all
658 (funcall do-replace)
659 (while (not (pcase (if replaced-this
660 (read-char-choice "[SPC ! q]" '(?\ ?! ?q ?n))
661 (read-char-choice
662 (concat "Replace this occurrence"
663 (if (or (string-match-p "\n" to-insert)
664 (< 40 (length to-insert)))
665 "" (format " with `%s'" to-insert))
666 "? [y SPC r ! q]" )
667 '(?y ?n ?r ?\ ?! ?q)))
668 (?r (funcall do-replace)
669 nil)
670 (?y (funcall do-replace)
671 t)
672 ((or ?\ ?n)
673 (unless replaced-this (cl-incf nbr-skipped))
674 t)
675 (?! (unless replaced-this
676 (funcall do-replace))
677 (setq replace-all t)
678 t)
679 (?q (setq done t)
680 t)))))
681 (unless (or done (eobp)) (el-search--skip-expression nil t)))))
682 (el-search-hl-remove)
683 (goto-char opoint)
684 (message "Replaced %d matches%s"
685 nbr-replaced
686 (if (zerop nbr-skipped) ""
687 (format " (%d skipped)" nbr-skipped)))))
688
689 ;; We need a variable for the initial contents because we want to `call-interactively'
690 ;; `el-search-query-replace-read-args'
691 (defvar el-search--initial-mb-contents nil)
692
693 (defun el-search-query-replace-read-args ()
694 (barf-if-buffer-read-only)
695 (let* ((from (el-search--read-pattern "Replace from: "))
696 (to (let ((el-search--initial-mb-contents nil))
697 (el-search--read-pattern "Replace with result of evaluation of: " from))))
698 (list (el-search--wrap-pattern (read from)) (read to)
699 (with-temp-buffer
700 (insert to)
701 (el-search--create-read-map 1)))))
702
703 ;;;###autoload
704 (defun el-search-query-replace (from to &optional mapping)
705 "Replace some occurrences of FROM pattern with evaluated TO."
706 (interactive (el-search-query-replace-read-args))
707 (setq this-command 'el-search-query-replace) ;in case we come from isearch
708 (setq el-search-current-pattern from)
709 (barf-if-buffer-read-only)
710 (el-search-search-and-replace-pattern from to mapping))
711
712 (defun el-search--take-over-from-isearch ()
713 (let ((other-end isearch-other-end)
714 (input isearch-string))
715 (isearch-exit)
716 (when (and other-end (< other-end (point)))
717 (goto-char other-end))
718 input))
719
720 ;;;###autoload
721 (defun el-search-search-from-isearch ()
722 ;; FIXME: an interesting alternative would be to really integrate it
723 ;; with Isearch, using `isearch-search-fun-function'.
724 ;; Alas, this is not trivial if we want to transfer our optimizations.
725 (interactive)
726 (let ((el-search--initial-mb-contents (concat "'" (el-search--take-over-from-isearch))))
727 ;; use `call-interactively' so we get recorded in `extended-command-history'
728 (call-interactively #'el-search-pattern)))
729
730 ;;;###autoload
731 (defun el-search-replace-from-isearch ()
732 (interactive)
733 (let ((el-search--initial-mb-contents (concat "'" (el-search--take-over-from-isearch))))
734 (call-interactively #'el-search-query-replace)))
735
736
737
738 (provide 'el-search)
739 ;;; el-search.el ends here