saving + message after completion

SilenNaihin · SilenNaihin · commit c451161a30c8 · 2023-11-28T17:53:51.000-05:00
diff --git a/src/addImportStatement/index.ts b/src/addImportStatement/index.ts
@@ -29,7 +29,9 @@ export default function addImportStatement(
   // Insert the import statement at the top of the document
   const edit = new vscode.WorkspaceEdit();
   edit.insert(document.uri, new vscode.Position(0, 0), importStatement);
-  vscode.workspace.applyEdit(edit);
+  vscode.workspace.applyEdit(edit).then(() => {
+    saveDocument(document);
+  });
 }
 
 function calculateRelativePath(
@@ -66,3 +68,16 @@ function calculateRelativePath(
 
   return relativePath;
 }
+
+function saveDocument(document: vscode.TextDocument) {
+  if (document.isDirty) {
+    document.save().then(
+      () => {
+        vscode.window.showInformationMessage('Stack created successfully 🎉');
+      },
+      (err) => {
+        vscode.window.showErrorMessage(`Error saving document: ${err}`);
+      }
+    );
+  }
+}
