archbyte 0.2.2 → 0.2.5

package/dist/cli/config.js

@@ -1,9 +1,7 @@
 import * as fs from "fs";
-import * as path from "path";
 import chalk from "chalk";
-const CONFIG_DIR = path.join(process.env.HOME ?? process.env.USERPROFILE ?? ".", ".archbyte");
-const CONFIG_PATH = path.join(CONFIG_DIR, "config.json");
-const VALID_PROVIDERS = ["anthropic", "openai", "google", "ollama"];
+import { CONFIG_DIR, CONFIG_PATH } from "./constants.js";
+const VALID_PROVIDERS = ["anthropic", "openai", "google"];
 export async function handleConfig(options) {
     const [action, key, value] = options.args;
     if (!action || action === "show") {
@@ -13,7 +11,7 @@ export async function handleConfig(options) {
     if (action === "set") {
         if (!key || !value) {
             console.error(chalk.red("Usage: archbyte config set <key> <value>"));
-            console.error(chalk.gray("  Keys: provider, api-key, ollama-url"));
+            console.error(chalk.gray("  Keys: provider, api-key, model"));
             process.exit(1);
         }
         setConfig(key, value);
@@ -32,10 +30,10 @@ export async function handleConfig(options) {
         return;
     }
     console.error(chalk.red(`Unknown action: ${action}`));
-    console.error(chalk.gray("  archbyte config show       — show current config"));
-    console.error(chalk.gray("  archbyte config set <k> <v> — set a config value"));
-    console.error(chalk.gray("  archbyte config get <k>     — get a config value"));
-    console.error(chalk.gray("  archbyte config path        — show config file path"));
+    console.error(chalk.gray("  archbyte config show          show current config"));
+    console.error(chalk.gray("  archbyte config set <k> <v>   set a config value"));
+    console.error(chalk.gray("  archbyte config get <k>       get a config value"));
+    console.error(chalk.gray("  archbyte config path          show config file path"));
     process.exit(1);
 }
 function loadConfig() {
@@ -57,75 +55,105 @@ function saveConfig(config) {
 }
 function showConfig() {
     const config = loadConfig();
+    const profiles = (config.profiles ?? {});
     console.log();
     console.log(chalk.bold.cyan("ArchByte Configuration"));
     console.log(chalk.gray(`Config file: ${CONFIG_PATH}`));
     console.log();
-    if (!config.provider && !config.apiKey) {
-        console.log(chalk.yellow("No configuration found. Set up with:"));
-        console.log(chalk.gray("  archbyte config set provider anthropic"));
-        console.log(chalk.gray("  archbyte config set api-key sk-ant-..."));
-        console.log();
-        console.log(chalk.gray("Or use environment variables:"));
-        console.log(chalk.gray("  export ARCHBYTE_PROVIDER=anthropic"));
-        console.log(chalk.gray("  export ARCHBYTE_API_KEY=sk-ant-..."));
+    const configured = Object.keys(profiles).filter((p) => profiles[p]?.apiKey);
+    if (!config.provider && !configured.length && !config.apiKey) {
+        console.log(chalk.yellow("No configuration found. Run:"));
+        console.log(chalk.gray("  archbyte init"));
         return;
     }
     console.log(`  ${chalk.bold("provider")}: ${config.provider ?? chalk.gray("not set")}`);
-    console.log(`  ${chalk.bold("api-key")}:  ${config.apiKey ? maskKey(config.apiKey) : chalk.gray("not set")}`);
-    if (config.model) {
-        console.log(`  ${chalk.bold("model")}:    ${config.model}`);
+    // Show profiles
+    if (configured.length > 0) {
+        console.log();
+        for (const name of configured) {
+            const p = profiles[name];
+            const active = name === config.provider ? chalk.green(" (active)") : "";
+            console.log(`  ${chalk.bold(name)}${active}`);
+            console.log(`    api-key: ${p.apiKey ? maskKey(p.apiKey) : chalk.gray("not set")}`);
+            if (p.model)
+                console.log(`    model:   ${p.model}`);
+        }
     }
-    if (config.ollamaBaseUrl) {
-        console.log(`  ${chalk.bold("ollama-url")}: ${config.ollamaBaseUrl}`);
+    else if (config.apiKey) {
+        // Legacy flat config
+        console.log(`  ${chalk.bold("api-key")}:  ${maskKey(config.apiKey)}`);
+        if (config.model) {
+            console.log(`  ${chalk.bold("model")}:    ${config.model}`);
+        }
     }
     console.log();
 }
 function setConfig(key, value) {
     const config = loadConfig();
+    if (!config.profiles)
+        config.profiles = {};
+    const profiles = config.profiles;
     switch (key) {
         case "provider": {
             if (!VALID_PROVIDERS.includes(value)) {
                 console.error(chalk.red(`Invalid provider: ${value}. Must be: ${VALID_PROVIDERS.join(", ")}`));
                 process.exit(1);
             }
-            const oldProvider = config.provider;
             config.provider = value;
-            // Clear model override when switching providers — model names are provider-specific
-            if (oldProvider && oldProvider !== value && config.model) {
-                delete config.model;
-                console.log(chalk.yellow(`Cleared model override (was for ${oldProvider})`));
+            // Check if this provider has a stored profile
+            if (profiles[value]?.apiKey) {
+                console.log(chalk.green(`Switched to ${value} (credentials on file)`));
+            }
+            else {
+                console.log(chalk.yellow(`Switched to ${value}. Run archbyte init to configure API key.`));
             }
             break;
         }
         case "api-key":
         case "apiKey":
-        case "key":
-            config.apiKey = value;
+        case "key": {
+            const activeProvider = config.provider;
+            if (!activeProvider) {
+                console.error(chalk.red("Set a provider first: archbyte config set provider <name>"));
+                process.exit(1);
+            }
+            if (!profiles[activeProvider])
+                profiles[activeProvider] = {};
+            profiles[activeProvider].apiKey = value;
             break;
-        case "model":
+        }
+        case "model": {
+            const activeProvider2 = config.provider;
+            if (!activeProvider2) {
+                console.error(chalk.red("Set a provider first: archbyte config set provider <name>"));
+                process.exit(1);
+            }
+            if (!profiles[activeProvider2])
+                profiles[activeProvider2] = {};
             if (value === "" || value === "default") {
-                delete config.model;
+                delete profiles[activeProvider2].model;
                 saveConfig(config);
                 console.log(chalk.green("Cleared model override (will use provider defaults)"));
                 return;
             }
-            config.model = value;
-            break;
-        case "ollama-url":
-        case "ollamaUrl":
-            config.ollamaBaseUrl = value;
+            profiles[activeProvider2].model = value;
             break;
+        }
         default:
             console.error(chalk.red(`Unknown config key: ${key}`));
-            console.error(chalk.gray("  Valid keys: provider, api-key, model, ollama-url"));
+            console.error(chalk.gray("  Valid keys: provider, api-key, model"));
             process.exit(1);
     }
     saveConfig(config);
-    console.log(chalk.green(`Set ${key} = ${key.includes("key") ? maskKey(value) : value}`));
+    if (key !== "provider") {
+        console.log(chalk.green(`Set ${key} = ${key.includes("key") ? maskKey(value) : value}`));
+    }
 }
 function getConfig(key) {
     const config = loadConfig();
+    const profiles = (config.profiles ?? {});
+    const active = config.provider;
+    const profile = active ? profiles[active] : undefined;
     switch (key) {
         case "provider":
             console.log(config.provider ?? "");
@@ -133,14 +161,10 @@ function getConfig(key) {
         case "api-key":
         case "apiKey":
         case "key":
-            console.log(config.apiKey ?? "");
+            console.log(profile?.apiKey ?? config.apiKey ?? "");
             break;
         case "model":
-            console.log(config.model ?? "");
-            break;
-        case "ollama-url":
-        case "ollamaUrl":
-            console.log(config.ollamaBaseUrl ?? "");
+            console.log(profile?.model ?? config.model ?? "");
             break;
         default:
             console.error(chalk.red(`Unknown config key: ${key}`));
@@ -154,12 +178,21 @@ function maskKey(key) {
 }
 /**
  * Resolve the full ArchByteConfig from config file + env vars.
+ * Supports profiles (new) and legacy flat config (backward compat).
  * Env vars override config file.
  */
 export function resolveConfig() {
     const config = loadConfig();
     const provider = process.env.ARCHBYTE_PROVIDER ?? config.provider;
-    const apiKey = process.env.ARCHBYTE_API_KEY ?? config.apiKey;
+    // Reject unknown providers (e.g. legacy "ollama")
+    if (provider && !VALID_PROVIDERS.includes(provider)) {
+        return null;
+    }
+    // Resolve API key + model from profiles first, then legacy flat keys
+    const profiles = (config.profiles ?? {});
+    const profile = provider ? profiles[provider] : undefined;
+    const apiKey = process.env.ARCHBYTE_API_KEY ?? profile?.apiKey ?? config.apiKey;
+    const model = process.env.ARCHBYTE_MODEL ?? profile?.model ?? config.model;
     // Auto-detect from known env vars if nothing explicit
     if (!provider && !apiKey) {
         if (process.env.ANTHROPIC_API_KEY) {
@@ -176,25 +209,11 @@ export function resolveConfig() {
         }
         return null;
     }
-    if (!provider)
-        return null;
-    // Ollama doesn't need an API key
-    if (provider === "ollama") {
-        return {
-            provider: "ollama",
-            apiKey: "",
-            model: process.env.ARCHBYTE_MODEL ?? config.model,
-            ollamaBaseUrl: process.env.OLLAMA_BASE_URL ??
-                config.ollamaBaseUrl ??
-                "http://localhost:11434",
-        };
-    }
-    if (!apiKey)
+    if (!provider || !apiKey)
         return null;
     return {
         provider,
         apiKey,
-        model: process.env.ARCHBYTE_MODEL ?? config.model,
-        ollamaBaseUrl: config.ollamaBaseUrl,
+        model: model,
     };
 }

package/dist/cli/constants.d.ts

@@ -0,0 +1,13 @@
+export declare const API_BASE: string;
+export declare const SITE_URL = "https://archbyte.heartbyte.io";
+export declare const DEFAULT_PORT = 3847;
+export declare const CLI_CALLBACK_PORT = 19274;
+export declare const CONFIG_DIR: string;
+export declare const CONFIG_PATH: string;
+export declare const CREDENTIALS_PATH: string;
+/** Project-local .archbyte directory name */
+export declare const PROJECT_DIR = ".archbyte";
+/** Timeout for non-critical network checks (license, version, gate) */
+export declare const NETWORK_TIMEOUT_MS = 5000;
+/** Timeout for OAuth callback server */
+export declare const OAUTH_TIMEOUT_MS = 60000;

package/dist/cli/constants.js

@@ -0,0 +1,20 @@
+// Shared constants for the ArchByte CLI.
+// Single source of truth for URLs, ports, paths, and timeouts.
+import * as path from "path";
+// ─── API ───
+export const API_BASE = process.env.ARCHBYTE_API_URL ?? "https://api.heartbyte.io";
+export const SITE_URL = "https://archbyte.heartbyte.io";
+// ─── Ports ───
+export const DEFAULT_PORT = 3847;
+export const CLI_CALLBACK_PORT = 19274;
+// ─── Paths ───
+export const CONFIG_DIR = path.join(process.env.HOME ?? process.env.USERPROFILE ?? ".", ".archbyte");
+export const CONFIG_PATH = path.join(CONFIG_DIR, "config.json");
+export const CREDENTIALS_PATH = path.join(CONFIG_DIR, "credentials.json");
+/** Project-local .archbyte directory name */
+export const PROJECT_DIR = ".archbyte";
+// ─── Timeouts ───
+/** Timeout for non-critical network checks (license, version, gate) */
+export const NETWORK_TIMEOUT_MS = 5000;
+/** Timeout for OAuth callback server */
+export const OAUTH_TIMEOUT_MS = 60_000;

package/dist/cli/diff.js

@@ -17,7 +17,7 @@ export async function handleDiff(options) {
     const currentArch = loadArchitectureFile(currentPath);
     const projectName = process.cwd().split("/").pop() || "project";
     console.log();
-    console.log(chalk.bold.cyan(`⚡ ArchByte Diff — ${projectName}`));
+    console.log(chalk.bold.cyan(`⚡ ArchByte Diff: ${projectName}`));
     console.log(chalk.gray(`  Baseline: ${options.baseline}`));
     console.log(chalk.gray(`  Current:  ${options.current || ".archbyte/architecture.json"}`));
     console.log();

package/dist/cli/export.js

@@ -104,7 +104,7 @@ function exportMarkdown(arch) {
     const realNodes = arch.nodes;
     const projectName = process.cwd().split("/").pop() || "project";
     const lines = [];
-    lines.push(`# Architecture — ${projectName}`);
+    lines.push(`# Architecture: ${projectName}`);
     lines.push("");
     lines.push(`> Generated by ArchByte on ${new Date().toISOString().slice(0, 10)}`);
     lines.push("");
@@ -127,7 +127,7 @@ function exportMarkdown(arch) {
     for (const node of realNodes) {
         const tech = node.techStack && node.techStack.length > 0
             ? node.techStack.join(", ")
-            : "—";
+            : "-";
         lines.push(`| ${node.label} | ${node.type} | ${node.layer} | ${tech} |`);
     }
     lines.push("");
@@ -145,7 +145,7 @@ function exportMarkdown(arch) {
         for (const edge of arch.edges) {
             const src = nodeMap.get(edge.source)?.label || edge.source;
             const tgt = nodeMap.get(edge.target)?.label || edge.target;
-            const label = edge.label ? ` — ${edge.label}` : "";
+            const label = edge.label ? `: ${edge.label}` : "";
             lines.push(`- **${src}** → **${tgt}**${label}`);
         }
     }

package/dist/cli/gate.js

@@ -1,5 +1,5 @@
 import { loadCredentials, cacheVerifiedTier, resetOfflineActions, checkOfflineAction } from "./auth.js";
-const API_BASE = process.env.ARCHBYTE_API_URL ?? "https://api.heartbyte.io";
+import { API_BASE, NETWORK_TIMEOUT_MS } from "./constants.js";
 /**
  * Auth gate command for Claude Code integration.
  *
@@ -48,7 +48,7 @@ async function checkGate(action) {
                 "Content-Type": "application/json",
             },
             body: JSON.stringify({ action }),
-            signal: AbortSignal.timeout(5000),
+            signal: AbortSignal.timeout(NETWORK_TIMEOUT_MS),
         });
         if (res.status === 401) {
             const result = {
@@ -120,7 +120,7 @@ async function recordScan() {
                 "Content-Type": "application/json",
             },
             body: JSON.stringify({}),
-            signal: AbortSignal.timeout(5000),
+            signal: AbortSignal.timeout(NETWORK_TIMEOUT_MS),
         });
         console.log(JSON.stringify({ recorded: true }));
     }

package/dist/cli/generate.js

@@ -4,6 +4,7 @@ import chalk from "chalk";
 import { generateArchitecture } from "../server/src/generator/index.js";
 import { loadSpec, specToAnalysis } from "./yaml-io.js";
 import { spinner } from "./ui.js";
+import { DEFAULT_PORT } from "./constants.js";
 /**
  * Generate excalidraw diagram from analysis JSON
  */
@@ -233,7 +234,7 @@ export async function handleGenerate(options) {
         console.log();
         console.log(chalk.bold("Next steps:"));
         console.log(chalk.gray(`  1. Run ${chalk.cyan("archbyte serve")} to start the visualization server`));
-        console.log(chalk.gray(`  2. Open ${chalk.cyan("http://localhost:3847")} to view and adjust the diagram`));
+        console.log(chalk.gray(`  2. Open ${chalk.cyan(`http://localhost:${DEFAULT_PORT}`)} to view and adjust the diagram`));
         console.log();
     }
     catch (error) {

package/dist/cli/license-gate.js

@@ -1,6 +1,6 @@
 import chalk from "chalk";
 import { loadCredentials, cacheVerifiedTier, resetOfflineActions, checkOfflineAction } from "./auth.js";
-const API_BASE = process.env.ARCHBYTE_API_URL ?? "https://api.heartbyte.io";
+import { API_BASE, NETWORK_TIMEOUT_MS } from "./constants.js";
 /**
  * Pre-flight license check. Must be called before scan/analyze/generate.
  *
@@ -22,10 +22,10 @@ export async function requireLicense(action) {
         console.error();
         console.error(chalk.red("Authentication required."));
         console.error();
-        console.error(chalk.gray("Sign in to use ArchByte:"));
+        console.error(chalk.gray("Sign in or create a free account:"));
         console.error(chalk.gray("  archbyte login"));
         console.error();
-        console.error(chalk.gray("Basic tier includes unlimited scans."));
+        console.error(chalk.gray("Free tier includes unlimited scans. No credit card required."));
         process.exit(1);
     }
     // Token expired locally
@@ -44,7 +44,7 @@ export async function requireLicense(action) {
                 "Content-Type": "application/json",
             },
             body: JSON.stringify({ action }),
-            signal: AbortSignal.timeout(5000),
+            signal: AbortSignal.timeout(NETWORK_TIMEOUT_MS),
         });
         if (res.status === 401) {
             console.error();
@@ -112,7 +112,7 @@ export async function recordUsage(meta) {
                 "Content-Type": "application/json",
             },
             body: JSON.stringify(meta),
-            signal: AbortSignal.timeout(5000),
+            signal: AbortSignal.timeout(NETWORK_TIMEOUT_MS),
         });
     }
     catch {

package/dist/cli/mcp-server.d.ts

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