import type { ColumnDef } from '@tanstack/react-table';
import { Key, Pencil, Trash2 } from 'lucide-react';
import { GenderBadge } from '@/components/gender-badge';
import { RowActions } from '@/components/row-actions';
import { StudentStatusBadge } from '@/components/student-status-badge';

export type Student = {
    id: number;
    username: string;
    email: string;
    profile: { full_name: string; phone_number: string; gender: string } | null;
    student: {
        student_number: string;
        department: { name: string } | null;
        enrollment_year: number;
        status: string | null;
    } | null;
};

type CreateColumnsParams = {
    handleEdit: (student: Student) => void;
    handleDeleteClick: (student: Student) => void;
    handleResetPassword: (student: Student) => void;
};

export function createStudentColumns(
    params: CreateColumnsParams,
): ColumnDef<Student>[] {
    const { handleEdit, handleDeleteClick, handleResetPassword } = params;

    return [
        {
            id: 'identity',
            header: () => <span>NIM</span>,
            cell: ({ row }) => (
                <div className="flex flex-col">
                    <span>{row.original.student?.student_number ?? '-'}</span>
                    <span className="text-sm text-muted-foreground">
                        {row.original.profile?.full_name ?? '-'}
                    </span>
                </div>
            ),
        },
        {
            id: 'account',
            header: () => <span>Akun</span>,
            cell: ({ row }) => (
                <div className="flex flex-col">
                    <span>{row.original.username}</span>
                    <span className="text-sm text-muted-foreground">
                        {row.original.email}
                    </span>
                </div>
            ),
        },
        {
            accessorKey: 'student.department.name',
            header: () => <span>Jurusan</span>,
            cell: ({ row }) => row.original.student?.department?.name ?? '-',
        },
        {
            accessorKey: 'student.enrollment_year',
            header: () => <span>Angkatan</span>,
            cell: ({ row }) => row.original.student?.enrollment_year ?? '-',
        },
        {
            accessorKey: 'student.status',
            header: () => <span>Status</span>,
            cell: ({ row }) => (
                <StudentStatusBadge status={row.original.student?.status} />
            ),
        },
        {
            accessorKey: 'profile.phone_number',
            header: () => <span>Nomor Telepon</span>,
            cell: ({ row }) => row.original.profile?.phone_number ?? '-',
        },
        {
            accessorKey: 'profile.gender',
            header: () => <span>Jenis Kelamin</span>,
            cell: ({ row }) => (
                <GenderBadge gender={row.original.profile?.gender} />
            ),
        },
        {
            id: 'actions',
            header: () => <span className="block text-center">Aksi</span>,
            meta: {
                className: 'w-[100px] text-center',
                headerClassName: 'w-[100px] text-center',
            },
            cell: ({ row }) => {
                const student = row.original;

                return (
                    <RowActions
                        actions={[
                            {
                                label: 'Edit',
                                icon: <Pencil className="h-4 w-4" />,
                                onClick: () => handleEdit(student),
                            },
                            {
                                label: 'Reset Password',
                                icon: <Key className="h-4 w-4" />,
                                onClick: () => handleResetPassword(student),
                            },
                            {
                                label: 'Hapus',
                                icon: (
                                    <Trash2 className="h-4 w-4 text-destructive" />
                                ),
                                onClick: () => handleDeleteClick(student),
                            },
                        ]}
                    />
                );
            },
        },
    ];
}
