diff --git a/packages/devui-vue/devui/tree/src/components/tree-node.tsx b/packages/devui-vue/devui/tree/src/components/tree-node.tsx index fa1df6dde5..f5e70e5d0f 100644 --- a/packages/devui-vue/devui/tree/src/components/tree-node.tsx +++ b/packages/devui-vue/devui/tree/src/components/tree-node.tsx @@ -1,8 +1,9 @@ import { defineComponent, inject, PropType, toRefs } from 'vue'; -import { USE_TREE_TOKEN } from '../const'; +import { NODE_HEIGHT, USE_TREE_TOKEN } from '../const'; import { IInnerTreeNode } from '../core/use-tree-types'; import DTreeNodeToggle from './tree-node-toggle'; import { Checkbox } from '../../../checkbox'; +import useTreeNode from './use-tree-node'; export default defineComponent({ name: 'DTreeNode', @@ -15,22 +16,32 @@ export default defineComponent({ const { data } = toRefs(props); const { selectNode, toggleCheckNode } = inject(USE_TREE_TOKEN); + const { + nodeClass, + nodeStyle, + nodeContentClass, + nodeTitleClass, + nodeVLineClass, + nodeVLineStyle, + nodeHLineClass, + } = useTreeNode(data); + return () => { return ( -