nolo-cli 0.7.0 → 0.8.0-alpha.11

package/chunk-XBIUN5CR.js

@@ -0,0 +1,191 @@
+import {
+  listUserRecordsFromServers,
+  readLiveDbRecordAfterTombstoneMerge
+} from "./chunk-GCMRDRCQ.js";
+import {
+  agentRecordHasConfiguredCredential
+} from "./chunk-GGXABIPP.js";
+
+// packages/cli/agentListHelpers.ts
+function parseAgentRecordId(privateKey, explicitId) {
+  if (explicitId) return explicitId;
+  const agentMatch = privateKey.match(/^agent-[^-]+-(.+)$/i);
+  return agentMatch?.[1] ?? "";
+}
+function buildCompanionKeys(rawId, userId) {
+  return {
+    privateKey: `agent-${userId}-${rawId}`,
+    publicKey: `agent-pub-${rawId}`
+  };
+}
+function normalizeListedAgent(record) {
+  const privateKey = typeof record?.dbKey === "string" ? record.dbKey : "";
+  const explicitId = typeof record?.id === "string" && record.id ? record.id : void 0;
+  const ownerUserId = typeof record?.userId === "string" ? record.userId : "";
+  const rawId = explicitId || (ownerUserId && privateKey.startsWith(`agent-${ownerUserId}-`) ? privateKey.slice(`agent-${ownerUserId}-`.length) : parseAgentRecordId(privateKey, explicitId));
+  if (!privateKey || !rawId || !ownerUserId) return null;
+  const keys = buildCompanionKeys(rawId, ownerUserId);
+  const credentialConfigured = agentRecordHasConfiguredCredential(record);
+  const credentialRef = typeof record?.credentialRef === "string" && record.credentialRef ? record.credentialRef : void 0;
+  const apiKeyRef = typeof record?.apiKeyRef === "string" && record.apiKeyRef ? record.apiKeyRef : void 0;
+  return {
+    id: rawId,
+    privateKey,
+    publicKey: keys.publicKey,
+    name: typeof record?.name === "string" && record.name ? record.name : "(unnamed)",
+    model: typeof record?.model === "string" && record.model ? record.model : "-",
+    updatedAt: typeof record?.updatedAt === "string" || typeof record?.updatedAt === "number" ? record.updatedAt : typeof record?.createdAt === "string" || typeof record?.createdAt === "number" ? record.createdAt : typeof record?.created === "string" || typeof record?.created === "number" ? record.created : null,
+    isPublicFlag: !!record?.isPublic,
+    publicRecordExists: false,
+    type: typeof record?.type === "string" ? record.type : null,
+    tools: Array.isArray(record?.tools) ? record.tools.filter((tool) => typeof tool === "string") : [],
+    credentialConfigured,
+    credentialRef,
+    apiKeyRef,
+    ...typeof record?.apiSource === "string" && record.apiSource ? { apiSource: record.apiSource } : {},
+    ...typeof record?.cliProvider === "string" && record.cliProvider ? { cliProvider: record.cliProvider } : {}
+  };
+}
+function sortListedAgents(agents) {
+  agents.sort((a, b) => {
+    const ta = a.updatedAt == null ? 0 : new Date(a.updatedAt).getTime();
+    const tb = b.updatedAt == null ? 0 : new Date(b.updatedAt).getTime();
+    return tb - ta;
+  });
+  return agents;
+}
+function parseAgentListArgs(args) {
+  return {
+    wantJson: args.includes("--json"),
+    publicOnly: args.includes("--public-only"),
+    idsOnly: args.includes("--ids-only")
+  };
+}
+async function listLocalCachedAgents(args) {
+  const privateRecords = /* @__PURE__ */ new Map();
+  const publicKeys = /* @__PURE__ */ new Set();
+  for await (const [key, value] of args.db.iterator({ gte: "", lte: "\uFFFF" })) {
+    if (typeof key !== "string" || !value || typeof value !== "object") continue;
+    if (key.startsWith(`agent-${args.userId}-`)) {
+      privateRecords.set(key, { ...value, dbKey: key });
+      continue;
+    }
+    if (key.startsWith("agent-pub-")) {
+      publicKeys.add(key);
+    }
+  }
+  const agents = sortListedAgents(
+    [...privateRecords.values()].map((record) => normalizeListedAgent(record)).filter((agent) => agent != null)
+  );
+  for (const agent of agents) {
+    agent.publicRecordExists = publicKeys.has(agent.publicKey);
+  }
+  return agents;
+}
+async function hasReadableRecord(args) {
+  try {
+    await args.readDbRecord(args);
+    return true;
+  } catch {
+    return false;
+  }
+}
+async function listRemoteAgents(args) {
+  const recordGroups = [await args.queryUserRecords({ ...args, type: "agent" })];
+  const agents = sortListedAgents(
+    recordGroups.flat().map((record) => normalizeListedAgent(record)).filter((agent) => agent != null)
+  );
+  await Promise.all(
+    agents.map(async (agent) => {
+      agent.publicRecordExists = await hasReadableRecord({
+        authToken: args.authToken,
+        dbKey: agent.publicKey,
+        fallbackFetchImpl: args.fallbackFetchImpl,
+        fetchImpl: args.fetchImpl,
+        serverUrl: args.serverUrl,
+        readDbRecord: args.readDbRecord
+      });
+    })
+  );
+  return agents;
+}
+async function listRemoteAgentsAcrossServers(args) {
+  const remoteResult = await listUserRecordsFromServers({
+    authToken: args.authToken,
+    fallbackFetchImpl: args.fallbackFetchImpl,
+    fetchImpl: args.fetchImpl,
+    label: "agent query",
+    serverUrls: args.serverUrls,
+    type: "agent",
+    userId: args.userId
+  });
+  const agents = sortListedAgents(
+    remoteResult.records.map((record) => normalizeListedAgent(record)).filter((agent) => agent != null)
+  );
+  return { agents, failures: remoteResult.failures };
+}
+async function listFavoriteAgentIdsAcrossServers(args) {
+  const favoritedAtByKey = {};
+  await Promise.all(
+    args.serverUrls.map(async (serverUrl) => {
+      try {
+        const res = await args.fetchImpl(`${serverUrl}/rpc/listFavorites`, {
+          method: "POST",
+          headers: {
+            "Content-Type": "application/json",
+            Authorization: `Bearer ${args.authToken}`
+          },
+          body: JSON.stringify({ targetType: "agent" })
+        });
+        if (!res.ok) return;
+        const data = await res.json().catch(() => ({}));
+        const items = Array.isArray(data?.items) ? data.items : [];
+        for (const item of items) {
+          const id = typeof item?.id === "string" ? item.id : "";
+          if (!id) continue;
+          const at = Number(item?.favoritedAt) || 0;
+          favoritedAtByKey[id] = Math.max(
+            favoritedAtByKey[id] ?? 0,
+            at || Date.now()
+          );
+        }
+        const ids = Array.isArray(data?.ids) ? data.ids : [];
+        for (const id of ids) {
+          if (typeof id === "string" && id && !(id in favoritedAtByKey)) {
+            favoritedAtByKey[id] = 1;
+          }
+        }
+      } catch {
+      }
+    })
+  );
+  return favoritedAtByKey;
+}
+async function decorateAgentsWithPublicStatusAcrossServers(args) {
+  await Promise.all(
+    args.agents.map(async (agent) => {
+      try {
+        const result = await readLiveDbRecordAfterTombstoneMerge({
+          authToken: args.authToken,
+          dbKey: agent.publicKey,
+          fallbackFetchImpl: args.fallbackFetchImpl,
+          fetchImpl: args.fetchImpl,
+          serverUrls: args.serverUrls
+        });
+        agent.publicRecordExists = result.record?.isPublic === true;
+      } catch {
+        agent.publicRecordExists = false;
+      }
+    })
+  );
+  return args.agents;
+}
+
+export {
+  parseAgentListArgs,
+  listLocalCachedAgents,
+  listRemoteAgents,
+  listRemoteAgentsAcrossServers,
+  listFavoriteAgentIdsAcrossServers,
+  decorateAgentsWithPublicStatusAcrossServers
+};