network-tools-server/index.js

const fs      = require("fs");
const path    = require("path");
const yaml    = require("js-yaml");
const express = require("express");
const http    = require("http");
const { Server } = require("socket.io");
const cors  = require("cors");
const axios = require("axios");

// ─── Config ───────────────────────────────────────────────────────────────────
const configPath = path.join(__dirname, "config.yml");
const config     = yaml.load(fs.readFileSync(configPath, "utf8"));

const PORT      = config.server?.port || 4000;
const HOSTS     = config.targets?.hosts     || [{ url: "http://127.0.0.1:7088", name: "my-server", metadata: {} }];
const ENDPOINTS = config.targets?.endpoints || [];
const TIMEOUT   = config.targets?.timeout   || 120_000;

// ─── Check name → endpoint suffix mapping ────────────────────────────────────
// Allows the client to pass ["ping", "curl", "tracert"] in any casing
const CHECK_MAP = {
  ping:       "/testmontools/ping",
  curl:       "/testmontools/curl",
  tracert:    "/testmontools/traceroute",
  traceroute: "/testmontools/traceroute",
};

/**
 * Resolve the list of endpoints to run.
 * @param {string[]} checks  - e.g. ["ping", "curl"] or [] (= all)
 * @returns {string[]} deduplicated endpoint paths present in ENDPOINTS
 */
function resolveEndpoints(checks) {
  if (!Array.isArray(checks) || checks.length === 0) {
    return ENDPOINTS; // run everything declared in config
  }

  const requested = new Set(
    checks
      .map((c) => CHECK_MAP[c.toLowerCase().trim()])
      .filter(Boolean)
  );

  // keep only endpoints that are both requested AND declared in config
  return ENDPOINTS.filter((ep) => requested.has(ep));
}

console.log(`[config] loaded from ${configPath}`);
console.log(`[config] hosts (${HOSTS.length}):`, HOSTS.map(h => `${h.name} (${h.url})`));
console.log(`[config] endpoints:`, ENDPOINTS);

// ─── Express + Socket.io ──────────────────────────────────────────────────────
const app    = express();
const server = http.createServer(app);
const io     = new Server(server, {
  cors: { origin: "*", methods: ["GET", "POST"] },
});

app.use(cors());
app.use(express.json());

// ─── POST /api/parallel-fetch ─────────────────────────────────────────────────
app.post("/api/parallel-fetch", async (req, res) => {
  const { socketId, url, timeout = TIMEOUT, checks = [] } = req.body;

  if (!socketId || !url) {
    return res.status(400).json({ error: "socketId and url are required" });
  }

  const socket = io.sockets.sockets.get(socketId);
  if (!socket) {
    return res.status(400).json({ error: `Socket ${socketId} not found` });
  }

  const activeEndpoints = resolveEndpoints(checks);

  if (activeEndpoints.length === 0) {
    return res.status(400).json({
      error: "No valid endpoints resolved. Check 'checks' values or config endpoints.",
      validCheckNames: Object.keys(CHECK_MAP),
    });
  }

  res.json({ status: "started", endpoints: activeEndpoints });

  const tasks = HOSTS.flatMap((host) =>
    activeEndpoints.map((endpoint) => ({ host, endpoint }))
  );

  socket.emit("batch:start", { url, total: tasks.length, endpoints: activeEndpoints });

  const promises = tasks.map(({ host, endpoint }) => {
    const fullUrl = host.url + endpoint;
    const start   = Date.now();

    return axios({
      method: "POST",
      url: fullUrl,
      headers: { "Content-Type": "application/json" },
      data: { url },
      timeout,
    })
      .then((response) => {
        const elapsed = Date.now() - start;
        console.log(`[OK]    ${host.name} ${endpoint} (${elapsed}ms)`);

        socket.emit("tool:response", {
          host:       host.url,
          serverName: host.name,
          metadata:   host.metadata ?? {},   // ← метаданные хоста
          endpoint,
          ok:         true,
          status:     response.status,
          data:       response.data,
          elapsed,
        });
      })
      .catch((err) => {
        const elapsed = Date.now() - start;
        console.log(`[ERROR] ${host.name} ${endpoint} (${elapsed}ms) — ${err.message}`);

        socket.emit("tool:error", {
          host:       host.url,
          serverName: host.name,
          metadata:   host.metadata ?? {},   // ← метаданные хоста
          endpoint,
          ok:         false,
          error:      err.message,
          code:       err.code || null,
          elapsed,
        });
      });
  });

  Promise.allSettled(promises).then(() => {
    console.log("=== ALL DONE ===");
    socket.emit("batch:done", { url, total: tasks.length });
  });
});

// ─── Health ───────────────────────────────────────────────────────────────────
app.get("/health", (_req, res) =>
  res.json({ ok: true, hosts: HOSTS, endpoints: ENDPOINTS, checkMap: CHECK_MAP })
);

// ─── Socket lifecycle ─────────────────────────────────────────────────────────
io.on("connection", (socket) => {
  console.log(`[socket] connected    ${socket.id}`);
  socket.on("disconnect", () =>
    console.log(`[socket] disconnected ${socket.id}`)
  );
});

// ─── Start ────────────────────────────────────────────────────────────────────
server.listen(PORT, () =>
  console.log(`Server running on http://localhost:${PORT}`)
);