owl-bt/src/client/app/tree/treeModel/treeNodeDtoConverter.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 () {

  /**
   * Converts node (tree) to dto
   */
  class TreeNodeDtoConverter {

    constructor(_, TreeItemPropertyDtoConverter) {
      this._ = _;
      this._TreeItemPropertyDtoConverter = TreeItemPropertyDtoConverter;
    }

    convert(node) {
      let dto = {};
      angular.extend(dto, node);

      dto = this._removeInternalProperties(dto);

      if (dto.properties) {
        dto.properties = this._TreeItemPropertyDtoConverter.convertToDto(dto.properties);
        if (_.isEmpty(dto.properties)) {
          delete dto.properties;
        }
      }
      if (dto.decorators) {
        dto.decorators = dto.decorators.map(d => this.convertSubItem(d));
      }
      if (dto.services) {
        dto.services = dto.services.map(s => this.convertSubItem(s));
      }
      if (dto.childNodes) {
        dto.childNodes = dto.childNodes.map(n => this.convert(n));
      }

      this._clearEmptyArrays(dto);

      return dto;
    }

    convertSubItem(item) {
      let dto = {};
      angular.extend(dto, item);
      this._clearEmptyArrays(dto);
      dto = this._removeInternalProperties(dto);
      if (dto.properties) {
        dto.properties = this._TreeItemPropertyDtoConverter.convertToDto(dto.properties);
        if (_.isEmpty(dto.properties)) {
          delete dto.properties;
        }
      }
      return dto;
    }

    _clearEmptyArrays(obj) {
      for (let prop in obj) {
        if (obj.hasOwnProperty(prop)) {
          let value = obj[prop];
          if (value && _.isArray(value) && !value.length) {
            delete obj[prop];
          }
        }
      }
    }

    _removeInternalProperties(obj) {
      return this._.pickBy(obj, (v, key) => key[0] !== '$');
    }

  }

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