import { useState, useRef } from "react";
import { FileText, UploadCloud, X } from "lucide-react";
import { LuSendHorizontal } from "react-icons/lu";
import SubmitStatusTable from "./SubmissionStatusTable";
import { useSubmission } from "../../../../hooks/useTasks";
import type { TaskSubmissionData } from "../../../../types/task.types";
import { customToast } from "../../../utils/showToast";

interface TaskFormFileProps {
  taskId: string;
  projectId?: string;
  deadline?: Date;
  initialSubmission?: TaskSubmissionData | null;
  isProjectActive?: boolean;
}

export default function TaskFormFile({ taskId, projectId, deadline, initialSubmission, isProjectActive = true }: TaskFormFileProps) {
  const { submission, loading, submitFile, updateFile, remove } = useSubmission(taskId, initialSubmission, projectId);
  const [files, setFiles] = useState<File[]>([]);
  const [isDragActive, setIsDragActive] = useState(false);
  const [isEditing, setIsEditing] = useState(false);
  const fileInputRef = useRef<HTMLInputElement>(null);

  const validateAndAddFiles = (selectedFiles: File[]) => {
    const allowedExtensions = ["pdf"];
    const validFiles: File[] = [];
    selectedFiles.forEach((file) => {
      const ext = file.name.split(".").pop()?.toLowerCase();
      if (ext && allowedExtensions.includes(ext)) {
        if (file.size <= 5 * 1024 * 1024) {
          validFiles.push(file);
        } else {
          customToast.error(
            "File terlalu besar!",
            `File ${file.name} melebihi batas ukuran 5MB.`
          );
        }
      }
    });
    if (validFiles.length > 0) setFiles((prev) => [...prev, ...validFiles]);
  };

  const handleDrag = (e: React.DragEvent) => {
    e.preventDefault(); e.stopPropagation();
    setIsDragActive(e.type === "dragenter" || e.type === "dragover");
  };

  const handleDrop = (e: React.DragEvent) => {
    e.preventDefault(); e.stopPropagation();
    setIsDragActive(false);
    if (e.dataTransfer.files.length > 0) validateAndAddFiles(Array.from(e.dataTransfer.files));
  };

  const removeFile = (index: number, e: React.MouseEvent) => {
    e.stopPropagation();
    setFiles((prev) => prev.filter((_, i) => i !== index));
  };

  const handleSubmit = async (e: React.FormEvent) => {
    e.preventDefault();
    if (files.length === 0) {
      customToast.error(
        "File belum dipilih!",
        "Choose your file before submit it!"
      );
      return;
    }

    const res = await customToast.promise(
      isEditing ? updateFile(submission!.id, files) : submitFile(files),
      {
        loading: isEditing ? "Memperbarui file..." : "Mengirim file...",
        success: (res) => ({
          title: isEditing ? "File berhasil diperbarui!" : "File berhasil dikirim!",
          description: res?.message ??
              (isEditing
                ? "Perubahan submission Anda telah disimpan."
                : "Link berhasil dikirim."),
        }),
        error: (err) => ({
          title: isEditing ? "Gagal memperbarui file!" : "Gagal mengirim file!",
          description:
            err?.response?.data?.message ||
            err?.message ||
            "Terjadi kesalahan saat memproses file Anda.",
        }),
      }
    );

    if (res) {
      setIsEditing(false);
      setFiles([]);
    }
  };

  const handleDelete = async () => {
    try {
      const ok = await remove(submission!.id);
      if (ok) {
        setFiles([]);
        setIsEditing(false);
        customToast.success(
          "Submission berhasil dihapus!",
          ok.message
        );
      }
    } catch (err: any) {
      customToast.error(
        "Gagal menghapus submission!",
        err?.response?.data?.message || "Terjadi kesalahan saat menghapus submission."
      );
    }
  };

  const handleCancelEdit = () => {
    setFiles([]);
    setIsEditing(false);
  };

  if (submission && !isEditing) {
    return (
      <SubmitStatusTable
        type="file"
        files={submission.files}
        submittedAt={new Date(submission.submitted_at)}
        deadline={deadline}
        onEdit={() => setIsEditing(true)}
        onDelete={handleDelete}
        disabled={!isProjectActive} 
      />
    );
  }

  return (
    <form onSubmit={handleSubmit} className="space-y-4">
      <input
        type="file"
        multiple
        ref={fileInputRef}
        onChange={(e) => e.target.files && validateAndAddFiles(Array.from(e.target.files))}
        accept=".pdf"
        className="hidden"
      />
      <div
        onDragEnter={handleDrag} onDragOver={handleDrag}
        onDragLeave={handleDrag} onDrop={handleDrop}
        onClick={() => fileInputRef.current?.click()}
        className={`w-full min-h-[120px] border border-dashed rounded-xl p-4 bg-box-secondary cursor-pointer transition-all ${isDragActive ? "border-[#B30000] bg-red-50/40" : "border-foreground hover:border-gray-500"
          }`}
      >
        {files.length > 0 ? (
          <div className="space-y-2">
            {files.map((file, index) => (
              <div key={index} className="flex items-center justify-between text-sm font-medium text-gray-800" onClick={(e) => e.stopPropagation()}>
                <div className="flex items-center gap-2">
                  <FileText className="w-4 h-4 text-red-700 stroke-[2.5]" />
                  {file.name}
                </div>
                <button type="button" onClick={(e) => removeFile(index, e)} className="p-1 text-gray-400 hover:text-red-600 hover:bg-red-50 rounded-md transition-colors">
                  <X className="w-4 h-4" />
                </button>
              </div>
            ))}
          </div>
        ) : isEditing && submission?.files && submission.files.length > 0 ? (
          <div className="flex flex-col items-center justify-center h-full space-y-2 text-center py-6">
            <div className="space-y-0.5">
              {submission.files.map((f) => (
                <div key={f.id} className="flex items-center gap-2 text-[11px]">
                  <FileText className="w-4 h-4 text-red-700" />
                  {f.original_name}
                </div>
              ))}
            </div>
            <p className="text-[14px] text-gray-400 font-medium pointer-events-none">
              Upload ulang files yang akan diganti ke dalam submissions, atau klik batal
            </p>
          </div>
        ) : (
          <div className="flex flex-col items-center justify-center h-full space-y-2 pointer-events-none text-center py-6">
            <UploadCloud className="w-8 h-8 text-gray-400 mx-auto stroke-[1.5]" />
            <p className="text-sm font-bold text-gray-700">Click to upload <span className="font-medium text-gray-500">or drag & drop</span></p>
            <p className="text-[11px] text-gray-400 font-semibold">PDF, DOCX, or ZIP (Max.5MB)</p>
          </div>
        )}
      </div>

      <div className="flex gap-2">
        {isEditing && (
          <button
            type="button"
            onClick={handleCancelEdit}
            className="w-full bg-white border border-gray-300 hover:bg-gray-50 text-gray-700 font-bold py-3 rounded-xl transition-colors"
          >
            Batal
          </button>
        )}
        <button
          type="submit" disabled={loading || !isProjectActive}
          className="w-full bg-[#B30000] hover:bg-[#990000] disabled:opacity-60 text-white font-bold py-3 rounded-xl transition-colors shadow-md flex items-center justify-center gap-2 tracking-wide cursor-pointer disabled:cursor-not-allowed disabled:hover:bg-[#990000]"
        >
          {loading ? "Mengirim..." : isEditing ? "Update Submission" : "Save Changes"}
          <LuSendHorizontal className="w-4 h-4 stroke-[2.5]" />
        </button>
      </div>
    </form>
  );
}