import type { GetTableRowsArgs } from './table-rows-query'

type TableRowKeyArgs = Omit<GetTableRowsArgs, 'table'> & { table?: { id?: number } }

export const tableRowKeys = {
  tableRows: (
    projectRef?: string,
    { table, roleImpersonationState, ...args }: TableRowKeyArgs = {}
  ) =>
    [
      'projects',
      projectRef,
      'table-rows',
      table?.id,
      'rows',
      { roleImpersonation: roleImpersonationState?.role, ...args },
    ] as const,
  tableRowsCount: (projectRef?: string, { table, ...args }: TableRowKeyArgs = {}) =>
    ['projects', projectRef, 'table-rows', table?.id, 'count', args] as const,
  tableRowsAndCount: (projectRef?: string, tableId?: number) =>
    ['projects', projectRef, 'table-rows', tableId] as const,
}