import React from 'react'
import { Tree, TreeNodeEvent } from 'react-icecream'
import { RightIcon, DownIcon } from 'react-icecream/icons'

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

export default function Demo() {
  function renderSwitcherIcon({ isLeaf, expanded }: TreeNodeEvent) {
    if (isLeaf) {
      return null
    }
    return expanded ? <DownIcon /> : <RightIcon />
  }

  return (
    <Tree
      switcherIcon={renderSwitcherIcon}
      treeData={treeData}
      defaultExpandedKeys={['1', '2']}
    />
  )
}