julefmt: improve short variable declaration support

julelang · Aug 12, 2024 · 1fd3e60 · 1fd3e60
1 parent ac91290
commit 1fd3e60
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/src/format.jule b/src/format.jule
@@ -7,6 +7,7 @@ use ast for std::jule::ast::{
 }
 use lex for std::jule::lex::{
     Token,
+    TokenId,
     TokenKind,
 }
 use std::jule::types::{TypeKind}
@@ -666,7 +667,7 @@ impl formatter {
                 self.write("static ")
             } else if d.Constant {
                 self.write("const ")
-            } else {
+            } else if d.Setter == nil || d.Setter.Id == TokenId.Eq {
                 self.write("let ")
             }
         }
@@ -682,7 +683,9 @@ impl formatter {
             self.formatType(d.Kind)
         }
         if d.Expr != nil {
-            self.write(" = ")
+            self.write(" ")
+            self.write(d.Setter.Kind)
+            self.write(" ")
             self.formatExpr(d.Expr)
         }
     }