import { TableHeaderRow } from '@devexpress/dx-react-grid-material-ui'
import { TableCell } from 'material-ui/Table'

import './WithHeaderRow.less'

export default withHeaderRow = (WrappedComponent) =>
  observer class Cell extends Component
    render: =>
      <TableHeaderRow.Cell {...@props} className="slap-table-header-cell" />

  observer class WithHeaderRow extends Component
    render: =>
      <WrappedComponent
        {...@props}
        tableHeaderRowComponent={<TableHeaderRow {...@props.tableHeaderRowProps} cellComponent={Cell} />}
      />