'use client'

import { useState } from 'react'
import { Button } from './ui/button'
import { Input } from './ui/input'
import {
  Dialog,
  DialogContent,
  DialogDescription,
  DialogHeader,
  DialogTitle,
  DialogTrigger,
} from './ui/dialog'
import { Settings, Plus, Palette, Trash2, Sparkles } from 'lucide-react'
import { LABEL_COLORS, LabelColorName } from '@/lib/types'
import { cn } from '@/lib/utils'
import { useNotebooks } from '@/context/notebooks-context'
import { useLanguage } from '@/lib/i18n'
import { useRefresh } from '@/lib/use-refresh'

export interface LabelManagementDialogProps {
  open?: boolean
  onOpenChange?: (open: boolean) => void
}

export function LabelManagementDialog(props: LabelManagementDialogProps = {}) {
  const { open, onOpenChange } = props
  const { labels, isLoading: loading, addLabel, updateLabel, deleteLabel } = useNotebooks()
  const { t, language } = useLanguage()
  const { refreshLabels } = useRefresh()
  const [confirmDeleteId, setConfirmDeleteId] = useState<string | null>(null)
  const [newLabel, setNewLabel] = useState('')
  const [editingColorId, setEditingColorId] = useState<string | null>(null)

  const controlled = open !== undefined && onOpenChange !== undefined

  const handleAddLabel = async () => {
    const trimmed = newLabel.trim()
    if (trimmed) {
      try {
        await addLabel(trimmed, 'gray')
        refreshLabels()
        setNewLabel('')
      } catch (error) {
        console.error('Failed to add label:', error)
      }
    }
  }

  const handleDeleteLabel = async (id: string) => {
    try {
      const labelToDelete = labels.find(l => l.id === id)
      await deleteLabel(id)
      refreshLabels()
      if (labelToDelete) {
        window.dispatchEvent(new CustomEvent('label-deleted', { detail: { name: labelToDelete.name } }))
      }
      setConfirmDeleteId(null)
    } catch (error) {
      console.error('Failed to delete label:', error)
    }
  }

  const handleChangeColor = async (id: string, color: LabelColorName) => {
    try {
      await updateLabel(id, { color })
      refreshLabels()
      setEditingColorId(null)
    } catch (error) {
      console.error('Failed to update label color:', error)
    }
  }

  const dialogContent = (
      <DialogContent
        className="max-w-md"
        dir={language === 'fa' || language === 'ar' ? 'rtl' : 'ltr'}
        onInteractOutside={(event) => {
          const target = event.target as HTMLElement;
          const isSonnerElement =
            target.closest('[data-sonner-toast]') ||
            target.closest('[data-sonner-toaster]') ||
            target.closest('[data-icon]') ||
            target.closest('[data-content]') ||
            target.closest('[data-description]') ||
            target.closest('[data-title]') ||
            target.closest('[data-button]');
          if (isSonnerElement) {
            event.preventDefault();
            return;
          }
          if (target.getAttribute('data-sonner-toaster') !== null) {
            event.preventDefault();
            return;
          }
        }}
      >
        <DialogHeader>
          <DialogTitle>{t('labels.editLabels')}</DialogTitle>
          <DialogDescription>
            {t('labels.editLabelsDescription')}
          </DialogDescription>
        </DialogHeader>

        <div className="space-y-4 py-4">
          <div className="flex gap-2">
            <Input
              placeholder={t('labels.newLabelPlaceholder')}
              value={newLabel}
              onChange={(e) => setNewLabel(e.target.value)}
              onKeyDown={(e) => {
                if (e.key === 'Enter') {
                  e.preventDefault()
                  handleAddLabel()
                }
              }}
            />
            <Button onClick={handleAddLabel} size="icon">
              <Plus className="h-4 w-4" />
            </Button>
          </div>

          <div className="max-h-[60vh] overflow-y-auto space-y-2">
            {loading ? (
              <p className="text-sm text-muted-foreground">{t('labels.loading')}</p>
            ) : labels.length === 0 ? (
              <p className="text-sm text-muted-foreground">{t('labels.noLabelsFound')}</p>
            ) : (
              labels.map((label) => {
                const colorClasses = LABEL_COLORS[label.color]
                const isEditing = editingColorId === label.id
                const isAI = label.type === 'ai'

                return (
                  <div key={label.id} className="flex items-center justify-between p-2 rounded-md hover:bg-accent/50 group">
                    <div className="flex items-center gap-3 flex-1 relative">
                      {isAI ? (
                        <Sparkles className={cn("h-4 w-4", "text-memento-blue")} />
                      ) : (
                        <div className={cn("h-3 w-3 rounded-full", colorClasses.bg)} />
                      )}
                      <span className="font-medium text-sm">{label.name}</span>
                      {isAI && (
                        <span className="text-[8px] px-1.5 py-0.5 rounded-full bg-memento-blue/10 text-memento-blue font-bold uppercase">IA</span>
                      )}

                      {isEditing && (
                        <div className="absolute z-20 top-8 left-0 bg-popover text-popover-foreground border border-border rounded-lg shadow-xl p-3 animate-in fade-in zoom-in-95 w-48">
                          <div className="grid grid-cols-5 gap-2">
                            {(Object.keys(LABEL_COLORS) as LabelColorName[]).map((color) => {
                              const classes = LABEL_COLORS[color]
                              return (
                                <button
                                  key={color}
                                  className={cn(
                                    'h-7 w-7 rounded-full border-2 transition-all hover:scale-110',
                                    classes.bg,
                                    label.color === color ? 'border-foreground dark:border-foreground ring-2 ring-offset-1' : 'border-transparent'
                                  )}
                                  onClick={() => handleChangeColor(label.id, color)}
                                  title={color}
                                />
                              )
                            })}
                          </div>
                        </div>
                      )}
                    </div>

                    {confirmDeleteId === label.id ? (
                      <div className="flex items-center gap-2">
                        <span className="text-xs text-red-500 font-medium">{t('labels.confirmDeleteShort') || 'Confirmer ?'}</span>
                        <Button variant="ghost" size="sm" className="h-7 px-2 text-xs" onClick={() => setConfirmDeleteId(null)}>
                          {t('common.cancel') || 'Annuler'}
                        </Button>
                        <Button variant="destructive" size="sm" className="h-7 px-2 text-xs" onClick={() => handleDeleteLabel(label.id)}>
                          {t('common.delete') || 'Supprimer'}
                        </Button>
                      </div>
                    ) : (
                      <div className="flex items-center gap-1 opacity-0 group-hover:opacity-100 transition-opacity">
                        <Button
                          variant="ghost"
                          size="icon"
                          className="h-8 w-8 text-muted-foreground hover:text-foreground"
                          onClick={() => setEditingColorId(isEditing ? null : label.id)}
                          title={t('labels.changeColor')}
                        >
                          <Palette className="h-4 w-4" />
                        </Button>
                        <Button
                          variant="ghost"
                          size="icon"
                          className="h-8 w-8 text-red-400 hover:text-red-600 hover:bg-red-50 dark:hover:bg-red-950/20"
                          onClick={() => setConfirmDeleteId(label.id)}
                          title={t('labels.deleteTooltip')}
                        >
                          <Trash2 className="h-4 w-4" />
                        </Button>
                      </div>
                    )}
                  </div>
                )
                })
              )}
          </div>
        </div>
      </DialogContent>
  )

  if (controlled) {
    return (
      <Dialog open={open} onOpenChange={onOpenChange}>
        {dialogContent}
      </Dialog>
    )
  }

  return (
    <Dialog>
      <DialogTrigger asChild>
        <Button variant="ghost" size="icon" title={t('labels.manage')}>
          <Settings className="h-5 w-5" />
        </Button>
      </DialogTrigger>
      {dialogContent}
    </Dialog>
  )
}