<template>
  <div class="space-y-3">
    <div class="flex justify-end">
      <UTooltip :text="$t('applicationForms.formElements.table.enlargeTable')">
        <UButton
          icon="i-lucide-maximize-2"
          color="neutral"
          variant="ghost"
          size="xs"
          :aria-label="$t('applicationForms.formElements.table.enlargeTable')"
          @click="fullscreenOpen = true"
        />
      </UTooltip>
    </div>

    <!-- Regular table -->
    <TheTableContent
      :table-data="tableData"
      :table-columns="tableColumns"
      :data-columns="dataColumns"
      :form-options="formOptions"
      :disabled="disabled"
      :can-modify-rows="canModifyRows"
      :get-column-options="getColumnOptions"
      @update:cell="updateCell"
      @update:cell-value="updateCellValue"
      @update:checkbox-cell="updateCheckboxCell"
      @add-row="addRow"
      @remove-row="removeRow"
    />

    <!-- Fullscreen Modal -->
    <UModal
      v-model:open="fullscreenOpen"
      :title="$t('applicationForms.formElements.table.enlargeTable')"
      class="min-w-3/4"
    >
      <template #content>
        <div class="flex flex-col h-full">
          <!-- Modal header -->
          <div class="flex items-center justify-between p-4 border-b border-default">
            <h2 class="text-lg font-semibold">{{ $t('applicationForms.formElements.table.editTable') }}</h2>
            <UButton
              icon="i-lucide-x"
              color="neutral"
              variant="ghost"
              size="sm"
              :aria-label="$t('common.close')"
              @click="fullscreenOpen = false"
            />
          </div>

          <!-- Modal body with full-width table -->
          <div class="flex-1 overflow-auto p-4">
            <TheTableContent
              :table-data="tableData"
              :table-columns="tableColumns"
              :data-columns="dataColumns"
              :form-options="formOptions"
              :disabled="disabled"
              :can-modify-rows="canModifyRows"
              :get-column-options="getColumnOptions"
              add-row-button-class="mt-4"
              @update:cell="updateCell"
              @update:cell-value="updateCellValue"
              @update:checkbox-cell="updateCheckboxCell"
              @add-row="addRow"
              @remove-row="removeRow"
            />
          </div>
        </div>
      </template>
    </UModal>
  </div>
</template>

<script setup lang="ts">
import type { FormElementDto, FormOptionDto, TableRowPresetDto } from '~~/.api-client'
import type { TableColumn } from '@nuxt/ui'
import { useTableCrossReferences } from '~/composables/useTableCrossReferences'
import { useFormElementVisibility } from '~/composables/useFormElementVisibility'

const props = defineProps<{
  formOptions: FormOptionDto[]
  disabled?: boolean
  allFormElements?: FormElementDto[]
  tableRowPreset?: TableRowPresetDto
}>()

const { isFormOptionVisible } = useFormElementVisibility()

const emit = defineEmits<{
  (e: 'update:formOptions', value: FormOptionDto[]): void
}>()

const { getReferencedColumnValues, getConstrainedColumnValues, applyRowPresets } = useTableCrossReferences()

const fullscreenOpen = ref(false)

const canModifyRows = computed(() => {
  if (!props.tableRowPreset) return true
  return props.tableRowPreset.canAddRows !== false
})

// Watch for changes in source table and apply row presets reactively
const sourceTableOptions = computed(() => {
  if (!props.tableRowPreset?.sourceTableReference || !props.allFormElements) return null
  const sourceTable = props.allFormElements.find(
    (el) => el.reference === props.tableRowPreset?.sourceTableReference && el.type === 'TABLE'
  )
  return sourceTable?.options
})

watch(
  sourceTableOptions,
  () => {
    if (!sourceTableOptions.value || !props.tableRowPreset || !props.allFormElements) return

    const updatedOptions = applyRowPresets(props.tableRowPreset, props.formOptions, props.allFormElements)

    const hasChanges = updatedOptions.some((opt, idx) => opt.value !== props.formOptions[idx]?.value)
    if (hasChanges) {
      emit('update:formOptions', updatedOptions)
    }
  },
  { immediate: true, deep: true }
)

