update import writer

NethermindEth · Feb 15, 2023 · cc406a8 · cc406a8
1 parent ba9c2a7
commit cc406a8
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/src/cairoWriter/utils.ts b/src/cairoWriter/utils.ts
@@ -24,10 +24,12 @@ export function writeImports(imports: Map<string, Set<string>>): string {
     );
   }
   return [...imports.entries()]
-    .map(
-      ([location, importedSymbols]) =>
-        `from ${location} import ${[...importedSymbols.keys()].join(', ')}`,
-    )
+    .map(([location, importedSymbols]) => {
+      const toImport = [...importedSymbols.keys()];
+      const toImportFormatted =
+        toImport.length === 1 ? toImport[0] : '{' + toImport.join(', ') + '}';
+      return `use ${location.replaceAll('.', '::')}::${toImportFormatted};`;
+    })
     .join('\n');
 }