51 lines
1.4 KiB
TypeScript
51 lines
1.4 KiB
TypeScript
import { ref } from "vue"
|
|
import type { Ref } from "vue"
|
|
import { useTreeDataStore } from "@/store"
|
|
import { storeToRefs } from "pinia"
|
|
/**
|
|
* 辅助搜索：递归函数
|
|
*/
|
|
const loop = (itemdata: any[], searchKey: Ref<string>) => {
|
|
const result: any[] = []
|
|
itemdata.forEach((item) => {
|
|
if (item.title.indexOf(searchKey.value.replace(" ", "")) > -1) {
|
|
result.push({ ...item })
|
|
} else if (item.children) {
|
|
const filterdata = loop(item.children, searchKey)
|
|
if (filterdata.length) {
|
|
result.push({
|
|
...item,
|
|
children: filterdata
|
|
})
|
|
}
|
|
}
|
|
})
|
|
return result
|
|
}
|
|
|
|
/**
|
|
* 搜索v-model/搜索按钮点击
|
|
*/
|
|
export default function useSearchNodes() {
|
|
// global
|
|
const treeDataStore = useTreeDataStore()
|
|
const { treeData } = storeToRefs(treeDataStore)
|
|
// refs
|
|
const searchKey = ref("")
|
|
// 点击搜索事件
|
|
const handleSearchTreeDataClick = () => {
|
|
// 返回过滤后的treeData
|
|
// treeDataStore.originTreeData
|
|
if (searchKey.value) {
|
|
treeData.value = treeDataStore.originTreeData
|
|
;(treeData as Ref<any[]>).value = loop(treeDataStore.treeData, searchKey)
|
|
} else {
|
|
treeData.value = treeDataStore.originTreeData
|
|
}
|
|
}
|
|
return {
|
|
searchKey,
|
|
handleSearchTreeDataClick
|
|
}
|
|
}
|