type CellValue = string | string[] | boolean
type TableRowData = Record<string, CellValue>

interface DataColumn {
  key: string
  colIndex: number
}

// Filter columns based on visibility conditions
interface VisibleColumn {
  option: FormOptionDto
  originalIndex: number
}

const visibleColumns = computed<VisibleColumn[]>(() => {
  return props.formOptions
    .map((option, index) => ({ option, originalIndex: index }))
    .filter(({ option }) => {
      if (!option.visibilityConditions || !props.allFormElements) {
        return true
      }
      return isFormOptionVisible(option.visibilityConditions, props.allFormElements)
    })
})

const dataColumns = computed<DataColumn[]>(() =>
  visibleColumns.value.map(({ originalIndex }) => ({
    key: `col_${originalIndex}`,
    colIndex: originalIndex
  }))
)

const tableColumns = computed<TableColumn<TableRowData>[]>(() => {
  const columns: TableColumn<TableRowData>[] = visibleColumns.value.map(({ option, originalIndex }) => ({
    accessorKey: `col_${originalIndex}`,
    header: option.label || ''
  }))

  // Only show actions column if not disabled AND rows can be modified
  if (!props.disabled && canModifyRows.value) {
    columns.push({
      id: 'actions',
      header: ''
    })
  }

  return columns
})

const tableData = computed<TableRowData[]>(() => {
  if (props.formOptions.length === 0) return []

  const columnData: CellValue[][] = props.formOptions.map((option, colIndex) => {
    try {
      const parsed = JSON.parse(option.value || '[]')
      if (!Array.isArray(parsed)) return []

      // For multi-select columns, each cell value is already an array
      // For checkbox columns, each cell value is a boolean
      // For single-select columns, each cell value is a string
      if (isColumnMultipleAllowed(colIndex)) {
        return parsed.map((val: CellValue) => (Array.isArray(val) ? val : []))
      }
      if (isColumnCheckbox(colIndex)) {
        return parsed.map((val: CellValue) => val === true)
      }
      return parsed
    } catch {
      return []
    }
  })

  const rowCount = Math.max(...columnData.map((col) => col.length), 0)

  const rows: TableRowData[] = []
  for (let rowIndex = 0; rowIndex < rowCount; rowIndex++) {
    const row: TableRowData = {}
    props.formOptions.forEach((_, colIndex) => {
      const cellValue = columnData[colIndex]?.[rowIndex]
      if (isColumnMultipleAllowed(colIndex)) {
        row[`col_${colIndex}`] = Array.isArray(cellValue) ? cellValue : []
      } else if (isColumnCheckbox(colIndex)) {
        row[`col_${colIndex}`] = cellValue === true
      } else {
        row[`col_${colIndex}`] = typeof cellValue === 'string' ? cellValue : ''
      }
    })
    rows.push(row)
  }

  return rows
})

function isColumnMultipleAllowed(colIndex: number): boolean {
  const option = props.formOptions[colIndex]
  return option?.columnConfig?.isMultipleAllowed === true
}

function isColumnCheckbox(colIndex: number): boolean {
  const option = props.formOptions[colIndex]
  return option?.columnConfig?.isCheckbox === true
}

