@leeandrew94/ccm 0.2.1 → 0.3.1

package/README.md

@@ -35,14 +35,14 @@ rm -rf ~/.ccm
 ## Quick Start
 
 ```bash
-# Add a profile (interactive prompts for Base URL / Token / Model)
-ccm add mimo
+# Interactive setup wizard (recommended)
+ccm init
 
 # Launch
 ccm mimo
 
 # Open another terminal, switch to a different model
-ccm add deepseek
+ccm init
 ccm deepseek
 
 # See what's running
@@ -53,6 +53,7 @@ ccm ps
 
 | Command | Description |
 |---|---|
+| `ccm init` | Interactive setup wizard |
 | `ccm <name>` | Load profile and launch claude |
 | `ccm add <name>` | Add a new profile |
 | `ccm edit <name>` | Edit an existing profile |

package/README.zh.md

@@ -35,14 +35,14 @@ rm -rf ~/.ccm
 ## 快速开始
 
 ```bash
-# 添加 profile（交互式输入 Base URL / Token / Model）
-ccm add mimo
+# 交互式初始化（推荐）
+ccm init
 
 # 启动
 ccm mimo
 
 # 新开终端，换个模型
-ccm add deepseek
+ccm init
 ccm deepseek
 
 # 查看谁在跑
@@ -53,6 +53,7 @@ ccm ps
 
 | 命令 | 说明 |
 |---|---|
+| `ccm init` | 交互式初始化向导 |
 | `ccm <name>` | 加载 profile 并启动 claude |
 | `ccm add <name>` | 新增 profile |
 | `ccm edit <name>` | 修改 profile |

package/bin/ccm.js

