Fix a bug when formatting consecutive -define directives with singl…

…e-quoted macro names.
sile · Aug 26, 2024 · 5c057c3 · 5c057c3
1 parent 3216ca8
commit 5c057c3
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/efmt_core/src/items/module.rs b/efmt_core/src/items/module.rs
@@ -121,7 +121,11 @@ impl<'a> FormatState<'a> {
         let indent = self
             .pending_constants
             .iter()
-            .map(|define| "-define(".len() + define.macro_name().len() + ", ".len())
+            .map(|define| {
+                let macro_name_len = define.macro_name_token().end_position().offset()
+                    - define.macro_name_token().start_position().offset();
+                "-define(".len() + macro_name_len + ", ".len()
+            })
             .max()
             .expect("unreachable");
 

diff --git a/tests/testdata/aligned_macro.erl b/tests/testdata/aligned_macro.erl
@@ -23,3 +23,7 @@
 %% bar
 -define(ccc,  3).
 -define(dddd, 4).
+
+-define('meow-1', "Meow").
+-define('meow-2', "Meow Meow").
+-define('meow-3', "Meow Meow Meow").