"use client"

import React from "react"
import { useForm } from "react-hook-form"
import { zodResolver } from "@hookform/resolvers/zod"
import { z } from "zod"
import {
  Dialog,
  DialogContent,
  DialogDescription,
  DialogHeader,
  DialogTitle,
  DialogFooter
} from "@/components/ui/dialog"
import {
  Form,
  FormControl,
  FormField,
  FormItem,
  FormLabel,
  FormMessage,
} from "@/components/ui/form"
import { Button } from "@/components/ui/button"
import { Progress } from "@/components/ui/progress"
import { 
  Upload, 
  FileText, 
  X, 
  Loader2, 
  CheckCircle,
  Paperclip
} from "lucide-react"
import { toast } from "sonner"
import { useSession } from "next-auth/react"

/* -------------------------------------------------------------------------------------------------
 * Schema & Types
 * -----------------------------------------------------------------------------------------------*/

// 파일 검증 스키마
const MAX_FILE_SIZE = 1024 * 1024 * 1024 // 50MB
const ACCEPTED_FILE_TYPES = [
  'application/pdf',
  'application/vnd.openxmlformats-officedocument.wordprocessingml.document',
  'application/vnd.openxmlformats-officedocument.spreadsheetml.sheet',
  'application/vnd.ms-excel',
  'image/jpeg',
  'image/png',
  'image/gif',
  'text/plain',
  'application/zip',
  'application/x-zip-compressed',
  // Presentations
  'application/vnd.ms-powerpoint',                                           // .ppt
  'application/vnd.openxmlformats-officedocument.presentationml.presentation', // .pptx
  // CAD and Drawing files
  'application/acad',                                                        // .dwg
  'image/vnd.dwg',                                                          // .dwg (alternative MIME)
  'application/x-autocad',                                                  // .dwg (alternative MIME)
  'application/dxf',                                                        // .dxf
  'image/vnd.dxf',                                                          // .dxf (alternative MIME)
  'application/x-dxf',                                                      // .dxf (alternative MIME)
  'application/step',                                                       // .step/.stp
  'application/sla',                                                        // .stl
  'model/stl',                                                              // .stl (alternative MIME)
  'application/iges',                                                       // .iges/.igs
]

const attachmentUploadSchema = z.object({
  attachments: z
    .array(z.instanceof(File))
    .min(1, "Please upload at least 1 file")
    // .max(10, "Maximum 10 files can be uploaded")
    .refine(
      (files) => files.every((file) => file.size <= MAX_FILE_SIZE),
      "File size must be 50MB or less"
    )
    // .refine(
    //   (files) => files.every((file) => ACCEPTED_FILE_TYPES.includes(file.type)),
    //   "Unsupported file format"
    // ),
})

interface AddAttachmentDialogProps {
  open: boolean
  onOpenChange: (open: boolean) => void
  revisionId: number
  revisionName: string
  onSuccess?: (result?: any) => void
}

/* -------------------------------------------------------------------------------------------------
 * File Upload Component
 * -----------------------------------------------------------------------------------------------*/
