edit variable definition

src/misc/b3type.ts

@@ -30,6 +30,12 @@ export interface VarDef {
   desc: string;
 }
 
+export interface ImportDef {
+  path: string;
+  date?: number;
+  vars: VarDef[];
+}
+
 export interface TreeModel {
   name: string;
   desc?: string;

src/misc/b3util.ts

@@ -1,6 +1,7 @@
 import * as fs from "fs";
-import { NodeDef } from "../behavior3/src/behavior3";
+import { type NodeDef } from "../behavior3/src/behavior3";
 import {
+  ImportDef,
   isBoolType,
   isEnumType,
   isExprType,
@@ -12,9 +13,10 @@ import {
   NodeModel,
   TreeGraphData,
   TreeModel,
+  VarDef,
 } from "./b3type";
 import Path from "./path";
-import { readJson } from "./util";
+import { readJson, readTree } from "./util";
 
 export class NodeDefs extends Map<string, NodeDef> {
   get(key: string): NodeDef {
@@ -603,8 +605,7 @@ export const createTreeData = (
     parsingStack.push(node.path);
     try {
       const subtreePath = workdir + "/" + node.path;
-      const str = fs.readFileSync(subtreePath, "utf8");
-      treeData = createTreeData(JSON.parse(str).root, treeData.id, calcSize);
+      treeData = createTreeData(readTree(subtreePath).root, treeData.id, calcSize);
       treeData.lastModified = fs.statSync(subtreePath).mtimeMs;
       treeData.path = node.path;
       treeData.debug = node.debug;
@@ -685,3 +686,36 @@ export const createNewTree = (path: string) => {
 export const isTreeFile = (path: string) => {
   return path.toLocaleLowerCase().endsWith(".json");
 };
+
+export const loadImportedVars = (imports: ImportDef[]) => {
+  for (const importDef of imports) {
+    // TODO: check file date
+    // TODO: use cache
+    if (importDef.vars.length) {
+      continue;
+    }
+    const filter: Map<string, boolean> = new Map();
+    const vars: VarDef[] = [];
+    const load = (path: string) => {
+      if (filter.has(path)) {
+        return;
+      }
+      filter.set(path, true);
+      const model: TreeModel = readTree(`${workdir}/${path}`);
+      model.declare?.vars?.forEach((v) => vars.push(v));
+      model.declare?.imports?.forEach((v) => load(v));
+      console.log("load imported vars:", path);
+    };
+    load(importDef.path);
+    filter.clear();
+    importDef.vars = vars
+      .filter((v) => {
+        if (!filter.has(v.name)) {
+          filter.set(v.name, true);
+          return true;
+        }
+        return false;
+      })
+      .sort((a, b) => a.name.localeCompare(b.name));
+  }
+};

src/misc/util.ts

@@ -1,10 +1,16 @@
 import * as fs from "fs";
+import type { TreeModel } from "./b3type";
 
 export const readJson = (path: string) => {
   const str = fs.readFileSync(path, "utf-8");
   return JSON.parse(str);
 };
 
+export const readTree = (path: string) => {
+  const str = fs.readFileSync(path, "utf-8");
+  return JSON.parse(str) as TreeModel;
+};
+
 export const writeJson = (path: string, data: any) => {
   const str = JSON.stringify(data, undefined, 2);
   fs.writeFileSync(path, str, "utf-8");