archbyte 0.2.2 → 0.2.5

package/dist/cli/setup.js

@@ -1,20 +1,38 @@
 import * as fs from "fs";
 import * as path from "path";
 import { fileURLToPath } from "url";
+import { execSync } from "child_process";
 import chalk from "chalk";
 import { resolveModel } from "../agents/runtime/types.js";
 import { createProvider } from "../agents/providers/router.js";
 import { select, spinner, confirm } from "./ui.js";
+import { CONFIG_DIR, CONFIG_PATH } from "./constants.js";
 const __filename = fileURLToPath(import.meta.url);
 const __dirname = path.dirname(__filename);
-const CONFIG_DIR = path.join(process.env.HOME ?? process.env.USERPROFILE ?? ".", ".archbyte");
-const CONFIG_PATH = path.join(CONFIG_DIR, "config.json");
 const PROVIDERS = [
-    { name: "anthropic", label: "Anthropic", hint: "Claude Opus / Sonnet" },
-    { name: "openai", label: "OpenAI", hint: "GPT-4o / o1" },
+    { name: "anthropic", label: "Anthropic", hint: "Claude Sonnet / Opus" },
+    { name: "openai", label: "OpenAI", hint: "Codex / GPT-5.2" },
     { name: "google", label: "Google", hint: "Gemini Flash / Pro" },
-    { name: "ollama", label: "Ollama", hint: "Local models (no key needed)" },
 ];
