jquery easyui tree的搜索树节点方法

/**
* 1）扩展jquery easyui tree的节点检索方法。使用方法如下：
* $("#treeId").tree("search", searchText);
* 其中，treeId为easyui tree的根UL元素的ID，searchText为检索的文本。
* 如果searchText为空或""，将恢复展示所有节点为正常状态
*/
(function($) {

$.extend($.fn.tree.methods, {
/**
* 扩展easyui tree的搜索方法
* @param tree easyui tree的根DOM节点(UL节点)的jQuery对象
* @param searchText 检索的文本
* @param this-context easyui tree的tree对象
*/
search: function(jqTree, searchText) {
//easyui tree的tree对象。可以通过tree.methodName(jqTree)方式调用easyui tree的方法
var tree = this;

//获取所有的树节点
var nodeList = getAllNodes(jqTree, tree);

//如果没有搜索条件，则展示所有树节点
searchText = $.trim(searchText);
if (searchText == "") {
for (var i=0; i<nodeList.length; i++) {
$(".tree-node-targeted", nodeList[i].target).removeClass("tree-node-targeted");
$(nodeList[i].target).show();
}
//展开已选择的节点（如果之前选择了）
var selectedNode = tree.getSelected(jqTree);
if (selectedNode) {
tree.expandTo(jqTree, selectedNode.target);
}
return;
}

//搜索匹配的节点并高亮显示
var matchedNodeList = [];
if (nodeList && nodeList.length>0) {
var node = null;
for (var i=0; i<nodeList.length; i++) {
node = nodeList[i];
if (isMatch(searchText, node.text)) {
matchedNodeList.push(node);
}
}

//隐藏所有节点
for (var i=0; i<nodeList.length; i++) {
$(".tree-node-targeted", nodeList[i].target).removeClass("tree-node-targeted");
$(nodeList[i].target).hide();
}

//折叠所有节点
tree.collapseAll(jqTree);

//展示所有匹配的节点以及父节点
for (var i=0; i<matchedNodeList.length; i++) {
showMatchedNode(jqTree, tree, matchedNodeList[i]);
}
}
},

/**
* 展示节点的子节点（子节点有可能在搜索的过程中被隐藏了）
* @param node easyui tree节点
*/
showChildren: function(jqTree, node) {
//easyui tree的tree对象。可以通过tree.methodName(jqTree)方式调用easyui tree的方法
var tree = this;

//展示子节点
if (!tree.isLeaf(jqTree, node.target)) {
var children = tree.getChildren(jqTree, node.target);
if (children && children.length>0) {
for (var i=0; i<children.length; i++) {
if ($(children[i].target).is(":hidden")) {
$(children[i].target).show();
}
}
}
}
},

/**
* 将滚动条滚动到指定的节点位置，使该节点可见（如果有滚动条才滚动，没有滚动条就不滚动）
* @param param {
*    treeContainer: easyui tree的容器（即存在滚动条的树容器）。如果为null，则取easyui tree的根UL节点的父节点。
*    targetNode:  将要滚动到的easyui tree节点。如果targetNode为空，则默认滚动到当前已选中的节点，如果没有选中的节点，则不滚动
* }
*/
scrollTo: function(jqTree, param) {
//easyui tree的tree对象。可以通过tree.methodName(jqTree)方式调用easyui tree的方法
var tree = this;

//如果node为空，则获取当前选中的node
var targetNode = param && param.targetNode ? param.targetNode : tree.getSelected(jqTree);

if (targetNode != null) {
//判断节点是否在可视区域
var root = tree.getRoot(jqTree);
var $targetNode = $(targetNode.target);
var container = param && param.treeContainer ? param.treeContainer : jqTree.parent();
var containerH = container.height();
var nodeOffsetHeight = $targetNode.offset().top - container.offset().top;
if (nodeOffsetHeight > (containerH - 30)) {
var scrollHeight = container.scrollTop() + nodeOffsetHeight - containerH + 30;
container.scrollTop(scrollHeight);
}
}
}
});

/**
* 展示搜索匹配的节点
*/
function showMatchedNode(jqTree, tree, node) {
//展示所有父节点
$(node.target).show();
$(".tree-title", node.target).addClass("tree-node-targeted");
var pNode = node;
while ((pNode = tree.getParent(jqTree, pNode.target))) {
$(pNode.target).show();
}
//展开到该节点
tree.expandTo(jqTree, node.target);
//如果是非叶子节点，需折叠该节点的所有子节点
if (!tree.isLeaf(jqTree, node.target)) {
tree.collapse(jqTree, node.target);
}
}

/**
* 判断searchText是否与targetText匹配
* @param searchText 检索的文本
* @param targetText 目标文本
* @return true-检索的文本与目标文本匹配；否则为false.
*/
function isMatch(searchText, targetText) {
return $.trim(targetText)!="" && targetText.indexOf(searchText)!=-1;
}

/**
* 获取easyui tree的所有node节点
*/
function getAllNodes(jqTree, tree) {
var allNodeList = jqTree.data("allNodeList");
if (!allNodeList) {
var roots = tree.getRoots(jqTree);
allNodeList = getChildNodeList(jqTree, tree, roots);
jqTree.data("allNodeList", allNodeList);
}
return allNodeList;
}

/**
* 定义获取easyui tree的子节点的递归算法
*/
function getChildNodeList(jqTree, tree, nodes) {
var childNodeList = [];
if (nodes && nodes.length>0) {
var node = null;
for (var i=0; i<nodes.length; i++) {
node = nodes[i];
childNodeList.push(node);
if (!tree.isLeaf(jqTree, node.target)) {
var children = tree.getChildren(jqTree, node.target);
childNodeList = childNodeList.concat(getChildNodeList(jqTree, tree, children));
}
}
}
return childNodeList;
}
})(jQuery);