Fix thread-first overrides

cespare · Sep 17, 2021 · 5932ea3 · 5932ea3
1 parent 67936d1
commit 5932ea3
Show file tree

Hide file tree

Showing 3 changed files with 19 additions and 0 deletions.
diff --git a/cljfmt/cljfmt.go b/cljfmt/cljfmt.go
@@ -192,6 +192,7 @@ func (c *config) processFile(filename string, in io.Reader) error {
 	p := format.NewPrinter(&buf2)
 	p.IndentChar = ' '
 	p.IndentOverrides = c.indentOverrides
+	p.ThreadFirstStyleOverrides = c.threadFirstOverrides
 	p.Transforms = c.transforms
 	if err := p.PrintTree(t); err != nil {
 		return err

diff --git a/format/format_test.go b/format/format_test.go
@@ -126,6 +126,16 @@ func TestIssue41(t *testing.T) {
 	testChangeCustom(t, file, file, f)
 }
 
+func TestThreadFirstOverride(t *testing.T) {
+	const file = "custom/threadfirst.clj"
+	f := func(p *Printer) {
+		p.ThreadFirstStyleOverrides = map[string]ThreadFirstStyle{
+			"-?>": ThreadFirstNormal,
+		}
+	}
+	testChangeCustom(t, file, file, f)
+}
+
 func testFixture(t *testing.T, filename string) {
 	testChange(t, filename, filename)
 }

diff --git a/format/testdata/custom/threadfirst.clj b/format/testdata/custom/threadfirst.clj
@@ -0,0 +1,8 @@
+(-?> a
+     b
+     (cond->
+       "1"
+         (foo 1)
+       "2"
+         (foo 2)
+       :default))