owl-bt/src/client/app/tree/treeModel/treeNodeByIdStore.service.js

owl-bt is editor for Behavior trees. It has been inspired by Unreal engine behavior trees in a way, that it supports special node items like decorators and services. This makes trees smaller and much more readable.

'use strict';

(function() {

  class TreeNodeByIdStore {

    addNode(node) {
      if (node) {
        if (!this.nodesById) {
          this.nodesById = {};
        }
        let existingNode = this.nodesById[node.$meta.id];
        if (existingNode &&  existingNode !== node) {
          throw new Error('Duplicate node id');
        }
        this.nodesById[node.$meta.id] = node;
      }
    }

    removeNode(node){
      if(node && this.nodesById){
        delete this.nodesById[node.$meta.id];
      }
    }

    getNode(nodeId) {
      if (this.nodesById) {
        return this.nodesById[nodeId];
      }
    }

  }

  angular.module('editorApp')
    .service('TreeNodeByIdStore', TreeNodeByIdStore);
})();