import { observable } from "mobx";
import { Tree } from "banyan-immutable-tree/lib/immutable_tree";
import { Node } from "banyan-immutable-tree/lib/immutable_node";

export default class TreeStore {
    @observable public tree: Tree;

    constructor(tree: Tree) {
        this.tree = tree;
    }

    public select(node: Node) {
        this.tree = this.tree.selectNode(node.get("id"));
    }

    public toggle(node: Node) {
        this.tree = this.tree.toggleNode(node.get("id"));
    }

    public handleKey(key: string): any {
        const [is_handled, tree] = this.tree.handleKey(key);

        if (is_handled) {
            this.tree = tree;

            return true;
        }
    }
}