Structure definitiona formatting fixes

maccasoft · Jun 12, 2024 · 381f5f3 · 381f5f3
1 parent be04811
commit 381f5f3
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 1 deletion.
diff --git a/modules/spin-tools/src-tests/com/maccasoft/propeller/spin2/Spin2FormatterTest.java b/modules/spin-tools/src-tests/com/maccasoft/propeller/spin2/Spin2FormatterTest.java
@@ -582,4 +582,18 @@ void testExpressions() {
             + "", text);
     }
 
+    @Test
+    void testStructureDefinition() {
+        Formatter subject = new Spin2Formatter();
+        String text = subject.format(""
+            + "CON point(x,y)\n"
+            + " line(point a,point b,color)\n"
+            + "");
+        Assertions.assertEquals(""
+            + "CON\n"
+            + "    point(x, y)\n"
+            + "    line(point a, point b, color)\n"
+            + "", text);
+    }
+
 }
diff --git a/modules/spin-tools/src/com/maccasoft/propeller/Formatter.java b/modules/spin-tools/src/com/maccasoft/propeller/Formatter.java
@@ -244,6 +244,10 @@ void formatConstant(FormatterStringBuilder sb) {
                     }
                     break;
                 case 4:
+                    if ("(".equals(token.getText()) || ")".equals(token.getText())) {
+                        sb.append(token);
+                        break;
+                    }
                     if ("=".equals(token.getText())) {
                         sb.append(" ");
                         sb.append(token);
@@ -260,7 +264,7 @@ void formatConstant(FormatterStringBuilder sb) {
                         state = 6;
                         break;
                     }
-                    if (sb.lastChar() != ' ') {
+                    if (sb.lastChar() != ' ' && sb.lastChar() != '(') {
                         sb.append(" ");
                     }
                     sb.append(token);