import { useFormik } from "formik";
import * as Yup from "yup";
import { useState, useEffect } from "react";
import Modal from "../../components/Modals";
import Select from "../../components/Select";
import Datepicker from "../../components/Datepicker";

interface Props {
    open: boolean;
    onClose: () => void;
    onSave: (data: any) => Promise<void>; // Make it return a promise
    studentName?: string;
}

const AddExpense = ({ open, onClose, onSave, studentName }: Props) => {
    const [loading, setLoading] = useState(false);

    const validationSchema = Yup.object({
        expense_name: Yup.string().required("Expense name is required").min(3, "Minimum 3 characters"),
        category: Yup.string().required("Category is required"),
        amount: Yup.number()
            .required("Amount is required")
            .positive("Amount must be positive")
            .typeError("Amount must be a number"),
        start_date: Yup.string().required("Date is required"),
    });

    const formik = useFormik({
        initialValues: {
            expense_name: "",
            category: "",
            amount: "",
            start_date: "",
            description: "",
            student_name: studentName || "",
        },
        validationSchema,
        enableReinitialize: true,
        onSubmit: async (values, { resetForm }) => {
            setLoading(true);
            try {
                // Don't call onClose here - let the parent handle it
                await onSave({
                    ...values,
                    student_name: studentName,
                });
                resetForm();
                // Remove onClose() from here
            } catch (error) {
                console.error("Error saving expense:", error);
            } finally {
                setLoading(false);
            }
        },
    });

    // Auto-fill expense name with student name when provided
    useEffect(() => {
        if (studentName && studentName.trim() !== "" && open) {
            formik.setFieldValue("expense_name", studentName);
        }
    }, [studentName, open]);

    const CategoryOption = [
        { value: "Rent", label: "Rent" },
        { value: "Utilities", label: "Utilities" },
        { value: "Salaries", label: "Salaries" },
        { value: "Marketing", label: "Marketing" },
        { value: "Supplies", label: "Supplies" },
        { value: "Maintenance", label: "Maintenance" },
        { value: "Other", label: "Other" },
    ];

    return (
        <Modal
            open={open}
            onClose={onClose}
            title={studentName ? `Add Expense - ${studentName}` : "Add Expense"}
            maxWidth="max-w-2xl"
            footer={
                <>
                    <button
                        onClick={onClose}
                        className="btn btn-outline-danger"
                        type="button"
                        disabled={loading}
                    >
                        Cancel
                    </button>
                    <button
                        onClick={() => formik.handleSubmit()}
                        className="btn btn-primary"
                        type="button"
                        disabled={loading}
                    >
                        {loading ? 'Adding...' : 'Add Expense'}
                    </button>
                </>
            }
        >
            <form className="grid grid-cols-12 gap-4">
                {/* Show student name if provided */}
                {studentName && (
                    <div className="col-span-12">
                        <div className="p-3 bg-blue-50 rounded-md border border-blue-200">
                            <p className="text-sm text-blue-700">
                                Student: <span className="font-semibold">{studentName}</span>
                            </p>
                        </div>
                    </div>
                )}

                <div className="col-span-6">
                    <label className="block text-sm font-medium mb-1">Expense Name <span className="text-red-500">*</span></label>
                    <input
                        name="expense_name"
                        className="form-input w-full"
                        value={formik.values.expense_name}
                        onChange={formik.handleChange}
                        onBlur={formik.handleBlur}
                        placeholder="Enter expense name"
                    />
                    {formik.touched.expense_name && formik.errors.expense_name && (
                        <p className="text-red-500 text-xs mt-1">{formik.errors.expense_name}</p>
                    )}
                </div>

                <div className="col-span-6">
                    <label className="block text-sm font-medium mb-1">Category <span className="text-red-500">*</span></label>
                    <Select
                        options={CategoryOption}
                        value={CategoryOption.find(o => o.value === formik.values.category)}
                        onChange={(val: any) => {
                            formik.setFieldValue("category", val?.value || "");
                            setTimeout(() => {
                                formik.setFieldTouched("category", true);
                                formik.validateField("category");
                            }, 0);
                        }}
                        placeholder="Select category"
                    />
                    {formik.touched.category && formik.errors.category && (
                        <p className="text-red-500 text-xs mt-1">{formik.errors.category}</p>
                    )}
                </div>

                <div className="col-span-6">
                    <label className="block text-sm font-medium mb-1">Amount (₹) <span className="text-red-500">*</span></label>
                    <input
                        name="amount"
                        type="number"
                        className="form-input w-full"
                        value={formik.values.amount}
                        onChange={formik.handleChange}
                        onBlur={formik.handleBlur}
                        placeholder="Enter amount"
                    />
                    {formik.touched.amount && formik.errors.amount && (
                        <p className="text-red-500 text-xs mt-1">{formik.errors.amount}</p>
                    )}
                </div>

                <div className="col-span-6">
                    <label className="block text-sm font-medium mb-1">Date <span className="text-red-500">*</span></label>
                    <Datepicker
                        value={formik.values.start_date || undefined}
                        onChange={(val) => formik.setFieldValue("start_date", val)}
                        placeholder="Select date"
                    />
                    {formik.touched.start_date && formik.errors.start_date && (
                        <p className="text-red-500 text-xs mt-1">{formik.errors.start_date}</p>
                    )}
                </div>

                <div className="col-span-12">
                    <label className="block text-sm font-medium mb-1">Description</label>
                    <textarea
                        name="description"
                        rows={3}
                        className="form-textarea w-full"
                        value={formik.values.description}
                        onChange={formik.handleChange}
                        onBlur={formik.handleBlur}
                        placeholder="Enter description (optional)"
                    />
                </div>
            </form>
        </Modal>
    );
};

export default AddExpense;