/** * Copyright (c) 2015, Facebook, Inc. All rights reserved. * * @providesModule NavigationTreeNode * @flow * @typechecks */ 'use strict'; var invariant = require('invariant'); var immutable = require('immutable'); var {List} = immutable; /** * Utility to build a tree of nodes. * Note that this tree does not perform cyclic redundancy check * while appending child node. */ class NavigationTreeNode { __parent: ?NavigationTreeNode; _children: List; _value: any; constructor(value: any) { this.__parent = null; this._children = new List(); this._value = value; } getValue(): any { return this._value; } getParent(): ?NavigationTreeNode { return this.__parent; } getChildrenCount(): number { return this._children.size; } getChildAt(index: number): ?NavigationTreeNode { return index > -1 && index < this._children.size ? this._children.get(index) : null; } appendChild(child: NavigationTreeNode): void { if (child.__parent) { child.__parent.removeChild(child); } child.__parent = this; this._children = this._children.push(child); } removeChild(child: NavigationTreeNode): void { var index = this._children.indexOf(child); invariant( index > -1, 'The node to be removed is not a child of this node.' ); child.__parent = null; this._children = this._children.splice(index, 1); } indexOf(child: NavigationTreeNode): number { return this._children.indexOf(child); } forEach(callback: Function, context: any): void { this._children.forEach(callback, context); } map(callback: Function, context: any): Array { return this._children.map(callback, context).toJS(); } some(callback: Function, context: any): boolean { return this._children.some(callback, context); } } module.exports = NavigationTreeNode;