+const PROVIDER_MODELS = {
+    anthropic: [
+        { id: "", label: "Default (recommended)", hint: "Haiku for fast agents, Sonnet for connections" },
+        { id: "claude-haiku-4-5-20251001", label: "Claude Haiku 4.5", hint: "Fastest, cheapest" },
+        { id: "claude-sonnet-4-5-20250929", label: "Claude Sonnet 4.5", hint: "Balanced, great quality" },
+        { id: "claude-opus-4-6", label: "Claude Opus 4.6", hint: "Most capable" },
+    ],
+    openai: [
+        { id: "", label: "Default (recommended)", hint: "Codex for fast agents, GPT-5.2 for connections" },
+        { id: "gpt-5.2-codex", label: "GPT-5.2 Codex", hint: "Fast, coding-optimized" },
+        { id: "gpt-5.2", label: "GPT-5.2", hint: "General purpose, thinking model" },
+        { id: "o3", label: "o3", hint: "Reasoning model" },
+    ],
+    google: [
+        { id: "", label: "Default (recommended)", hint: "Flash for fast agents, Pro for connections" },
+        { id: "gemini-2.5-flash", label: "Gemini 2.5 Flash", hint: "Fastest, cheapest" },
+        { id: "gemini-2.5-pro", label: "Gemini 2.5 Pro", hint: "Most capable" },
+    ],
+};
 function loadConfig() {
     try {
         if (fs.existsSync(CONFIG_PATH)) {
@@ -75,40 +93,8 @@ function askHidden(prompt) {
         stdin.on("data", onData);
     });
 }
-async function fetchOllamaModels(baseUrl) {
+async function validateProviderSilent(providerName, apiKey) {
     try {
-        const controller = new AbortController();
-        const timeout = setTimeout(() => controller.abort(), 5000);
-        const res = await fetch(`${baseUrl}/api/tags`, { signal: controller.signal });
-        clearTimeout(timeout);
-        if (!res.ok)
-            return [];
-        const data = await res.json();
-        return (data.models ?? []).map((m) => m.name);
-    }
-    catch {
-        return [];
-    }
-}
-async function validateProviderSilent(providerName, apiKey, ollamaBaseUrl) {
-    try {
-        if (providerName === "ollama") {
-            const url = ollamaBaseUrl ?? "http://localhost:11434";
-            const controller = new AbortController();
-            const timeout = setTimeout(() => controller.abort(), 5000);
-            try {
-                const res = await fetch(url, { signal: controller.signal });
-                clearTimeout(timeout);
-                if (!res.ok)
-                    return false;
-                // Model listing happens in setup flow, not here
-                return true;
-            }
-            catch {
-                clearTimeout(timeout);
-                return false;
-            }
-        }
         const provider = createProvider({ provider: providerName, apiKey });
         const model = resolveModel(providerName, "fast");
         await provider.chat({
@@ -123,61 +109,151 @@ async function validateProviderSilent(providerName, apiKey, ollamaBaseUrl) {
         return false;
     }
 }
+function getProfiles(config) {
+    return config.profiles ?? {};
+}
+function isInPath(cmd) {
+    try {
+        execSync(`which ${cmd}`, { stdio: "ignore" });
+        return true;
+    }
+    catch {
+        return false;
+    }
+}
 export async function handleSetup() {
     console.log();
     console.log(chalk.bold.cyan("ArchByte Setup"));
     console.log(chalk.gray("Configure your model provider and API key.\n"));
+    // Detect AI coding tools — suggest MCP instead of BYOK
+    const hasClaude = isInPath("claude");
+    const codexDir = path.join(process.env.HOME ?? process.env.USERPROFILE ?? ".", ".codex");
+    const hasCodex = fs.existsSync(codexDir);
+    if (hasClaude || hasCodex) {
+        const tools = [hasClaude && "Claude Code", hasCodex && "Codex CLI"].filter(Boolean).join(" and ");
+        console.log(chalk.cyan(`  Detected ${tools} on this machine.`));
+        console.log(chalk.white(`  You can use ArchByte directly through MCP. No API key needed.`));
+        console.log(chalk.white(`  Your AI tool already provides the model, so you skip the BYOK step.`));
+        console.log();
+        console.log(chalk.white(`  Run: `) + chalk.bold.cyan(`archbyte mcp install`));
+        console.log();
+        const continueIdx = await select("Continue with BYOK setup anyway?", [
+            `Skip ${chalk.gray("(use MCP instead, recommended)")}`,
+            `Continue ${chalk.gray("(set up your own API key)")}`,
+        ]);
+        if (continueIdx === 0) {
+            console.log();
+            console.log(chalk.gray("  Run `archbyte mcp install` to configure MCP for your AI tool."));
+            console.log();
+            console.log(chalk.gray("  Then open your AI tool and try:"));
+            console.log(chalk.cyan('    "Analyze the architecture of this project"'));
+            console.log(chalk.cyan('    "Export the architecture as a Mermaid diagram"'));
+            console.log(chalk.cyan('    "Show me the architecture stats"'));
+            console.log();
+            return;
+        }
+        console.log();
+    }
     const config = loadConfig();
-    // Show current config if exists
-    if (config.provider) {
-        console.log(chalk.gray(`  Current provider: ${config.provider}`));
-        if (config.apiKey) {
-            console.log(chalk.gray(`  Current API key:  ${maskKey(config.apiKey)}`));
+    const profiles = getProfiles(config);
+    // Migrate legacy flat config → profiles
+    if (!config.profiles && config.provider && config.apiKey) {
+        const legacy = config.provider;
+        const validNames = PROVIDERS.map((p) => p.name);
+        if (validNames.includes(legacy)) {
+            // Known provider — migrate to profile
+            profiles[legacy] = { apiKey: config.apiKey };
+            if (config.model)
+                profiles[legacy].model = config.model;
+            config.profiles = profiles;
+        }
+        // Clean up legacy flat keys regardless
+        delete config.apiKey;
+        delete config.model;
+        delete config.ollamaBaseUrl;
+        if (!validNames.includes(legacy)) {
+            delete config.provider;
+        }
+    }
+    // Show current config + configured profiles
+    const configured = Object.keys(profiles).filter((p) => profiles[p]?.apiKey);
+    if (configured.length > 0) {
+        console.log(chalk.gray(`  Active provider: ${config.provider ?? "none"}`));
+        for (const p of configured) {
+            const active = p === config.provider ? chalk.green(" (active)") : "";
+            const model = profiles[p].model ? chalk.gray(` model: ${profiles[p].model}`) : "";
+            console.log(chalk.gray(`  ${p}: ${maskKey(profiles[p].apiKey)}${model}${active}`));
         }
         console.log();
     }
-    // Step 1: Choose provider with arrow-key selection
+    // Step 1: Choose provider
     const idx = await select("Choose your model provider:", PROVIDERS.map((p) => {
-        const current = config.provider === p.name ? chalk.green(" (current)") : "";
-        return `${p.label} ${chalk.gray("— " + p.hint)}${current}`;
+        const active = config.provider === p.name ? chalk.green(" (active)") : "";
+        const hasKey = profiles[p.name]?.apiKey ? chalk.cyan(" ✓ configured") : "";
+        return `${p.label} ${chalk.gray(p.hint)}${hasKey}${active}`;
     }));
     const provider = PROVIDERS[idx].name;
     config.provider = provider;
     const selected = PROVIDERS[idx];
     console.log(chalk.green(`\n  ✓ Provider: ${selected.label}`));
-    // Step 2: API key / model selection
-    if (provider === "ollama") {
-        config.ollamaBaseUrl = config.ollamaBaseUrl ?? "http://localhost:11434";
-        console.log(chalk.gray(`  Ollama URL: ${config.ollamaBaseUrl}`));
-        // Let user pick from installed models
-        const models = await fetchOllamaModels(config.ollamaBaseUrl);
-        if (models.length > 0) {
-            const modelIdx = await select("\n  Choose a model:", models.map((m) => m));
-            config.model = models[modelIdx];
-            console.log(chalk.green(`  ✓ Model: ${config.model}`));
+    // Step 2: API key (show existing if profile exists)
+    const existing = profiles[provider];
+    let apiKey;
+    if (existing?.apiKey) {
+        console.log(chalk.gray(`\n  Existing key: ${maskKey(existing.apiKey)}`));
+        const keepIdx = await select("  Update API key?", [
+            `Keep existing ${chalk.gray("(" + maskKey(existing.apiKey) + ")")}`,
+            "Enter new key",
+        ]);
+        if (keepIdx === 0) {
+            apiKey = existing.apiKey;
+            console.log(chalk.green(`  ✓ API key: ${maskKey(apiKey)} (kept)`));
         }
         else {
-            console.log(chalk.yellow(`  Warning: no models installed.`));
-            console.log(chalk.gray(`  Run: ollama pull <model-name>`));
+            console.log();
+            apiKey = await askHidden(chalk.bold("  API key: "));
+            if (!apiKey) {
+                console.log(chalk.red("  No API key entered. Setup cancelled."));
+                process.exit(1);
+            }
+            console.log(chalk.green(`  ✓ API key: ${maskKey(apiKey)}`));
         }
     }
     else {
-        // Clear Ollama model override — model names are provider-specific
-        if (config.model) {
-            delete config.model;
-        }
         console.log();
-        const apiKey = await askHidden(chalk.bold("  API key: "));
+        apiKey = await askHidden(chalk.bold("  API key: "));
         if (!apiKey) {
             console.log(chalk.red("  No API key entered. Setup cancelled."));
             process.exit(1);
         }
-        config.apiKey = apiKey;
         console.log(chalk.green(`  ✓ API key: ${maskKey(apiKey)}`));
     }
+    // Save to profile
+    if (!profiles[provider])
+        profiles[provider] = { apiKey: "" };
+    profiles[provider].apiKey = apiKey;
+    // Step 3: Model selection
+    const models = PROVIDER_MODELS[provider];
+    if (models) {
+        const currentModel = profiles[provider].model;
+        const modelIdx = await select("\n  Choose a model:", models.map((m) => {
+            const current = currentModel === m.id && m.id ? chalk.green(" (current)") : "";
+            return `${m.label} ${chalk.gray(m.hint)}${current}`;
+        }));
+        const chosen = models[modelIdx];
+        if (chosen.id) {
+            profiles[provider].model = chosen.id;
+            console.log(chalk.green(`  ✓ Model: ${chosen.label}`));
+        }
+        else {
+            delete profiles[provider].model;
+            console.log(chalk.green(`  ✓ Model: Default (auto per agent tier)`));
+        }
+    }
+    config.profiles = profiles;
     // Validate provider with spinner
     const s = spinner("Validating credentials");
-    let isValid = await validateProviderSilent(provider, config.apiKey ?? "", config.ollamaBaseUrl);
+    let isValid = await validateProviderSilent(provider, apiKey);
     s.stop(isValid ? "valid" : "invalid", isValid ? "green" : "red");
     // Retry loop on failure
     if (!isValid) {
@@ -186,24 +262,28 @@ export async function handleSetup() {
             if (!await confirm("  Retry?"))
                 break;
             retries++;
-            if (provider !== "ollama") {
-                const newKey = await askHidden(chalk.bold("  API key: "));
-                if (newKey) {
-                    config.apiKey = newKey;
-                    console.log(chalk.green(`  ✓ API key: ${maskKey(newKey)}`));
-                }
+            const newKey = await askHidden(chalk.bold("  API key: "));
+            if (newKey) {
+                profiles[provider].apiKey = newKey;
+                console.log(chalk.green(`  ✓ API key: ${maskKey(newKey)}`));
             }
             const s2 = spinner("Validating credentials");
-            isValid = await validateProviderSilent(provider, config.apiKey ?? "", config.ollamaBaseUrl);
+            isValid = await validateProviderSilent(provider, profiles[provider].apiKey);
             s2.stop(isValid ? "valid" : "invalid", isValid ? "green" : "red");
         }
         if (!isValid) {
             console.log(chalk.yellow("  Warning: credentials unverified. Saving anyway."));
         }
     }
+    // Clean up legacy top-level keys
+    delete config.apiKey;
+    delete config.model;
+    delete config.ollamaBaseUrl;
     // Save config
     saveConfig(config);
     console.log(chalk.gray(`\n  Config saved to ${CONFIG_PATH}`));
+    console.log(chalk.gray("  Your API key is stored locally on this machine and never sent to ArchByte."));
+    console.log(chalk.gray("  All model calls go directly from your machine to your provider."));
     // Generate archbyte.yaml in .archbyte/ if it doesn't exist
     const projectDir = process.cwd();
     const archbyteDir = path.join(projectDir, ".archbyte");
@@ -236,5 +316,6 @@ export async function handleSetup() {
     console.log(chalk.green("  Setup complete!"));
     console.log();
     console.log(chalk.bold("  Next: ") + chalk.cyan("archbyte run") + chalk.bold(" to analyze your codebase."));
+    console.log(chalk.gray("        Run from your project root, or use ") + chalk.cyan("archbyte run -d /path/to/project"));
     console.log();
 }

package/dist/cli/stats.js

@@ -17,7 +17,7 @@ export async function handleStats(options) {
     // Auto-detect project name from cwd
     const projectName = process.cwd().split("/").pop() || "project";
     console.log();
-    console.log(chalk.bold.cyan(`⚡ ArchByte Stats — ${projectName}`));
+    console.log(chalk.bold.cyan(`⚡ ArchByte Stats: ${projectName}`));
     console.log();
     // ── Scan Info (from metadata.json, fallback to analysis.json) ──
     const metadataJsonPath = path.join(process.cwd(), ".archbyte", "metadata.json");

package/dist/cli/ui.js

@@ -89,8 +89,8 @@ export function select(prompt, options) {
                 cleanup();
                 resolve(selected);
             }
-            else if (data === "\x03") {
-                // Ctrl+C
+            else if (data === "\x03" || data === "q" || data === "Q") {
+                // Ctrl+C or q to quit
                 cleanup();
                 process.exit(0);
             }
@@ -173,7 +173,7 @@ export function confirm(prompt) {
                 process.stdout.write("n\n");
                 resolve(false);
             }
-            else if (data === "\x03") {
+            else if (data === "\x03" || data === "q" || data === "Q") {
                 process.stdout.write("\n");
                 process.exit(0);
             }

package/dist/cli/validate.js

@@ -225,7 +225,7 @@ export async function handleValidate(options) {
 function printValidationReport(result) {
     const projectName = process.cwd().split("/").pop() || "project";
     console.log();
-    console.log(chalk.bold.cyan(`⚡ ArchByte Validate — ${projectName}`));
+    console.log(chalk.bold.cyan(`⚡ ArchByte Validate: ${projectName}`));
     console.log();
     // Group violations by rule for display
     const ruleViolations = new Map();
@@ -248,7 +248,7 @@ function printValidationReport(result) {
     }
     console.log();
     const resultStr = result.errors > 0 ? chalk.red("FAIL") : chalk.green("PASS");
-    console.log(`  Result: ${result.warnings} warning${result.warnings !== 1 ? "s" : ""}, ${result.errors} error${result.errors !== 1 ? "s" : ""} — ${resultStr}`);
+    console.log(`  Result: ${result.warnings} warning${result.warnings !== 1 ? "s" : ""}, ${result.errors} error${result.errors !== 1 ? "s" : ""} ${resultStr}`);
     console.log();
 }
 function printRuleResult(rule, level, count, violations) {

package/dist/cli/version.d.ts

@@ -0,0 +1,2 @@
+export declare function handleVersion(): Promise<void>;
+export declare function handleUpdate(): Promise<void>;

package/dist/cli/version.js

@@ -0,0 +1,84 @@
+import * as fs from "fs";
+import * as path from "path";
+import { fileURLToPath } from "url";
+import chalk from "chalk";
+import { execSync } from "child_process";
+import { NETWORK_TIMEOUT_MS } from "./constants.js";
+const __filename = fileURLToPath(import.meta.url);
+const __dirname = path.dirname(__filename);
+function getLocalVersion() {
+    const pkgPath = path.resolve(__dirname, "../../package.json");
+    const pkg = JSON.parse(fs.readFileSync(pkgPath, "utf-8"));
+    return pkg.version;
+}
+export async function handleVersion() {
+    const version = getLocalVersion();
+    console.log();
+    console.log(chalk.bold.cyan("ArchByte") + chalk.gray(` v${version}`));
+    console.log();
+    console.log(`  ${chalk.bold("version")}:  ${version}`);
+    console.log(`  ${chalk.bold("node")}:     ${process.versions.node}`);
+    console.log(`  ${chalk.bold("platform")}: ${process.platform} ${process.arch}`);
+    // Show install path
+    try {
+        const which = execSync("which archbyte", { encoding: "utf-8" }).trim();
+        console.log(`  ${chalk.bold("binary")}:   ${which}`);
+    }
+    catch {
+        // not globally installed or `which` not available
+    }
+    console.log();
+}
+async function fetchLatestVersion() {
+    try {
+        const controller = new AbortController();
+        const timeout = setTimeout(() => controller.abort(), NETWORK_TIMEOUT_MS);
+        const res = await fetch("https://registry.npmjs.org/archbyte/latest", {
+            signal: controller.signal,
+        });
+        clearTimeout(timeout);
+        if (!res.ok)
+            return null;
+        const data = (await res.json());
+        return data.version ?? null;
+    }
+    catch {
+        return null;
+    }
+}
+export async function handleUpdate() {
+    const current = getLocalVersion();
+    console.log();
+    console.log(chalk.bold.cyan("ArchByte Update"));
+    console.log(chalk.gray(`  Current version: ${current}`));
+    console.log(chalk.gray("  Checking for updates..."));
+    const latest = await fetchLatestVersion();
+    if (!latest) {
+        console.log(chalk.yellow("  Could not reach npm registry. Update manually:"));
+        console.log(chalk.cyan("  npm install -g archbyte@latest"));
+        console.log();
+        return;
+    }
+    if (latest === current) {
+        console.log(chalk.green(`  You're on the latest version (${current}).`));
+        console.log();
+        return;
+    }
+    console.log(chalk.yellow(`  New version available: ${current} → ${chalk.bold(latest)}`));
+    console.log();
+    // Run the update
+    console.log(chalk.gray("  Updating..."));
+    try {
+        execSync("npm install -g archbyte@latest", {
+            stdio: "inherit",
+        });
+        console.log();
+        console.log(chalk.green(`  Updated to v${latest}`));
+    }
+    catch {
+        console.log();
+        console.log(chalk.red("  Update failed. Try manually:"));
+        console.log(chalk.cyan("  npm install -g archbyte@latest"));
+    }
+    console.log();
+}

package/dist/cli/workflow.js

@@ -349,7 +349,7 @@ async function runWorkflow(workflow) {
         // Check dependencies
         const unmetDeps = step.needs.filter((dep) => state.steps[dep]?.status !== "completed");
         if (unmetDeps.length > 0) {
-            console.log(chalk.yellow(`  [skip] ${step.name} — waiting on: ${unmetDeps.join(", ")}`));
+            console.log(chalk.yellow(`  [skip] ${step.name} (waiting on: ${unmetDeps.join(", ")})`));
             state.steps[step.id] = { status: "pending" };
             saveState(state);
             continue;
@@ -414,7 +414,7 @@ function listWorkflows() {
     const workflows = loadWorkflows();
     const projectName = process.cwd().split("/").pop() || "project";
     console.log();
-    console.log(chalk.bold.cyan(`  ArchByte Workflows — ${projectName}`));
+    console.log(chalk.bold.cyan(`  ArchByte Workflows: ${projectName}`));
     console.log();
     for (const w of workflows) {
         const state = loadState(w.id);
@@ -428,7 +428,7 @@ function listWorkflows() {
         const builtinTag = BUILTIN_WORKFLOWS.some((b) => b.id === w.id)
             ? chalk.gray(" (built-in)")
             : "";
-        console.log(`  ${chalk.bold(w.id)}${builtinTag} — ${statusStr}`);
+        console.log(`  ${chalk.bold(w.id)}${builtinTag} ${statusStr}`);
         console.log(chalk.gray(`    ${w.description}`));
         console.log(chalk.gray(`    Steps: ${w.steps.map((s) => s.id).join(" → ")}`));
         console.log();
@@ -464,7 +464,7 @@ function showWorkflow(id) {
         }
         else if (stepState?.status === "failed") {
             icon = chalk.red("FAIL");
-            statusLabel = chalk.red(` — ${stepState.error?.slice(0, 60)}`);
+            statusLabel = chalk.red(` ${stepState.error?.slice(0, 60)}`);
         }
         const depsStr = step.needs.length > 0
             ? chalk.gray(` [needs: ${step.needs.join(", ")}]`)
@@ -481,7 +481,7 @@ function showStatus() {
     const workflows = loadWorkflows();
     const projectName = process.cwd().split("/").pop() || "project";
     console.log();
-    console.log(chalk.bold.cyan(`  Workflow Status — ${projectName}`));
+    console.log(chalk.bold.cyan(`  Workflow Status: ${projectName}`));
     console.log();
     let anyActive = false;
     for (const w of workflows) {
@@ -499,7 +499,7 @@ function showStatus() {
             statusIcon = chalk.green("done");
         if (state.status === "failed")
             statusIcon = chalk.red("failed");
-        console.log(`  ${chalk.bold(w.id)} [${bar}] ${pct}% — ${statusIcon}`);
+        console.log(`  ${chalk.bold(w.id)} [${bar}] ${pct}% ${statusIcon}`);
         console.log(chalk.gray(`    ${completed}/${total} steps complete`));
         console.log();
     }
@@ -583,7 +583,7 @@ export async function handleWorkflow(options) {
         }
         const projectName = process.cwd().split("/").pop() || "project";
         console.log();
-        console.log(chalk.bold.cyan(`  ArchByte Workflow: ${workflow.name} — ${projectName}`));
+        console.log(chalk.bold.cyan(`  ArchByte Workflow: ${workflow.name} | ${projectName}`));
         console.log(chalk.gray(`  ${workflow.description}`));
         console.log(chalk.gray(`  Steps: ${workflow.steps.length}`));
         await runWorkflow(workflow);

package/dist/cli/yaml-io.js

@@ -36,7 +36,7 @@ export function writeSpec(rootDir, spec) {
     if (!fs.existsSync(dir)) {
         fs.mkdirSync(dir, { recursive: true });
     }
-    const header = `# Generated by ArchByte — ${new Date().toISOString().split("T")[0]}\n# Human-editable. Changes are preserved on re-scan.\n\n`;
+    const header = `# Generated by ArchByte ${new Date().toISOString().split("T")[0]}\n# Human-editable. Changes are preserved on re-scan.\n\n`;
     const yamlStr = yaml.dump(spec, {
         indent: 2,
         lineWidth: 120,

package/dist/server/src/index.js

@@ -135,6 +135,46 @@ function createHttpServer() {
             res.end(JSON.stringify(currentArchitecture || { nodes: [], edges: [] }));
             return;
         }
+        // API: Update node positions (save layout changes)
+        if (url === "/api/update-positions" && req.method === "POST") {
+            let body = "";
+            req.on("data", (chunk) => { body += chunk.toString(); });
+            req.on("end", async () => {
+                try {
+                    const { updates } = JSON.parse(body);
+                    if (!updates || !Array.isArray(updates)) {
+                        res.writeHead(400, { "Content-Type": "application/json" });
+                        res.end(JSON.stringify({ error: "Missing updates array" }));
+                        return;
+                    }
+                    // Read current architecture file
+                    const content = await readFile(config.diagramPath, "utf-8");
+                    const arch = JSON.parse(content);
+                    // Apply position/dimension updates
+                    for (const update of updates) {
+                        const node = arch.nodes.find((n) => n.id === update.id);
+                        if (node) {
+                            node.x = update.x;
+                            node.y = update.y;
+                            node.width = update.width;
+                            node.height = update.height;
+                        }
+                    }
+                    // Mark layout as user-saved so UI uses these positions
+                    arch.layoutSaved = true;
+                    // Write back
+                    await writeFile(config.diagramPath, JSON.stringify(arch, null, 2), "utf-8");
+                    currentArchitecture = arch;
+                    res.writeHead(200, { "Content-Type": "application/json" });
+                    res.end(JSON.stringify({ ok: true, updated: updates.length }));
+                }
+                catch (err) {
+                    res.writeHead(500, { "Content-Type": "application/json" });
+                    res.end(JSON.stringify({ error: "Failed to save positions" }));
+                }
+            });
+            return;
+        }
         // API: Health
         if (url === "/health") {
             res.writeHead(200, { "Content-Type": "application/json" });
@@ -1344,4 +1384,16 @@ export async function startServer(cfg) {
     setupWatcher();
     console.error(`[archbyte] Serving ${config.name}`);
     console.error(`[archbyte] Diagram: ${config.diagramPath}`);
+    // Listen for 'q' keypress to quit gracefully
+    if (process.stdin.isTTY) {
+        process.stdin.setRawMode(true);
+        process.stdin.resume();
+        process.stdin.setEncoding("utf8");
+        process.stdin.on("data", (key) => {
+            if (key === "q" || key === "Q" || key === "\x03") {
+                process.emit("SIGINT");
+            }
+        });
+        console.error(`[archbyte] Press q to quit`);
+    }
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "archbyte",
-  "version": "0.2.2",
+  "version": "0.2.5",
   "description": "ArchByte - See what agents build. As they build it.",
   "type": "module",
   "bin": {
@@ -37,11 +37,13 @@
   "dependencies": {
     "@anthropic-ai/sdk": "^0.74.0",
     "@google/generative-ai": "^0.24.1",
+    "@modelcontextprotocol/sdk": "^1.26.0",
     "chalk": "^5.3.0",
     "chokidar": "^3.5.3",
     "commander": "^12.0.0",
     "js-yaml": "^4.1.1",
-    "openai": "^6.19.0"
+    "openai": "^6.19.0",
+    "zod": "^4.3.6"
   },
   "devDependencies": {
     "@types/js-yaml": "^4.0.9",