Add symbols to identify entity types

Author: knolleary

lib/NRConfigNode.js

@@ -1,4 +1,5 @@
 const NRObject = require("./NRObject.js");
+const Symbols = require("./Symbols");
 
 /**
  * NRConfigNode - Description
@@ -15,6 +16,7 @@ class NRConfigNode extends NRObject {
      */
     constructor(config) {
         super(config);
+        this.TYPE = Symbols.ConfigNode;
         this.users = new Set();
     }
 

lib/NRFlow.js

@@ -1,5 +1,5 @@
 const NRContainer = require("./NRContainer.js");
-
+const Symbols = require("./Symbols");
 
 /**
  * NRFlow - Description
@@ -17,6 +17,7 @@ class NRFlow extends NRContainer {
      */
     constructor(config) {
         super(config);
+        this.TYPE = Symbols.Flow;
         this.disabled = !!config.disabled;
         this.info = config.info;
 

lib/NRGroup.js

@@ -1,4 +1,5 @@
 const NRContainer = require("./NRContainer.js");
+const Symbols = require("./Symbols");
 
 /**
  * NRGroup - Description
@@ -14,6 +15,7 @@ class NRGroup extends NRContainer {
      */
     constructor(config) {
         super(config);
+        this.TYPE = Symbols.Group;
         this.w = config.w;
         this.h = config.h;
         this.groupId = config.g;
@@ -37,7 +39,7 @@ class NRGroup extends NRContainer {
         this.nodes.set(node.id, node);
         node.setGroup(this);
     }
-    
+
     export() {
         let obj = super.export();
         if (this.groupId) {

lib/NRNode.js

@@ -1,4 +1,5 @@
 const NRObject = require("./NRObject.js");
+const Symbols = require("./Symbols");
 
 /**
  * NRNode - Description
@@ -22,6 +23,7 @@ class NRNode extends NRObject {
      */
     constructor(config) {
         super(config);
+        this.TYPE = Symbols.Node;
         this.x = config.x;
         this.y = config.y;
         this.groupId = config.g;

lib/NRSubflow.js

@@ -1,4 +1,5 @@
 const NRFlow = require("./NRFlow.js");
+const Symbols = require("./Symbols");
 
 /**
  * NRSubflow - Description
@@ -14,6 +15,8 @@ class NRSubflow extends NRFlow {
      */
     constructor(config) {
         super(config);
+        this.TYPE = Symbols.Subflow;
+
         this.instances = new Map();
         this._ownProperties = ["category", "in", "out", "env", "meta", "color", "inputLabels", "outputLabels", "icon"];
         this._ownProperties.forEach(prop => {

lib/NRWire.js

@@ -1,3 +1,5 @@
+const Symbols = require("./Symbols");
+
 /**
  * NRWire - Description
  * @property {NRNode} sourceNode - source of the wire

lib/Symbols.js

@@ -0,0 +1,7 @@
+module.exports = {
+    Node: Symbol("NRNode"),
+    ConfigNode: Symbol("NRConfigNode"),
+    Group: Symbol("NRGroup"),
+    Flow: Symbol("NRFlow"),
+    Subflow: Symbol("NRSubflow")
+}

lib/index.js

@@ -12,5 +12,6 @@ function parseFlow(flowConfig) {
 }
 
 module.exports = {
-    parseFlow: parseFlow
+    parseFlow: parseFlow,
+    types: require("./Symbols")
 }

test/test.js

@@ -1,3 +1,4 @@
+const fs = require("fs");
 const assert = require("assert");
 const fp = require("../lib/index.js");
 
@@ -10,13 +11,23 @@ describe("Flow Parser tests", function() {
     })
 
     function runTests(filename) {
-        const fs = require("fs");
-
-
         const original = JSON.parse(fs.readFileSync(filename,"utf-8"));
         const flowSet = fp.parseFlow(original);
         const generated = flowSet.export();
 
+        // flowSet.walk(function(obj) {
+        //     if (obj.type === 'tab') {
+        //         console.log("FLOW",obj.type)
+        //     } else if (obj.type === 'subflow') {
+        //         console.log("SUBFLOW",obj.type)
+        //     } else if (obj.type === 'group') {
+        //         console.log("GROUP",obj.type)
+        //     } else if (obj.constructor.name === 'NRConfigNode') {
+        //         console.log("CONFIG NODE",obj.type)
+        //     } else {
+        //         console.log("NODE",obj.type)
+        //     }
+        // })
 
         const knownIds = original.map(n => n.id).sort();