src/components/forms/PinSetupForm.tsx

'use client'

import { useState } from 'react'
import { useForm } from 'react-hook-form'
import { zodResolver } from '@hookform/resolvers/zod'
import { useRouter, useSearchParams } from 'next/navigation'
import { Button } from 'components/ui/button'
import { z } from 'zod'
import { pinEntrySchema, type PinEntryForm as PinEntryFormType } from 'lib/validations/code'

export function PinSetupForm() {
  const [loading, setLoading] = useState(false)
  const [error, setError] = useState<string | null>(null)
  const [skipPin, setSkipPin] = useState(false)
  const [manualCode, setManualCode] = useState('')
  const router = useRouter()
  const searchParams = useSearchParams()
  const codeFromParams = searchParams.get('code')

  // Define schema typed for the form to ensure TypeScript knows pin + confirmPin exist
  const setupSchema = pinEntrySchema.extend({
    confirmPin: pinEntrySchema.shape.pin
  }).refine((data) => data.pin === data.confirmPin, {
    message: "PIN se neshoduje",
    path: ["confirmPin"],
  })

  const {
    register,
    handleSubmit,
    formState: { errors }
  } = useForm<z.infer<typeof setupSchema>>({
    resolver: zodResolver(setupSchema)
  })

  // Use effectiveCode: prefer query param, otherwise manual input
  const effectiveCode = (codeFromParams || manualCode || '').toUpperCase()

  const onSubmit = async (data: PinEntryFormType | { pin?: string }) => {
    const code = effectiveCode
    if (!code) {
      setError('Chybí kód místnosti')
      return
    }

    try {
      setLoading(true)
      setError(null)

      const response = await fetch('/api/codes/setup', {
        method: 'POST',
        credentials: 'include',
        headers: { 'Content-Type': 'application/json' },
        body: JSON.stringify({
          code: code.toUpperCase(),
          pin: (data as any).pin ? (data as any).pin : undefined,
          userAgent: navigator.userAgent
        })
      })

      const result = await response.json()

      if (!response.ok) {
        setError(result.error || 'Nastala chyba při nastavování')
        return
      }

      if (result.roomId) {
        router.push(`/room/${result.roomId}?code=${code}`)
      }
    } catch (e) {
      console.error('PinSetup submit error:', e)
      setError('Došlo k neočekávané chybě')
    } finally {
      setLoading(false)
    }
  }

  const handleSkipPin = async () => {
    // Ensure we have a code
    if (!effectiveCode) {
      setError('Zadejte prosím kód místnosti před pokračováním bez PIN.')
      return
    }
    setSkipPin(true)
    await onSubmit({ pin: '' })
  }

  if (skipPin) {
    return (
      <div className="space-y-4">
        <div className="p-4 bg-yellow-50 rounded-lg">
          <p className="text-yellow-800 text-sm">
            <strong>Upozornění:</strong> Bez PIN ochrany se o místnost přijdete při smazání cookies prohlížeče.
          </p>
        </div>
        <Button onClick={handleSkipPin} className="w-full" disabled={loading} variant="secondary" >
          {loading ? 'Vytvářím místnost...' : 'Pokračovat bez PIN'}
        </Button>
      </div>
    )
  }

  return (
    <div className="space-y-6">
      <div className="p-4 bg-blue-50 rounded-lg">
        <h3 className="font-medium text-blue-900 mb-2">Ochrana místnosti</h3>
        <p className="text-blue-800 text-sm">
          Nastavte si 5-číselný PIN pro ochranu přístupu do této místnosti.
          Bez PIN se o přístup přijdete při smazání cookies.
        </p>
      </div>

      {/* If no code in query, allow user to type it */}
      {!codeFromParams && (
        <div>
          <label htmlFor="manualCode" className="block text-sm font-medium text-gray-700 mb-1">
            Kód místnosti (5 znaků)
          </label>
          <input
            value={manualCode}
            onChange={(e) => setManualCode(e.target.value.toUpperCase())}
            type="text"
            id="manualCode"
            maxLength={5}
            placeholder="A1B2C"
            className="w-full px-3 py-2 border border-gray-300 rounded-md focus:outline-none focus:ring-2 focus:ring-brand-500 focus:border-brand-500 uppercase text-center font-mono"
            style={{ textTransform: 'uppercase' }}
          />
          <p className="mt-1 text-xs text-gray-500">Zadejte kód místnosti, který chcete vytvořit nebo použít.</p>
        </div>
      )}

      <form onSubmit={handleSubmit(onSubmit)} className="space-y-4">
        <div>
          <label htmlFor="pin" className="block text-sm font-medium text-gray-700 mb-2">
            Nový PIN (5 číslic)
          </label>
          <input
            {...register('pin')}
            type="password"
            id="pin"
            maxLength={5}
            placeholder="12345"
            autoComplete="new-password"
            inputMode="numeric"
            pattern="[0-9]*"
            aria-label="Nový PIN (5 číslic)"
            className="w-full px-3 py-2 border border-gray-300 rounded-md focus:outline-none focus:ring-2 focus:ring-brand-500 focus:border-brand-500 text-center text-xl font-mono tracking-widest"
          />
          {errors?.pin && (
            <p className="mt-1 text-sm text-red-600">{errors.pin.message}</p>
          )}
        </div>

        <div>
          <label htmlFor="confirmPin" className="block text-sm font-medium text-gray-700 mb-2">
            Potvrdit PIN
          </label>
          <input
            {...register('confirmPin')}
            type="password"
            id="confirmPin"
            maxLength={5}
            placeholder="12345"
            autoComplete="new-password"
            inputMode="numeric"
            pattern="[0-9]*"
            aria-label="Potvrzení PIN"
            className="w-full px-3 py-2 border border-gray-300 rounded-md focus:outline-none focus:ring-2 focus:ring-brand-500 focus:border-brand-500 text-center text-xl font-mono tracking-widest"
          />
          {errors?.confirmPin && (
            <p className="mt-1 text-sm text-red-600">{errors.confirmPin.message}</p>
          )}
        </div>

        {error && (
          <div className="p-3 text-sm text-red-600 bg-red-50 rounded-md">
            {error}
          </div>
        )}

        <div className="space-y-2">
          <Button type="submit" className="w-full" disabled={loading}>
            {loading ? 'Nastavuji PIN...' : 'Vytvořit místnost s PIN'}
          </Button>
          
          <button
            type="button"
            onClick={handleSkipPin}
            className="w-full px-4 py-2 text-sm text-gray-600 hover:text-gray-800 border border-gray-300 rounded-md hover:bg-gray-50"
          >
            Pokračovat bez PIN ochrany
          </button>
        </div>
      </form>

      <div className="text-center">
        <button
          type="button"
          onClick={() => router.push('/')}
          className="text-sm text-gray-600 hover:text-gray-800"
        >
          ← Zpět na zadání kódu
        </button>
      </div>
    </div>
  )
}