MikrotikManager/backend/src/services/bgp-peers-live.ts

import { eq } from "drizzle-orm"
import { db } from "../db/index.js"
import { servers } from "../db/schema.js"
import type { BgpSessionRead } from "../types/server.js"
import { parseBgpSessions } from "./bgp-parse-sessions.js"
import { MikrotikClient } from "./mikrotik.js"

/** Как в UI `/alerts`: `serverName: peerName`. */
export function bgpPeerAlertKey(s: BgpSessionRead): string {
  const sn = String(s.serverName ?? "").trim()
  const nm = String(s.name ?? "").trim()
  return `${sn}: ${nm}`.trim()
}

/** Агрегат BGP-сессий с включённых серверов — как `/api/bgp/sessions`. */
export async function fetchBgpSessionsForAlerts(): Promise<BgpSessionRead[]> {
  const allServers = await db.select().from(servers).where(eq(servers.enabled, true))
  const results = await Promise.all(
    allServers.map(async (server) => {
      try {
        const client = MikrotikClient.fromServer(server)
        const raw = await client.getBgpSessions()
        return parseBgpSessions(server, raw)
      } catch {
        return [] as BgpSessionRead[]
      }
    }),
  )
  return results.flat()
}

export async function countBgpSessions(): Promise<number> {
  try {
    const result = await Promise.race([
      fetchBgpSessionsForAlerts(),
      new Promise<null>((resolve) => setTimeout(() => resolve(null), 8_000)),
    ])
    if (!result) return 0
    return result.length
  } catch {
    return 0
  }
}