import { AlertDialog, AlertDialogAction, AlertDialogCancel, AlertDialogContent, AlertDialogDescription, AlertDialogFooter, AlertDialogHeader, AlertDialogTitle, } from "../ui/alert-dialog"; import { DropdownMenu, DropdownMenuContent, DropdownMenuItem, DropdownMenuTrigger, } from "@/components/ui/dropdown-menu"; import { Tooltip, TooltipContent, TooltipTrigger } from "../ui/tooltip"; import { LuCopy, LuPencil } from "react-icons/lu"; import { FaDrawPolygon, FaObjectGroup } from "react-icons/fa"; import { BsPersonBoundingBox } from "react-icons/bs"; import { HiOutlineDotsVertical, HiTrash } from "react-icons/hi"; import { isDesktop, isMobile } from "react-device-detect"; import { flattenPoints, parseCoordinates, toRGBColorString, } from "@/utils/canvasUtil"; import { Polygon, PolygonType } from "@/types/canvas"; import { useCallback, useContext, useMemo, useState } from "react"; import axios from "axios"; import { Toaster } from "@/components/ui/sonner"; import { toast } from "sonner"; import useSWR from "swr"; import { FrigateConfig } from "@/types/frigateConfig"; import { reviewQueries } from "@/utils/zoneEdutUtil"; import IconWrapper from "../ui/icon-wrapper"; import { StatusBarMessagesContext } from "@/context/statusbar-provider"; import { buttonVariants } from "../ui/button"; import { Trans, useTranslation } from "react-i18next"; type PolygonItemProps = { polygon: Polygon; index: number; hoveredPolygonIndex: number | null; setHoveredPolygonIndex: (index: number | null) => void; setActivePolygonIndex: (index: number | undefined) => void; setEditPane: (type: PolygonType) => void; handleCopyCoordinates: (index: number) => void; }; export default function PolygonItem({ polygon, index, hoveredPolygonIndex, setHoveredPolygonIndex, setActivePolygonIndex, setEditPane, handleCopyCoordinates, }: PolygonItemProps) { const { t } = useTranslation("views/settings"); const { data: config, mutate: updateConfig } = useSWR("config"); const [deleteDialogOpen, setDeleteDialogOpen] = useState(false); const { addMessage } = useContext(StatusBarMessagesContext)!; const [isLoading, setIsLoading] = useState(false); const cameraConfig = useMemo(() => { if (polygon?.camera && config) { return config.cameras[polygon.camera]; } }, [polygon, config]); const polygonTypeIcons = { zone: FaDrawPolygon, motion_mask: FaObjectGroup, object_mask: BsPersonBoundingBox, }; const PolygonItemIcon = polygon ? polygonTypeIcons[polygon.type] : undefined; const saveToConfig = useCallback( async (polygon: Polygon) => { if (!polygon || !cameraConfig) { return; } let url = ""; if (polygon.type == "zone") { const { alertQueries, detectionQueries } = reviewQueries( polygon.name, false, false, polygon.camera, cameraConfig?.review.alerts.required_zones || [], cameraConfig?.review.detections.required_zones || [], ); url = `cameras.${polygon.camera}.zones.${polygon.name}${alertQueries}${detectionQueries}`; } if (polygon.type == "motion_mask") { const filteredMask = ( Array.isArray(cameraConfig.motion.mask) ? cameraConfig.motion.mask : [cameraConfig.motion.mask] ).filter((_, currentIndex) => currentIndex !== polygon.typeIndex); url = filteredMask .map((pointsArray) => { const coordinates = flattenPoints( parseCoordinates(pointsArray), ).join(","); return `cameras.${polygon?.camera}.motion.mask=${coordinates}&`; }) .join(""); if (!url) { // deleting last mask url = `cameras.${polygon?.camera}.motion.mask&`; } } if (polygon.type == "object_mask") { let configObject; let globalMask = false; // global mask on camera for all objects if (!polygon.objects.length) { configObject = cameraConfig.objects.mask; globalMask = true; } else { configObject = cameraConfig.objects.filters[polygon.objects[0]].mask; } if (!configObject) { return; } const globalObjectMasksArray = Array.isArray(cameraConfig.objects.mask) ? cameraConfig.objects.mask : cameraConfig.objects.mask ? [cameraConfig.objects.mask] : []; let filteredMask; if (globalMask) { filteredMask = ( Array.isArray(configObject) ? configObject : [configObject] ).filter((_, currentIndex) => currentIndex !== polygon.typeIndex); } else { filteredMask = ( Array.isArray(configObject) ? configObject : [configObject] ) .filter((mask) => !globalObjectMasksArray.includes(mask)) .filter((_, currentIndex) => currentIndex !== polygon.typeIndex); } url = filteredMask .map((pointsArray) => { const coordinates = flattenPoints( parseCoordinates(pointsArray), ).join(","); return globalMask ? `cameras.${polygon?.camera}.objects.mask=${coordinates}&` : `cameras.${polygon?.camera}.objects.filters.${polygon.objects[0]}.mask=${coordinates}&`; }) .join(""); if (!url) { // deleting last mask url = globalMask ? `cameras.${polygon?.camera}.objects.mask&` : `cameras.${polygon?.camera}.objects.filters.${polygon.objects[0]}.mask`; } } setIsLoading(true); await axios .put(`config/set?${url}`, { requires_restart: 0 }) .then((res) => { if (res.status === 200) { toast.success( t("masksAndZones.form.polygonDrawing.delete.success", { name: polygon?.name, }), { position: "top-center", }, ); updateConfig(); } else { toast.error( t("toast.save.error.title", { ns: "common", errorMessage: res.statusText, }), { position: "top-center", }, ); } }) .catch((error) => { const errorMessage = error.response?.data?.message || error.response?.data?.detail || "Unknown error"; toast.error( t("toast.save.error.title", { errorMessage, ns: "common" }), { position: "top-center", }, ); }) .finally(() => { setIsLoading(false); }); }, [updateConfig, cameraConfig, t], ); const handleDelete = () => { setActivePolygonIndex(undefined); saveToConfig(polygon); addMessage( "masks_zones", t("masksAndZones.restart_required"), undefined, "masks_zones", ); }; return ( <>
setHoveredPolygonIndex(index)} onMouseLeave={() => setHoveredPolygonIndex(null)} style={{ backgroundColor: hoveredPolygonIndex === index ? toRGBColorString(polygon.color, false) : "", }} >
{PolygonItemIcon && ( )}

{polygon.name}

setDeleteDialogOpen(!deleteDialogOpen)} > {t("masksAndZones.form.polygonDrawing.delete.title")} masksAndZones.form.polygonDrawing.delete.desc {t("button.cancel", { ns: "common" })} {t("button.delete", { ns: "common" })} {isMobile && ( <> { setActivePolygonIndex(index); setEditPane(polygon.type); }} > {t("button.edit", { ns: "common" })} handleCopyCoordinates(index)} > {t("button.copy", { ns: "common" })} setDeleteDialogOpen(true)} > {t("button.delete", { ns: "common" })} )} {!isMobile && hoveredPolygonIndex === index && (
{ setActivePolygonIndex(index); setEditPane(polygon.type); }} /> {t("button.edit", { ns: "common" })} handleCopyCoordinates(index)} /> {t("button.copyCoordinates", { ns: "common" })} !isLoading && setDeleteDialogOpen(true)} /> {t("button.delete", { ns: "common" })}
)}
); }