/* Extending/depending on:
~ = modified function (when updating from SVN be sure to check these for changes, especially to Ext.tree.TreeNodeUI.render() )
+ = added function

TreeSelectionModel.js
    Ext.tree.CheckNodeMultiSelectionModel : ~init(), ~onNodeClick(), +extendSelection(), ~onKeyDown()

TreeNodeUI.js
    Ext.tree.CheckboxNodeUI : ~render(), +checked(), +check(), +toggleCheck()

TreePanel.js
    Ext.tree.TreePanel : +getChecked()

TreeLoader.js
    Ext.tree.TreeLoader : ~createNode()

*/

/**
 * 原始功能只能實現(xiàn)選中或不選兩種狀態(tài)
 * 但我們現(xiàn)在需要非葉子節(jié)點可以實現(xiàn)三種狀態(tài)，既：
 *   節(jié)點的子節(jié)點都被選中時，顯示都被選中的圖標
 *   節(jié)點的子節(jié)點都沒選中時，顯示都沒選中的圖標
 *   節(jié)點的子節(jié)點部分選中時，顯示部分選中的圖標
 *
 *   顯示都被選中，和部分選中時，當前節(jié)點的狀態(tài)是被選中
 *   顯示都沒選中時，當前節(jié)點的狀態(tài)是沒選中
 *
 *   顯示都被選中時，單擊當前節(jié)點，觸發(fā)的事件是，當前節(jié)點變成未選中狀態(tài)，所有子節(jié)點變成未選中狀態(tài)
 *   顯示都沒選中時，單擊當前節(jié)點，觸發(fā)的事件是，當前節(jié)點變成都選中狀態(tài)，所有子節(jié)點變成選中狀態(tài)
 *   顯示部分選中時，單擊當前節(jié)點，觸發(fā)的事件是，當前節(jié)點變成都選中狀態(tài)，所有子節(jié)點變成選中狀態(tài)
 *
 *   子節(jié)點變成未選中狀態(tài)時，對應(yīng)父節(jié)點判斷所有子節(jié)點是否未選中，如果都未選中，變成未選中狀態(tài)，否則變成部分選中狀態(tài)
 *   子節(jié)點變成選中狀態(tài)時，對應(yīng)父節(jié)點判斷所有子節(jié)點是否都選中，如果都選中，變成都選中狀態(tài)，否則變成部分選中狀態(tài)
 *
 * @return {Array} 選中節(jié)點的id數(shù)組
 */
Ext.tree.TreePanel.prototype.getChecked = function(node){
    var checked = [], i;
    if( typeof node == 'undefined' ) {
        node = this.rootVisible ? this.getRootNode() : this.getRootNode().firstChild;
    }

    if( node.attributes.checked ) {
        checked.push(node.id);
        if( !node.isLeaf() ) {
            for( i = 0; i < node.childNodes.length; i++ ) {
                checked = checked.concat( this.getChecked(node.childNodes[i]) );
            }
        }
    }
    return checked;
};

/**
 * @class Ext.tree.CustomUITreeLoader
 * @extends Ext.tree.TreeLoader
 * 重寫createNode()，強制uiProvider是任意的TreeNodeUI來保存廣度
 */
Ext.tree.CustomUITreeLoader = function() {
    Ext.tree.CustomUITreeLoader.superclass.constructor.apply(this, arguments);
};

Ext.extend(Ext.tree.CustomUITreeLoader, Ext.tree.TreeLoader, {
    createNode : function(attr){
        Ext.apply(attr, this.baseAttr || {});

        if(this.applyLoader !== false){
            attr.loader = this;
        }

        // 如果uiProvider是字符串，那么要么從uiProviders數(shù)組里取一個對應(yīng)的，要么解析字符串獲得uiProvider
        if(typeof attr.uiProvider == 'string'){
            attr.uiProvider = this.uiProviders[attr.uiProvider] || eval(attr.uiProvider);
        }

        // 返回的時候，如果是葉子，就返回普通TreeNode，如果不是葉子，就返回異步讀取樹
        return(attr.leaf ? new Ext.tree.TreeNode(attr) : new Ext.tree.AsyncTreeNode(attr));
    }
});


/**
 * @class Ext.tree.CheckboxNodeUI
 * @extends Ext.tree.TreeNodeUI
 * 給所有節(jié)點添加checkbox
 */
Ext.tree.CheckboxNodeUI = function() {
    Ext.tree.CheckboxNodeUI.superclass.constructor.apply(this, arguments);
};

