squarify again, now with numeric prefix

1 files changed, 23 insertions, 12 deletions

diff --git a/elisp/geiser-mode.el b/elisp/geiser-mode.el
index 3575474..1645ba2 100644
--- a/elisp/geiser-mode.el
+++ b/elisp/geiser-mode.el
@@ -176,22 +176,33 @@ With prefix, try to enter the current's buffer module."
     (goto-char (point-max))
     (pop-to-buffer b)))
 
-(defun geiser-squarify ()
-  "Toggle between () and [] for current form."
-  (interactive)
-  (let ((pared (and (boundp 'paredit-mode) paredit-mode)))
+(defun geiser-squarify (n)
+  "Toggle between () and [] for current form.
+With numeric prefix, perform that many toggles, forward for
+positive values and backward for negative."
+  (interactive "p")
+  (let ((pared (and (boundp 'paredit-mode) paredit-mode))
+        (fwd (> n 0))
+        (steps (abs n)))
     (when pared (paredit-mode -1))
     (unwind-protect
         (save-excursion
           (unless (looking-at-p "\\s(") (backward-up-list))
-          (let ((p (point))
-                (round (looking-at-p "(")))
-            (forward-sexp)
-            (backward-delete-char 1)
-            (insert (if round "]" ")"))
-            (goto-char p)
-            (delete-char 1)
-            (insert (if round "[" "("))))
+          (while (> steps 0)
+            (let ((p (point))
+                  (round (looking-at-p "(")))
+              (forward-sexp)
+              (backward-delete-char 1)
+              (insert (if round "]" ")"))
+              (goto-char p)
+              (delete-char 1)
+              (insert (if round "[" "("))
+              (setq steps (1- steps))
+              (backward-char)
+              (condition-case nil
+                  (progn (when fwd (forward-sexp 2))
+                         (backward-sexp))
+                (error (setq steps 0))))))
       (when pared (paredit-mode 1)))))