'use client';

import { useState } from 'react';
import { useQuery, useMutation, useQueryClient } from '@tanstack/react-query';
import { ColumnDef } from '@tanstack/react-table';
import Link from 'next/link';
import { MoreHorizontal, FileText, Plus, Trash2 } from 'lucide-react';

import { documentsApi } from '@/lib/api';
import type { Document } from '@/types/models';
import { DataTable } from '@/components/data-table';
import { Button } from '@/components/ui/button';
import { Badge } from '@/components/ui/badge';
import {
  DropdownMenu,
  DropdownMenuContent,
  DropdownMenuItem,
  DropdownMenuLabel,
  DropdownMenuTrigger,
} from '@/components/ui/dropdown-menu';
import {
  AlertDialog,
  AlertDialogAction,
  AlertDialogCancel,
  AlertDialogContent,
  AlertDialogDescription,
  AlertDialogFooter,
  AlertDialogHeader,
  AlertDialogTitle,
} from '@/components/ui/alert-dialog';
import { toast } from 'sonner';

const documentTypeColors: Record<string, 'default' | 'secondary' | 'destructive' | 'outline'> = {
  warranty: 'default',
  receipt: 'secondary',
  contract: 'destructive',
  insurance: 'outline',
  manual: 'secondary',
  general: 'outline',
};

const columns: ColumnDef<Document>[] = [
  {
    accessorKey: 'id',
    header: 'ID',
    cell: ({ row }) => (
      <span className="font-mono text-sm text-muted-foreground">
        {row.original.id}
      </span>
    ),
  },
  {
    accessorKey: 'title',
    header: 'Title',
    cell: ({ row }) => (
      <div className="flex items-center gap-2">
        <FileText className="h-4 w-4 text-muted-foreground" />
        <Link
          href={`/documents/${row.original.id}`}
          className="font-medium text-blue-600 hover:underline"
        >
          {row.original.title}
        </Link>
      </div>
    ),
  },
  {
    accessorKey: 'residence_name',
    header: 'Residence',
  },
  {
    accessorKey: 'document_type',
    header: 'Type',
    cell: ({ row }) => (
      <Badge variant={documentTypeColors[row.original.document_type] || 'outline'}>
        {row.original.document_type}
      </Badge>
    ),
  },
  {
    accessorKey: 'vendor',
    header: 'Vendor',
    cell: ({ row }) => row.original.vendor || '-',
  },
  {
    accessorKey: 'expiry_date',
    header: 'Expires',
    cell: ({ row }) => {
      const expiry = row.original.expiry_date;
      if (!expiry) return '-';
      const date = new Date(expiry);
      const isExpired = date < new Date();
      return (
        <span className={isExpired ? 'text-red-500' : ''}>
          {date.toLocaleDateString()}
        </span>
      );
    },
  },
  {
    accessorKey: 'is_active',
    header: 'Status',
    cell: ({ row }) => (
      <Badge variant={row.original.is_active ? 'default' : 'secondary'}>
        {row.original.is_active ? 'Active' : 'Inactive'}
      </Badge>
    ),
  },
  {
    accessorKey: 'created_at',
    header: 'Created',
    cell: ({ row }) => new Date(row.original.created_at).toLocaleDateString(),
  },
  {
    id: 'actions',
    cell: ({ row }) => {
      const document = row.original;
      return (
        <DropdownMenu>
          <DropdownMenuTrigger asChild>
            <Button variant="ghost" className="h-8 w-8 p-0">
              <span className="sr-only">Open menu</span>
              <MoreHorizontal className="h-4 w-4" />
            </Button>
          </DropdownMenuTrigger>
          <DropdownMenuContent align="end">
            <DropdownMenuLabel>Actions</DropdownMenuLabel>
            <DropdownMenuItem asChild>
              <Link href={`/documents/${document.id}`}>View details</Link>
            </DropdownMenuItem>
            <DropdownMenuItem asChild>
              <Link href={`/documents/${document.id}/edit`}>Edit</Link>
            </DropdownMenuItem>
          </DropdownMenuContent>
        </DropdownMenu>
      );
    },
  },
];

export default function DocumentsPage() {
  const queryClient = useQueryClient();
  const [page, setPage] = useState(1);
  const [pageSize, setPageSize] = useState(20);
  const [search, setSearch] = useState('');
  const [selectedRows, setSelectedRows] = useState<Document[]>([]);
  const [showDeleteDialog, setShowDeleteDialog] = useState(false);

  const { data, isLoading } = useQuery({
    queryKey: ['documents', { page, per_page: pageSize, search }],
    queryFn: () => documentsApi.list({ page, per_page: pageSize, search: search || undefined }),
  });

  const bulkDeleteMutation = useMutation({
    mutationFn: (ids: number[]) => documentsApi.bulkDelete(ids),
    onSuccess: () => {
      queryClient.invalidateQueries({ queryKey: ['documents'] });
      setSelectedRows([]);
      setShowDeleteDialog(false);
      toast.success(`${selectedRows.length} document(s) deactivated`);
    },
    onError: () => {
      toast.error('Failed to deactivate documents');
    },
  });

  const handleBulkDelete = () => {
    const ids = selectedRows.map((d) => d.id);
    bulkDeleteMutation.mutate(ids);
  };

  return (
    <div className="space-y-4">
      <div className="flex items-center justify-between">
        <div>
          <h1 className="text-2xl font-bold">Documents</h1>
          <p className="text-muted-foreground">Manage documents, warranties, and receipts</p>
        </div>
        <div className="flex items-center gap-2">
          {selectedRows.length > 0 && (
            <Button
              variant="destructive"
              size="sm"
              onClick={() => setShowDeleteDialog(true)}
            >
              <Trash2 className="h-4 w-4 mr-2" />
              Deactivate ({selectedRows.length})
            </Button>
          )}
          <Button asChild>
            <Link href="/documents/new">
              <Plus className="h-4 w-4 mr-2" />
              New Document
            </Link>
          </Button>
        </div>
      </div>

      <DataTable
        columns={columns}
        data={data?.data ?? []}
        totalCount={data?.total ?? 0}
        page={page}
        pageSize={pageSize}
        onPageChange={setPage}
        onPageSizeChange={(size) => {
          setPageSize(size);
          setPage(1);
        }}
        searchValue={search}
        onSearchChange={(value) => {
          setSearch(value);
          setPage(1);
        }}
        searchPlaceholder="Search documents..."
        isLoading={isLoading}
        enableRowSelection
        onRowSelectionChange={setSelectedRows}
      />

      <AlertDialog open={showDeleteDialog} onOpenChange={setShowDeleteDialog}>
        <AlertDialogContent>
          <AlertDialogHeader>
            <AlertDialogTitle>Deactivate Documents</AlertDialogTitle>
            <AlertDialogDescription>
              Are you sure you want to deactivate {selectedRows.length} document(s)?
              This will mark them as inactive but not permanently delete them.
            </AlertDialogDescription>
          </AlertDialogHeader>
          <AlertDialogFooter>
            <AlertDialogCancel>Cancel</AlertDialogCancel>
            <AlertDialogAction
              onClick={handleBulkDelete}
              className="bg-red-600 hover:bg-red-700"
            >
              {bulkDeleteMutation.isPending ? 'Deactivating...' : 'Deactivate'}
            </AlertDialogAction>
          </AlertDialogFooter>
        </AlertDialogContent>
      </AlertDialog>
    </div>
  );
}
