"use client"

import { Button } from "@/components/ui/button"
import { AppSidebar } from "@/components/app-sidebar"
import { SiteHeader } from "@/components/site-header"
import { SidebarInset, SidebarProvider } from "@/components/ui/sidebar"
import Link from "next/link"
import { useEffect, useState } from "react"
import { useTranslation } from "react-i18next"
import { DataTableContracts } from "@/components/data-table-contracts"
import { ColumnDef } from "@tanstack/react-table"
import { Badge } from "@/components/ui/badge"
import { IconCircleCheckFilled, IconLoader, IconEye } from "@tabler/icons-react"
import { Input } from "@/components/ui/input"
import { Combobox } from "@/components/combobox"

type ContractType = 
  | "CTR_RO"
  | "CTR_ENG"
  | "CTR_BLINGV"
  | "CTR_EUROCONSULT_ENG"
  | "CTR_MAGIC_ENG"
  | "CTR_AGROCHIRNOGI_RO"
  | "CTR_AGROCHIRNOGI_BGV"
  | "CTR_ANDERSONS_RO"
  | "CTR_BUNGE_RO"
  | "CTR_FINANTARE_ACHIZITIE_AAYLEX_RO"
  | "CTR_FINANTARE_VANZARE_AAYLEX_RO"
  | "CTR_GOZURI_GGI_RO"
  | "CTR_GGI_RO"
  | "CTR_GGI_ENG"

interface Draft {
  id: number
  draftNumber: string
  company: string
  status: string
  type: ContractType
  createdDate: string
  lastModified: string
  createdBy: string
  goods: string
  quantity: number
  price: number
  currency: string
  paymentTerms: string
  deliveryTerms: string
  notes: string
}

export default function DraftsListPage() {
  const [drafts, setDrafts] = useState<Draft[]>([])
  const [searchQuery, setSearchQuery] = useState("")
  const { t } = useTranslation()
  const [selectedCompany, setSelectedCompany] = useState("all")

  useEffect(() => {
    fetch("/contracts/api/drafts")
      .then(res => res.json())
      .then(setDrafts)
  }, [])

  const companies = [...new Set(drafts.map(draft => draft.company))];

  const filteredDrafts = drafts.filter(draft => {
    const matchesSearch = 
      draft.draftNumber.toLowerCase().includes(searchQuery.toLowerCase()) ||
      draft.company.toLowerCase().includes(searchQuery.toLowerCase()) ||
      draft.type.toLowerCase().includes(searchQuery.toLowerCase());

    const matchesCompany = selectedCompany === "all" || draft.company === selectedCompany; // Check for selected company

    return matchesSearch && matchesCompany; // Combine filters
  });

  const columns: ColumnDef<Draft>[] = [
    {
      accessorKey: "draftNumber",
      header: t("drafts.draftnumber"),
    },
    {
      accessorKey: "company",
      header: t("drafts.company"),
    },
    {
      accessorKey: "status",
      header: t("drafts.status"),
      cell: ({ row }) => {
        const statusKey = row.original.status.replace(/\s+/g, '').toLowerCase();
        let icon = null;
        switch (statusKey) {
          case "active":
            icon = <IconCircleCheckFilled className="h-4 w-4 text-green-500" />;
            break;
          case "inprogress":
            icon = <IconLoader className="h-4 w-4 text-blue-500 animate-spin" />;
            break;
          case "underreview":
            icon = <IconLoader className="h-4 w-4 text-yellow-500 animate-spin" />;
            break;
          case "pendingapproval":
            icon = <IconLoader className="h-4 w-4 text-purple-500 animate-spin" />;
            break;
          default:
            icon = <IconLoader className="h-4 w-4 text-gray-500 animate-spin" />;
        }
        return (
          <Badge variant="outline" className="flex items-center gap-1">
            {icon}
            {t(`drafts.statusLabels.${statusKey}`)}
          </Badge>
        );
      },
    },
    {
      accessorKey: "type",
      header: t("drafts.type"),
      cell: ({ row }) => (
        <span>{t(`drafts.types.${row.original.type}`)}</span>
      ),
    },
    {
      accessorKey: "goods",
      header: t("drafts.goods"),
    },
    {
      accessorKey: "quantity",
      header: t("drafts.quantity"),
      cell: ({ row }) => `${row.original.quantity.toLocaleString()} MT`,
    },
    {
      accessorKey: "price",
      header: t("drafts.price"),
      cell: ({ row }) => `${row.original.currency} ${row.original.price.toLocaleString()}`,
    },
    {
      accessorKey: "createdDate",
      header: t("drafts.createddate"),
      cell: ({ row }) => (
        <span>{new Date(row.original.createdDate).toLocaleDateString()}</span>
      ),
    },
    {
      accessorKey: "lastModified",
      header: t("drafts.lastmodified"),
      cell: ({ row }) => (
        <span>{new Date(row.original.lastModified).toLocaleDateString()}</span>
      ),
    },
    {
      accessorKey: "createdBy",
      header: t("drafts.createdby"),
    },
    {
      id: "actions",
      header: t("drafts.action"),
      cell: ({ row }) => (
        <div className="flex">
          <Link href={`/contracts/drafts/${row.original.id}`}>
            <Button size="sm" variant="outline"><IconEye className="h-4 w-4" />{t("drafts.view")}</Button>
          </Link>
        </div>
      ),
    },
  ]

  const companyOptions = [
    { value: "all", label: "All Companies" },
    ...companies.map(company => ({ value: company, label: company }))
  ];

  return (
    <SidebarProvider
      style={{
        "--sidebar-width": "calc(var(--spacing) * 72)",
        "--header-height": "calc(var(--spacing) * 12)",
      } as React.CSSProperties}
    >
      <AppSidebar variant="inset" />
      <SidebarInset>
        <SiteHeader title={t("drafts.draftlist")} />
        <div className="flex flex-1 flex-col">
          <div className="@container/main flex flex-1 flex-col gap-2">
            <div className="flex flex-col gap-4 py-4 md:gap-6 md:py-6">
              <div className="flex justify-between items-center px-4 lg:px-6">
                <h2 className="text-2xl font-bold">{t("drafts.draftlist")}</h2>
                <Link href="/contracts/drafts/new">
                  <Button>{t("drafts.newdraft")}</Button>
                </Link>
              </div>
              <div className="px-4 lg:px-6">
                <div className="flex items-center gap-4 mb-4 justify-between">
                  <Input
                    placeholder={t("drafts.search")}
                    value={searchQuery}
                    onChange={(e) => setSearchQuery(e.target.value)}
                    className="max-w-sm"
                  />
                  <Combobox
                    items={companyOptions}
                    value={selectedCompany}
                    onValueChange={setSelectedCompany}
                    placeholder="Select company"
                    emptyText="No companies found"
                  />
                </div>
                <div className="rounded-md border">
                  <DataTableContracts 
                    data={filteredDrafts} 
                    columns={columns}
                  />
                </div>
              </div>
            </div>
          </div>
        </div>
      </SidebarInset>
    </SidebarProvider>
  )
}