Fix clojure-ts-add-arity bug when body has more than one expression

rrudakov · bbatsov · commit 32490c80fad7 · 2025-08-15T07:41:10.000+03:00
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -9,7 +9,8 @@
 - [#116](https://github.com/clojure-emacs/clojure-ts-mode/pull/116): Extend built-in completion to complete all imported symbols from an `ns`
   form.
 - Add documentation and bug reporting commands from `clojure-mode`.
-- Add some ns manipulation functions from `clojure-mode`.
+- [#118](https://github.com/clojure-emacs/clojure-ts-mode/pull/118): Add some ns manipulation functions from `clojure-mode`.
+- Fix a bug in `clojure-ts-add-arity` when body has more than one expression.
 
 ## 0.5.1 (2025-06-17)
 
diff --git a/clojure-ts-mode.el b/clojure-ts-mode.el
@@ -2325,7 +2325,12 @@ type, etc.  See `treesit-thing-settings' for more details."
     (when same-line-p
       (newline-and-indent))
     (when single-arity-p
-      (insert-pair 2 ?\( ?\))
+      (save-excursion
+        (backward-up-list)
+        (forward-list)
+        (down-list -1)
+        (insert ")"))
+      (insert "(")
       (backward-up-list))
     (insert "([])\n")
     ;; Put the point between square brackets.
diff --git a/test/clojure-ts-mode-refactor-add-arity-test.el b/test/clojure-ts-mode-refactor-add-arity-test.el
@@ -80,6 +80,22 @@
 
     (clojure-ts-add-arity))
 
+  (when-refactoring-with-point-it "should handle a single-arity defn with multiple body expressions"
+    "(defn fo|o
+  ^{:bla \"meta\"}
+  [arg]
+  body
+  second-expr)"
+
+    "(defn foo
+  ^{:bla \"meta\"}
+  ([|])
+  ([arg]
+   body
+   second-expr))"
+
+    (clojure-ts-add-arity))
+
   (when-refactoring-with-point-it "should add an arity to a multi-arity defn"
     "(defn foo
   ([arg1])
diff --git a/test/samples/refactoring.clj b/test/samples/refactoring.clj
@@ -133,7 +133,8 @@
 (defn foo
   ^{:bla "meta"}
   [arg]
-  body)
+  body
+  second-expr)
 
 (if ^boolean (= 2 2)
   true
