Keep [private] attribute when formatting assignments (#2592)

casey · Jan 22, 2025 · a6ab720 · a6ab720
1 parent ab3da9b
commit a6ab720
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 0 deletions.
diff --git a/src/assignment.rs b/src/assignment.rs
@@ -5,9 +5,14 @@ pub(crate) type Assignment<'src> = Binding<'src, Expression<'src>>;
 
 impl Display for Assignment<'_> {
   fn fmt(&self, f: &mut Formatter) -> fmt::Result {
+    if self.private {
+      writeln!(f, "[private]")?;
+    }
+
     if self.export {
       write!(f, "export ")?;
     }
+
     write!(f, "{} := {}", self.name, self.value)
   }
 }
diff --git a/tests/format.rs b/tests/format.rs
@@ -1155,3 +1155,22 @@ fn if_else() {
     )
     .run();
 }
+
+#[test]
+fn private_variable() {
+  Test::new()
+    .justfile(
+      "
+        [private]
+        foo := 'bar'
+      ",
+    )
+    .arg("--dump")
+    .stdout(
+      "
+        [private]
+        foo := 'bar'
+      ",
+    )
+    .run();
+}