@@ -443,6 +443,213 @@ function cmdList() {
 `);
 }
 
+// src/commands/init.ts
+import https from "https";
+import http from "http";
+function httpRequest(url, token, timeout = 8e3) {
+  return new Promise((resolve) => {
+    const parsed = new URL(url);
+    const mod = parsed.protocol === "https:" ? https : http;
+    const req = mod.request(url, {
+      method: "GET",
+      headers: { "Authorization": `Bearer ${token}`, "Content-Type": "application/json" },
+      timeout
+    }, (res) => {
+      let body = "";
+      res.on("data", (chunk) => {
+        body += chunk;
+      });
+      res.on("end", () => {
+        try {
+          resolve({ status: res.statusCode || 0, body: JSON.parse(body) });
+        } catch {
+          resolve({ status: res.statusCode || 0, body });
+        }
+      });
+    });
+    req.on("error", () => resolve({ status: 0, body: {} }));
+    req.on("timeout", () => {
+      req.destroy();
+      resolve({ status: 0, body: {} });
+    });
+    req.end();
+  });
+}
+async function fetchModels(baseUrl, token) {
+  const url = baseUrl.replace(/\/+$/, "");
+  for (const suffix of ["/anthropic", "/v1/chat/completions"]) {
+    if (url.endsWith(suffix)) return [];
+  }
+  const endpoints = [`${url}/models`, `${url}/v1/models`];
+  for (const ep of endpoints) {
+    const result = await httpRequest(ep, token);
+    if (result.status === 200 && result.body) {
+      const data = result.body.data || result.body.models;
+      if (Array.isArray(data)) {
+        return data.map((m) => m.id || m.name).filter(Boolean).sort();
+      }
+    }
+  }
+  return [];
+}
+async function selectModel(models, label) {
+  if (models.length === 0) {
+    return await ask("Model name");
+  }
+  console.log(`
+  \x1B[1m${label}\x1B[0m (\u2191\u2193 navigate, Enter to select):
+`);
+  const PAGE = 12;
+  let offset = 0;
+  let selected = 0;
+  function render() {
+    process.stdout.write("\x1B[?25l");
+    const start = offset;
+    const end = Math.min(offset + PAGE, models.length);
+    for (let i = start; i < end; i++) {
+      process.stdout.write("\x1B[2K");
+      if (i === selected) {
+        console.log(`  \x1B[36m\u276F\x1B[0m \x1B[1m${models[i]}\x1B[0m`);
+      } else {
+        console.log(`    \x1B[90m${models[i]}\x1B[0m`);
+      }
+    }
+    process.stdout.write("\x1B[2K");
+    if (models.length > PAGE) {
+      console.log(`  \x1B[90m${selected + 1}/${models.length} \u2014 \u2191\u2193 scroll, Enter select\x1B[0m`);
+    }
+    const lines = end - start + (models.length > PAGE ? 1 : 0);
+    process.stdout.write(`\x1B[${lines}A`);
+  }
+  return new Promise((resolve) => {
+    process.stdin.setRawMode(true);
+    process.stdin.resume();
+    process.stdin.setEncoding("utf-8");
+    render();
+    let buf = "";
+    process.stdin.on("data", function onData(key) {
+      if (key === "\x1B[A") {
+        selected = Math.max(0, selected - 1);
+        if (selected < offset) offset = selected;
+        render();
+      } else if (key === "\x1B[B") {
+        selected = Math.min(models.length - 1, selected + 1);
+        if (selected >= offset + PAGE) offset = selected - PAGE + 1;
+        render();
+      } else if (key === "\r" || key === "\n") {
+        const lines = Math.min(PAGE, models.length - offset) + (models.length > PAGE ? 1 : 0);
+        for (let i = 0; i < lines; i++) {
+          process.stdout.write("\x1B[2K\n");
+        }
+        process.stdout.write(`\x1B[${lines}A`);
+        for (let i = 0; i < lines; i++) {
+          process.stdout.write("\x1B[2K\n");
+        }
+        process.stdout.write(`\x1B[${lines}A`);
+        process.stdout.write(`  \x1B[36m\u276F\x1B[0m \x1B[1m${models[selected]}\x1B[0m
+`);
+        process.stdout.write("\x1B[?25h");
+        process.stdin.setRawMode(false);
+        process.stdin.removeListener("data", onData);
+        process.stdin.pause();
+        resolve(models[selected]);
+      } else if (key === "") {
+        process.stdout.write("\x1B[?25h");
+        process.exit(0);
+      }
+    });
+  });
+}
+async function cmdInit() {
+  console.log();
+  console.log("  \x1B[1;36m\u250C\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2510\x1B[0m");
+  console.log("  \x1B[1;36m\u2502\x1B[0m  ccm - Claude Code Model Manager         \x1B[1;36m\u2502\x1B[0m");
+  console.log("  \x1B[1;36m\u2502\x1B[0m  \x1B[90mSet up your model profile\x1B[0m                \x1B[1;36m\u2502\x1B[0m");
+  console.log("  \x1B[1;36m\u2514\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2518\x1B[0m");
+  console.log();
+  if (!whichClaude()) {
+    warn("Claude Code is not installed.");
+    info("Install it first: npm install -g @anthropic-ai/claude-code\n");
+    const proceed = await ask("Continue anyway? [y/N]");
+    if (proceed.toLowerCase() !== "y") return;
+  }
+  const baseUrl = await ask("API Base URL (*)");
+  if (!baseUrl) {
+    err("Base URL is required.");
+    return;
+  }
+  const token = await ask("API Key (*)");
+  if (!token) {
+    err("API Key is required.");
+    return;
+  }
+  console.log();
+  info("Validating connection...");
+  const isAnthropic = baseUrl.toLowerCase().includes("anthropic");
+  let models = [];
+  if (!isAnthropic) {
+    models = await fetchModels(baseUrl, token);
+    if (models.length > 0) {
+      ok(`Found \x1B[1m${models.length}\x1B[0m models`);
+    } else {
+      info("Could not fetch model list (enter manually)");
+    }
+  }
+  let model;
+  if (isAnthropic) {
+    const anthropicModels = [
+      "claude-sonnet-4-20250514",
+      "claude-opus-4-20250514",
+      "claude-haiku-4-20250501"
+    ];
+    model = await selectModel(anthropicModels, "Select a model");
+  } else if (models.length > 0) {
+    model = await selectModel(models, "Select a model");
+  } else {
+    model = await ask("Model name (*)");
+    if (!model) {
+      err("Model name is required.");
+      return;
+    }
+  }
+  console.log();
+  info("Optional: model mappings for Haiku/Sonnet/Opus (Enter to skip)\n");
+  const haiku = await ask("Haiku model");
+  const sonnet = await ask("Sonnet model");
+  const opus = await ask("Opus model");
+  console.log();
+  const name = await ask("Profile name (*)");
+  if (!name) {
+    err("Profile name is required.");
+    return;
+  }
+  if (profileExists(name)) {
+    warn(`Profile '${name}' already exists.`);
+    const overwrite = await ask("Overwrite? [y/N]");
+    if (overwrite.toLowerCase() !== "y") {
+      info("Cancelled.");
+      return;
+    }
+  }
+  const profile = {
+    ANTHROPIC_BASE_URL: baseUrl,
+    ANTHROPIC_AUTH_TOKEN: token,
+    ANTHROPIC_MODEL: model
+  };
+  if (haiku) profile.ANTHROPIC_DEFAULT_HAIKU_MODEL = haiku;
+  if (sonnet) profile.ANTHROPIC_DEFAULT_SONNET_MODEL = sonnet;
+  if (opus) profile.ANTHROPIC_DEFAULT_OPUS_MODEL = opus;
+  addProfile(name, profile);
+  console.log();
+  ok(`Profile '\x1B[1m${name}\x1B[0m' saved!
+`);
+  console.log(`  \x1B[36mccm ${name}\x1B[0m          launch with this profile`);
+  console.log(`  \x1B[36mccm edit ${name}\x1B[0m      edit configuration`);
+  console.log(`  \x1B[36mccm test ${name}\x1B[0m      test connection`);
+  console.log(`  \x1B[36mccm balance ${name}\x1B[0m   check credits`);
+  console.log();
+}
+
 // src/commands/runtime.ts
 function cmdPs() {
   const runs = getAllRuns();
@@ -494,12 +701,12 @@ function cmdCheck() {
 }
 
 // src/commands/diagnose.ts
-import https from "https";
-import http from "http";
-function httpRequest(url, token, method = "GET", timeout = 1e4, headers = {}) {
+import https2 from "https";
+import http2 from "http";
+function httpRequest2(url, token, method = "GET", timeout = 1e4, headers = {}) {
   return new Promise((resolve) => {
     const parsed = new URL(url);
-    const mod = parsed.protocol === "https:" ? https : http;
+    const mod = parsed.protocol === "https:" ? https2 : http2;
     const reqHeaders = {
       "Authorization": `Bearer ${token}`,
       "x-api-key": token,
@@ -536,7 +743,7 @@ function isAnthropicCompat(baseUrl) {
 async function testConnection(baseUrl, token) {
   const url = baseUrl.replace(/\/+$/, "");
   if (isAnthropicCompat(baseUrl)) {
-    const result = await httpRequest(`${url}/v1/messages`, token, "POST", 1e4, {
+    const result = await httpRequest2(`${url}/v1/messages`, token, "POST", 1e4, {
       "anthropic-version": "2023-06-01"
     });
     if (result.status === 200 || result.status === 400) return { ok: true, endpoint: "/v1/messages" };
@@ -544,14 +751,14 @@ async function testConnection(baseUrl, token) {
     return { ok: false, error: result.error || `HTTP ${result.status}` };
   }
   for (const endpoint of ["/v1/models", "/models"]) {
-    const result = await httpRequest(`${url}${endpoint}`, token);
+    const result = await httpRequest2(`${url}${endpoint}`, token);
     if (result.status === 200) return { ok: true, endpoint };
     if (result.status === 401 || result.status === 403) return { ok: false, error: "Authentication failed (invalid token)" };
   }
   try {
     const baseParts = url.split("//");
     const hostUrl = `${baseParts[0]}//${baseParts[1].split("/")[0]}/`;
-    const result = await httpRequest(hostUrl, token);
+    const result = await httpRequest2(hostUrl, token);
     if (result.status > 0) return { ok: true, endpoint: "(host reachable)" };
   } catch {
   }
@@ -568,10 +775,11 @@ async function queryBalance(baseUrl, token) {
     `${url}/dashboard/billing/usage`,
     `${url}/api/user/balance`,
     `${url}/billing/usage`,
-    `${url}/v1/dashboard/billing/usage`
+    `${url}/v1/dashboard/billing/usage`,
+    `${url}/auth/key`
   ];
   for (const ep of endpoints) {
-    const result = await httpRequest(ep, token);
+    const result = await httpRequest2(ep, token);
     if (result.status === 200) return result.body;
   }
   return null;
@@ -636,6 +844,23 @@ function formatBalanceCell(data, statusWidth) {
     const bal = fmtAmount(data.total_available, currency);
     return [`\x1B[1;32m${bal}\x1B[0m`, "\x1B[32m\u25CF active\x1B[0m"];
   }
+  if ("data" in data && typeof data.data === "object" && data.data !== null && "usage" in data.data && "limit" in data.data) {
+    const d = data.data;
+    const used = parseFloat(d.usage);
+    const limit = parseFloat(d.limit);
+    const remaining = limit - used;
+    const balanceStr = `\x1B[1;32m${fmtAmount(remaining.toFixed(4))}\x1B[0m`;
+    let statusStr;
+    if (limit > 0) {
+      const ratio = used / limit;
+      const pct = ratio * 100;
+      const barStr = bar(ratio, 12);
+      statusStr = `${barStr} ${pct.toFixed(1)}% used`;
+    } else {
+      statusStr = d.is_free_tier ? "\x1B[32m\u25CF free tier\x1B[0m" : "\x1B[32m\u25CF active\x1B[0m";
+    }
+    return [balanceStr, statusStr];
+  }
   if ("data" in data && typeof data.data === "object" && data.data !== null) {
     const d = data.data;
     const granted = d.total_granted;
@@ -756,7 +981,7 @@ function padAnsi2(text, width) {
 }
 
 // src/completions.ts
-var COMMANDS = ["add", "edit", "rm", "list", "ls", "ps", "kill", "check", "test", "balance", "bal", "config", "completions"];
+var COMMANDS = ["init", "add", "edit", "rm", "list", "ls", "ps", "kill", "check", "test", "balance", "bal", "config", "completions", "sessions"];
 function cmdCompletions(args) {
   const shell = args.shell || detectShell();
   if (shell === "zsh") {
@@ -1126,7 +1351,7 @@ function cleanupOldTrash() {
 
 // src/session-server.ts
 import { exec } from "child_process";
-import http2 from "http";
+import http3 from "http";
 function openBrowser(url) {
   const cmd = process.platform === "darwin" ? `open "${url}"` : `xdg-open "${url}"`;
   exec(cmd);
@@ -1137,7 +1362,7 @@ function jsonResponse(res, data, status = 200) {
 }
 function startSessionServer(port = 13501) {
   return new Promise((resolve) => {
-    const server = http2.createServer((req, res) => {
+    const server = http3.createServer((req, res) => {
       const url = new URL(req.url || "/", `http://localhost:${port}`);
       if (req.method === "GET" && (url.pathname === "/" || /^\/session\/[0-9a-f-]+$/.test(url.pathname))) {
         res.writeHead(200, { "Content-Type": "text/html; charset=utf-8" });
@@ -1305,10 +1530,37 @@ function escHtml(s) {
 
 function renderMarkdown(text) {
   if (!text) return '';
-  var h = escHtml(text);
   var BT = String.fromCharCode(96);
-  h = h.replace(new RegExp(BT + BT + BT + '(\\\\w*)\\\\n([\\\\s\\\\S]*?)' + BT + BT + BT, 'g'), function(m, lang, code) { return '<pre><code>' + highlightCode(code) + '</code></pre>'; });
-  h = h.replace(new RegExp(BT + '([^\\\\n]+?)' + BT, 'g'), '<code>$1</code>');
+  var h = text;
+
+  // Placeholder delimiters: Unicode PUA (Private Use Area) codepoints.
+  // Control chars (\\x00, \\x01) would be replaced with \uFFFD (\uFFFD) by the
+  // browser's HTML parser, so we use printable PUA chars that survive intact.
+  var PS = '\uE000';  // placeholder start
+  var PE = '\uE001';  // placeholder end
+
+  // Step 1: Extract fenced code blocks from RAW text (before any HTML escaping)
+  // Apply syntax highlighting to raw code so highlightCode regexes work correctly
+  var codeBlocks = [];
+  h = h.replace(new RegExp(BT + BT + BT + '(\\\\w*)\\\\n([\\\\s\\\\S]*?)' + BT + BT + BT, 'g'), function(m, lang, code) {
+    var idx = codeBlocks.length;
+    codeBlocks.push('<pre><code>' + highlightCode(code) + '</code></pre>');
+    return PS + 'CCM_CB' + idx + PE;
+  });
+
+  // Step 2: Extract inline code from RAW text
+  // escHtml the code content so it displays literally
+  var inlineCodes = [];
+  h = h.replace(new RegExp(BT + '([^\\\\n]+?)' + BT, 'g'), function(m, code) {
+    var idx = inlineCodes.length;
+    inlineCodes.push('<code>' + escHtml(code) + '</code>');
+    return PS + 'CCM_IC' + idx + PE;
+  });
+
+  // Step 3: Now HTML-escape the remaining text (blockquotes, etc.)
+  h = escHtml(h);
+
+  // Step 4: Apply markdown formatting
   h = h.replace(/^### (.+)$/gm, '<h3>$1</h3>');
   h = h.replace(/^## (.+)$/gm, '<h2>$1</h2>');
   h = h.replace(/^# (.+)$/gm, '<h1>$1</h1>');
@@ -1340,16 +1592,56 @@ function renderMarkdown(text) {
   h = h.replace(/(<\\/blockquote>)<\\/p>/g, '$1');
   h = h.replace(/<p>(<hr>)/g, '$1');
   h = h.replace(/(<hr>)<\\/p>/g, '$1');
+
+  // Step 5: Restore code blocks and inline code
+  h = h.replace(new RegExp(PS + 'CCM_CB(\\d+)' + PE, 'g'), function(m, idx) { return codeBlocks[parseInt(idx)]; });
+  h = h.replace(new RegExp(PS + 'CCM_IC(\\d+)' + PE, 'g'), function(m, idx) { return inlineCodes[parseInt(idx)]; });
+
   return h;
 }
 
 function highlightCode(code) {
   var h = code;
-  h = h.replace(/(\\/\\/.*$)/gm, '<span class="cm">$1</span>');
-  h = h.replace(/(#.*$)/gm, '<span class="cm">$1</span>');
-  h = h.replace(/("(?:[^"\\\\\\\\]|\\\\\\\\.)*"|'(?:[^'\\\\\\\\]|\\\\\\\\.)*')/g, '<span class="str">$1</span>');
+  var p = [];
+
+  // Same PUA delimiters used in renderMarkdown
+  var PS = '\uE000';
+  var PE = '\uE001';
+
+  // Protect strings with placeholders (before other regexes, to avoid
+  // keywords / comments inside strings being highlighted)
+  h = h.replace(/("[^"]*"|'[^']*')/g, function(m) {
+    var i = p.length;
+    p.push('<span class="str">' + escHtml(m) + '</span>');
+    return PS + 'HL' + i + PE;
+  });
+
+  // Protect // comments
+  h = h.replace(/(\\/\\/.*$)/gm, function(m) {
+    var i = p.length;
+    p.push('<span class="cm">' + escHtml(m) + '</span>');
+    return PS + 'HL' + i + PE;
+  });
+
+  // Protect # comments
+  h = h.replace(/(#.*$)/gm, function(m) {
+    var i = p.length;
+    p.push('<span class="cm">' + escHtml(m) + '</span>');
+    return PS + 'HL' + i + PE;
+  });
+
+  // HTML-escape remaining literal code characters
+  h = escHtml(h);
+
+  // Keywords (safe: placeholders contain no keywords)
   h = h.replace(/\\b(import|export|from|const|let|var|function|return|if|else|for|while|class|extends|new|async|await|try|catch|throw|switch|case|break|default|typeof|instanceof|void|null|undefined|true|false|def|self|print|raise|with|as|in|not|and|or|is|lambda|yield|assert|del|global|nonlocal|pass|elif|except|finally)\\b/g, '<span class="kw">$1</span>');
+
+  // Numbers
   h = h.replace(/\\b(\\d+\\.?\\d*)\\b/g, '<span class="num">$1</span>');
+
+  // Restore protected fragments
+  h = h.replace(new RegExp(PS + 'HL(\\d+)' + PE, 'g'), function(m, i) { return p[parseInt(i)]; });
+
   return h;
 }
 
@@ -1469,6 +1761,7 @@ body {
 .main-header h1 {
   font-size: 16px; font-weight: 600; margin-bottom: 10px;
   line-height: 1.5; color: var(--td-text-primary);
+  overflow: hidden; text-overflow: ellipsis; white-space: nowrap;
 }
 .main-header .meta {
   display: flex; gap: 18px; font-size: 12px;
@@ -1903,6 +2196,7 @@ var VERSION = pkg.version;
 var KNOWN_COMMANDS = /* @__PURE__ */ new Set([
   "_launch",
   "_register",
+  "init",
   "add",
   "edit",
   "rm",
@@ -1923,6 +2217,7 @@ function printHelp() {
 Usage: ccm <command> [options]
 
 Commands:
+  init              Interactive setup wizard
   add <name>        Add a new model profile
   edit <name>       Edit an existing profile
   rm <name>         Delete a profile
@@ -2004,6 +2299,9 @@ async function main() {
     case "_register":
       cmdRegister({ name: args.name, pid: args.pid, tty: args.tty });
       break;
+    case "init":
+      await cmdInit();
+      break;
     case "add":
       await cmdAdd({ name: args.name });
       break;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@leeandrew94/ccm",
-  "version": "0.2.1",
+  "version": "0.3.1",
   "description": "Claude Code Model Manager - switch between AI models per terminal",
   "type": "module",
   "bin": {