viber-channel 0.4.0 → 0.5.0

package/lib/connect.ts

@@ -0,0 +1,189 @@
+/**
+ * connect.ts — one-shot claim flow that turns a Viber connection URL into
+ * `.viber/auth.json` so the channel can be started.
+ *
+ * Invoked from the CLI as:
+ *   bunx viber-channel connect https://viber.dgypx.dev/connect/<claim_id>
+ *
+ * Pure logic: takes a URL + a target folder, talks to the Worker, writes the
+ * three files (`.viber/auth.json`, `.viber/readme.md`, `.gitignore` entry).
+ * No MCP, no SSE.
+ */
+
+import {
+  appendFileSync,
+  existsSync,
+  mkdirSync,
+  readFileSync,
+  writeFileSync,
+} from "node:fs";
+import { join } from "node:path";
+import { clientFingerprint } from "./fingerprint.js";
+
+interface InitResponse {
+  confirm_url: string;
+  wait_token: string;
+}
+
+interface ConsumedResponse {
+  status: "consumed";
+  project_id: number;
+  project_name: string;
+  user_email: string;
+  project_token: string;
+  issued_at: string;
+  rotates_after?: string;
+  target_conversation_id?: string | null;
+  readme_md: string;
+}
+
+interface PendingResponse {
+  status: "pending";
+}
+
+interface TerminalResponse {
+  status: "rejected" | "expired";
+  reason?: string;
+}
+
+type PollResponse = ConsumedResponse | PendingResponse | TerminalResponse;
+
+const POLL_INTERVAL_MS = 2000;
+const MAX_POLL_DURATION_MS = 10 * 60 * 1000;
+
+function parseClaimUrl(input: string): { baseUrl: string; claimId: string } {
+  let url: URL;
+  try {
+    url = new URL(input);
+  } catch {
+    throw new Error(`Invalid URL: ${input}`);
+  }
+  const match = url.pathname.match(/^\/connect\/([a-f0-9-]+)\/?$/i);
+  if (!match) {
+    throw new Error(
+      `Not a Viber claim URL. Expected format: https://<host>/connect/<claim_id>`,
+    );
+  }
+  return { baseUrl: `${url.protocol}//${url.host}`, claimId: match[1] };
+}
+
+function sleep(ms: number): Promise<void> {
+  return new Promise((resolve) => setTimeout(resolve, ms));
+}
+
+function writeAuthJson(
+  cwd: string,
+  data: ConsumedResponse,
+  fingerprint: string,
+): void {
+  const viberDir = join(cwd, ".viber");
+  mkdirSync(viberDir, { recursive: true });
+
+  const auth: Record<string, unknown> = {
+    schema_version: 1,
+    project_id: data.project_id,
+    project_name: data.project_name,
+    user_email: data.user_email,
+    project_token: data.project_token,
+    client_fingerprint: fingerprint,
+    issued_at: data.issued_at,
+    rotates_after: data.rotates_after,
+  };
+  // Only emit target_conversation_id when the server explicitly returned a
+  // non-null value (attach-to-project flow). Absent / null → omit to keep
+  // the file minimal for the common "fresh standalone conversation" case.
+  if (
+    data.target_conversation_id !== undefined &&
+    data.target_conversation_id !== null
+  ) {
+    auth.target_conversation_id = data.target_conversation_id;
+  }
+
+  writeFileSync(
+    join(viberDir, "auth.json"),
+    JSON.stringify(auth, null, 2),
+    "utf-8",
+  );
+  writeFileSync(join(viberDir, "readme.md"), data.readme_md, "utf-8");
+
+  // Append `.viber/` to .gitignore if not already present.
+  const gitignorePath = join(cwd, ".gitignore");
+  let current = "";
+  if (existsSync(gitignorePath)) {
+    current = readFileSync(gitignorePath, "utf-8");
+  }
+  const alreadyIgnored = current
+    .split(/\r?\n/)
+    .some((line) => {
+      const trimmed = line.trim();
+      return trimmed === ".viber/" || trimmed === ".viber";
+    });
+  if (!alreadyIgnored) {
+    const prefix = current === "" || current.endsWith("\n") ? "" : "\n";
+    appendFileSync(gitignorePath, `${prefix}.viber/\n`);
+  }
+}
+
+/**
+ * Run the claim flow. Writes .viber/auth.json + .viber/readme.md + .gitignore
+ * entry on success. Exits the process on any terminal outcome.
+ */
+export async function runConnect(
+  claimUrl: string,
+  cwd: string = process.cwd(),
+): Promise<void> {
+  const { baseUrl, claimId } = parseClaimUrl(claimUrl);
+
+  const fingerprint = clientFingerprint(cwd);
+  process.stderr.write(
+    `[viber-channel] fingerprint=${fingerprint.slice(0, 8)}…\n`,
+  );
+
+  const initResp = await fetch(`${baseUrl}/api/connect/${claimId}`, {
+    method: "POST",
+    headers: { "Content-Type": "application/json" },
+    body: JSON.stringify({ client_fingerprint: fingerprint }),
+  });
+  if (!initResp.ok) {
+    const body = await initResp.text().catch(() => "");
+    throw new Error(`Claim init failed (HTTP ${initResp.status}): ${body}`);
+  }
+  const init = (await initResp.json()) as InitResponse;
+  process.stdout.write(
+    `\nConfirm this connection in your browser:\n  ${init.confirm_url}\n\n`,
+  );
+  process.stdout.write(`Polling for confirmation (up to 10 minutes)…\n`);
+
+  const pollUrl = `${baseUrl}/api/connect/${claimId}?wait_token=${encodeURIComponent(init.wait_token)}`;
+  const deadline = Date.now() + MAX_POLL_DURATION_MS;
+  while (Date.now() < deadline) {
+    await sleep(POLL_INTERVAL_MS);
+    const pollResp = await fetch(pollUrl);
+    if (!pollResp.ok) {
+      const body = await pollResp.text().catch(() => "");
+      throw new Error(`Poll failed (HTTP ${pollResp.status}): ${body}`);
+    }
+    const data = (await pollResp.json()) as PollResponse;
+    if (data.status === "consumed") {
+      writeAuthJson(cwd, data, fingerprint);
+      process.stdout.write(
+        `\n✓ Connected as ${data.user_email} to project '${data.project_name}'.\n` +
+          `  Wrote ${join(cwd, ".viber", "auth.json")}\n\n` +
+          `Next steps:\n` +
+          `  1. Register the channel MCP server (one-time per machine):\n` +
+          `       claude mcp add viber-channel --scope user -- bunx viber-channel@latest\n` +
+          `  2. Restart Claude Code in this directory.\n\n`,
+      );
+      return;
+    }
+    if (data.status === "rejected" || data.status === "expired") {
+      throw new Error(
+        `Connection ${data.status}${data.reason ? `: ${data.reason}` : ""}. Get a fresh URL from ${baseUrl}/projects.`,
+      );
+    }
+    // "pending" — keep polling
+  }
+  throw new Error(
+    `Confirmation timed out after ${MAX_POLL_DURATION_MS / 60_000} minutes. Get a fresh URL from ${baseUrl}/projects.`,
+  );
+}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "viber-channel",
-  "version": "0.4.0",
+  "version": "0.5.0",
   "description": "Voice + text MCP channel between a Claude Code session and the Viber UI (https://viber.dgypx.dev). Push transcripts to Claude; send_message tool delivers text back to the UI.",
   "type": "module",
   "bin": {

package/viber-channel.ts

@@ -14,6 +14,44 @@ import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js"
 import { ListToolsRequestSchema, CallToolRequestSchema } from "@modelcontextprotocol/sdk/types.js";
 import { mkdirSync, writeFileSync, readFileSync, unlinkSync } from "node:fs";
 import { join } from "node:path";
+import { runConnect } from "./lib/connect.ts";
+
+// ---- CLI subcommand dispatch (must happen before lock acquire + loadAuth) ----
+//
+// `bunx viber-channel connect <claim_url>` runs the one-shot claim flow and
+// exits — it does not start the MCP server. Any other invocation (no args, or
+// the unknown subcommand) falls through to the MCP server below.
+{
+  const sub = process.argv[2];
+  if (sub === "--help" || sub === "-h") {
+    process.stdout.write(
+      `viber-channel — voice + text MCP channel for Claude Code\n\n` +
+      `USAGE\n` +
+      `  bunx viber-channel connect <claim_url>   one-shot auth flow; writes .viber/auth.json\n` +
+      `  bunx viber-channel                       start the MCP server (used by Claude Code)\n\n` +
+      `Get <claim_url> from https://viber.dgypx.dev/projects → Create connection.\n`,
+    );
+    process.exit(0);
+  }
+  if (sub === "connect") {
+    const claimUrl = process.argv[3];
+    if (!claimUrl) {
+      process.stderr.write(
+        `[viber-channel] Missing claim URL.\n` +
+        `Usage: bunx viber-channel connect <claim_url>\n` +
+        `Get the URL from https://viber.dgypx.dev/projects → Create connection.\n`,
+      );
+      process.exit(1);
+    }
+    try {
+      await runConnect(claimUrl);
+      process.exit(0);
+    } catch (err) {
+      process.stderr.write(`[viber-channel] ${err instanceof Error ? err.message : String(err)}\n`);
+      process.exit(1);
+    }
+  }
+}
 
 // Lock file path: %APPDATA%/viber/ (Windows) or ~/.config/viber/ (Linux/Mac)
 const LOCK_DIR =
@@ -136,7 +174,7 @@ const label = process.env.VIBER_CONVERSATION_LABEL ?? defaultLabel(process.cwd()
 // ---- MCP server ----
 
 const mcp = new Server(
-  { name: "viber-channel", version: "0.3.0" },
+  { name: "viber-channel", version: "0.5.0" },
   {
     capabilities: {
       experimental: { "claude/channel": {} },