@nokai/cli 1.0.0 → 1.2.0

package/README.md

@@ -4,190 +4,33 @@
 
 [![License: MIT](https://img.shields.io/badge/License-MIT-blue.svg)](LICENSE)
 
-NokAI is an MCP server that enables **peer-to-peer communication between AI coding assistants**. Connect your Claude Code with your teammate's Cursor, exchange questions about each other's codebases, and even send action requests — all through encrypted WebRTC data channels.
+NokAI lets AI coding assistants talk to each other. Connect your Claude Code with your teammate's Cursor, ask questions about their codebase, share files, and send action requests — without leaving your editor.
 
-**No code or conversation data ever touches the server.** The signaling server only brokers the initial handshake.
-
-## Quick Start
+## Get Started
 
 ```bash
-# Install globally
-npm install -g nokai
-
-# Add to Claude Code MCP settings
-# See docs/setup/ for your specific tool
-```
-
-### Connect with a Teammate
-
-```
-You: "Connect me with my teammate"
-NokAI: 🔗 Connection code: 847293 (expires in 5 minutes)
-
---- Your teammate ---
-
-Teammate: "Join NokAI with code 847293"
-NokAI: ✓ Connected with Michael!
-```
-
-### Ask Questions Across Codebases
-
-```
-You: "Ask Sara what the message type for /cmd_vel is
-      and build me the Flutter model for it"
-
-NokAI: → Asking Sara...
-       ← Sara: geometry_msgs/msg/Twist at 10Hz
-              Fields: linear.x (float64), angular.z (float64)
-       
-       Building your Flutter model class...
-       ✓ Created lib/models/twist_model.dart
+npm install -g @nokai/cli
+nokai
 ```
 
-## Supported Platforms
-
-| Platform | Status | Setup Guide |
-|---|---|---|
-| Claude Code (CLI & VS Code) | ✅ Full support | [docs/setup/claude-code.md](docs/setup/claude-code.md) |
-| Cursor | ✅ Full support | [docs/setup/cursor.md](docs/setup/cursor.md) |
-| Windsurf (Codeium) | ✅ Full support | [docs/setup/windsurf.md](docs/setup/windsurf.md) |
-| Any MCP tool | ✅ Full support | [docs/setup/generic-mcp.md](docs/setup/generic-mcp.md) |
-
-## Tools (13)
-
-### Connection
-
-| Tool | Description |
-|---|---|
-| `nokai_connect` | Generate a 6-digit OTP connection code |
-| `nokai_join` | Join using an OTP code |
-| `nokai_reconnect` | Reconnect to a saved peer (no new OTP) |
-| `nokai_disconnect` | Disconnect from a peer |
-| `nokai_connections` | List all active and saved connections |
-
-### Communication
+The setup wizard will ask your name, which AI tool you use, and configure everything automatically. No JSON editing required.
 
-| Tool | Description |
-|---|---|
-| `nokai_ask` | Ask a question to a peer's AI agent |
-| `nokai_respond` | Reply to an incoming question |
-| `nokai_tell` | Send an action request (requires read-write permission) |
-
-### Management
-
-| Tool | Description |
-|---|---|
-| `nokai_permissions` | View/change permission levels (read-only, read-write, ask-only) |
-| `nokai_approve` | Toggle manual approval mode for incoming queries |
-| `nokai_history` | Browse past message exchanges |
-| `nokai_audit` | View audit log of all events |
-| `nokai_profile` | Manage agent profile and capability cards |
-
-## Architecture
-
-```
-Your Machine                          Teammate's Machine
-┌────────────────┐                    ┌────────────────┐
-│ AI Assistant    │                    │ AI Assistant    │
-│ (Claude/Cursor) │                    │ (Claude/Cursor) │
-│   ↕ MCP        │                    │   ↕ MCP        │
-│ NokAI Server   │◄══════════════════►│ NokAI Server   │
-│ (local)        │  Encrypted P2P     │ (local)        │
-└────────────────┘  (WebRTC)          └────────────────┘
-         │                                     │
-         └──────── Signaling Server ───────────┘
-                   (handshake only)
-```
+## What You Can Do
 
-- **P2P by default** — messages flow directly between machines
-- **TURN fallback** — relay for firewalled networks (env vars)
-- **Offline queue** — messages queued locally, delivered on reconnect
-- **End-to-end encrypted** — WebRTC DTLS + Ed25519 signatures
+- **Ask across codebases** — query a teammate's agent about their code
+- **Send action requests** — ask a peer's agent to make changes
+- **Share files** — send files and project context directly to another agent
+- **Connect your own projects** — link your mobile and web sessions
+- **Discover peers** — see who's online and reconnect instantly
 
-## Security
+## Works With
 
-| Feature | Implementation |
-|---|---|
-| Agent identity | Ed25519 keypair per agent |
-| Connection auth | 6-digit OTP, 5-min expiry, single-use |
-| Transport | WebRTC DTLS encryption |
-| Server access | Signaling server never sees message content |
-| Local storage | Private keys at `~/.nokai/profile.json` (chmod 600) |
-| Reconnection | Peer confirmation required, public key verification |
+Claude Code, Cursor, Windsurf, and any MCP-compatible tool.
 
-## Self-Hosting the Signaling Server
+## Full Docs
 
-The signaling server runs on Cloudflare Workers (free tier):
-
-```bash
-cd packages/signaling-server
-npx wrangler d1 create nokai-signaling  # Create D1 database
-# Update wrangler.toml with the database ID
-npx wrangler deploy                      # Deploy to Cloudflare
-```
-
-Set your deployed URL as `NOKAI_SIGNALING_URL` in your MCP config.
-
-## Environment Variables
-
-| Variable | Required | Default | Description |
-|---|---|---|---|
-| `NOKAI_AGENT_NAME` | No | "Agent" | Your agent's display name |
-| `NOKAI_SIGNALING_URL` | No | — | Signaling server WebSocket URL |
-| `NOKAI_TURN_SERVER` | No | — | TURN relay server URL |
-| `NOKAI_TURN_USERNAME` | No | — | TURN server username |
-| `NOKAI_TURN_PASSWORD` | No | — | TURN server password |
-
-## Development
-
-```bash
-# Clone and install
-git clone https://github.com/heiisenberrg/nokAI.git
-cd nokAI
-npm install
-
-# Build all packages
-npm run build
-
-# Run tests (125 tests)
-npm test
-
-# Dev mode (MCP server)
-cd packages/mcp-server && npm run dev
-```
-
-## Project Structure
-
-```
-nokai/
-├── packages/
-│   ├── shared/              # Shared types, constants, type guards
-│   ├── mcp-server/          # MCP server + P2P engine (13 tools)
-│   │   ├── src/tools/       # Tool handlers
-│   │   └── tests/           # 100+ tests
-│   └── signaling-server/    # Cloudflare Workers signaling server
-└── docs/setup/              # Platform-specific setup guides
-```
-
-## Roadmap
-
-- [x] v0.1.0 — Core: connect, ask, respond via P2P
-- [x] v0.2.0 — Reconnection, history, TURN, offline queue, approval mode
-- [x] v0.3.0 — Actions (tell), permission levels
-- [x] v0.4.0 — Audit logs, agent profiles
-- [x] v1.0.0 — Multi-platform, npm packaging
-- [ ] v1.1.0 — Web dashboard for connection management
-- [ ] v1.2.0 — Agent capability discovery within organizations
-- [ ] v2.0.0 — Cross-organization connections
+https://nokai.pages.dev
 
 ## License
 
-MIT — see [LICENSE](LICENSE)
-
-## Contributing
-
-Contributions welcome! Please open an issue first to discuss.
-
----
-
-Built with frustration at copy-pasting between Slack and Claude.
+MIT

package/dist/cli.d.ts

@@ -0,0 +1,2 @@
+export declare function runCli(): Promise<void>;
+//# sourceMappingURL=cli.d.ts.map

package/dist/cli.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"cli.d.ts","sourceRoot":"","sources":["../src/cli.ts"],"names":[],"mappings":"AAuPA,wBAAsB,MAAM,IAAI,OAAO,CAAC,IAAI,CAAC,CAS5C"}

package/dist/cli.js

@@ -0,0 +1,239 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.runCli = runCli;
+const node_readline_1 = require("node:readline");
+const node_fs_1 = require("node:fs");
+const node_path_1 = require("node:path");
+const profile_js_1 = require("./profile.js");
+const rl = (0, node_readline_1.createInterface)({ input: process.stdin, output: process.stdout });
+function ask(question) {
+    return new Promise((resolve) => {
+        rl.question(question, (answer) => resolve(answer.trim()));
+    });
+}
+function choose(question, options) {
+    return new Promise((resolve) => {
+        console.log(`\n  ${question}\n`);
+        options.forEach((opt, i) => {
+            console.log(`    ${i + 1}) ${opt}`);
+        });
+        console.log();
+        rl.question("  Enter number: ", (answer) => {
+            const idx = parseInt(answer.trim(), 10) - 1;
+            if (idx >= 0 && idx < options.length) {
+                resolve(options[idx]);
+            }
+            else {
+                resolve(options[0]);
+            }
+        });
+    });
+}
+function getPlatforms() {
+    const home = process.env.HOME || process.env.USERPROFILE || "~";
+    return [
+        {
+            name: "Claude Code",
+            configPath: (0, node_path_1.join)(home, ".claude.json"),
+            configDir: (0, node_path_1.join)(home),
+        },
+        {
+            name: "Cursor",
+            configPath: (0, node_path_1.join)(process.cwd(), ".cursor", "mcp.json"),
+            configDir: (0, node_path_1.join)(process.cwd(), ".cursor"),
+        },
+        {
+            name: "Windsurf",
+            configPath: (0, node_path_1.join)(home, ".codeium", "windsurf", "mcp_config.json"),
+            configDir: (0, node_path_1.join)(home, ".codeium", "windsurf"),
+        },
+        {
+            name: "VS Code (Extension)",
+            configPath: "",
+            configDir: "",
+        },
+        {
+            name: "Other",
+            configPath: "",
+            configDir: "",
+        },
+    ];
+}
+function writeMcpConfig(platform, agentName, signalingUrl) {
+    if (!platform.configPath)
+        return false;
+    try {
+        // Ensure directory exists
+        if (platform.configDir && !(0, node_fs_1.existsSync)(platform.configDir)) {
+            (0, node_fs_1.mkdirSync)(platform.configDir, { recursive: true });
+        }
+        // Read existing config or create new
+        let config = {};
+        if ((0, node_fs_1.existsSync)(platform.configPath)) {
+            try {
+                config = JSON.parse((0, node_fs_1.readFileSync)(platform.configPath, "utf-8"));
+            }
+            catch {
+                config = {};
+            }
+        }
+        // Add/update mcpServers.nokai
+        if (!config.mcpServers || typeof config.mcpServers !== "object") {
+            config.mcpServers = {};
+        }
+        config.mcpServers.nokai = {
+            command: "nokai",
+            args: ["serve"],
+            env: {
+                NOKAI_AGENT_NAME: agentName,
+                NOKAI_SIGNALING_URL: signalingUrl,
+            },
+        };
+        (0, node_fs_1.writeFileSync)(platform.configPath, JSON.stringify(config, null, 2), "utf-8");
+        return true;
+    }
+    catch {
+        return false;
+    }
+}
+const SIGNALING_URL = "wss://nokai-signaling.ajaykkumar-rajen.workers.dev";
+async function firstTimeSetup() {
+    console.log("\n  ✨ Welcome to NokAI! Let's set you up.\n");
+    // Ask for agent name
+    const name = await ask("  What would you like to name your agent? ");
+    if (!name) {
+        console.log("  Agent name is required.");
+        rl.close();
+        return;
+    }
+    // Choose platform
+    const platforms = getPlatforms();
+    const platformNames = platforms.map((p) => p.name);
+    const chosen = await choose("Which AI tool are you using?", platformNames);
+    const platform = platforms.find((p) => p.name === chosen) || platforms[platforms.length - 1];
+    // Create profile
+    const profileDir = (0, profile_js_1.getProfileDir)();
+    const profile = (0, profile_js_1.createProfile)(name, profileDir);
+    console.log(`\n  ✓ Agent created: ${name}`);
+    console.log(`  ✓ Your NokAI ID: ${profile.nokaiId}`);
+    // Configure MCP
+    if (platform.configPath) {
+        const confirm = await ask(`\n  Add NokAI to ${platform.name} config? (${platform.configPath}) [Y/n] `);
+        if (confirm.toLowerCase() !== "n") {
+            const success = writeMcpConfig(platform, name, SIGNALING_URL);
+            if (success) {
+                console.log(`  ✓ MCP configuration added to ${platform.name}`);
+            }
+            else {
+                console.log(`  ✗ Failed to write config. You can add it manually.`);
+                showManualConfig(name);
+            }
+        }
+        else {
+            showManualConfig(name);
+        }
+    }
+    else if (platform.name === "VS Code (Extension)") {
+        console.log(`\n  Install the NokAI VS Code extension and set your Signaling URL in settings.`);
+        console.log(`  Signaling URL: ${SIGNALING_URL}`);
+    }
+    else {
+        showManualConfig(name);
+    }
+    console.log(`\n  You're ready! Restart your AI tool to activate NokAI.`);
+    console.log(`  Share your NokAI ID with teammates: ${profile.nokaiId}\n`);
+    rl.close();
+}
+function showManualConfig(agentName) {
+    console.log(`\n  Add this to your MCP configuration:\n`);
+    console.log(JSON.stringify({
+        mcpServers: {
+            nokai: {
+                command: "nokai",
+                args: ["serve"],
+                env: {
+                    NOKAI_AGENT_NAME: agentName,
+                    NOKAI_SIGNALING_URL: SIGNALING_URL,
+                },
+            },
+        },
+    }, null, 2).split("\n").map((l) => `  ${l}`).join("\n"));
+}
+async function returningUser() {
+    const profileDir = (0, profile_js_1.getProfileDir)();
+    const profile = (0, profile_js_1.loadProfile)(profileDir);
+    if (!profile) {
+        return firstTimeSetup();
+    }
+    console.log(`\n  Welcome back, ${profile.name}! (${profile.nokaiId})\n`);
+    const action = await choose("What would you like to do?", [
+        "Connect to a teammate",
+        "Connect my own projects",
+        "Show my NokAI ID",
+        "View connections",
+        "Reconfigure MCP settings",
+        "Exit",
+    ]);
+    switch (action) {
+        case "Connect to a teammate": {
+            const peerId = await ask("\n  Enter teammate's NokAI ID (e.g., NKI-XXXX-XXXX): ");
+            if (peerId) {
+                console.log(`\n  To connect, use this in your AI tool:`);
+                console.log(`  nokai_connect mode=teammate id=${peerId}\n`);
+            }
+            break;
+        }
+        case "Connect my own projects": {
+            console.log(`\n  To connect your sessions, use this in your AI tool:`);
+            console.log(`  nokai_connect mode=self\n`);
+            console.log(`  This will list your active sessions and let you connect them.\n`);
+            break;
+        }
+        case "Show my NokAI ID": {
+            console.log(`\n  NokAI ID:    ${profile.nokaiId}`);
+            console.log(`  Agent Name:  ${profile.name}`);
+            console.log(`  Device ID:   ${profile.deviceId}\n`);
+            console.log(`  Share your NokAI ID with teammates so they can connect to you.\n`);
+            break;
+        }
+        case "View connections": {
+            console.log(`\n  To view connections, use nokai_connections in your AI tool.\n`);
+            break;
+        }
+        case "Reconfigure MCP settings": {
+            const platforms = getPlatforms();
+            const platformNames = platforms.map((p) => p.name);
+            const chosen = await choose("Which AI tool?", platformNames);
+            const platform = platforms.find((p) => p.name === chosen) || platforms[platforms.length - 1];
+            if (platform.configPath) {
+                const success = writeMcpConfig(platform, profile.name, SIGNALING_URL);
+                if (success) {
+                    console.log(`\n  ✓ MCP configuration updated for ${platform.name}`);
+                    console.log(`  Restart your AI tool to apply changes.\n`);
+                }
+                else {
+                    console.log(`\n  ✗ Failed to write config.`);
+                    showManualConfig(profile.name);
+                }
+            }
+            else {
+                showManualConfig(profile.name);
+            }
+            break;
+        }
+        case "Exit":
+            break;
+    }
+    rl.close();
+}
+async function runCli() {
+    const profileDir = (0, profile_js_1.getProfileDir)();
+    const profile = (0, profile_js_1.loadProfile)(profileDir);
+    if (!profile) {
+        await firstTimeSetup();
+    }
+    else {
+        await returningUser();
+    }
+}
+//# sourceMappingURL=cli.js.map

package/dist/cli.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"cli.js","sourceRoot":"","sources":["../src/cli.ts"],"names":[],"mappings":";;AAuPA,wBASC;AAhQD,iDAAgD;AAChD,qCAA6E;AAC7E,yCAAiC;AACjC,6CAA0F;AAE1F,MAAM,EAAE,GAAG,IAAA,+BAAe,EAAC,EAAE,KAAK,EAAE,OAAO,CAAC,KAAK,EAAE,MAAM,EAAE,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC;AAE7E,SAAS,GAAG,CAAC,QAAgB;IAC3B,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE;QAC7B,EAAE,CAAC,QAAQ,CAAC,QAAQ,EAAE,CAAC,MAAM,EAAE,EAAE,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;IAC5D,CAAC,CAAC,CAAC;AACL,CAAC;AAED,SAAS,MAAM,CAAC,QAAgB,EAAE,OAAiB;IACjD,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE;QAC7B,OAAO,CAAC,GAAG,CAAC,OAAO,QAAQ,IAAI,CAAC,CAAC;QACjC,OAAO,CAAC,OAAO,CAAC,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE;YACzB,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,KAAK,GAAG,EAAE,CAAC,CAAC;QACtC,CAAC,CAAC,CAAC;QACH,OAAO,CAAC,GAAG,EAAE,CAAC;QACd,EAAE,CAAC,QAAQ,CAAC,kBAAkB,EAAE,CAAC,MAAM,EAAE,EAAE;YACzC,MAAM,GAAG,GAAG,QAAQ,CAAC,MAAM,CAAC,IAAI,EAAE,EAAE,EAAE,CAAC,GAAG,CAAC,CAAC;YAC5C,IAAI,GAAG,IAAI,CAAC,IAAI,GAAG,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC;gBACrC,OAAO,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC;YACxB,CAAC;iBAAM,CAAC;gBACN,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC;YACtB,CAAC;QACH,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC;AAQD,SAAS,YAAY;IACnB,MAAM,IAAI,GAAG,OAAO,CAAC,GAAG,CAAC,IAAI,IAAI,OAAO,CAAC,GAAG,CAAC,WAAW,IAAI,GAAG,CAAC;IAChE,OAAO;QACL;YACE,IAAI,EAAE,aAAa;YACnB,UAAU,EAAE,IAAA,gBAAI,EAAC,IAAI,EAAE,cAAc,CAAC;YACtC,SAAS,EAAE,IAAA,gBAAI,EAAC,IAAI,CAAC;SACtB;QACD;YACE,IAAI,EAAE,QAAQ;YACd,UAAU,EAAE,IAAA,gBAAI,EAAC,OAAO,CAAC,GAAG,EAAE,EAAE,SAAS,EAAE,UAAU,CAAC;YACtD,SAAS,EAAE,IAAA,gBAAI,EAAC,OAAO,CAAC,GAAG,EAAE,EAAE,SAAS,CAAC;SAC1C;QACD;YACE,IAAI,EAAE,UAAU;YAChB,UAAU,EAAE,IAAA,gBAAI,EAAC,IAAI,EAAE,UAAU,EAAE,UAAU,EAAE,iBAAiB,CAAC;YACjE,SAAS,EAAE,IAAA,gBAAI,EAAC,IAAI,EAAE,UAAU,EAAE,UAAU,CAAC;SAC9C;QACD;YACE,IAAI,EAAE,qBAAqB;YAC3B,UAAU,EAAE,EAAE;YACd,SAAS,EAAE,EAAE;SACd;QACD;YACE,IAAI,EAAE,OAAO;YACb,UAAU,EAAE,EAAE;YACd,SAAS,EAAE,EAAE;SACd;KACF,CAAC;AACJ,CAAC;AAED,SAAS,cAAc,CAAC,QAAwB,EAAE,SAAiB,EAAE,YAAoB;IACvF,IAAI,CAAC,QAAQ,CAAC,UAAU;QAAE,OAAO,KAAK,CAAC;IAEvC,IAAI,CAAC;QACH,0BAA0B;QAC1B,IAAI,QAAQ,CAAC,SAAS,IAAI,CAAC,IAAA,oBAAU,EAAC,QAAQ,CAAC,SAAS,CAAC,EAAE,CAAC;YAC1D,IAAA,mBAAS,EAAC,QAAQ,CAAC,SAAS,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QACrD,CAAC;QAED,qCAAqC;QACrC,IAAI,MAAM,GAA4B,EAAE,CAAC;QACzC,IAAI,IAAA,oBAAU,EAAC,QAAQ,CAAC,UAAU,CAAC,EAAE,CAAC;YACpC,IAAI,CAAC;gBACH,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,IAAA,sBAAY,EAAC,QAAQ,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC,CAAC;YAClE,CAAC;YAAC,MAAM,CAAC;gBACP,MAAM,GAAG,EAAE,CAAC;YACd,CAAC;QACH,CAAC;QAED,8BAA8B;QAC9B,IAAI,CAAC,MAAM,CAAC,UAAU,IAAI,OAAO,MAAM,CAAC,UAAU,KAAK,QAAQ,EAAE,CAAC;YAChE,MAAM,CAAC,UAAU,GAAG,EAAE,CAAC;QACzB,CAAC;QAEA,MAAM,CAAC,UAAsC,CAAC,KAAK,GAAG;YACrD,OAAO,EAAE,OAAO;YAChB,IAAI,EAAE,CAAC,OAAO,CAAC;YACf,GAAG,EAAE;gBACH,gBAAgB,EAAE,SAAS;gBAC3B,mBAAmB,EAAE,YAAY;aAClC;SACF,CAAC;QAEF,IAAA,uBAAa,EAAC,QAAQ,CAAC,UAAU,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC,EAAE,OAAO,CAAC,CAAC;QAC7E,OAAO,IAAI,CAAC;IACd,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,KAAK,CAAC;IACf,CAAC;AACH,CAAC;AAED,MAAM,aAAa,GAAG,oDAAoD,CAAC;AAE3E,KAAK,UAAU,cAAc;IAC3B,OAAO,CAAC,GAAG,CAAC,6CAA6C,CAAC,CAAC;IAE3D,qBAAqB;IACrB,MAAM,IAAI,GAAG,MAAM,GAAG,CAAC,4CAA4C,CAAC,CAAC;IACrE,IAAI,CAAC,IAAI,EAAE,CAAC;QACV,OAAO,CAAC,GAAG,CAAC,2BAA2B,CAAC,CAAC;QACzC,EAAE,CAAC,KAAK,EAAE,CAAC;QACX,OAAO;IACT,CAAC;IAED,kBAAkB;IAClB,MAAM,SAAS,GAAG,YAAY,EAAE,CAAC;IACjC,MAAM,aAAa,GAAG,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;IACnD,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,8BAA8B,EAAE,aAAa,CAAC,CAAC;IAC3E,MAAM,QAAQ,GAAG,SAAS,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,CAAC,IAAI,SAAS,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;IAE7F,iBAAiB;IACjB,MAAM,UAAU,GAAG,IAAA,0BAAa,GAAE,CAAC;IACnC,MAAM,OAAO,GAAG,IAAA,0BAAa,EAAC,IAAI,EAAE,UAAU,CAAC,CAAC;IAEhD,OAAO,CAAC,GAAG,CAAC,wBAAwB,IAAI,EAAE,CAAC,CAAC;IAC5C,OAAO,CAAC,GAAG,CAAC,sBAAsB,OAAO,CAAC,OAAO,EAAE,CAAC,CAAC;IAErD,gBAAgB;IAChB,IAAI,QAAQ,CAAC,UAAU,EAAE,CAAC;QACxB,MAAM,OAAO,GAAG,MAAM,GAAG,CAAC,oBAAoB,QAAQ,CAAC,IAAI,aAAa,QAAQ,CAAC,UAAU,UAAU,CAAC,CAAC;QACvG,IAAI,OAAO,CAAC,WAAW,EAAE,KAAK,GAAG,EAAE,CAAC;YAClC,MAAM,OAAO,GAAG,cAAc,CAAC,QAAQ,EAAE,IAAI,EAAE,aAAa,CAAC,CAAC;YAC9D,IAAI,OAAO,EAAE,CAAC;gBACZ,OAAO,CAAC,GAAG,CAAC,kCAAkC,QAAQ,CAAC,IAAI,EAAE,CAAC,CAAC;YACjE,CAAC;iBAAM,CAAC;gBACN,OAAO,CAAC,GAAG,CAAC,sDAAsD,CAAC,CAAC;gBACpE,gBAAgB,CAAC,IAAI,CAAC,CAAC;YACzB,CAAC;QACH,CAAC;aAAM,CAAC;YACN,gBAAgB,CAAC,IAAI,CAAC,CAAC;QACzB,CAAC;IACH,CAAC;SAAM,IAAI,QAAQ,CAAC,IAAI,KAAK,qBAAqB,EAAE,CAAC;QACnD,OAAO,CAAC,GAAG,CAAC,iFAAiF,CAAC,CAAC;QAC/F,OAAO,CAAC,GAAG,CAAC,oBAAoB,aAAa,EAAE,CAAC,CAAC;IACnD,CAAC;SAAM,CAAC;QACN,gBAAgB,CAAC,IAAI,CAAC,CAAC;IACzB,CAAC;IAED,OAAO,CAAC,GAAG,CAAC,2DAA2D,CAAC,CAAC;IACzE,OAAO,CAAC,GAAG,CAAC,yCAAyC,OAAO,CAAC,OAAO,IAAI,CAAC,CAAC;IAC1E,EAAE,CAAC,KAAK,EAAE,CAAC;AACb,CAAC;AAED,SAAS,gBAAgB,CAAC,SAAiB;IACzC,OAAO,CAAC,GAAG,CAAC,2CAA2C,CAAC,CAAC;IACzD,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,SAAS,CAAC;QACzB,UAAU,EAAE;YACV,KAAK,EAAE;gBACL,OAAO,EAAE,OAAO;gBAChB,IAAI,EAAE,CAAC,OAAO,CAAC;gBACf,GAAG,EAAE;oBACH,gBAAgB,EAAE,SAAS;oBAC3B,mBAAmB,EAAE,aAAa;iBACnC;aACF;SACF;KACF,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;AAC3D,CAAC;AAED,KAAK,UAAU,aAAa;IAC1B,MAAM,UAAU,GAAG,IAAA,0BAAa,GAAE,CAAC;IACnC,MAAM,OAAO,GAAG,IAAA,wBAAW,EAAC,UAAU,CAAC,CAAC;IACxC,IAAI,CAAC,OAAO,EAAE,CAAC;QACb,OAAO,cAAc,EAAE,CAAC;IAC1B,CAAC;IAED,OAAO,CAAC,GAAG,CAAC,qBAAqB,OAAO,CAAC,IAAI,MAAM,OAAO,CAAC,OAAO,KAAK,CAAC,CAAC;IAEzE,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,4BAA4B,EAAE;QACxD,uBAAuB;QACvB,yBAAyB;QACzB,kBAAkB;QAClB,kBAAkB;QAClB,0BAA0B;QAC1B,MAAM;KACP,CAAC,CAAC;IAEH,QAAQ,MAAM,EAAE,CAAC;QACf,KAAK,uBAAuB,CAAC,CAAC,CAAC;YAC7B,MAAM,MAAM,GAAG,MAAM,GAAG,CAAC,uDAAuD,CAAC,CAAC;YAClF,IAAI,MAAM,EAAE,CAAC;gBACX,OAAO,CAAC,GAAG,CAAC,2CAA2C,CAAC,CAAC;gBACzD,OAAO,CAAC,GAAG,CAAC,oCAAoC,MAAM,IAAI,CAAC,CAAC;YAC9D,CAAC;YACD,MAAM;QACR,CAAC;QACD,KAAK,yBAAyB,CAAC,CAAC,CAAC;YAC/B,OAAO,CAAC,GAAG,CAAC,yDAAyD,CAAC,CAAC;YACvE,OAAO,CAAC,GAAG,CAAC,6BAA6B,CAAC,CAAC;YAC3C,OAAO,CAAC,GAAG,CAAC,mEAAmE,CAAC,CAAC;YACjF,MAAM;QACR,CAAC;QACD,KAAK,kBAAkB,CAAC,CAAC,CAAC;YACxB,OAAO,CAAC,GAAG,CAAC,oBAAoB,OAAO,CAAC,OAAO,EAAE,CAAC,CAAC;YACnD,OAAO,CAAC,GAAG,CAAC,kBAAkB,OAAO,CAAC,IAAI,EAAE,CAAC,CAAC;YAC9C,OAAO,CAAC,GAAG,CAAC,kBAAkB,OAAO,CAAC,QAAQ,IAAI,CAAC,CAAC;YACpD,OAAO,CAAC,GAAG,CAAC,oEAAoE,CAAC,CAAC;YAClF,MAAM;QACR,CAAC;QACD,KAAK,kBAAkB,CAAC,CAAC,CAAC;YACxB,OAAO,CAAC,GAAG,CAAC,mEAAmE,CAAC,CAAC;YACjF,MAAM;QACR,CAAC;QACD,KAAK,0BAA0B,CAAC,CAAC,CAAC;YAChC,MAAM,SAAS,GAAG,YAAY,EAAE,CAAC;YACjC,MAAM,aAAa,GAAG,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;YACnD,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,gBAAgB,EAAE,aAAa,CAAC,CAAC;YAC7D,MAAM,QAAQ,GAAG,SAAS,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,CAAC,IAAI,SAAS,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;YAE7F,IAAI,QAAQ,CAAC,UAAU,EAAE,CAAC;gBACxB,MAAM,OAAO,GAAG,cAAc,CAAC,QAAQ,EAAE,OAAO,CAAC,IAAI,EAAE,aAAa,CAAC,CAAC;gBACtE,IAAI,OAAO,EAAE,CAAC;oBACZ,OAAO,CAAC,GAAG,CAAC,uCAAuC,QAAQ,CAAC,IAAI,EAAE,CAAC,CAAC;oBACpE,OAAO,CAAC,GAAG,CAAC,4CAA4C,CAAC,CAAC;gBAC5D,CAAC;qBAAM,CAAC;oBACN,OAAO,CAAC,GAAG,CAAC,+BAA+B,CAAC,CAAC;oBAC7C,gBAAgB,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;gBACjC,CAAC;YACH,CAAC;iBAAM,CAAC;gBACN,gBAAgB,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;YACjC,CAAC;YACD,MAAM;QACR,CAAC;QACD,KAAK,MAAM;YACT,MAAM;IACV,CAAC;IAED,EAAE,CAAC,KAAK,EAAE,CAAC;AACb,CAAC;AAEM,KAAK,UAAU,MAAM;IAC1B,MAAM,UAAU,GAAG,IAAA,0BAAa,GAAE,CAAC;IACnC,MAAM,OAAO,GAAG,IAAA,wBAAW,EAAC,UAAU,CAAC,CAAC;IAExC,IAAI,CAAC,OAAO,EAAE,CAAC;QACb,MAAM,cAAc,EAAE,CAAC;IACzB,CAAC;SAAM,CAAC;QACN,MAAM,aAAa,EAAE,CAAC;IACxB,CAAC;AACH,CAAC"}

