workbench: support reloading nodes from workspace.json and register as callback to backend with ChangeNotifier. set callback to window.reloadNodes in browser backend for testing.

Author: progrium

lib/backend/browser.ts

@@ -1,19 +1,25 @@
 import {RawNode} from "../model/mod.ts";
-
+import { SearchIndex, FileStore, ChangeNotifier } from "./mod.ts";
 
 export class BrowserBackend {
   auth: null;
   index: SearchIndex;
   files: FileStore;
+  changes?: ChangeNotifier;
 
   constructor() {
     this.auth = null;
-    this.files = new FileStore();
+    this.files = new LocalStorageFileStore();
     if (window.MiniSearch) {
       this.index = new SearchIndex_MiniSearch();
     } else {
       this.index = new SearchIndex_Dumb();
     }
+    this.changes = {
+      registerNotifier(cb: (nodeIDs: string[]) => void) {
+        window.reloadNodes = cb;
+      }
+    };
   }
 }
 
@@ -84,8 +90,8 @@ export class SearchIndex_Dumb {
 
 
 
-export class FileStore {
-  async readFile(path: string): string|null {
+export class LocalStorageFileStore {
+  async readFile(path: string): Promise<string|null> {
     return localStorage.getItem(`treehouse:${path}`);
   }
 

lib/backend/github.ts

@@ -262,7 +262,7 @@ export class GitHubBackend {
   }
 
 
-  async readFile(path: string): string|null {
+  async readFile(path: string): Promise<string|null> {
     try {
       const resp = await this.client.rest.repos.getContent({
         owner: this.user?.userID(), 

lib/backend/mod.ts

@@ -44,5 +44,5 @@ export interface FileStore {
 }
 
 export interface ChangeNotifier {
-  onNodeChange(cb: (nodeIDs: string[]) => void);
+  registerNotifier(cb: (nodeIDs: string[]) => void);
 }

lib/workbench/workbench.ts

@@ -54,7 +54,7 @@ export class Workbench {
     this.menus = new MenuRegistry();
 
     this.backend = backend;
-    this.workspace = new Workspace(backend.files);
+    this.workspace = new Workspace(backend.files, backend.changes);
 
     this.context = {node: null};
     this.panels = [];

lib/workbench/workspace.ts

@@ -1,4 +1,4 @@
-import { FileStore } from "../backend/mod.ts";
+import { FileStore, ChangeNotifier } from "../backend/mod.ts";
 import { Bus, Node, RawNode } from "../model/mod.ts";
 import { Path } from "./mod.ts";
 import * as module from "../model/module/mod.ts";
@@ -20,12 +20,16 @@ export class Workspace {
   expanded: { [key: string]: { [key: string]: boolean } }; // [rootid][id]
   settings: {};
 
-  constructor(fs: FileStore) {
+  constructor(fs: FileStore, changes?: ChangeNotifier) {
     this.fs = fs;
     this.bus = new module.Bus();
     this.expanded = {};
     this.settings = {};
 
+    if (changes) {
+      changes.registerNotifier(this.reload.bind(this));
+    }
+
     this.writeDebounce = debounce(async (path, contents) => {
       try {
         await this.fs.writeFile(path, contents);
@@ -60,16 +64,28 @@ export class Workspace {
     }
   }
 
+  migrateRawNode(n: RawNode): RawNode {
+    if (n.Name === "treehouse.SearchNode") {
+      n.Name = "treehouse.SmartNode";
+    }
+    return n;
+  }
+
+  async reload(nodeIDs: string[]) {
+    let doc = JSON.parse(await this.fs.readFile("workspace.json") || "{}");
+    if (doc.nodes) {
+      doc.nodes = doc.nodes.filter(n => nodeIDs.includes(n.ID));
+      doc.nodes = doc.nodes.map(this.migrateRawNode);
+      this.bus.import(doc.nodes);
+      m.redraw();
+      console.log(`Reloaded ${doc.nodes.length} nodes.`);
+    }
+  }
+
   async load() {
     let doc = JSON.parse(await this.fs.readFile("workspace.json") || "{}");
     if (doc.nodes) {
-      doc.nodes = doc.nodes.map(n => {
-        // any node migrations:
-        if (n.Name === "treehouse.SearchNode") {
-          n.Name = "treehouse.SmartNode";
-        }
-        return n;
-      })
+      doc.nodes = doc.nodes.map(this.migrateRawNode);
       this.bus.import(doc.nodes);
       console.log(`Loaded ${doc.nodes.length} nodes.`);
     }