Function useTableColumns

apps/sim/lib/table/hooks/use-table-columns.ts:12–36

({ tableId, includeBuiltIn = false }: UseTableColumnsOptions)

/** Fetches table schema columns as dropdown options. */
export function useTableColumns({ tableId, includeBuiltIn = false }: UseTableColumnsOptions) {
  const workspaceId = useWorkflowRegistry((state) => state.hydration.workspaceId)
  const { data: tableData } = useTable(workspaceId ?? undefined, tableId ?? undefined)

  const schemaColumns = useMemo<ColumnOption[]>(
    () =>
      (tableData?.schema?.columns || []).map((col) => ({
        value: col.name,
        label: col.name,
      })),
    [tableData]
  )

  return useMemo(() => {
    if (includeBuiltIn) {
      const builtInCols = [
        { value: 'createdAt', label: 'createdAt' },
        { value: 'updatedAt', label: 'updatedAt' },
      ]
      return [...schemaColumns, ...builtInCols]
    }

    return schemaColumns
  }, [includeBuiltIn, schemaColumns])
}