@fibscope/agent 0.1.0

package/lib/pulse/snapshot.mjs

@@ -0,0 +1,155 @@
+import { FiberCliProvider, FiberProviderError, FiberRpcProvider } from "./providers.mjs";
+
+export async function collectFiberSnapshot(providerOrOptions, maybeOptions = {}) {
+  const provider = isProvider(providerOrOptions) ? providerOrOptions : undefined;
+  const options = isProvider(providerOrOptions) ? maybeOptions : (providerOrOptions ?? maybeOptions);
+  const selected = provider ? { provider, failures: [] } : await selectActiveFiberProvider(options);
+  if (!selected.provider) return offlineSnapshot(selected.failures);
+
+  try {
+    const timestamp = Date.now();
+    const [nodeInfo, peers, channels] = await Promise.all([
+      selected.provider.getNodeInfo(),
+      selected.provider.getPeers(),
+      selected.provider.getChannels(),
+    ]);
+    const node = normalizeCollectedNode(nodeInfo, peers, channels, timestamp);
+    const snapshot = {
+      timestamp,
+      provider: providerInfo(selected.provider),
+      node,
+      peers,
+      channels,
+      payments: [],
+      collectionErrors: [],
+    };
+    snapshot.payments = await collectPayments(selected.provider, options, snapshot);
+    return snapshot;
+  } catch (error) {
+    return offlineSnapshot([{ provider: selected.provider.label, message: errorMessage(error) }]);
+  }
+}
+
+export async function collectNode(provider) {
+  return provider.getNodeInfo();
+}
+
+export async function collectPeers(provider) {
+  return provider.getPeers();
+}
+
+export async function collectChannels(provider) {
+  return provider.getChannels();
+}
+
+export async function collectPayments(provider, options = {}, snapshot = {}) {
+  if (typeof options.paymentProvider === "function") {
+    return (await options.paymentProvider(snapshot)) ?? [];
+  }
+  if (typeof provider.getPayments === "function") {
+    try {
+      return (await provider.getPayments()) ?? [];
+    } catch (error) {
+      snapshot.collectionErrors?.push?.({ provider: provider.label, collector: "payments", message: errorMessage(error) });
+      return [];
+    }
+  }
+  return [];
+}
+
+export async function selectActiveFiberProvider(options = {}) {
+  if (options.provider) return { provider: options.provider, failures: [] };
+  const candidates = buildProviderCandidates(options);
+  const failures = [];
+  for (const provider of candidates) {
+    try {
+      await provider.getNodeInfo();
+      return { provider, failures };
+    } catch (error) {
+      failures.push({ provider: provider.label, message: errorMessage(error) });
+    }
+  }
+  return { failures };
+}
+
+export function buildProviderCandidates(options = {}) {
+  const providers = [];
+  const timeoutMs = options.timeoutMs ?? 1_500;
+  const rpcToken = options.rpcToken ?? process.env.PULSE_FIBER_RPC_TOKEN;
+  const rpcHeaders = {
+    ...(options.rpcHeaders ?? {}),
+    ...(rpcToken ? { authorization: `Bearer ${rpcToken}` } : {}),
+  };
+  const rpcUrls = unique([
+    options.rpcUrl,
+    process.env.PULSE_FIBER_RPC_URL,
+    process.env.FIBER_RPC_URL,
+    ...(options.rpcUrls ?? []),
+    "http://127.0.0.1:8227",
+    "http://127.0.0.1:8237",
+  ].filter(Boolean));
+  providers.push(...rpcUrls.map((rpcUrl) => new FiberRpcProvider({
+    rpcUrl,
+    headers: rpcHeaders,
+    timeoutMs,
+  })));
+  const cliBins = options.cliBin || process.env.PULSE_FNN_CLI || process.env.FNN_CLI
+    ? [options.cliBin ?? process.env.PULSE_FNN_CLI ?? process.env.FNN_CLI ?? "fnn-cli"]
+    : ["fnn-cli", "fiber-cli"];
+  providers.push(...cliBins.map((bin) => new FiberCliProvider({
+    bin,
+    baseArgs: options.baseArgs,
+    timeoutMs,
+  })));
+  return providers;
+}
+
+export function offlineSnapshot(failures = []) {
+  const timestamp = Date.now();
+  return {
+    timestamp,
+    provider: null,
+    node: {
+      nodeId: "",
+      version: "",
+      online: false,
+      peerCount: 0,
+      channelCount: 0,
+      lastUpdated: timestamp,
+    },
+    peers: [],
+    channels: [],
+    payments: [],
+    collectionErrors: failures,
+  };
+}
+
+function normalizeCollectedNode(node, peers, channels, timestamp) {
+  return {
+    ...node,
+    online: true,
+    peerCount: peers.length,
+    channelCount: channels.length,
+    lastUpdated: timestamp,
+  };
+}
+
+function providerInfo(provider) {
+  return {
+    kind: provider.kind ?? "unknown",
+    label: provider.label ?? "unknown",
+  };
+}
+
+function isProvider(value) {
+  return Boolean(value && typeof value === "object" && "getNodeInfo" in value && "getPeers" in value && "getChannels" in value);
+}
+
+function errorMessage(error) {
+  if (error instanceof FiberProviderError) return error.message;
+  return error instanceof Error ? error.message : String(error);
+}
+
+function unique(values) {
+  return [...new Set(values)];
+}