import { useState, useRef } from 'react'
import { Button, Flex, Space, Modal, Dropdown, message } from 'antd'
import NewDocumentSize from './DocumentSizes/NewDocumentSize'
import ObjectTable from '../common/ObjectTable'
import PlusIcon from '../../Icons/PlusIcon'
import ReloadIcon from '../../Icons/ReloadIcon'
import useColumnVisibility from '../hooks/useColumnVisibility'
import GridIcon from '../../Icons/GridIcon'
import ListIcon from '../../Icons/ListIcon'
import useViewMode from '../hooks/useViewMode'
import ColumnViewButton from '../common/ColumnViewButton'

const DocumentSizes = () => {
  const [messageApi, contextHolder] = message.useMessage()
  const [newDocumentSizeOpen, setNewDocumentSizeOpen] = useState(false)
  const tableRef = useRef()
  const [viewMode, setViewMode] = useViewMode('documentSize')
  const [columnVisibility, setColumnVisibility] =
    useColumnVisibility('documentSize')

  const actionItems = {
    items: [
      {
        label: 'New Document Size',
        key: 'newDocumentSize',
        icon: <PlusIcon />
      },
      { type: 'divider' },
      {
        label: 'Reload List',
        key: 'reloadList',
        icon: <ReloadIcon />
      }
    ],
    onClick: ({ key }) => {
      if (key === 'reloadList') {
        tableRef.current?.reload()
      } else if (key === 'newDocumentSize') {
        setNewDocumentSizeOpen(true)
      }
    }
  }

  return (
    <>
      <Flex vertical={'true'} gap='large'>
        {contextHolder}
        <Flex justify={'space-between'}>
          <Space size='small'>
            <Dropdown menu={actionItems}>
              <Button>Actions</Button>
            </Dropdown>
            <ColumnViewButton
              type='documentSize'
              loading={false}
              collapseState={columnVisibility}
              updateCollapseState={setColumnVisibility}
            />
          </Space>
          <Space>
            <Button
              icon={viewMode === 'cards' ? <ListIcon /> : <GridIcon />}
              onClick={() =>
                setViewMode(viewMode === 'cards' ? 'list' : 'cards')
              }
            />
          </Space>
        </Flex>
        <ObjectTable
          ref={tableRef}
          visibleColumns={columnVisibility}
          type='documentSize'
          cards={viewMode === 'cards'}
        />
      </Flex>
      <Modal
        open={newDocumentSizeOpen}
        onCancel={() => setNewDocumentSizeOpen(false)}
        footer={null}
        destroyOnHidden={true}
        width={700}
      >
        <NewDocumentSize
          onOk={() => {
            setNewDocumentSizeOpen(false)
            messageApi.success('New document size created successfully.')
            tableRef.current?.reload()
          }}
          reset={!newDocumentSizeOpen}
        />
      </Modal>
    </>
  )
}

export default DocumentSizes