package/dist/connection-manager.d.ts

@@ -1,13 +1,19 @@
 import { EventEmitter } from "node:events";
 import type { AgentProfile, PeerMessage, ConnectionRecord } from "./shared/index.js";
 import { type NokaiDB } from "./db.js";
+import type { SharedFile, ProjectContext } from "./shared/index.js";
 export declare class ConnectionManager extends EventEmitter {
     private db;
     private profile;
     private activePeers;
     private incomingHandler;
     private heldQueries;
-    constructor(db: NokaiDB, profile: AgentProfile);
+    private signalingUrl;
+    private pendingBytes;
+    private transferReportTimer;
+    constructor(db: NokaiDB, profile: AgentProfile, signalingUrl?: string);
+    private trackBytes;
+    private flushTransferStats;
     registerPeer(connectionId: string, peerName: string, peerPublicKey: string): void;
     setPeerSend(connectionId: string, sendFn: (msg: string) => void): void;
     private findPeerByName;
@@ -15,6 +21,8 @@ export declare class ConnectionManager extends EventEmitter {
     sendResponse(peerName: string, replyTo: string, content: string, sourceFiles?: string[]): PeerMessage;
     sendAction(peerName: string, content: string, context?: string): PeerMessage;
     sendActionResult(peerName: string, replyTo: string, content: string, success: boolean, sourceFiles?: string[]): PeerMessage;
+    sendShareFiles(peerName: string, files: SharedFile[]): PeerMessage;
+    sendShareContext(peerName: string, context: ProjectContext): PeerMessage;
     handleIncomingMessage(connectionId: string, rawData: string): void;
     onIncomingMessage(handler: (msg: PeerMessage, peerName: string) => void): void;
     getActiveConnections(): Array<{

package/dist/connection-manager.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"connection-manager.d.ts","sourceRoot":"","sources":["../src/connection-manager.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AAC3C,OAAO,KAAK,EAAE,YAAY,EAAE,WAAW,EAAE,gBAAgB,EAAE,MAAM,mBAAmB,CAAC;AACrF,OAAO,EAAE,KAAK,OAAO,EAAE,MAAM,SAAS,CAAC;AAkBvC,qBAAa,iBAAkB,SAAQ,YAAY;IACjD,OAAO,CAAC,EAAE,CAAU;IACpB,OAAO,CAAC,OAAO,CAAe;IAC9B,OAAO,CAAC,WAAW,CAAsC;IACzD,OAAO,CAAC,eAAe,CAA+D;IACtF,OAAO,CAAC,WAAW,CAAuG;gBAE9G,EAAE,EAAE,OAAO,EAAE,OAAO,EAAE,YAAY;IAM9C,YAAY,CAAC,YAAY,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,EAAE,aAAa,EAAE,MAAM,GAAG,IAAI;IAqBjF,WAAW,CAAC,YAAY,EAAE,MAAM,EAAE,MAAM,EAAE,CAAC,GAAG,EAAE,MAAM,KAAK,IAAI,GAAG,IAAI;IAOtE,OAAO,CAAC,cAAc;IAOtB,SAAS,CAAC,QAAQ,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,MAAM,GAAG,WAAW;IA4B3E,YAAY,CAAC,QAAQ,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,WAAW,CAAC,EAAE,MAAM,EAAE,GAAG,WAAW;IA2BrG,UAAU,CAAC,QAAQ,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,MAAM,GAAG,WAAW;IAkC5E,gBAAgB,CAAC,QAAQ,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,WAAW,CAAC,EAAE,MAAM,EAAE,GAAG,WAAW;IAwB3H,qBAAqB,CAAC,YAAY,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,GAAG,IAAI;IAwBlE,iBAAiB,CAAC,OAAO,EAAE,CAAC,GAAG,EAAE,WAAW,EAAE,QAAQ,EAAE,MAAM,KAAK,IAAI,GAAG,IAAI;IAI9E,oBAAoB,IAAI,KAAK,CAAC;QAAE,YAAY,EAAE,MAAM,CAAC;QAAC,QAAQ,EAAE,MAAM,CAAA;KAAE,CAAC;IAOzE,cAAc,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO;IAazC,YAAY,CAAC,YAAY,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,IAAI,GAAG,IAAI;IAO7D,oBAAoB,CAAC,YAAY,EAAE,MAAM,EAAE,MAAM,EAAE,CAAC,GAAG,EAAE,MAAM,KAAK,IAAI,GAAG,IAAI;IAyB/E,YAAY,CAAC,YAAY,EAAE,MAAM,EAAE,iBAAiB,EAAE,MAAM,GAAG,IAAI;IAKnE,oBAAoB,CAAC,YAAY,EAAE,MAAM,GAAG,MAAM;IAelD,mBAAmB,IAAI,gBAAgB,EAAE;IAIzC,mBAAmB,CAAC,QAAQ,EAAE,MAAM,GAAG,gBAAgB,GAAG,IAAI;IAK9D,SAAS,CAAC,QAAQ,EAAE,MAAM,EAAE,KAAK,EAAE;QAAE,EAAE,EAAE,MAAM,CAAC;QAAC,OAAO,EAAE,MAAM,CAAC;QAAC,SAAS,EAAE,MAAM,CAAA;KAAE,GAAG,IAAI;IAM5F,cAAc,CAAC,QAAQ,EAAE,MAAM,GAAG,KAAK,CAAC;QAAE,EAAE,EAAE,MAAM,CAAC;QAAC,OAAO,EAAE,MAAM,CAAC;QAAC,QAAQ,EAAE,MAAM,CAAC;QAAC,SAAS,EAAE,MAAM,CAAA;KAAE,CAAC;IAI7G,eAAe,CAAC,QAAQ,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,GAAG,OAAO;IAS7D,aAAa,IAAI,IAAI;CAGtB"}
+{"version":3,"file":"connection-manager.d.ts","sourceRoot":"","sources":["../src/connection-manager.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AAC3C,OAAO,KAAK,EAAE,YAAY,EAAE,WAAW,EAAE,gBAAgB,EAAE,MAAM,mBAAmB,CAAC;AACrF,OAAO,EAAE,KAAK,OAAO,EAAE,MAAM,SAAS,CAAC;AAWvC,OAAO,KAAK,EAAE,UAAU,EAAE,cAAc,EAAE,MAAM,mBAAmB,CAAC;AAuBpE,qBAAa,iBAAkB,SAAQ,YAAY;IACjD,OAAO,CAAC,EAAE,CAAU;IACpB,OAAO,CAAC,OAAO,CAAe;IAC9B,OAAO,CAAC,WAAW,CAAsC;IACzD,OAAO,CAAC,eAAe,CAA+D;IACtF,OAAO,CAAC,WAAW,CAAuG;IAC1H,OAAO,CAAC,YAAY,CAAS;IAC7B,OAAO,CAAC,YAAY,CAAK;IACzB,OAAO,CAAC,mBAAmB,CAA+C;gBAE9D,EAAE,EAAE,OAAO,EAAE,OAAO,EAAE,YAAY,EAAE,YAAY,CAAC,EAAE,MAAM;IAYrE,OAAO,CAAC,UAAU;YAIJ,kBAAkB;IAQhC,YAAY,CAAC,YAAY,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,EAAE,aAAa,EAAE,MAAM,GAAG,IAAI;IA0BjF,WAAW,CAAC,YAAY,EAAE,MAAM,EAAE,MAAM,EAAE,CAAC,GAAG,EAAE,MAAM,KAAK,IAAI,GAAG,IAAI;IAOtE,OAAO,CAAC,cAAc;IAOtB,SAAS,CAAC,QAAQ,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,MAAM,GAAG,WAAW;IA6B3E,YAAY,CAAC,QAAQ,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,WAAW,CAAC,EAAE,MAAM,EAAE,GAAG,WAAW;IA4BrG,UAAU,CAAC,QAAQ,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,MAAM,GAAG,WAAW;IAmC5E,gBAAgB,CAAC,QAAQ,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,WAAW,CAAC,EAAE,MAAM,EAAE,GAAG,WAAW;IAyB3H,cAAc,CAAC,QAAQ,EAAE,MAAM,EAAE,KAAK,EAAE,UAAU,EAAE,GAAG,WAAW;IA0BlE,gBAAgB,CAAC,QAAQ,EAAE,MAAM,EAAE,OAAO,EAAE,cAAc,GAAG,WAAW;IA0BxE,qBAAqB,CAAC,YAAY,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,GAAG,IAAI;IAyBlE,iBAAiB,CAAC,OAAO,EAAE,CAAC,GAAG,EAAE,WAAW,EAAE,QAAQ,EAAE,MAAM,KAAK,IAAI,GAAG,IAAI;IAI9E,oBAAoB,IAAI,KAAK,CAAC;QAAE,YAAY,EAAE,MAAM,CAAC;QAAC,QAAQ,EAAE,MAAM,CAAA;KAAE,CAAC;IAOzE,cAAc,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO;IAazC,YAAY,CAAC,YAAY,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,IAAI,GAAG,IAAI;IAO7D,oBAAoB,CAAC,YAAY,EAAE,MAAM,EAAE,MAAM,EAAE,CAAC,GAAG,EAAE,MAAM,KAAK,IAAI,GAAG,IAAI;IAyB/E,YAAY,CAAC,YAAY,EAAE,MAAM,EAAE,iBAAiB,EAAE,MAAM,GAAG,IAAI;IAKnE,oBAAoB,CAAC,YAAY,EAAE,MAAM,GAAG,MAAM;IAelD,mBAAmB,IAAI,gBAAgB,EAAE;IAIzC,mBAAmB,CAAC,QAAQ,EAAE,MAAM,GAAG,gBAAgB,GAAG,IAAI;IAK9D,SAAS,CAAC,QAAQ,EAAE,MAAM,EAAE,KAAK,EAAE;QAAE,EAAE,EAAE,MAAM,CAAC;QAAC,OAAO,EAAE,MAAM,CAAC;QAAC,SAAS,EAAE,MAAM,CAAA;KAAE,GAAG,IAAI;IAM5F,cAAc,CAAC,QAAQ,EAAE,MAAM,GAAG,KAAK,CAAC;QAAE,EAAE,EAAE,MAAM,CAAC;QAAC,OAAO,EAAE,MAAM,CAAC;QAAC,QAAQ,EAAE,MAAM,CAAC;QAAC,SAAS,EAAE,MAAM,CAAA;KAAE,CAAC;IAI7G,eAAe,CAAC,QAAQ,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,GAAG,OAAO;IAS7D,aAAa,IAAI,IAAI;CAGtB"}

package/dist/connection-manager.js

@@ -3,16 +3,47 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.ConnectionManager = void 0;
 const node_events_1 = require("node:events");
 const message_protocol_js_1 = require("./message-protocol.js");
+async function reportStats(signalingUrl, endpoint, body) {
+    try {
+        const httpUrl = signalingUrl.replace("wss://", "https://").replace("ws://", "http://");
+        await fetch(`${httpUrl}/stats/${endpoint}`, {
+            method: "POST",
+            headers: { "Content-Type": "application/json" },
+            body: JSON.stringify(body),
+        });
+    }
+    catch {
+        // Stats reporting is best-effort, don't break on failure
+    }
+}
 class ConnectionManager extends node_events_1.EventEmitter {
     db;
     profile;
     activePeers = new Map(); // connectionId -> peer
     incomingHandler = null;
     heldQueries = new Map();
-    constructor(db, profile) {
+    signalingUrl;
+    pendingBytes = 0;
+    transferReportTimer = null;
+    constructor(db, profile, signalingUrl) {
         super();
         this.db = db;
         this.profile = profile;
+        this.signalingUrl = signalingUrl || "";
+        // Periodically flush transfer stats (every 60 seconds)
+        if (this.signalingUrl) {
+            this.transferReportTimer = setInterval(() => this.flushTransferStats(), 60000);
+        }
+    }
+    trackBytes(size) {
+        this.pendingBytes += size;
+    }
+    async flushTransferStats() {
+        if (this.pendingBytes > 0 && this.signalingUrl) {
+            const bytes = this.pendingBytes;
+            this.pendingBytes = 0;
+            await reportStats(this.signalingUrl, "transfer", { bytes });
+        }
     }
     registerPeer(connectionId, peerName, peerPublicKey) {
         this.activePeers.set(connectionId, {
@@ -32,6 +63,10 @@ class ConnectionManager extends node_events_1.EventEmitter {
             status: "active",
         });
         this.db.logEvent("connection_established", peerName, connectionId, null);
+        // Report session to stats
+        if (this.signalingUrl) {
+            reportStats(this.signalingUrl, "session", {});
+        }
     }
     setPeerSend(connectionId, sendFn) {
         const peer = this.activePeers.get(connectionId);
@@ -57,6 +92,7 @@ class ConnectionManager extends node_events_1.EventEmitter {
         const msg = (0, message_protocol_js_1.createQueryMessage)(this.profile.name, content, context);
         const serialized = (0, message_protocol_js_1.serializeMessage)(msg);
         peer.send(serialized);
+        this.trackBytes(serialized.length);
         this.db.logEvent("query_sent", peerName, peer.connectionId, content.slice(0, 100));
         this.db.saveMessage({
             id: msg.id,
@@ -81,6 +117,7 @@ class ConnectionManager extends node_events_1.EventEmitter {
         const msg = (0, message_protocol_js_1.createResponseMessage)(this.profile.name, replyTo, content, sourceFiles);
         const serialized = (0, message_protocol_js_1.serializeMessage)(msg);
         peer.send(serialized);
+        this.trackBytes(serialized.length);
         this.db.saveMessage({
             id: msg.id,
             connectionId: peer.connectionId,
@@ -109,6 +146,7 @@ class ConnectionManager extends node_events_1.EventEmitter {
         const msg = (0, message_protocol_js_1.createActionMessage)(this.profile.name, content, context);
         const serialized = (0, message_protocol_js_1.serializeMessage)(msg);
         peer.send(serialized);
+        this.trackBytes(serialized.length);
         this.db.logEvent("action_sent", peerName, peer.connectionId, content.slice(0, 100));
         this.db.saveMessage({
             id: msg.id,
@@ -130,6 +168,7 @@ class ConnectionManager extends node_events_1.EventEmitter {
         const msg = (0, message_protocol_js_1.createActionResultMessage)(this.profile.name, replyTo, content, success, sourceFiles);
         const serialized = (0, message_protocol_js_1.serializeMessage)(msg);
         peer.send(serialized);
+        this.trackBytes(serialized.length);
         this.db.saveMessage({
             id: msg.id,
             connectionId: peer.connectionId,
@@ -142,7 +181,52 @@ class ConnectionManager extends node_events_1.EventEmitter {
         });
         return msg;
     }
+    sendShareFiles(peerName, files) {
+        const peer = this.findPeerByName(peerName);
+        if (!peer || !peer.send) {
+            throw new Error(`No active connection with peer: ${peerName}`);
+        }
+        const msg = (0, message_protocol_js_1.createShareFileMessage)(this.profile.name, files);
+        const serialized = (0, message_protocol_js_1.serializeMessage)(msg);
+        peer.send(serialized);
+        this.trackBytes(serialized.length);
+        this.db.logEvent("share_files_sent", peerName, peer.connectionId, `${files.length} file(s)`);
+        this.db.saveMessage({
+            id: msg.id,
+            connectionId: peer.connectionId,
+            direction: "sent",
+            type: msg.type,
+            content: msg.content,
+            sourceFiles: JSON.stringify(files.map((f) => f.path)),
+            replyTo: null,
+            timestamp: msg.timestamp,
+        });
+        return msg;
+    }
+    sendShareContext(peerName, context) {
+        const peer = this.findPeerByName(peerName);
+        if (!peer || !peer.send) {
+            throw new Error(`No active connection with peer: ${peerName}`);
+        }
+        const msg = (0, message_protocol_js_1.createShareContextMessage)(this.profile.name, context);
+        const serialized = (0, message_protocol_js_1.serializeMessage)(msg);
+        peer.send(serialized);
+        this.trackBytes(serialized.length);
+        this.db.logEvent("share_context_sent", peerName, peer.connectionId, "project context");
+        this.db.saveMessage({
+            id: msg.id,
+            connectionId: peer.connectionId,
+            direction: "sent",
+            type: msg.type,
+            content: msg.content,
+            sourceFiles: null,
+            replyTo: null,
+            timestamp: msg.timestamp,
+        });
+        return msg;
+    }
     handleIncomingMessage(connectionId, rawData) {
+        this.trackBytes(rawData.length);
         const msg = (0, message_protocol_js_1.parseMessage)(rawData);
         if (!msg)
             return;