function getColumnOptions(colIndex: number, currentRowData?: TableRowData): string[] {
  const option = props.formOptions[colIndex]
  if (!option?.columnConfig || !props.allFormElements) {
    return []
  }

  const { columnConfig } = option
  const { rowConstraint } = columnConfig

  // If row constraint is configured, filter values based on current row's key value
  if (rowConstraint?.constraintTableReference && currentRowData) {
    const currentRowAsRecord: Record<string, string> = {}
    for (const [key, value] of Object.entries(currentRowData)) {
      currentRowAsRecord[key] =
        typeof value === 'string' ? value : Array.isArray(value) ? value.join(',') : String(value)
    }

    return getConstrainedColumnValues(
      columnConfig,
      currentRowAsRecord,
      rowConstraint.constraintTableReference,
      rowConstraint.constraintKeyColumnIndex ?? 0,
      rowConstraint.constraintValueColumnIndex ?? 1,
      props.allFormElements,
      rowConstraint.currentRowKeyColumnIndex
    )
  }

  return getReferencedColumnValues(columnConfig, props.allFormElements)
}

function updateCell(rowIndex: number, columnKey: string, value: string) {
  const colIndex = parseInt(columnKey.replace('col_', ''), 10)

  const updatedOptions = props.formOptions.map((option, index) => {
    if (index !== colIndex) return option

    let columnValues: CellValue[]
    try {
      columnValues = JSON.parse(option.value || '[]')
      if (!Array.isArray(columnValues)) columnValues = []
    } catch {
      columnValues = []
    }

    while (columnValues.length <= rowIndex) {
      columnValues.push('')
    }
    columnValues[rowIndex] = value

    return { ...option, value: JSON.stringify(columnValues) }
  })

  emit('update:formOptions', updatedOptions)
}

function updateCellValue(rowIndex: number, _columnKey: string, colIndex: number, value: string | string[]) {
  const updatedOptions = props.formOptions.map((option, index) => {
    if (index !== colIndex) return option

    let columnValues: CellValue[]
    try {
      columnValues = JSON.parse(option.value || '[]')
      if (!Array.isArray(columnValues)) columnValues = []
    } catch {
      columnValues = []
    }

    const isMultiple = isColumnMultipleAllowed(colIndex)
    while (columnValues.length <= rowIndex) {
      columnValues.push(isMultiple ? [] : '')
    }
    columnValues[rowIndex] = value

    return { ...option, value: JSON.stringify(columnValues) }
  })

  emit('update:formOptions', updatedOptions)
}

function updateCheckboxCell(rowIndex: number, colIndex: number, value: boolean) {
  const updatedOptions = props.formOptions.map((option, index) => {
    if (index !== colIndex) return option

    let columnValues: CellValue[]
    try {
      columnValues = JSON.parse(option.value || '[]')
      if (!Array.isArray(columnValues)) columnValues = []
    } catch {
      columnValues = []
    }

    while (columnValues.length <= rowIndex) {
      columnValues.push(false)
    }
    columnValues[rowIndex] = value

    return { ...option, value: JSON.stringify(columnValues) }
  })

  emit('update:formOptions', updatedOptions)
}

function addRow() {
  const updatedOptions = props.formOptions.map((option, colIndex) => {
    let columnValues: CellValue[]
    try {
      columnValues = JSON.parse(option.value || '[]')
      if (!Array.isArray(columnValues)) columnValues = []
    } catch {
      columnValues = []
    }

    // For multi-select columns, initialize with empty array
    // For checkbox columns, initialize with false
    // Otherwise empty string
    let emptyValue: CellValue = ''
    if (isColumnMultipleAllowed(colIndex)) {
      emptyValue = []
    } else if (isColumnCheckbox(colIndex)) {
      emptyValue = false
    }
    columnValues.push(emptyValue)

    return { ...option, value: JSON.stringify(columnValues) }
  })

  emit('update:formOptions', updatedOptions)
}

function removeRow(rowIndex: number) {
  const updatedOptions = props.formOptions.map((option) => {
    let columnValues: CellValue[]
    try {
      columnValues = JSON.parse(option.value || '[]')
      if (!Array.isArray(columnValues)) columnValues = []
    } catch {
      columnValues = []
    }

    columnValues.splice(rowIndex, 1)

    return { ...option, value: JSON.stringify(columnValues) }
  })

  emit('update:formOptions', updatedOptions)
}
</script>