Ext.extend(Ext.tree.CheckboxNodeUI, Ext.tree.TreeNodeUI, {
    /**
     * 重寫render()
     */
    render : function(bulkRender) {

        var n = this.node;
        /* 在未來的svn里，這個變成了n.ownerTree.innerCt.dom */
        var targetNode = n.parentNode ? n.parentNode.ui.getContainer() : n.ownerTree.container.dom;
        if (!this.rendered) {
            this.rendered = true;
            var a = n.attributes;

            // 為縮進添加緩存，在顯示非常大的樹的時候有幫助
            this.indentMarkup = "";
            if (n.parentNode) {
                // 根據(jù)父節(jié)點，計算子節(jié)點縮進
                this.indentMarkup = n.parentNode.ui.getChildIndent();
            }

            // modification，添加checkbox
            var buf = ['<li class="x-tree-node"><div class="x-tree-node-el ', n.attributes.cls,'">',
                '<span class="x-tree-node-indent">',this.indentMarkup,"</span>",
                '<img src="', this.emptyIcon, '" class="x-tree-ec-icon">',
                '<img src="', a.icon || this.emptyIcon, '" class="x-tree-node-icon',(a.icon ? " x-tree-node-inline-icon" : ""),(a.iconCls ? " "+a.iconCls : ""),'" unselectable="on">',
/*
                '<input class="l-tcb" type="checkbox" ', (a.checked ? "checked>" : '>'),
*/
                '<img src="',this.emptyIcon,'" class="',this.isAllChildrenChecked(),'">',

                '<a hidefocus="on" href="',a.href ? a.href : "#",'" ',
                 a.hrefTarget ? ' target="'+a.hrefTarget+'"' : "", '>',
                 '<span unselectable="on">',n.text,"</span></a></div>",
                '<ul class="x-tree-node-ct" style="display:none;"></ul>',
                "</li>"];

            if (bulkRender !== true && n.nextSibling && n.nextSibling.ui.getEl()) {
                this.wrap = Ext.DomHelper.insertHtml("beforeBegin", n.nextSibling.ui.getEl(), buf.join(""));
            } else {
                this.wrap = Ext.DomHelper.insertHtml("beforeEnd", targetNode, buf.join(""));
            }
            this.elNode = this.wrap.childNodes[0];
            this.ctNode = this.wrap.childNodes[1];
            var cs = this.elNode.childNodes;
            this.indentNode = cs[0];
            this.ecNode = cs[1];
            this.iconNode = cs[2];
            this.checkbox = cs[3]; // modification，添加checkbox
            this.checkboxImg = cs[3]; // 修改，添加仿造checkbox的圖片
            this.anchor = cs[4];
            this.textNode = cs[4].firstChild;
            if (a.qtip) {
                if (this.textNode.setAttributeNS) {
                    this.textNode.setAttributeNS("ext", "qtip", a.qtip);
                    if (a.qtipTitle) {
                        this.textNode.setAttributeNS("ext", "qtitle", a.qtipTitle);
                    }
                } else {
                    this.textNode.setAttribute("ext:qtip", a.qtip);
                    if (a.qtipTitle) {
                        this.textNode.setAttribute("ext:qtitle", a.qtipTitle);
                    }
                }
            } else if(a.qtipCfg) {
                a.qtipCfg.target = Ext.id(this.textNode);
                Ext.QuickTips.register(a.qtipCfg);
            }

            this.initEvents();

            // modification: 添加handlers，避免修改Ext.tree.TreeNodeUI
            Ext.fly(this.checkbox).on('click', this.check.createDelegate(this, [null]));
            n.on('dblclick', function(e) {
                if (this.isLeaf()) {
                    this.getUI().toggleCheck();
                }
            });

            if (!this.node.expanded) {
                this.updateExpandIcon();
            }
        } else {
            if (bulkRender === true) {
                targetNode.appendChild(this.wrap);
            }
        }
    }

    // 這個節(jié)點是否被選中了
    , checked : function() {
        // return this.checkbox.checked;
        return this.checkboxImg.className != "x-tree-node-checkbox-none";
    },

    // flag可能是：（NULL）以當前節(jié)點狀態(tài)為準判斷，ALL -> NONE -> SOME -> ALL
    // 否則按照設(shè)置的flag為準：SOME,ALL,NONE
    check : function(forParent, forChildren) {
        var flag = null;

        if (this.node.isLeaf) {
            flag = (this.checkboxImg.className == "x-tree-node-checkbox-all") ? "x-tree-node-checkbox-none" : "x-tree-node-checkbox-all";
        } else {
            if (this.checkboxImg.className == "x-tree-node-checkbox-all") {
                // 全部反選
                flag = "x-tree-node-checkbox-none";
            } else if (this.checkboxImg.className == "x-tree-node-checkbox-none") {
                // 全部選中
                flag = "x-tree-node-checkbox-some";
            } else {
                // 全部選中
                flag = "x-tree-node-checkbox-all";
            }
        }

        if (typeof forParent == "undefined" || forParent == null) {
            forParent = typeof this.node.parentNode != "undefined" && this.node.parentNode != null;
        }
        if (typeof forChildren == "undefined" || forChildren == null) {
            forChildren = !this.node.ifLeaf;
        }
        console.error(this);
        console.error(flag + "," + forParent + "," + forChildren);

        var n = this.node;
        var tree = n.getOwnerTree();
        var parentNode = n.parentNode;
        // 如果下級節(jié)點都尚未渲染過，就展開當前節(jié)點，并渲染下面的所有節(jié)點
        if (!n.isLeaf && !n.expanded && !n.childrenRendered) {
            n.expand(false, false, this.check.createDelegate(this, [forParent, forChildren]));
            return;
        }

        // 如果包含子節(jié)點
        if (forChildren && !n.isLeaf) {
            var cs = n.childNodes;
            for(var i = 0; i < cs.length; i++) {
                cs[i].getUI().checkChild(flag == "x-tree-node-checkbox-all");
            }
        }

        this.checkboxImg.className = "x-tree-node-checkbox-" + this.isAllChildrenChecked();
        if (this.checkboxImg.className == "x-tree-node-checkbox-none") {
            this.node.attributes.checked = false;
        } else {
            this.node.attributes.checked = true;
        }

        if (parentNode.getUI().checkParent) {
            parentNode.getUI().checkParent();