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

import { useState } from 'react'
import { X } from 'lucide-react'
import { cn } from '@/lib/utils'

interface SaveSessionAsTreeModalProps {
  isOpen: boolean
  onClose: () => void
  onSave: (data: { tree_name?: string; description?: string; status: 'draft' | 'published' }) => Promise<void>
  defaultTreeName?: string
  isSaving?: boolean
}

export function SaveSessionAsTreeModal({
  isOpen,
  onClose,
  onSave,
  defaultTreeName,
  isSaving = false
}: SaveSessionAsTreeModalProps) {
  const [treeName, setTreeName] = useState(defaultTreeName || '')
  const [description, setDescription] = useState('')
  const [status, setStatus] = useState<'draft' | 'published'>('draft')

  if (!isOpen) return null

  const handleSubmit = async (e: React.FormEvent) => {
    e.preventDefault()
    await onSave({
      tree_name: treeName.trim() || undefined,
      description: description.trim() || undefined,
      status
    })
  }

  return (
    <div className="fixed inset-0 z-50 flex items-center justify-center bg-black/80 backdrop-blur-sm">
      <div className="bg-card border border-border w-full max-w-lg rounded-2xl p-6 shadow-lg">
        {/* Header */}
        <div className="mb-4 flex items-center justify-between">
          <h2 className="text-lg font-semibold text-foreground">Save Session as Tree</h2>
          <button
            onClick={onClose}
            disabled={isSaving}
            className="rounded-full p-1 text-muted-foreground hover:bg-accent hover:text-foreground disabled:opacity-50"
          >
            <X className="h-5 w-5" />
          </button>
        </div>

        {/* Info */}
        <p className="mb-4 text-sm text-muted-foreground">
          Create a new tree from this session's path. The tree will be linked to the original tree as a fork.
        </p>

        {/* Form */}
        <form onSubmit={handleSubmit} className="space-y-4">
          {/* Tree Name */}
          <div>
            <label htmlFor="treeName" className="mb-1 block text-sm font-medium text-foreground">
              Tree Name <span className="text-muted-foreground">(optional)</span>
            </label>
            <input
              id="treeName"
              type="text"
              value={treeName}
              onChange={(e) => setTreeName(e.target.value)}
              placeholder={defaultTreeName || "Auto-generated if left blank"}
              disabled={isSaving}
              maxLength={255}
              className={cn(
                'w-full rounded-md border border-border bg-card px-3 py-2 text-sm text-foreground',
                'placeholder:text-muted-foreground',
                'focus:border-primary focus:outline-none focus:ring-1 focus:ring-primary/20',
                'disabled:opacity-50'
              )}
            />
          </div>

          {/* Description */}
          <div>
            <label htmlFor="description" className="mb-1 block text-sm font-medium text-foreground">
              Description <span className="text-muted-foreground">(optional)</span>
            </label>
            <textarea
              id="description"
              value={description}
              onChange={(e) => setDescription(e.target.value)}
              placeholder="Add a description for this tree"
              disabled={isSaving}
              rows={3}
              className={cn(
                'w-full rounded-md border border-border bg-card px-3 py-2 text-sm text-foreground',
                'placeholder:text-muted-foreground',
                'focus:border-primary focus:outline-none focus:ring-1 focus:ring-primary/20',
                'disabled:opacity-50'
              )}
            />
          </div>

          {/* Status */}
          <div>
            <label className="mb-2 block text-sm font-medium text-foreground">Status</label>
            <div className="flex gap-4">
              <label className="flex cursor-pointer items-center gap-2">
                <input
                  type="radio"
                  name="status"
                  value="draft"
                  checked={status === 'draft'}
                  onChange={() => setStatus('draft')}
                  disabled={isSaving}
                  className="h-4 w-4 border-border text-foreground focus:ring-2 focus:ring-primary/20 focus:ring-offset-0"
                />
                <span className="text-sm text-foreground">Draft</span>
              </label>
              <label className="flex cursor-pointer items-center gap-2">
                <input
                  type="radio"
                  name="status"
                  value="published"
                  checked={status === 'published'}
                  onChange={() => setStatus('published')}
                  disabled={isSaving}
                  className="h-4 w-4 border-border text-foreground focus:ring-2 focus:ring-primary/20 focus:ring-offset-0"
                />
                <span className="text-sm text-foreground">Published</span>
              </label>
            </div>
          </div>

          {/* Actions */}
          <div className="flex justify-end gap-2 pt-2">
            <button
              type="button"
              onClick={onClose}
              disabled={isSaving}
              className={cn(
                'rounded-md border border-border px-4 py-2 text-sm font-medium text-muted-foreground',
                'hover:bg-accent hover:text-foreground disabled:opacity-50'
              )}
            >
              Cancel
            </button>
            <button
              type="submit"
              disabled={isSaving}
              className={cn(
                'rounded-md bg-gradient-brand px-4 py-2 text-sm font-medium text-white shadow-lg shadow-primary/20',
                'hover:opacity-90 disabled:opacity-50'
              )}
            >
              {isSaving ? 'Saving...' : 'Save as Tree'}
            </button>
          </div>
        </form>
      </div>
    </div>
  )
}