add import/export

main.ts

@@ -16,6 +16,8 @@ import GPT3Tokenizer from "gpt3-tokenizer";
 import { Configuration, OpenAIApi } from "openai";
 import { v4 as uuidv4 } from "uuid";
 import * as _ from "lodash";
+import * as fs from "fs";
+const untildify = require("untildify") as any;
 
 const tokenizer = new GPT3Tokenizer({ type: "codex" });
 
@@ -29,16 +31,22 @@ interface LoomSettings {
 
   showSettings: boolean;
   cloneParentOnEdit: boolean;
+  showImport: boolean;
+  showExport: boolean;
 }
 
 const DEFAULT_SETTINGS: LoomSettings = {
   apiKey: "",
+
   model: "code-davinci-002",
   maxTokens: 60,
-  n: 5,
   temperature: 1,
+  n: 5,
+
   showSettings: false,
   cloneParentOnEdit: false,
+  showImport: false,
+  showExport: false,
 };
 
 type Color = "red" | "orange" | "yellow" | "green" | "blue" | "purple" | null;
@@ -690,6 +698,36 @@ export default class LoomPlugin extends Plugin {
       )
     );
 
+    this.registerEvent(
+      // @ts-ignore
+      this.app.workspace.on("loom:import", (pathName: string) => this.wftsar(
+        (file) => {
+          if (!pathName) return;
+
+          const rawPathName = untildify(pathName);
+          const json = fs.readFileSync(rawPathName, "utf8");
+          const data = JSON.parse(json);
+          this.state[file.path] = data;
+          new Notice("Imported from " + rawPathName);
+        }
+      ))
+    );
+
+    this.registerEvent(
+      // @ts-ignore
+      this.app.workspace.on("loom:export", (pathName: string) => this.wftsar(
+        (file) => {
+          if (!pathName) return;
+
+          const data = this.state[file.path];
+          const json = JSON.stringify(data, null, 2);
+          const rawPathName = untildify(pathName);
+          fs.writeFileSync(rawPathName, json);
+          new Notice("Exported to " + rawPathName);
+        }
+      ))
+    );
+
     this.registerEvent(
       this.app.workspace.on("file-open", (file) => {
         if (!file) return;
@@ -1051,7 +1089,7 @@ class LoomView extends ItemView {
       cls: "nav-buttons-container loom-buttons",
     });
 
-    const navButton = (
+    const settingNavButton = (
       setting: string,
       value: boolean,
       icon: string,
@@ -1067,23 +1105,65 @@ class LoomView extends ItemView {
       );
     };
 
-    navButton(
+    settingNavButton(
       "showSettings",
       settings.showSettings,
       "settings",
       "Show settings"
     );
-    navButton(
+    settingNavButton(
       "cloneParentOnEdit",
       settings.cloneParentOnEdit,
       "copy",
       "Don't allow nodes with children to be edited; clone them instead"
     );
+    settingNavButton(
+      "showImport",
+      settings.showImport,
+      "import",
+      "Import JSON"
+    );
+    settingNavButton(
+      "showExport",
+      settings.showExport,
+      "download",
+      "Export to JSON"
+    );
 
     // create the main container, which uses the `outline` class, which has
     // a margin visually consistent with other panes
     const container = this.containerEl.createDiv({ cls: "outline" });
 
+    // import/export
+
+    const importDiv = container.createDiv({
+      cls: `loom-zport${settings.showImport ? "" : " hidden"}`,
+    });
+
+    const importInput = importDiv.createEl("input", {
+      attr: { type: "text", placeholder: "Path of file to import" },
+    });
+    const importButton = importDiv.createEl("button", {});
+    setIcon(importButton, "import");
+    importButton.addEventListener("click", () =>
+      this.app.workspace.trigger("loom:import", importInput.value)
+    );
+
+    const exportDiv = container.createDiv({
+      cls: `loom-zport${settings.showExport ? "" : " hidden"}`,
+    });
+
+    const exportInput = exportDiv.createEl("input", {
+      attr: { type: "text", placeholder: "Path to export to" },
+    });
+    const exportButton = exportDiv.createEl("button", {});
+    setIcon(exportButton, "download");
+    exportButton.addEventListener("click", () =>
+      this.app.workspace.trigger("loom:export", exportInput.value)
+    );
+
+    container.createDiv({ cls: `loom-vspacer${settings.showImport || settings.showExport ? "" : " hidden"}` });
+
     // settings
 
     const settingsDiv = container.createDiv({

package-lock.json

@@ -13,6 +13,7 @@
 				"gpt3-tokenizer": "^1.1.5",
 				"lodash": "^4.17.21",
 				"openai": "^3.1.0",
+				"untildify": "^4.0.0",
 				"uuid": "^9.0.0"
 			},
 			"devDependencies": {
@@ -1842,6 +1843,14 @@
 				"node": ">=4.2.0"
 			}
 		},
+		"node_modules/untildify": {
+			"version": "4.0.0",
+			"resolved": "https://registry.npmjs.org/untildify/-/untildify-4.0.0.tgz",
+			"integrity": "sha512-KK8xQ1mkzZeg9inewmFVDNkg3l5LUhoq9kN6iWYB/CC9YMG8HA+c1Q8HwDe6dEX7kErrEVNVBO3fWsVq5iDgtw==",
+			"engines": {
+				"node": ">=8"
+			}
+		},
 		"node_modules/uri-js": {
 			"version": "4.4.1",
 			"resolved": "https://registry.npmjs.org/uri-js/-/uri-js-4.4.1.tgz",

package.json

@@ -28,6 +28,7 @@
 		"gpt3-tokenizer": "^1.1.5",
 		"lodash": "^4.17.21",
 		"openai": "^3.1.0",
+		"untildify": "^4.0.0",
 		"uuid": "^9.0.0"
 	}
 }

styles.css

@@ -6,6 +6,26 @@
   margin-top: 0.5em;
 }
 
+.loom-zport {
+  display: flex;
+  flex-direction: row;
+  align-items: center;
+  width: 100%;
+  margin-bottom: 0.5em;
+}
+
+.loom-zport input {
+  flex-grow: 1;
+}
+
+.loom-zport button {
+  margin-left: 0.5em;
+}
+
+.loom-vspacer {
+  padding-top: 0.5em;
+}
+
 .loom-settings {
   margin-bottom: 0.3em;
 }