resolutionflow/frontend/src/components/session/ForkTreeModal.tsx

import { useState } from 'react'
import { Modal } from '@/components/common/Modal'
import { GitFork } from 'lucide-react'
import { Button } from '@/components/ui/Button'
import { Input } from '@/components/ui/Input'
import { Textarea } from '@/components/ui/Textarea'

interface ForkTreeModalProps {
  isOpen: boolean
  onClose: () => void
  originalTreeName: string
  onFork: (name: string, description: string) => Promise<void>
  onSkip: () => void
}

export function ForkTreeModal({
  isOpen,
  onClose,
  originalTreeName,
  onFork,
  onSkip
}: ForkTreeModalProps) {
  const [name, setName] = useState(`${originalTreeName} (Custom)`)
  const [description, setDescription] = useState('')
  const [isSaving, setIsSaving] = useState(false)
  const [error, setError] = useState<string | null>(null)

  const handleFork = async () => {
    if (!name.trim()) {
      setError('Please enter a name for your tree')
      return
    }

    setIsSaving(true)
    setError(null)

    try {
      await onFork(name.trim(), description.trim())
    } catch (err) {
      setError('Failed to save tree. Please try again.')
      console.error(err)
    } finally {
      setIsSaving(false)
    }
  }

  const footer = (
    <div className="flex justify-end gap-3">
      <Button
        variant="ghost"
        onClick={onSkip}
        disabled={isSaving}
      >
        Skip
      </Button>
      <Button
        onClick={handleFork}
        disabled={!name.trim()}
        loading={isSaving}
      >
        <GitFork className="h-4 w-4" />
        Save as Personal Tree
      </Button>
    </div>
  )

  return (
    <Modal isOpen={isOpen} onClose={onClose} title="Save Custom Tree?" footer={footer}>
      <div className="space-y-4">
        <div className="flex items-start gap-3 rounded-lg bg-accent/50 p-4">
          <div className="flex h-10 w-10 shrink-0 items-center justify-center rounded-full bg-accent">
            <GitFork className="h-5 w-5 text-foreground" />
          </div>
          <div>
            <p className="font-medium text-foreground">You've created a custom troubleshooting path!</p>
            <p className="mt-1 text-sm text-muted-foreground">
              Save it as your own personal tree to reuse this troubleshooting flow in the future.
            </p>
          </div>
        </div>

        <div className="space-y-4">
          <div>
            <label htmlFor="tree-name" className="mb-1.5 block text-sm font-medium text-foreground">
              Tree Name <span className="text-red-400">*</span>
            </label>
            <Input
              id="tree-name"
              type="text"
              value={name}
              onChange={(e) => setName(e.target.value)}
              placeholder="My Custom Tree"
            />
          </div>

          <div>
            <label htmlFor="tree-description" className="mb-1.5 block text-sm font-medium text-foreground">
              Description <span className="text-muted-foreground">(optional)</span>
            </label>
            <Textarea
              id="tree-description"
              value={description}
              onChange={(e) => setDescription(e.target.value)}
              placeholder="Describe what this tree helps troubleshoot..."
              rows={3}
              className="resize-none"
            />
          </div>
        </div>

        {error && (
          <p className="text-sm text-red-400">{error}</p>
        )}

        <p className="text-xs text-muted-foreground">
          The new tree will include your custom steps and will be saved to your personal tree library.
        </p>
      </div>
    </Modal>
  )
}