woodstone 0.1.8 → 0.1.9

package/libs/scaffold.js

@@ -0,0 +1,80 @@
+import { Project, SourceFile, ImportDeclaration, ImportSpecifier } from "ts-morph";
+export default class Scaffold {
+    file;
+    project;
+    sourceFile;
+    constructor(file) {
+        this.file = file;
+        this.project = new Project({
+            compilerOptions: { allowJs: true }
+        });
+        this.sourceFile =
+            this.project.getSourceFile(this.file) ||
+                this.project.createSourceFile(this.file, "", { overwrite: false });
+    }
+    getImport(module) {
+        return this.sourceFile
+            .getImportDeclarations()
+            .find((d) => d.getModuleSpecifierValue() === module);
+    }
+    async commit() {
+        await this.project.save();
+    }
+    addNamedImport(name, module) {
+        const existing = this.getImport(module);
+        if (existing) {
+            const alreadyExists = existing
+                .getNamedImports()
+                .some((i) => i.getName() === name);
+            if (!alreadyExists) {
+                existing.addNamedImport(name);
+            }
+        }
+        else {
+            this.sourceFile.addImportDeclaration({
+                namedImports: [name],
+                moduleSpecifier: module
+            });
+        }
+    }
+    addDefaultImport(name, module) {
+        const existing = this.getImport(module);
+        if (existing) {
+            const currentDefault = existing.getDefaultImport();
+            if (!currentDefault) {
+                existing.setDefaultImport(name);
+            }
+            else if (currentDefault.getText() !== name) {
+                throw new Error(`Module "${module}" already has default import "${currentDefault.getText()}"`);
+            }
+        }
+        else {
+            this.sourceFile.addImportDeclaration({
+                defaultImport: name,
+                moduleSpecifier: module
+            });
+        }
+    }
+    removeNamedImport(name, module) {
+        const existing = this.getImport(module);
+        if (!existing)
+            return;
+        const named = existing
+            .getNamedImports()
+            .find((i) => i.getName() === name);
+        if (!named)
+            return;
+        named.remove();
+        if (existing.getNamedImports().length === 0 &&
+            !existing.getDefaultImport() &&
+            !existing.getNamespaceImport()) {
+            existing.remove();
+        }
+    }
+    removeImport(module) {
+        const existing = this.getImport(module);
+        if (!existing)
+            return;
+        existing.remove();
+    }
+}