import { defineCommands, type Extension } from '@prosekit/core'
import {
  addColumnAfter,
  addColumnBefore,
  addRowAfter,
  addRowBefore,
  deleteColumn,
  deleteRow,
  deleteTable,
  mergeCells,
  splitCell,
} from 'prosemirror-tables'

import { deleteCellSelection } from './table-commands/delete-cell-selection.ts'
import { exitTable } from './table-commands/exit-table.ts'
import { insertTable, type InsertTableOptions } from './table-commands/insert-table.ts'
import { moveTableColumn, type MoveTableColumnOptions } from './table-commands/move-table-column.ts'
import { moveTableRow, type MoveTableRowOptions } from './table-commands/move-table-row.ts'
import { selectTableCell, type SelectTableCellOptions } from './table-commands/select-table-cell.ts'
import { selectTableColumn, type SelectTableColumnOptions } from './table-commands/select-table-column.ts'
import { selectTableRow, type SelectTableRowOptions } from './table-commands/select-table-row.ts'
import { selectTable, type SelectTableOptions } from './table-commands/select-table.ts'

/**
 * @internal
 */
export type TableCommandsExtension = Extension<{
  Commands: {
    insertTable: [options: InsertTableOptions]
    exitTable: []

    selectTable: [options?: SelectTableOptions]
    selectTableCell: [options?: SelectTableCellOptions]
    selectTableColumn: [options?: SelectTableColumnOptions]
    selectTableRow: [options?: SelectTableRowOptions]

    addTableColumnBefore: []
    addTableColumnAfter: []
    addTableRowAbove: []
    addTableRowBelow: []

    deleteTable: []
    deleteTableColumn: []
    deleteTableRow: []
    deleteCellSelection: []

    mergeTableCells: []
    splitTableCell: []

    moveTableRow: [options: MoveTableRowOptions]
    moveTableColumn: [options: MoveTableColumnOptions]
  }
}>

/**
 * Adds commands for working with `table` nodes.
 */
export function defineTableCommands(): TableCommandsExtension {
  return defineCommands({
    insertTable,
    exitTable: () => exitTable,

    selectTable,
    selectTableCell,
    selectTableColumn,
    selectTableRow,

    addTableColumnBefore: () => addColumnBefore,
    addTableColumnAfter: () => addColumnAfter,
    addTableRowAbove: () => addRowBefore,
    addTableRowBelow: () => addRowAfter,

    deleteTable: () => deleteTable,
    deleteTableColumn: () => deleteColumn,
    deleteTableRow: () => deleteRow,
    deleteCellSelection: () => deleteCellSelection,

    mergeTableCells: () => mergeCells,
    splitTableCell: () => splitCell,

    moveTableRow,
    moveTableColumn,
  })
}