/**
 * desc: 自定义 `TreeNode` 图标，需要将 `showIcon` 设置为 `true`
 */
import React from 'react'
import { Tree, TreeNodeEvent } from 'react-icecream'
import { EyeThinIcon, EyeInvisibleThinIcon, FileIcon } from 'react-icecream/icons'

import { treeData } from './tree-data'

export default function Demo() {
  function renderIcon({ expanded, data }: TreeNodeEvent) {
    if (!data.children?.length) {
      return <FileIcon />
    }
    return expanded ? <EyeInvisibleThinIcon /> : <EyeThinIcon />
  }

  return (
    <Tree
      showIcon
      icon={renderIcon}
      treeData={treeData}
      defaultExpandedKeys={['1', '2']}
    />
  )
}