const columns = generateColumns(10)
const data = generateData(columns, 3)
const frozenData = generateData(columns, 1, 'frozen-row-')
const fixedColumns = columns.map((column, columnIndex) => {
  let frozen
  if (columnIndex < 2) frozen = Column.FrozenDirection.LEFT
  if (columnIndex > 8) frozen = Column.FrozenDirection.RIGHT
  return { ...column, frozen }
})
const expandColumnKey = 'column-0'
for (let i = 0; i < 3; i++) {
  data.push({
    ...data[0],
    id: `${data[0].id}-sub-${i}`,
    parentId: data[0].id,
    [expandColumnKey]: `Sub ${i}`,
  })
  data.push({
    ...data[2],
    id: `${data[2].id}-sub-${i}`,
    parentId: data[2].id,
    [expandColumnKey]: `Sub ${i}`,
  })
  data.push({
    ...data[2],
    id: `${data[2].id}-sub-sub-${i}`,
    parentId: `${data[2].id}-sub-${i}`,
    [expandColumnKey]: `Sub-Sub ${i}`,
  })
}
const treeData = unflatten(data)
export default () => (
  <Table
    fixed
    columns={fixedColumns}
    data={treeData}
    frozenData={frozenData}
    maxHeight={600}
    expandColumnKey={expandColumnKey}
    onScrollbarPresenceChange={action('onScrollbarPresenceChange')}
  />
)