265 lines
17 KiB
TypeScript
265 lines
17 KiB
TypeScript
"use client"
|
||
|
||
import { useCallback, useEffect, useRef, useState } from "react"
|
||
import { can } from "@/lib/auth"
|
||
import { emptyNetflowDraft, netflowDraftFromSettings, netflowDraftPatch, type NetflowDraft } from "@/shared/api/netflow-settings-form"
|
||
import { FormField, FormToggle } from "@/components/form-kit"
|
||
import { Alert, AlertDescription, AlertTitle } from "@/components/reui/alert"
|
||
import { Badge } from "@/components/reui/badge"
|
||
import { OpsPanel } from "@/components/ops-panel"
|
||
import { Button } from "@/components/ui/button"
|
||
import { Input } from "@/components/ui/input"
|
||
import { CodeExportSheet, type CodeExportFormat } from "@/components/reui-kit/code-export-sheet"
|
||
import type { TrafficFlowSettingsDto } from "@mmapp/contracts/traffic-flow"
|
||
import { GeoipSettingsSection } from "./geoip-settings-section"
|
||
import {
|
||
generateTrafficFlowKeys,
|
||
getTrafficFlowHostFiles,
|
||
getTrafficFlowSettings,
|
||
purgeTrafficFlowData,
|
||
putTrafficFlowSettings,
|
||
} from "@/shared/api/traffic-flow"
|
||
import { formatFlowPurgeResult, NetflowPurgeConfirm } from "@/components/traffic/netflow-purge-dialog"
|
||
import { KeyRoundIcon, DownloadIcon, InfoIcon, AlertCircleIcon } from "lucide-react"
|
||
|
||
const HOST_STEPS = [
|
||
"На хосте Docker (не в контейнере mmapp-backend): apt install wireguard (или эквивалент).",
|
||
"Скачайте wg-flow.conf и положите в /etc/wireguard/wg-flow.conf.",
|
||
"wg-quick up wg-flow (или systemctl enable --now wg-quick@wg-flow).",
|
||
"Firewall: разрешите UDP listen WireGuard. UDP 4739 наружу не открывайте.",
|
||
"В docker-compose у backend раскомментируйте bind IPFIX только на адресе wg-flow.",
|
||
"Проверка: wg show · ss -ulnp | grep 4739 · в этой панели — last datagram.",
|
||
]
|
||
|
||
function NetflowSettingsPanel(props: {backendUrl:string;enabled:boolean}) {
|
||
return <NetflowSettingsContent key={JSON.stringify([props.backendUrl,props.enabled])} {...props}/>
|
||
}
|
||
function NetflowSettingsContent({
|
||
backendUrl,
|
||
enabled,
|
||
}: {
|
||
backendUrl: string
|
||
enabled: boolean
|
||
}) {
|
||
const [settings,setSettings] = useState<TrafficFlowSettingsDto|null>(null)
|
||
const [draft,setDraft] = useState<NetflowDraft>(emptyNetflowDraft)
|
||
const dirty = useRef(new Set<keyof NetflowDraft>())
|
||
const [dirtyCount,setDirtyCount] = useState(0)
|
||
const [receivedAt,setReceivedAt] = useState<string|null>(null)
|
||
const [readError,setReadError] = useState<string|null>(null)
|
||
const [opError,setOpError] = useState<string|null>(null)
|
||
const [outcome,setOutcome] = useState<string|null>(null)
|
||
const [needsReview,setNeedsReview] = useState(false)
|
||
const [busy,setBusy] = useState(false)
|
||
const [readBusy,setReadBusy] = useState(false)
|
||
const [exportOpen,setExportOpen] = useState(false)
|
||
const [formats,setFormats] = useState<CodeExportFormat[]>([])
|
||
const [purgeOpen,setPurgeOpen] = useState(false)
|
||
const lock = useRef(false)
|
||
const reader = useRef<{active:boolean;revision:number;controller?:AbortController;operation?:AbortController}>({active:true,revision:0})
|
||
useEffect(()=>{const state=reader.current;state.active=true;return()=>{state.active=false;state.revision++;state.controller?.abort();state.operation?.abort()}},[])
|
||
const accept = useCallback((data:TrafficFlowSettingsDto)=>{
|
||
setSettings(data);setReceivedAt(new Date().toISOString());setReadError(null)
|
||
const fresh=netflowDraftFromSettings(data)
|
||
setDraft(current=>{const next={...fresh};for(const field of dirty.current)Object.assign(next,{[field]:current[field]});return next})
|
||
},[])
|
||
const load = useCallback(async()=>{
|
||
if(!enabled||!can("mm:traffic:read")||!reader.current.active||lock.current)return
|
||
const revision=++reader.current.revision;reader.current.controller?.abort()
|
||
const controller=new AbortController();reader.current.controller=controller
|
||
const current=()=>reader.current.active&&reader.current.revision===revision&&!controller.signal.aborted
|
||
setReadBusy(true)
|
||
try{const data=await getTrafficFlowSettings(backendUrl,{signal:controller.signal});if(current())accept(data)}
|
||
catch(error){if(current())setReadError(error instanceof Error?error.message:"Не удалось загрузить NetFlow")}
|
||
finally{if(current())setReadBusy(false)}
|
||
},[accept,backendUrl,enabled])
|
||
useEffect(()=>{queueMicrotask(()=>{void load()})},[load])
|
||
const writable=enabled&&can("mm:traffic:write")&&Boolean(settings)&&!readError
|
||
const disabled=!writable||busy
|
||
function change<K extends keyof NetflowDraft>(field:K,value:NetflowDraft[K]){dirty.current.add(field);setDirtyCount(dirty.current.size);setDraft(current=>({...current,[field]:value}))}
|
||
function begin(){
|
||
if(!reader.current.active||lock.current||!writable)return null
|
||
lock.current=true;reader.current.revision++;reader.current.controller?.abort();setReadBusy(false);setBusy(true);setOpError(null);setOutcome(null)
|
||
const controller=new AbortController();reader.current.operation=controller;return controller
|
||
}
|
||
function finish(){lock.current=false;if(reader.current.active)setBusy(false)}
|
||
const message=(error:unknown)=>error instanceof Error?error.message:"Не удалось выполнить операцию NetFlow"
|
||
async function handleSave(){
|
||
if(!dirty.current.size)return
|
||
let patch;try{patch=netflowDraftPatch(draft,dirty.current)}catch(error){setOpError(message(error));return}
|
||
const controller=begin();if(!controller)return
|
||
try{const data=await putTrafficFlowSettings(backendUrl,patch,{signal:controller.signal});if(!reader.current.active)return
|
||
dirty.current.clear();setDirtyCount(0);accept(data.settings);setOutcome("Сохранение настроек NetFlow подтверждено")
|
||
if(data.listenerError!==undefined)setOpError("Настройки сохранены; запуск приёма требует внимания: "+data.listenerError)
|
||
}catch(error){if(reader.current.active)setOpError(message(error)+". Подтверждение сохранения не получено; черновик сохранён.")}
|
||
finally{finish()}
|
||
}
|
||
async function handleKeys(){
|
||
if(needsReview)return;const controller=begin();if(!controller)return
|
||
try{const data=await generateTrafficFlowKeys(backendUrl,{signal:controller.signal});if(!reader.current.active)return;accept(data.settings);setOutcome(data.created?"Создание ключей NetFlow подтверждено":"Ключи NetFlow уже созданы")}
|
||
catch(error){if(reader.current.active){setNeedsReview(true);setOpError(message(error)+". Результат операции неизвестен; проверьте статус и журнал перед повтором.")}}
|
||
finally{finish()}
|
||
}
|
||
async function handlePurgeConfirm(){
|
||
if(needsReview)return;const controller=begin();if(!controller)return
|
||
try{const data=await purgeTrafficFlowData(backendUrl,{signal:controller.signal});if(!reader.current.active)return
|
||
setPurgeOpen(false);setOutcome("Сброс данных NetFlow подтверждён. "+formatFlowPurgeResult(data))
|
||
const warnings=[data.partitionError===undefined?null:"Обслуживание таблиц: "+data.partitionError,data.listenerError===undefined?null:"Запуск приёма: "+data.listenerError,data.eventError===undefined?null:"Запись журнала: "+data.eventError].filter(Boolean)
|
||
if(warnings.length)setOpError(warnings.join("; "))
|
||
}catch(error){if(reader.current.active){setPurgeOpen(false);setNeedsReview(true);setOpError(message(error)+". Результат операции неизвестен; проверьте статус и журнал перед повтором.")}}
|
||
finally{finish()}
|
||
if(reader.current.active)await load()
|
||
}
|
||
async function handleExport(){
|
||
if(!settings?.hasHostPrivateKey)return;const controller=begin();if(!controller)return
|
||
try{const data=await getTrafficFlowHostFiles(backendUrl,{signal:controller.signal});if(!reader.current.active)return;setFormats(data.files);setExportOpen(true)}
|
||
catch(error){if(reader.current.active)setOpError(message(error))}
|
||
finally{finish()}
|
||
}
|
||
|
||
return (
|
||
<>
|
||
<OpsPanel
|
||
title="Traffic Flow / NetFlow (IPFIX)"
|
||
description="Дополнение к сбору счётчиков REST. Приём только через WireGuard на хосте Docker MM."
|
||
contentClassName="px-5 py-4 flex flex-col gap-4"
|
||
>
|
||
<div className="flex flex-wrap gap-2"><Badge variant={settings?.listenerBound ? "success" : "secondary"}>{settings ? settings.listenerBound ? "Приём UDP: " + settings.listenerAddress : "Приём UDP не готов" : "Приём UDP: —"}</Badge></div>
|
||
<p role="status" className="text-xs text-muted-foreground">{!can("mm:traffic:read") ? "Нет доступа к данным NetFlow" : readBusy ? "Загрузка NetFlow…" : settings ? "Настройки NetFlow получены: " + new Date(receivedAt!).toLocaleString("ru-RU") : "Настройки NetFlow не получены"}</p>
|
||
{readError ? <Alert variant="warning"><AlertCircleIcon aria-hidden="true"/><AlertTitle>{settings ? "NetFlow не обновлён" : "NetFlow не получен"}</AlertTitle><AlertDescription>{readError}{settings ? ". Показаны последние подтверждённые данные: " + new Date(receivedAt!).toLocaleString("ru-RU") : ""}</AlertDescription></Alert> : null}
|
||
{opError ? <Alert variant="destructive"><AlertCircleIcon aria-hidden="true"/><AlertTitle>Операция NetFlow требует внимания</AlertTitle><AlertDescription>{opError}</AlertDescription></Alert> : null}
|
||
{outcome ? <p role="status" className="text-sm">{outcome}</p> : null}
|
||
{needsReview ? <div className="flex flex-col gap-2"><p className="text-xs text-muted-foreground">Проверьте журнал и состояние NetFlow. Повтор запустит новую операцию.</p><Button size="sm" variant="outline" aria-label="Разрешить новую операцию NetFlow" disabled={disabled || readBusy} onClick={() => { setNeedsReview(false);setOpError(null) }}>Разрешить повтор NetFlow</Button></div> : null}
|
||
<Alert>
|
||
<InfoIcon />
|
||
<AlertTitle>Ключи и UDP 4739</AlertTitle>
|
||
<AlertDescription>
|
||
Приватный ключ включён в экспорт. Файлы отражают сохранённые настройки. Порт IPFIX публикуйте только на адресе wg-flow, не на 0.0.0.0.
|
||
</AlertDescription>
|
||
</Alert>
|
||
|
||
<div className="flex items-center gap-3">
|
||
<FormToggle ariaLabel="Принимать IPFIX" disabled={disabled} checked={draft.enabled} onChange={value => change("enabled",value)} />
|
||
<span className="text-sm">Принимать IPFIX{settings ? "" : ": —"}</span>
|
||
</div>
|
||
|
||
<div className="grid grid-cols-1 sm:grid-cols-2 gap-3">
|
||
<FormField label="Collector IP" hint="Адрес в туннеле, куда JH шлёт flow">
|
||
<Input className="font-mono" aria-label="Collector IP" disabled={disabled} value={draft.collectorIp} onChange={(e) => change("collectorIp", e.target.value)} />
|
||
</FormField>
|
||
<FormField label="Префикс overlay">
|
||
<Input className="font-mono" aria-label="Префикс overlay" disabled={disabled} value={draft.prefix} onChange={(e) => change("prefix", e.target.value)} />
|
||
</FormField>
|
||
<FormField label="UDP IPFIX">
|
||
<Input className="font-mono" aria-label="UDP IPFIX" disabled={disabled} value={draft.flowListenPort} onChange={(e) => change("flowListenPort", e.target.value)} />
|
||
</FormField>
|
||
<FormField label="WG listen">
|
||
<Input className="font-mono" aria-label="WG listen" disabled={disabled} value={draft.wgListenPort} onChange={(e) => change("wgListenPort", e.target.value)} />
|
||
</FormField>
|
||
<FormField label="Публичный endpoint хоста MM" hint="IP или DNS, который видят JH">
|
||
<Input className="font-mono" aria-label="Публичный endpoint хоста MM" disabled={disabled} value={draft.publicEndpoint} onChange={(e) => change("publicEndpoint", e.target.value)} placeholder="203.0.113.10" />
|
||
</FormField>
|
||
<FormField label="Public key хоста">
|
||
<Input className="font-mono text-xs" aria-label="Public key хоста" readOnly value={settings ? settings.hostPublicKey || "Ключи не созданы" : "—"} />
|
||
</FormField>
|
||
<FormField label="Хранение (часов)">
|
||
<Input aria-label="Хранение (часов)" disabled={disabled} value={draft.retentionHours} onChange={(e) => change("retentionHours", e.target.value)} inputMode="numeric" />
|
||
</FormField>
|
||
<FormField label="Top-N разговоров">
|
||
<Input aria-label="Top-N разговоров" disabled={disabled} value={draft.topN} onChange={(e) => change("topN", e.target.value)} inputMode="numeric" />
|
||
</FormField>
|
||
<div className="sm:col-span-2 flex flex-col gap-2">
|
||
<div className="flex items-center gap-3">
|
||
<FormToggle
|
||
ariaLabel="Порог доли на карте" disabled={disabled} checked={draft.shareOn}
|
||
onChange={value => change("shareOn",value)}
|
||
/>
|
||
<span className="text-sm">Порог доли на карте{settings ? "" : ": —"}</span>
|
||
</div>
|
||
<FormField
|
||
label="Минимум % окна"
|
||
hint="Узел сервиса, если доля байт окна ≥ N%. Выключить — показать все распознанные бренды (макс. 20)"
|
||
>
|
||
<Input
|
||
aria-label="Минимум % окна" value={draft.mapServiceMinSharePct}
|
||
onChange={(e) => change("mapServiceMinSharePct", e.target.value)}
|
||
inputMode="decimal"
|
||
disabled={disabled || !draft.shareOn}
|
||
/>
|
||
</FormField>
|
||
</div>
|
||
</div>
|
||
|
||
<p className="text-xs text-muted-foreground break-words">
|
||
Последний пакет:{" "}
|
||
{settings?.lastDatagramAt
|
||
? new Date(settings.lastDatagramAt).toLocaleString("ru-RU")
|
||
: settings ? "Не получен" : "—"}
|
||
{settings?.lastExporterIp ? ` · ${settings.lastExporterIp}` : ""}
|
||
{settings?.lastError ? ` · ${settings.lastError}` : ""}
|
||
</p>
|
||
|
||
<div className="rounded-md border px-4 py-3 flex flex-col gap-2">
|
||
<p className="text-sm font-medium">Туннель на сервере Docker MM</p>
|
||
<ol className="text-xs text-muted-foreground flex flex-col gap-1.5 list-decimal pl-4">
|
||
{HOST_STEPS.map((s) => (
|
||
<li key={s}>{s}</li>
|
||
))}
|
||
</ol>
|
||
</div>
|
||
|
||
<div className="flex flex-wrap gap-2">
|
||
<Button size="sm" disabled={disabled || !dirtyCount} onClick={() => { void handleSave() }}>
|
||
Сохранить NetFlow
|
||
</Button>
|
||
<Button size="sm" variant="outline" disabled={disabled || needsReview} onClick={() => { void handleKeys() }}>
|
||
<KeyRoundIcon className="size-4" />
|
||
Ключи хоста
|
||
</Button>
|
||
<Button size="sm" variant="outline" disabled={disabled || !settings?.hasHostPrivateKey} aria-label="Скачать файлы хоста NetFlow" onClick={() => { void handleExport() }}>
|
||
<DownloadIcon className="size-4" />
|
||
Файлы хоста
|
||
</Button>
|
||
<Button size="sm" variant="outline" disabled={busy || readBusy || !enabled || !can("mm:traffic:read")} onClick={() => { void load() }}>Обновить статус NetFlow</Button>
|
||
</div>
|
||
|
||
<div className="flex flex-col sm:flex-row sm:items-center justify-between gap-3 rounded-md border border-destructive/30 px-4 py-3">
|
||
<div className="min-w-0">
|
||
<p className="text-sm font-medium">Сбросить данные NetFlow</p>
|
||
<p className="text-xs text-muted-foreground mt-0.5">
|
||
Удалит сессии и агрегаты в PostgreSQL. Ключи WG и пиры не трогает.
|
||
</p>
|
||
</div>
|
||
<Button
|
||
size="sm"
|
||
variant="destructive"
|
||
disabled={disabled || needsReview}
|
||
onClick={() => setPurgeOpen(true)}
|
||
>
|
||
Сбросить
|
||
</Button>
|
||
</div>
|
||
</OpsPanel>
|
||
|
||
<GeoipSettingsSection backendUrl={backendUrl} />
|
||
|
||
<CodeExportSheet
|
||
open={exportOpen}
|
||
onClose={() => { setExportOpen(false);setFormats([]) }}
|
||
title="Файлы для хоста Docker MM"
|
||
description="Сохранённые настройки; экспорт содержит приватный ключ."
|
||
formats={formats}
|
||
/>
|
||
|
||
<NetflowPurgeConfirm
|
||
open={purgeOpen}
|
||
busy={busy}
|
||
onConfirm={() => { void handlePurgeConfirm() }}
|
||
onCancel={() => { if (!busy) setPurgeOpen(false) }}
|
||
/>
|
||
</>
|
||
)
|
||
}
|
||
|
||
export { NetflowSettingsPanel }
|