0 上一期回顾

上一期直播我们完善了tree组件的样式，并增加了点选高亮、禁止展开/收起、禁止选择等功能。

当时还识别到了点选高亮功能use-highlight的一个小小的瑕疵点，这个瑕疵很快由pineaple同学进行了优化。

1 自定义图标

1.1 定义icon插槽

编写之前先思考下：

我们的插槽应该写在哪里？

tree.tsx中的renderIcon方法

const renderIcon = (item: TreeItem) => {
  return item.children
    ? <span class={item.disableToggle && 'toggle-disabled'} onClick={() => toggle(item)}>
        {
          // 增加插槽逻辑
          ctx.slots.icon 
            ? ctx.slots.icon(item)
            : item.open
              ? <IconOpen class='mr-xs' />
              : <IconClose class='mr-xs' />
        }
      </span>
    : <Indent />
}
复制代码

IconOpen.tsx

const IconOpen = () => {
  return (
    <svg
      class="icon"
      viewBox="0 0 1024 1024"
      version="1.1"
      xmlns="http://www.w3.org/2000/svg"
      p-id="4160"
      width="12"
      height="12"
    >
      <path d="M64 320l448 448 448-448z" fill="#8C92A4" p-id="4161"></path>
    </svg>
  )
}

export default IconOpen
复制代码

IconClose.tsx

const IconClose = () => {
  return (
    <svg
      class="icon"
      viewBox="0 0 1024 1024"
      version="1.1"
      xmlns="http://www.w3.org/2000/svg"
      p-id="4361"
      width="12"
      height="12"
    >
      <path d="M256 64l448 448-448 448z" fill="#8C92A4" p-id="4362"></path>
    </svg>
  )
}

export default IconClose
复制代码

1.2 使用插槽

JSX和SFC中插槽使用的写法对比。

JSX：

<d-tree data={data}>
  {{
    icon: (item: TreeItem) => (item.open ? <IconOpen /> : <IconClose />),
  }}
</d-tree>
复制代码

SFC：

<d-tree :data="data">
  <template #icon="item">
    <IconOpen v-if="item.open" />
    <IconClose v-else />
  </template>
</d-tree>
复制代码

效果如下：

2 节点勾选

2.1 新增checkable和checked

tree-types.ts

export const treeProps = {
  data: {
    type: Array as PropType<TreeData>,
    default: () => [],
  },
  
  // 新增
  checkable: {
    type: Boolean,
    default: false
  },
} as const

export interface TreeItem {
  label: string
  children?: TreeData
  open?: boolean
  disableToggle?: boolean,
  checked?: boolean, // 新增
  [key: string]: any
}
复制代码

2.2 添加Checkbox组件

tree.tsx

const { data, checkable } = toRefs(props)
复制代码

tree.tsx中的renderNode方法

{ renderIcon(item) }

// 新增
{ checkable.value && <d-checkbox v-model={item.checked} />}

<span class={['devui-tree-node__title', disabled && 'select-disabled']}>
  { label }
</span>
复制代码

2.3 配置checkable

// 增加`checkable={true}`参数
<d-tree data={data} checkable>
  {{
    icon: (node: TreeItem) => (node.open ? <IconOpen /> : <IconClose />),
  }}
</d-tree>
复制代码