pairai 0.2.2 → 0.2.4

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pairai",
-  "version": "0.2.2",
+  "version": "0.2.4",
   "description": "pairai CLI — connect AI agents to collaborate via the pairai hub",
   "type": "module",
   "license": "MIT",

package/pairai.ts

@@ -21,6 +21,7 @@ import { writeFileSync, mkdirSync, readFileSync, statSync, existsSync } from "no
 import { homedir } from "node:os";
 import { join, dirname } from "node:path";
 import { fileURLToPath } from "node:url";
+import { validateProvider, detectProvider, updateVersionInConfig, checkExistingConfig, formatKeyBackupBox } from "./lib.js";
 
 const __dirname = dirname(fileURLToPath(import.meta.url));
 const PKG = JSON.parse(readFileSync(join(__dirname, "package.json"), "utf-8"));
@@ -62,10 +63,7 @@ if (command === "upgrade") {
         try {
           if (!existsSync(p)) continue;
           const content = readFileSync(p, "utf-8");
-          const updated = content.replace(
-            new RegExp(`pairai@${VERSION.replace(/\./g, "\\.")}`, "g"),
-            `pairai@${latest}`,
-          );
+          const updated = updateVersionInConfig(content, latest);
           if (updated !== content) {
             writeFileSync(p, updated);
             console.log(`  Updated version in ${p}`);
@@ -80,11 +78,8 @@ if (command === "upgrade") {
   process.exit(0);
 }
 
-function detectProvider(): "claude" | "gemini" {
-  if (process.env.GEMINI_CLI) return "gemini";
-  try { if (statSync(".gemini").isDirectory()) return "gemini"; } catch {}
-  return "claude";
-}
+// detectProvider, validateProvider, updateVersionInConfig, checkExistingConfig,
+// formatKeyBackupBox are imported from ./lib.js
 
 // ── Setup: register + configure ──────────────────────────────────────────────
 
@@ -94,11 +89,13 @@ if (command === "setup") {
   const hubUrl = hubIdx !== -1 ? rest.splice(hubIdx, 2)[1] : "https://pairai.pro";
   const providerIdx = rest.indexOf("--provider");
   const providerArg = providerIdx !== -1 ? rest.splice(providerIdx, 2)[1] : undefined;
-  if (providerArg && providerArg !== "claude" && providerArg !== "gemini") {
-    console.error(`  Unknown provider "${providerArg}". Must be "claude" or "gemini".`);
-    process.exit(1);
+  if (providerArg) {
+    try { validateProvider(providerArg); } catch (e) {
+      console.error(`  ${(e as Error).message}`);
+      process.exit(1);
+    }
   }
-  const provider = providerArg ?? detectProvider();
+  const provider = (providerArg as "claude" | "gemini") ?? detectProvider();
   const globalIdx = rest.indexOf("--global");
   const useGlobal = globalIdx !== -1 ? (rest.splice(globalIdx, 1), true) : false;
   const agentName = rest.find((a) => !a.startsWith("--"));
@@ -111,21 +108,14 @@ if (command === "setup") {
   }
 
   // Check for existing config to avoid accidental overwrites
-  const existingConfigPath = provider === "gemini"
-    ? join(useGlobal ? join(homedir(), ".gemini") : join(process.cwd(), ".gemini"), "settings.json")
-    : join(process.cwd(), ".mcp.json");
-  const mcpKey = provider === "gemini" ? "pairai" : "pairai-channel";
-  if (!useForce && existsSync(existingConfigPath)) {
-    try {
-      const existing = JSON.parse(readFileSync(existingConfigPath, "utf-8"));
-      const servers = existing.mcpServers ?? {};
-      if (servers[mcpKey]) {
-        console.error(`\n  pairai is already configured in ${existingConfigPath}`);
-        console.error(`  Running setup again would overwrite the existing API key and config.`);
-        console.error(`\n  To force a fresh setup, run: npx pairai setup "${agentName}" --force\n`);
-        process.exit(1);
-      }
-    } catch {}
+  if (!useForce) {
+    const existingConfigPath = checkExistingConfig(provider, process.cwd(), homedir(), useGlobal);
+    if (existingConfigPath) {
+      console.error(`\n  pairai is already configured in ${existingConfigPath}`);
+      console.error(`  Running setup again would overwrite the existing API key and config.`);
+      console.error(`\n  To force a fresh setup, run: npx pairai setup "${agentName}" --force\n`);
+      process.exit(1);
+    }
   }
 
   console.log(`\n  Registering "${agentName}" on ${hubUrl}...\n`);
@@ -158,22 +148,8 @@ if (command === "setup") {
   const keyPath = join(keyDir, `${id}.pem`);
   writeFileSync(keyPath, privateKey, { mode: 0o600 });
   console.log(`  Private key: ${keyPath}`);
-  const lines = [
-    "BACK UP YOUR PRIVATE KEY",
-    "",
-    keyPath,
-    "",
-    "This key is stored only on your machine.",
-    "The hub never sees it. If lost, you must re-register",
-    "and re-pair — all encrypted history becomes unreadable.",
-    "",
-    "Copy it to a password manager or secure backup now.",
-  ];
-  const w = Math.max(...lines.map((l) => l.length)) + 2;
   console.log();
-  console.log(`  ┌${"─".repeat(w + 2)}┐`);
-  for (const l of lines) console.log(`  │  ${l.padEnd(w)}│`);
-  console.log(`  └${"─".repeat(w + 2)}┘`);
+  for (const line of formatKeyBackupBox(keyPath)) console.log(line);
   console.log();
 
   if (provider === "gemini") {
@@ -256,11 +232,13 @@ const { ListToolsRequestSchema, CallToolRequestSchema } = await import("@modelco
 const serveArgs = args.slice(1);
 const serveProviderIdx = serveArgs.indexOf("--provider");
 const serveProviderArg = serveProviderIdx !== -1 ? serveArgs[serveProviderIdx + 1] : undefined;
-if (serveProviderArg && serveProviderArg !== "claude" && serveProviderArg !== "gemini") {
-  console.error(`  Unknown provider "${serveProviderArg}". Must be "claude" or "gemini".`);
-  process.exit(1);
+if (serveProviderArg) {
+  try { validateProvider(serveProviderArg); } catch (e) {
+    console.error(`  ${(e as Error).message}`);
+    process.exit(1);
+  }
 }
-const serveProvider = serveProviderArg ?? "claude";
+const serveProvider = (serveProviderArg as "claude" | "gemini") ?? "claude";
 
 const HUB_URL = process.env.PAIRAI_HUB_URL ?? process.env.PAIRAI_URL ?? "https://pairai.pro";
 const API_KEY = process.env.PAIRAI_AGENT_CRED ?? process.env.PAIRAI_API_KEY;
@@ -436,6 +414,11 @@ mcp.setRequestHandler(ListToolsRequestSchema, async () => ({
         required: ["task_id", "status"],
       },
     },
+    {
+      name: "pairai_get_profile",
+      description: "Get your own agent profile — name, ID, description, capabilities.",
+      inputSchema: { type: "object" as const, properties: {} },
+    },
     {
       name: "pairai_list_connections",
       description: "List agents you are connected with.",
@@ -470,6 +453,66 @@ mcp.setRequestHandler(ListToolsRequestSchema, async () => ({
         required: ["code"],
       },
     },
+    {
+      name: "pairai_update_profile",
+      description: "Update your agent's profile — name, description, capabilities, and metadata. Returns the updated profile.",
+      inputSchema: {
+        type: "object" as const,
+        properties: {
+          name: { type: "string", description: "Display name (1-64 chars)" },
+          description: { type: "string", description: "What this agent does (max 500 chars)" },
+          capabilities: { type: "array", items: { type: "string" }, description: "List of capabilities, e.g. ['scheduling', 'code-review']" },
+          metadata: { type: "object", description: "Arbitrary JSON metadata (max 4KB)" },
+        },
+      },
+    },
+    {
+      name: "pairai_set_alias",
+      description: "Set a local alias for a connected agent. Only you see this alias. Set to null to clear.",
+      inputSchema: {
+        type: "object" as const,
+        properties: {
+          connection_id: { type: "string", description: "Connection ID" },
+          alias: { type: ["string", "null"], description: "Local alias, or null to clear" },
+        },
+        required: ["connection_id"],
+      },
+    },
+    {
+      name: "pairai_list_tasks",
+      description: "List all tasks you are involved in (as initiator or target).",
+      inputSchema: {
+        type: "object" as const,
+        properties: {
+          status: { type: "string", enum: ["submitted", "working", "input-required", "completed", "failed", "cancelled"], description: "Filter by status" },
+        },
+      },
+    },
+    {
+      name: "pairai_get_task",
+      description: "Get full details of a task including all messages.",
+      inputSchema: {
+        type: "object" as const,
+        properties: {
+          task_id: { type: "string", description: "Task ID" },
+        },
+        required: ["task_id"],
+      },
+    },
+    {
+      name: "pairai_upload_file",
+      description: "Upload a file to a task. Provide base64-encoded content.",
+      inputSchema: {
+        type: "object" as const,
+        properties: {
+          task_id: { type: "string", description: "Task ID" },
+          filename: { type: "string", description: "Original filename, e.g. photo.png" },
+          mime_type: { type: "string", description: "MIME type, e.g. image/png" },
+          base64_content: { type: "string", description: "Base64-encoded file content" },
+        },
+        required: ["task_id", "filename", "mime_type", "base64_content"],
+      },
+    },
     {
       name: "pairai_create_encrypted_task",
       description: "Create an encrypted task. Title and description are encrypted — the hub cannot read them.",
@@ -530,6 +573,11 @@ mcp.setRequestHandler(CallToolRequestSchema, async (req) => {
     return { content: [{ type: "text" as const, text: `Status → ${args.status}` }] };
   }
 
+  if (name === "pairai_get_profile") {
+    const data = await hubGet("/agents/me");
+    return { content: [{ type: "text" as const, text: JSON.stringify(data, null, 2) }] };
+  }
+
   if (name === "pairai_list_connections") {
     const data = await hubGet("/connections");
     return { content: [{ type: "text" as const, text: JSON.stringify(data, null, 2) }] };
@@ -584,6 +632,74 @@ mcp.setRequestHandler(CallToolRequestSchema, async (req) => {
     return { content: [{ type: "text" as const, text: JSON.stringify(data, null, 2) }] };
   }
 
+  if (name === "pairai_update_profile") {
+    const body: Record<string, unknown> = {};
+    if (args.name !== undefined) body.name = args.name;
+    if (args.description !== undefined) body.description = args.description;
+    if (args.capabilities !== undefined) body.capabilities = args.capabilities;
+    if (args.metadata !== undefined) body.metadata = args.metadata;
+    const data = await hubPatch("/agents/me", body);
+    return { content: [{ type: "text" as const, text: "Profile updated.\n" + JSON.stringify(data, null, 2) }] };
+  }
+
+  if (name === "pairai_set_alias") {
+    const { connection_id, alias } = args as { connection_id: string; alias?: string | null };
+    const data = await hubPatch(`/connections/${connection_id}`, { alias: alias ?? null });
+    return { content: [{ type: "text" as const, text: JSON.stringify(data, null, 2) }] };
+  }
+
+  if (name === "pairai_list_tasks") {
+    await loadPublicKeys();
+    const data = (await hubGet("/tasks")) as Array<{
+      id: string; status: string; title: string; encrypted?: boolean;
+      description?: string; descriptionKeys?: any; senderSignature?: string; initiatorAgentId?: string;
+    }>;
+    const filtered = args.status ? data.filter((t) => t.status === args.status) : data;
+    const decrypted = filtered.map((t) => {
+      if (t.encrypted) {
+        const desc = decryptTaskDescription(t, t.id);
+        return { ...t, title: desc.split("\n")[0] || t.title, description: desc };
+      }
+      return t;
+    });
+    return { content: [{ type: "text" as const, text: JSON.stringify(decrypted, null, 2) }] };
+  }
+
+  if (name === "pairai_get_task") {
+    await loadPublicKeys();
+    const data = (await hubGet(`/tasks/${args.task_id}`)) as {
+      id: string; encrypted?: boolean; description?: string; descriptionKeys?: any;
+      senderSignature?: string; initiatorAgentId?: string;
+      [key: string]: unknown;
+    };
+    const msgs = (await hubGet(`/tasks/${args.task_id}/messages`)) as Array<{
+      id: string; content: string; contentType: string; senderAgentId: string;
+      encryptedKeys?: any; senderSignature?: string;
+    }>;
+    if (data.encrypted) {
+      const desc = decryptTaskDescription(data, data.id);
+      data.description = desc;
+    }
+    const decryptedMsgs = msgs.map((m) => {
+      if (data.encrypted) {
+        const d = decryptMessage(m, data.id);
+        return { ...m, content: d.content, contentType: d.contentType };
+      }
+      return m;
+    });
+    return { content: [{ type: "text" as const, text: JSON.stringify({ ...data, messages: decryptedMsgs }, null, 2) }] };
+  }
+
+  if (name === "pairai_upload_file") {
+    const { task_id, filename, mime_type, base64_content } = args as {
+      task_id: string; filename: string; mime_type: string; base64_content: string;
+    };
+    const data = await hubPost(`/tasks/${task_id}/files/json`, {
+      filename, mimeType: mime_type, base64Content: base64_content,
+    });
+    return { content: [{ type: "text" as const, text: JSON.stringify(data, null, 2) }] };
+  }
+
   if (name === "pairai_create_encrypted_task") {
     if (!PRIVATE_KEY)
       return { content: [{ type: "text" as const, text: "No private key configured. Re-run setup." }] };
@@ -636,7 +752,7 @@ function decryptMessage(
   }
   try {
     const keys = typeof msg.encryptedKeys === "string" ? JSON.parse(msg.encryptedKeys) : msg.encryptedKeys;
-    const senderPub = pubKeyCache.get(msg.senderAgentId);
+    const senderPub = msg.senderAgentId === myAgentId ? myPublicKey : pubKeyCache.get(msg.senderAgentId);
     const myKey = keys[myAgentId];
     if (senderPub && myKey) {
       const plain = localDecrypt(msg.content, msg.senderSignature, taskId, senderPub, myKey);
@@ -659,7 +775,7 @@ function decryptTaskDescription(
   }
   try {
     const keys = typeof full.descriptionKeys === "string" ? JSON.parse(full.descriptionKeys) : full.descriptionKeys;
-    const senderPub = full.initiatorAgentId ? pubKeyCache.get(full.initiatorAgentId) : undefined;
+    const senderPub = full.initiatorAgentId === myAgentId ? myPublicKey : (full.initiatorAgentId ? pubKeyCache.get(full.initiatorAgentId) : undefined);
     const myKey = keys[myAgentId];
     if (senderPub && myKey) {
       const plain = localDecrypt(full.description, full.senderSignature, taskId, senderPub, myKey);
@@ -735,6 +851,7 @@ async function poll() {
         }>;
       };
 
+      if (!full?.messages) continue;
       for (const msg of full.messages.slice(-unread.count)) {
         const key = `msg:${msg.id}`;
         if (seenMessages.has(key)) continue;