function FileUploadArea({ 
  files, 
  onFilesChange 
}: { 
  files: File[]
  onFilesChange: (files: File[]) => void 
}) {
  const fileInputRef = React.useRef<HTMLInputElement>(null)

  const handleFileSelect = (event: React.ChangeEvent<HTMLInputElement>) => {
    const selectedFiles = Array.from(event.target.files || [])
    if (selectedFiles.length > 0) {
      onFilesChange([...files, ...selectedFiles])
    }
  }

  const handleDrop = (event: React.DragEvent<HTMLDivElement>) => {
    event.preventDefault()
    const droppedFiles = Array.from(event.dataTransfer.files)
    if (droppedFiles.length > 0) {
      onFilesChange([...files, ...droppedFiles])
    }
  }

  const handleDragOver = (event: React.DragEvent<HTMLDivElement>) => {
    event.preventDefault()
  }

  const removeFile = (index: number) => {
    onFilesChange(files.filter((_, i) => i !== index))
  }

  const formatFileSize = (bytes: number) => {
    if (bytes === 0) return '0 Bytes'
    const k = 1024
    const sizes = ['Bytes', 'KB', 'MB', 'GB']
    const i = Math.floor(Math.log(bytes) / Math.log(k))
    return parseFloat((bytes / Math.pow(k, i)).toFixed(2)) + ' ' + sizes[i]
  }

  return (
    <div className="space-y-4">
      <div
        className="border-2 border-dashed border-gray-300 rounded-lg p-6 text-center cursor-pointer hover:border-gray-400 transition-colors"
        onDrop={handleDrop}
        onDragOver={handleDragOver}
        onClick={() => fileInputRef.current?.click()}
      >
        <Paperclip className="mx-auto h-12 w-12 text-gray-400 mb-4" />
        <p className="text-sm text-gray-600 mb-2">
          Drag files to add here or click to select
        </p>
        <p className="text-xs text-gray-500">
          Supports PDF, Word, Excel, Image, Text, ZIP, CAD files (DWG, DXF, STEP, STL, IGES) (max 1GB)
        </p>
        <input
          ref={fileInputRef}
          type="file"
          multiple
          accept=".pdf,.doc,.docx,.xls,.xlsx,.jpg,.jpeg,.png,.gif,.txt,.zip,.dwg,.dxf,.step,.stp,.stl,.iges,.igs"
          onChange={handleFileSelect}
          className="hidden"
        />
      </div>

      {files.length > 0 && (
        <div className="space-y-2 max-h-40 overflow-y-auto overscroll-contain pr-2">
          <p className="text-sm font-medium">Selected Files ({files.length})</p>
          <div className="max-h-40 overflow-y-auto space-y-2">
            {files.map((file, index) => (
              <div
                key={index}
                className="flex items-center justify-between p-2 bg-gray-50 rounded border"
              >
                <div className="flex items-center space-x-2 flex-1">
                  <FileText className="h-4 w-4 text-gray-500" />
                  <div className="flex-1 min-w-0">
                    <p className="text-sm font-medium truncate" title={file.name}>
                      {file.name}
                    </p>
                    <p className="text-xs text-gray-500">
                      {formatFileSize(file.size)}
                    </p>
                  </div>
                </div>
                <Button
                  type="button"
                  variant="ghost"
                  size="sm"
                  onClick={() => removeFile(index)}
                  className="h-8 w-8 p-0"
                >
                  <X className="h-4 w-4" />
                </Button>
              </div>
            ))}
          </div>
        </div>
      )}
    </div>
  )
}

/* -------------------------------------------------------------------------------------------------
 * Main Dialog Component
 * -----------------------------------------------------------------------------------------------*/
