phorge-phorge/webroot/rsrc/js/application/diff/DiffTreeView.js

/**
 * @provides phabricator-diff-tree-view
 * @requires javelin-dom
 * @javelin
 */

JX.install('DiffTreeView', {

  construct: function() {
    this._keys = [];
    this._tree = this._newTreeNode(null, [], 0);
    this._nodes = {};
    this._paths = [];
  },

  members: {
    _node: null,
    _keys: null,
    _tree: null,
    _nodes: null,
    _dirty: false,
    _paths: null,
    _selectedPath: null,

    getNode: function() {
      if (!this._node) {
        this._node = JX.$N('ul');
      }

      if (this._dirty) {
        this.redraw();
      }

      return this._node;
    },

    addPath: function(path) {
      this._paths.push(path);

      var tree = this._getTree(this._tree, path.getPath(), 0);
      tree.pathObject = path;

      this._dirty = true;

      return this;
    },

    getPaths: function() {
      return this._paths;
    },

    setSelectedPath: function(path) {
      if (this._selectedPath) {
        this._selectedPath.setIsSelected(false);
        this._selectedPath = null;
      }

      if (path) {
        path.setIsSelected(true);
      }

      this._selectedPath = path;

      return this;
    },

    redraw: function() {
      if (!this._dirty) {
        return;
      }
      this._dirty = false;

      var ii;

      // For nodes which don't have a path object yet, build one.
      var tree;
      var trees = [];
      for (ii = 0; ii < this._keys.length; ii++) {
        var key = this._keys[ii];
        tree = this._nodes[key];
        var path = tree.pathObject;

        if (!path) {
          path = new JX.DiffPathView()
            .setPath(tree.parts);
          tree.pathObject = path;
        }

        trees.push(tree);
      }

      for (ii = 0; ii < trees.length; ii++) {
        tree = trees[ii];

        if (!tree.parent) {
          tree.depth = 0;
        } else {
          tree.depth = tree.parent.depth + 1;
        }

        tree.pathObject.setDepth((tree.depth - 1));
      }

      var nodes = [];
      for (ii = 0; ii < trees.length; ii++) {
        tree = trees[ii];
        nodes.push(tree.pathObject.getNode());
      }

      JX.DOM.setContent(this.getNode(), nodes);
    },

    _getTree: function(root, path, ii) {
      if (ii >= path.length) {
        return root;
      }

      var part = path[ii];

      if (!root.children.hasOwnProperty(part)) {
        root.children[part] = this._newTreeNode(root, path, ii);
        root.childCount++;
      }

      return this._getTree(root.children[part], path, ii + 1);
    },

    _newTreeNode: function(parent, path, ii) {
      var key;
      var parts;
      if (path.length) {
        parts = path.slice(0, ii + 1);
        key = parts.join('/');
        this._keys.push(key);
      } else {
        parts = [];
        key = null;
      }

      var node = {
        parent: parent,
        nodeKey: key,
        parts: parts,
        children: {},
        pathObject: null,
        childCount: 0,
        depth: 0
      };

      if (key !== null) {
        this._nodes[key] = node;
      }

      return node;
    }

  }

});