Fix import formatting. Closes #99.

datatrash · Mar 24, 2021 · a6e7d86 · a6e7d86
1 parent 8fcd161
commit a6e7d86
Show file tree

Hide file tree

Showing 3 changed files with 25 additions and 6 deletions.
diff --git a/src/formatting/mod.rs b/src/formatting/mod.rs
@@ -521,10 +521,17 @@ impl CodeFormatter {
     fn format_specific_import_args(&mut self, args: &[ArgItem<SpecificImportArg>]) -> Fmt {
         let mut fmt = Fmt::new();
         for (path, comma) in args {
-            fmt = fmt
-                .fmt(self, path.map(|p| p.to_string()))
-                .fmt(self, comma)
-                .spc_if_next();
+            fmt = fmt.fmt(self, &path.data.path.map(|p| p.to_string()));
+
+            if let Some((tag_as, id_as)) = &path.data.as_ {
+                fmt = fmt
+                    .spc()
+                    .fmt(self, tag_as)
+                    .spc()
+                    .fmt(self, &id_as.map(|p| p.to_string()));
+            }
+
+            fmt = fmt.fmt(self, comma).spc_if_next();
         }
         fmt
     }
@@ -1052,7 +1059,10 @@ mod tests {
     }
 
     fn get_source(src: &str) -> Arc<RefCell<dyn ParsingSource>> {
-        InMemoryParsingSource::new().add("test.asm", src).into()
+        InMemoryParsingSource::new()
+            .add("test.asm", src)
+            .add("other.asm", "nop")
+            .into()
     }
 
     // Cross-platform eq

diff --git a/test/cli/format/valid-formatted.asm b/test/cli/format/valid-formatted.asm
@@ -49,4 +49,10 @@ nop
 
 .macro MyMacro(arg1, arg2) {
     brk
-}
+}
+
+.import foo as bar from "other.asm" {
+    nop
+}
+
+.export data as boop
diff --git a/test/cli/format/valid-unformatted.asm b/test/cli/format/valid-unformatted.asm
@@ -28,3 +28,6 @@ start=$2000 + 4 -%00100}
  nop
 
  .macro MyMacro(arg1,arg2){brk}
+
+ .import   foo    as    bar    from    "other.asm" {nop}
+ .export    data   as    boop