Merge pull request #19 from isfopo/issue/Create-command-to-add-key

package.json

@@ -35,6 +35,11 @@
         "command": "environments.flip",
         "title": "Flip",
         "icon": "$(sync)"
+      },
+      {
+        "command": "environments.add",
+        "title": "Add",
+        "icon": "$(add)"
       }
     ],
     "menus": {
@@ -45,6 +50,11 @@
         }
       ],
       "view/item/context": [
+        {
+          "command": "environments.add",
+          "when": "viewItem == file",
+          "group": "inline"
+        },
         {
           "command": "environments.edit",
           "when": "viewItem == keyValue-string",
@@ -117,4 +127,4 @@
     "webpack": "^5.92.1",
     "webpack-cli": "^5.1.4"
   }
-}
+}

src/EnvironmentTreeviewProvider.ts

@@ -5,6 +5,19 @@ import type { EnvironmentContent, EnvironmentKeyValue } from "./types";
 export class EnvironmentTreeviewProvider
   implements vscode.TreeDataProvider<vscode.TreeItem>
 {
+  async add(element: EnvironmentFileTreeItem, key: string, value: string) {
+    let content = new TextDecoder().decode(
+      await vscode.workspace.fs.readFile(element.uri)
+    );
+
+    content += `${key}="${value}"\n`;
+
+    vscode.workspace.fs.writeFile(
+      element.uri,
+      new TextEncoder().encode(content)
+    );
+  }
+
   flip(element: EnvironmentKeyValueTreeItem) {
     this.edit(element, element.value.value === "true" ? "false" : "true");
     this.refresh();

src/extension.ts

@@ -1,5 +1,6 @@
 import * as vscode from "vscode";
 import {
+  EnvironmentFileTreeItem,
   EnvironmentKeyValueTreeItem,
   EnvironmentTreeviewProvider,
 } from "./EnvironmentTreeviewProvider";
@@ -14,6 +15,34 @@ export function activate(context: vscode.ExtensionContext) {
     treeDataProvider,
   });
 
+  vscode.commands.registerCommand(
+    "environments.add",
+    async (element: EnvironmentFileTreeItem) => {
+      const key = await vscode.window.showInputBox({
+        prompt: "Enter the key for the new environment variable",
+      });
+
+      if (!key) {
+        return;
+      } else if (key.includes(" ")) {
+        vscode.window.showErrorMessage(
+          "Environment variable keys cannot contain spaces"
+        );
+        return;
+      }
+
+      const value = await vscode.window.showInputBox({
+        prompt: `Enter the value for ${key}`,
+      });
+
+      if (!value) {
+        return;
+      }
+
+      treeDataProvider.add(element, key, value);
+    }
+  );
+
   vscode.commands.registerCommand("environments.refresh", () =>
     treeDataProvider.refresh()
   );