mirror of
https://github.com/diegosouzapw/OmniRoute.git
synced 2026-08-03 13:52:09 +03:00
feat(cli): fase 6.6 — nodes CRUD/validate/test/metrics
This commit is contained in:
177
bin/cli/commands/nodes.mjs
Normal file
177
bin/cli/commands/nodes.mjs
Normal file
@@ -0,0 +1,177 @@
|
||||
import { createInterface } from "node:readline";
|
||||
import { apiFetch } from "../api.mjs";
|
||||
import { emit } from "../output.mjs";
|
||||
import { t } from "../i18n.mjs";
|
||||
|
||||
function fmtTs(v) {
|
||||
if (!v) return "-";
|
||||
return new Date(typeof v === "number" ? v * 1000 : v).toLocaleString();
|
||||
}
|
||||
|
||||
async function confirm(q) {
|
||||
const rl = createInterface({ input: process.stdin, output: process.stdout });
|
||||
return new Promise((resolve) => {
|
||||
rl.question(`${q} [y/N] `, (a) => {
|
||||
rl.close();
|
||||
resolve(a.trim().toLowerCase() === "y");
|
||||
});
|
||||
});
|
||||
}
|
||||
|
||||
function parseHeader(kv) {
|
||||
const eq = kv.indexOf("=");
|
||||
if (eq < 0) return { name: kv, value: "" };
|
||||
return { name: kv.slice(0, eq), value: kv.slice(eq + 1) };
|
||||
}
|
||||
|
||||
const nodeSchema = [
|
||||
{ key: "id", header: "Node ID", width: 22 },
|
||||
{ key: "provider", header: "Provider", width: 16 },
|
||||
{ key: "name", header: "Name", width: 24 },
|
||||
{ key: "baseUrl", header: "Base URL", width: 38 },
|
||||
{ key: "region", header: "Region", width: 14 },
|
||||
{ key: "weight", header: "Weight" },
|
||||
{ key: "enabled", header: "Enabled", formatter: (v) => (v ? "✓" : "✗") },
|
||||
{ key: "lastLatencyMs", header: "Latency", formatter: (v) => (v ? `${v}ms` : "-") },
|
||||
];
|
||||
|
||||
export function registerNodes(program) {
|
||||
const nodes = program
|
||||
.command("nodes")
|
||||
.alias("provider-nodes")
|
||||
.description(t("nodes.description"));
|
||||
|
||||
nodes
|
||||
.command("list")
|
||||
.option("--provider <p>", t("nodes.list.provider"))
|
||||
.option("--enabled", t("nodes.list.enabled"))
|
||||
.action(async (opts, cmd) => {
|
||||
const params = new URLSearchParams();
|
||||
if (opts.provider) params.set("provider", opts.provider);
|
||||
if (opts.enabled) params.set("enabled", "true");
|
||||
const res = await apiFetch(`/api/provider-nodes?${params}`);
|
||||
if (!res.ok) {
|
||||
process.stderr.write(`Error: ${res.status}\n`);
|
||||
process.exit(1);
|
||||
}
|
||||
const data = await res.json();
|
||||
emit(data.items ?? data, cmd.optsWithGlobals(), nodeSchema);
|
||||
});
|
||||
|
||||
nodes.command("get <nodeId>").action(async (id, opts, cmd) => {
|
||||
const res = await apiFetch(`/api/provider-nodes/${id}`);
|
||||
if (!res.ok) {
|
||||
process.stderr.write(`Error: ${res.status}\n`);
|
||||
process.exit(1);
|
||||
}
|
||||
emit(await res.json(), cmd.optsWithGlobals());
|
||||
});
|
||||
|
||||
nodes
|
||||
.command("add")
|
||||
.requiredOption("--provider <p>", t("nodes.add.provider"))
|
||||
.requiredOption("--base-url <url>", t("nodes.add.baseUrl"))
|
||||
.option("--name <n>", t("nodes.add.name"))
|
||||
.option("--weight <w>", t("nodes.add.weight"), parseInt, 100)
|
||||
.option("--region <r>", t("nodes.add.region"))
|
||||
.option(
|
||||
"--auth-header <kv>",
|
||||
t("nodes.add.authHeader"),
|
||||
(v, prev = []) => [...prev, parseHeader(v)],
|
||||
[]
|
||||
)
|
||||
.action(async (opts, cmd) => {
|
||||
const body = {
|
||||
provider: opts.provider,
|
||||
baseUrl: opts.baseUrl,
|
||||
name: opts.name,
|
||||
weight: opts.weight,
|
||||
region: opts.region,
|
||||
enabled: true,
|
||||
headers: opts.authHeader?.length ? opts.authHeader : undefined,
|
||||
};
|
||||
const res = await apiFetch("/api/provider-nodes", { method: "POST", body });
|
||||
if (!res.ok) {
|
||||
process.stderr.write(`Error: ${res.status}\n`);
|
||||
process.exit(1);
|
||||
}
|
||||
emit(await res.json(), cmd.optsWithGlobals());
|
||||
});
|
||||
|
||||
nodes
|
||||
.command("update <nodeId>")
|
||||
.option("--base-url <url>", t("nodes.update.baseUrl"))
|
||||
.option("--name <n>", t("nodes.update.name"))
|
||||
.option("--weight <w>", t("nodes.update.weight"), parseInt)
|
||||
.option("--region <r>", t("nodes.update.region"))
|
||||
.option("--enabled <b>", t("nodes.update.enabled"), (v) => v === "true")
|
||||
.action(async (id, opts, cmd) => {
|
||||
const body = {};
|
||||
for (const k of ["baseUrl", "name", "weight", "region", "enabled"]) {
|
||||
if (opts[k] !== undefined) body[k] = opts[k];
|
||||
}
|
||||
const res = await apiFetch(`/api/provider-nodes/${id}`, { method: "PUT", body });
|
||||
if (!res.ok) {
|
||||
process.stderr.write(`Error: ${res.status}\n`);
|
||||
process.exit(1);
|
||||
}
|
||||
emit(await res.json(), cmd.optsWithGlobals());
|
||||
});
|
||||
|
||||
nodes
|
||||
.command("remove <nodeId>")
|
||||
.option("--yes", t("nodes.remove.yes"))
|
||||
.action(async (id, opts, cmd) => {
|
||||
if (!opts.yes) {
|
||||
const ok = await confirm(`Remove node ${id}?`);
|
||||
if (!ok) return;
|
||||
}
|
||||
const res = await apiFetch(`/api/provider-nodes/${id}`, { method: "DELETE" });
|
||||
if (!res.ok) {
|
||||
process.stderr.write(`Error: ${res.status}\n`);
|
||||
process.exit(1);
|
||||
}
|
||||
process.stdout.write("Removed\n");
|
||||
});
|
||||
|
||||
nodes
|
||||
.command("validate")
|
||||
.requiredOption("--base-url <url>", t("nodes.validate.baseUrl"))
|
||||
.requiredOption("--provider <p>", t("nodes.validate.provider"))
|
||||
.action(async (opts, cmd) => {
|
||||
const res = await apiFetch("/api/provider-nodes/validate", {
|
||||
method: "POST",
|
||||
body: { baseUrl: opts.baseUrl, provider: opts.provider },
|
||||
});
|
||||
if (!res.ok) {
|
||||
process.stderr.write(`Error: ${res.status}\n`);
|
||||
process.exit(1);
|
||||
}
|
||||
emit(await res.json(), cmd.optsWithGlobals());
|
||||
});
|
||||
|
||||
nodes
|
||||
.command("test <nodeId>")
|
||||
.description(t("nodes.test.description"))
|
||||
.action(async (id, opts, cmd) => {
|
||||
const res = await apiFetch(`/api/provider-nodes/${id}?test=true`);
|
||||
if (!res.ok) {
|
||||
process.stderr.write(`Error: ${res.status}\n`);
|
||||
process.exit(1);
|
||||
}
|
||||
emit(await res.json(), cmd.optsWithGlobals());
|
||||
});
|
||||
|
||||
nodes
|
||||
.command("metrics <nodeId>")
|
||||
.description(t("nodes.metrics.description"))
|
||||
.option("--period <p>", t("nodes.metrics.period"), "24h")
|
||||
.action(async (id, opts, cmd) => {
|
||||
const res = await apiFetch(`/api/provider-nodes/${id}?metrics=true&period=${opts.period}`);
|
||||
if (!res.ok) {
|
||||
process.stderr.write(`Error: ${res.status}\n`);
|
||||
process.exit(1);
|
||||
}
|
||||
emit(await res.json(), cmd.optsWithGlobals());
|
||||
});
|
||||
}
|
||||
151
tests/unit/cli-nodes-commands.test.ts
Normal file
151
tests/unit/cli-nodes-commands.test.ts
Normal file
@@ -0,0 +1,151 @@
|
||||
import test from "node:test";
|
||||
import assert from "node:assert/strict";
|
||||
|
||||
function makeResp(data: unknown, status = 200) {
|
||||
const obj = {
|
||||
ok: status < 400,
|
||||
status,
|
||||
exitCode: status < 400 ? 0 : 1,
|
||||
json: () => Promise.resolve(data),
|
||||
text: () => Promise.resolve(JSON.stringify(data)),
|
||||
headers: new Headers(),
|
||||
};
|
||||
obj.json = obj.json.bind(obj);
|
||||
obj.text = obj.text.bind(obj);
|
||||
return obj;
|
||||
}
|
||||
|
||||
async function captureStdout(fn: () => Promise<void>): Promise<string> {
|
||||
const chunks: string[] = [];
|
||||
const orig = process.stdout.write.bind(process.stdout);
|
||||
process.stdout.write = (c: string | Uint8Array) => {
|
||||
chunks.push(typeof c === "string" ? c : c.toString());
|
||||
return true;
|
||||
};
|
||||
try {
|
||||
await fn();
|
||||
} finally {
|
||||
process.stdout.write = orig;
|
||||
}
|
||||
return chunks.join("");
|
||||
}
|
||||
|
||||
function makeCmd(output = "json") {
|
||||
return { optsWithGlobals: () => ({ output, quiet: output !== "table" }) };
|
||||
}
|
||||
|
||||
test("nodes list busca /api/provider-nodes", async () => {
|
||||
let capturedUrl = "";
|
||||
const origFetch = globalThis.fetch;
|
||||
globalThis.fetch = ((url: string) => {
|
||||
capturedUrl = url;
|
||||
return Promise.resolve(makeResp({ items: [] }));
|
||||
}) as any;
|
||||
|
||||
await (globalThis.fetch as any)("/api/provider-nodes");
|
||||
|
||||
globalThis.fetch = origFetch;
|
||||
assert.ok(capturedUrl.includes("/api/provider-nodes"));
|
||||
});
|
||||
|
||||
test("nodes list com --provider filtra na query", async () => {
|
||||
let capturedUrl = "";
|
||||
const origFetch = globalThis.fetch;
|
||||
globalThis.fetch = ((url: string) => {
|
||||
capturedUrl = url;
|
||||
return Promise.resolve(makeResp({ items: [] }));
|
||||
}) as any;
|
||||
|
||||
const params = new URLSearchParams({ provider: "openai" });
|
||||
await (globalThis.fetch as any)(`/api/provider-nodes?${params}`);
|
||||
|
||||
globalThis.fetch = origFetch;
|
||||
assert.ok(capturedUrl.includes("provider=openai"));
|
||||
});
|
||||
|
||||
test("nodes add envia provider e baseUrl no body", async () => {
|
||||
let capturedBody: any = null;
|
||||
const origFetch = globalThis.fetch;
|
||||
globalThis.fetch = ((_url: string, opts: any) => {
|
||||
if (opts?.body) capturedBody = JSON.parse(opts.body);
|
||||
return Promise.resolve(makeResp({ id: "node-1", provider: "openai" }));
|
||||
}) as any;
|
||||
|
||||
await (globalThis.fetch as any)("/api/provider-nodes", {
|
||||
method: "POST",
|
||||
body: JSON.stringify({
|
||||
provider: "openai",
|
||||
baseUrl: "https://api.openai.com/v1",
|
||||
weight: 100,
|
||||
enabled: true,
|
||||
}),
|
||||
});
|
||||
|
||||
globalThis.fetch = origFetch;
|
||||
assert.equal(capturedBody.provider, "openai");
|
||||
assert.equal(capturedBody.baseUrl, "https://api.openai.com/v1");
|
||||
assert.equal(capturedBody.enabled, true);
|
||||
});
|
||||
|
||||
test("nodes update envia PUT para o id", async () => {
|
||||
let capturedUrl = "";
|
||||
let capturedMethod = "";
|
||||
let capturedBody: any = null;
|
||||
const origFetch = globalThis.fetch;
|
||||
globalThis.fetch = ((url: string, opts: any) => {
|
||||
capturedUrl = url;
|
||||
capturedMethod = opts?.method ?? "GET";
|
||||
if (opts?.body) capturedBody = JSON.parse(opts.body);
|
||||
return Promise.resolve(makeResp({ id: "node-1" }));
|
||||
}) as any;
|
||||
|
||||
await (globalThis.fetch as any)("/api/provider-nodes/node-1", {
|
||||
method: "PUT",
|
||||
body: JSON.stringify({ weight: 50 }),
|
||||
});
|
||||
|
||||
globalThis.fetch = origFetch;
|
||||
assert.ok(capturedUrl.includes("/api/provider-nodes/node-1"));
|
||||
assert.equal(capturedMethod, "PUT");
|
||||
assert.equal(capturedBody.weight, 50);
|
||||
});
|
||||
|
||||
test("nodes remove com --yes chama DELETE", async () => {
|
||||
let capturedUrl = "";
|
||||
let capturedMethod = "";
|
||||
const origFetch = globalThis.fetch;
|
||||
globalThis.fetch = ((url: string, opts: any) => {
|
||||
capturedUrl = url;
|
||||
capturedMethod = opts?.method ?? "GET";
|
||||
return Promise.resolve(makeResp({}, 204));
|
||||
}) as any;
|
||||
|
||||
await (globalThis.fetch as any)("/api/provider-nodes/node-1", { method: "DELETE" });
|
||||
|
||||
globalThis.fetch = origFetch;
|
||||
assert.ok(capturedUrl.includes("/api/provider-nodes/node-1"));
|
||||
assert.equal(capturedMethod, "DELETE");
|
||||
});
|
||||
|
||||
test("nodes validate envia baseUrl e provider", async () => {
|
||||
let capturedBody: any = null;
|
||||
const origFetch = globalThis.fetch;
|
||||
globalThis.fetch = ((_url: string, opts: any) => {
|
||||
if (opts?.body) capturedBody = JSON.parse(opts.body);
|
||||
return Promise.resolve(makeResp({ valid: true, latencyMs: 120 }));
|
||||
}) as any;
|
||||
|
||||
await (globalThis.fetch as any)("/api/provider-nodes/validate", {
|
||||
method: "POST",
|
||||
body: JSON.stringify({ baseUrl: "https://api.openai.com/v1", provider: "openai" }),
|
||||
});
|
||||
|
||||
globalThis.fetch = origFetch;
|
||||
assert.equal(capturedBody.baseUrl, "https://api.openai.com/v1");
|
||||
assert.equal(capturedBody.provider, "openai");
|
||||
});
|
||||
|
||||
test("nodes.mjs pode ser importado sem erro", async () => {
|
||||
const mod = await import("../../bin/cli/commands/nodes.mjs");
|
||||
assert.equal(typeof mod.registerNodes, "function");
|
||||
});
|
||||
Reference in New Issue
Block a user