export function AddAttachmentDialog({
    open,
    onOpenChange,
    revisionId,
    revisionName,
    onSuccess
  }: AddAttachmentDialogProps) {
    const [isUploading, setIsUploading] = React.useState(false)
    const [uploadProgress, setUploadProgress] = React.useState(0)
    const { data: session } = useSession()
  
    const userName = React.useMemo(() => {
      return session?.user?.name ? session.user.name : null;
    }, [session]);
  
    type AttachmentUploadSchema = z.infer<typeof attachmentUploadSchema>
  
    const form = useForm<AttachmentUploadSchema>({
      resolver: zodResolver(attachmentUploadSchema),
      defaultValues: {
        attachments: [],
      },
    })
  
    const watchedFiles = form.watch("attachments")
  
    const handleDialogClose = () => {
      if (!isUploading) {
        form.reset()
        setUploadProgress(0)
        onOpenChange(false)
      }
    }
  
    const onSubmit = async (data: AttachmentUploadSchema) => {
      setIsUploading(true)
      setUploadProgress(0)
  
      try {
        const formData = new FormData()
        formData.append("revisionId", String(revisionId))
        formData.append("uploaderName", userName || "evcp")
  
        // 파일들 추가
        data.attachments.forEach((file) => {
          formData.append("attachments", file)
        })
  
        // 진행률 업데이트 시뮬레이션
        const totalSize = data.attachments.reduce((sum, file) => sum + file.size, 0)
        let uploadedSize = 0
        
        const progressInterval = setInterval(() => {
          uploadedSize += totalSize * 0.1
          const progress = Math.min((uploadedSize / totalSize) * 100, 90)
          setUploadProgress(progress)
        }, 300)
  
        const response = await fetch('/api/revision-attachment', {
          method: 'POST',
          body: formData,
        })
  
        clearInterval(progressInterval)
  
        if (!response.ok) {
          const errorData = await response.json()
          throw new Error(errorData.error || errorData.details || 'Failed to upload attachments.')
        }
  
        const result = await response.json()
        setUploadProgress(100)
  
        toast.success(
          result.message || 
          `${result.data?.uploadedFiles?.length || 0} attachments added.`
        )
  
        console.log('✅ Attachment upload successful:', result)
        
        setTimeout(() => {
          handleDialogClose()
          onSuccess?.(result)
        }, 1000)
  
      } catch (error) {
        console.error('❌ Attachment upload error:', error)
        toast.error(error instanceof Error ? error.message : "An error occurred while uploading attachments")
      } finally {
        setIsUploading(false)
        setTimeout(() => setUploadProgress(0), 2000)
      }
    }
  
    return (
      <Dialog open={open} onOpenChange={handleDialogClose}>
        <DialogContent className="max-w-2xl max-h-[80vh] flex flex-col overflow-hidden">
          {/* 고정 헤더 */}
          <DialogHeader className="flex-shrink-0 pb-4 border-b">
            <DialogTitle className="flex items-center gap-2">
              <Paperclip className="h-5 w-5" />
              Add Attachments
            </DialogTitle>
            <DialogDescription className="text-sm">
              Upload additional attachments to revision {revisionName}
            </DialogDescription>
          </DialogHeader>
  
          <Form {...form}>
            <form onSubmit={form.handleSubmit(onSubmit)} className="flex flex-col flex-1 overflow-hidden">
              {/* 스크롤 가능한 중간 영역 */}
              <div className="flex-1 overflow-y-auto px-1 py-4 space-y-6">
                {/* 파일 업로드 */}
                <FormField
                  control={form.control}
                  name="attachments"
                  render={({ field }) => (
                    <FormItem>
                      <FormLabel className="required">Attachments</FormLabel>
                      <FormControl>
                        <FileUploadArea
                          files={watchedFiles || []}
                          onFilesChange={field.onChange}
                        />
                      </FormControl>
                      <FormMessage />
                    </FormItem>
                  )}
                />
  
                {/* 업로드 진행률 */}
                {isUploading && (
                  <div className="space-y-2">
                    <div className="flex items-center justify-between text-sm">
                      <span>Upload Progress</span>
                      <span>{uploadProgress.toFixed(0)}%</span>
                    </div>
                    <Progress value={uploadProgress} className="w-full" />
                    {uploadProgress === 100 && (
                      <div className="flex items-center gap-2 text-sm text-green-600">
                        <CheckCircle className="h-4 w-4" />
                        <span>Upload Complete</span>
                      </div>
                    )}
                  </div>
                )}
              </div>
  
              {/* 고정 버튼 영역 */}
              <DialogFooter>
                <Button
                  type="button"
                  variant="outline"
                  onClick={handleDialogClose}
                  disabled={isUploading}
                >
                  Cancel
                </Button>
                <Button
                  type="submit"
                  disabled={isUploading || !form.formState.isValid}
                  className="min-w-[120px]"
                >
                  {isUploading ? (
                    <>
                      <Loader2 className="mr-2 h-4 w-4 animate-spin" />
                      Uploading...
                    </>
                  ) : (
                    <>
                      <Upload className="mr-2 h-4 w-4" />
                      Add
                    </>
                  )}
                </Button>
              </DialogFooter>
            </form>
          </Form>
        </DialogContent>
      </Dialog>
    )
  }