MikrotikManager/components/data-grids/wireguard-data-grid.tsx

"use client"

import { useMemo } from "react"
import {
  type ColumnDef,
  getCoreRowModel,
  getExpandedRowModel,
  getSortedRowModel,
  useReactTable,
} from "@tanstack/react-table"
import type { WireGuardInterface } from "@/lib/data"
import { Flag } from "@/components/flag"
import { cn } from "@/lib/utils"
import { Button } from "@/components/ui/button"
import {
  DropdownMenu,
  DropdownMenuContent,
  DropdownMenuItem,
  DropdownMenuSeparator,
  DropdownMenuTrigger,
} from "@/components/ui/dropdown-menu"
import { DataGridShell } from "@/components/data-grids/shared/data-grid-shell"
import {
  DATA_GRID_CELL_PAD,
  DATA_GRID_CELL_PAD_FIRST,
  DATA_GRID_CELL_PAD_LAST,
} from "@/components/data-grids/shared/data-grid-layout"
import { DataGridSortHeader } from "@/components/data-grids/shared/data-grid-sort-header"
import { WireGuardPeersDetail } from "@/components/data-grids/wireguard-peers-detail"
import { EmptyState } from "@/components/empty-state"
import {
  ChevronDownIcon,
  ChevronRightIcon,
  CodeXmlIcon,
  MoreHorizontalIcon,
  PencilIcon,
  PlusIcon,
  PowerIcon,
  ShieldCheckIcon,
  Trash2Icon,
} from "lucide-react"

export interface WgIfaceWithServer extends WireGuardInterface {
  serverId: string
  serverName: string
  serverCountry: string
}

interface WireguardDataGridProps {
  interfaces: WgIfaceWithServer[]
  onExport: (iface: WgIfaceWithServer) => void
}

function WireguardDataGrid({ interfaces, onExport }: WireguardDataGridProps) {
  const columns = useMemo<ColumnDef<WgIfaceWithServer>[]>(
    () => [
      {
        id: "name",
        accessorKey: "name",
        header: ({ column }) => (
          <DataGridSortHeader column={column} title="Интерфейс / Сервер" className="ml-1" />
        ),
        cell: ({ row }) => {
          const iface = row.original
          const expanded = row.getIsExpanded()
          const onlinePeers = iface.peers.filter((p) => !!p.latestHandshake).length
          return (
            <div className="flex items-start gap-2 min-w-0">
              {expanded ? (
                <ChevronDownIcon className="size-3.5 mt-0.5 shrink-0 text-muted-foreground" />
              ) : (
                <ChevronRightIcon className="size-3.5 mt-0.5 shrink-0 text-muted-foreground/40" />
              )}
              <div className="min-w-0">
                <div className="flex items-center gap-2">
                  <span
                    className={cn(
                      "size-2 rounded-full shrink-0",
                      iface.status === "up" ? "bg-emerald-500 animate-pulse" : "bg-red-500",
                    )}
                  />
                  <span className="font-mono font-semibold text-sm">{iface.name}</span>
                </div>
                <div className="flex items-center gap-1.5 mt-0.5 text-[11px] text-muted-foreground font-mono">
                  <Flag code={iface.serverCountry} size={12} />
                  {iface.serverName}
                </div>
                <p className="sr-only">
                  {onlinePeers}/{iface.peers.length} пиров
                </p>
              </div>
            </div>
          )
        },
        meta: {
          headerTitle: "Интерфейс / Сервер",
          headerClassName: DATA_GRID_CELL_PAD_FIRST,
          cellClassName: DATA_GRID_CELL_PAD_FIRST,
          expandedContent: (row: WgIfaceWithServer) => (
            <WireGuardPeersDetail peers={row.peers} />
          ),
        },
      },
      {
        id: "listenPort",
        accessorKey: "listenPort",
        header: ({ column }) => <DataGridSortHeader column={column} title="Порт" />,
        cell: ({ row }) => (
          <span className="font-mono text-sm">{row.original.listenPort}</span>
        ),
        meta: {
          headerTitle: "Порт",
          headerClassName: DATA_GRID_CELL_PAD,
          cellClassName: cn(DATA_GRID_CELL_PAD, "text-center"),
        },
      },
      {
        id: "mtu",
        accessorKey: "mtu",
        header: ({ column }) => <DataGridSortHeader column={column} title="MTU" />,
        cell: ({ row }) => <span className="font-mono text-sm">{row.original.mtu}</span>,
        meta: {
          headerTitle: "MTU",
          headerClassName: DATA_GRID_CELL_PAD,
          cellClassName: cn(DATA_GRID_CELL_PAD, "text-center"),
        },
      },
      {
        id: "peers",
        header: () => (
          <span className="text-xs font-medium text-muted-foreground">Пиры</span>
        ),
        enableSorting: false,
        cell: ({ row }) => {
          const iface = row.original
          const onlinePeers = iface.peers.filter((p) => !!p.latestHandshake).length
          return (
            <span className="font-mono text-sm text-center block">
              <span className="text-emerald-600 dark:text-emerald-400">{onlinePeers}</span>
              <span className="text-muted-foreground">/{iface.peers.length}</span>
            </span>
          )
        },
        meta: {
          headerTitle: "Пиры",
          headerClassName: DATA_GRID_CELL_PAD,
          cellClassName: cn(DATA_GRID_CELL_PAD, "text-center"),
        },
      },
      {
        id: "status",
        accessorKey: "status",
        header: ({ column }) => <DataGridSortHeader column={column} title="Статус" />,
        cell: ({ row }) => (
          <span
            className={cn(
              "text-[11px] font-mono px-2 py-0.5 rounded border",
              row.original.status === "up"
                ? "bg-emerald-500/10 text-emerald-600 dark:text-emerald-400 border-emerald-500/20"
                : "bg-red-500/10 text-red-500 border-red-500/20",
            )}
          >
            {row.original.status === "up" ? "UP" : "DOWN"}
          </span>
        ),
        meta: {
          headerTitle: "Статус",
          headerClassName: DATA_GRID_CELL_PAD,
          cellClassName: DATA_GRID_CELL_PAD,
        },
      },
      {
        id: "actions",
        header: () => <span className="sr-only">Действия</span>,
        cell: ({ row }) => {
          const iface = row.original
          return (
            <div
              className="flex justify-end"
              onClick={(e) => e.stopPropagation()}
              onKeyDown={(e) => e.stopPropagation()}
            >
              <DropdownMenu>
                <DropdownMenuTrigger
                  render={
                    <Button
                      type="button"
                      variant="outline"
                      size="icon"
                      className={cn(
                        "size-8 shrink-0 border-border/60 bg-background/80 text-muted-foreground shadow-none",
                        "opacity-0 transition-[opacity,background-color,color,border-color]",
                        "group-hover/row:opacity-100 focus-visible:opacity-100",
                        "data-popup-open:opacity-100 data-popup-open:bg-muted",
                      )}
                      onClick={(e) => e.stopPropagation()}
                      aria-label={`Действия: ${iface.name}`}
                    >
                      <MoreHorizontalIcon className="size-4" />
                    </Button>
                  }
                />
                <DropdownMenuContent side="bottom" align="end">
                  <DropdownMenuItem onClick={() => onExport(iface)}>
                    <CodeXmlIcon className="size-4" />
                    Экспорт .rsc
                  </DropdownMenuItem>
                  <DropdownMenuItem>
                    <PencilIcon className="size-4" />
                    Редактировать
                  </DropdownMenuItem>
                  <DropdownMenuItem>
                    <PlusIcon className="size-4" />
                    Добавить пира
                  </DropdownMenuItem>
                  <DropdownMenuSeparator />
                  <DropdownMenuItem>
                    <PowerIcon className="size-4" />
                    {iface.enabled ? "Отключить" : "Включить"}
                  </DropdownMenuItem>
                  <DropdownMenuSeparator />
                  <DropdownMenuItem variant="destructive">
                    <Trash2Icon className="size-4" />
                    Удалить
                  </DropdownMenuItem>
                </DropdownMenuContent>
              </DropdownMenu>
            </div>
          )
        },
        enableSorting: false,
        size: 56,
        meta: {
          headerClassName: DATA_GRID_CELL_PAD_LAST,
          cellClassName: DATA_GRID_CELL_PAD_LAST,
        },
      },
    ],
    [onExport],
  )

  const table = useReactTable({
    data: interfaces,
    columns,
    getCoreRowModel: getCoreRowModel(),
    getSortedRowModel: getSortedRowModel(),
    getExpandedRowModel: getExpandedRowModel(),
    getRowId: (row) => row.id,
    getRowCanExpand: () => true,
  })

  if (interfaces.length === 0) {
    return (
      <EmptyState
        icon={<ShieldCheckIcon className="size-4" />}
        title="Нет WireGuard интерфейсов"
        description="Добавьте первый интерфейс или проверьте поиск"
        className="border-0 py-16"
      />
    )
  }

  return (
    <DataGridShell
      table={table}
      recordCount={interfaces.length}
      onRowClick={(row) => table.getRow(row.id).toggleExpanded()}
      tableClassNames={{
        headerRow: "border-b border-border",
        bodyRow: cn("group/row", "[&[data-disabled=true]]:opacity-50"),
      }}
    />
  )
}

export { WireguardDataGrid, type WireguardDataGridProps }