@noobdemon/noob-cli 1.10.20 → 1.11.1

package/src/api.js

@@ -1,7 +1,7 @@
 // Client for the noob gateway (claude-code-proxy worker). The gateway validates
 // the API key against Supabase, enforces plan limits, and hides the real
 // upstream. The CLI only ever sees the gateway URL + the user's key.
-import { config } from "./config.js";
+import { config } from './config.js';
 
 // ── memoryToken: per-session random token for upstream conversation state.
 // Browser sends something like "uuid_uuid" (two v4 UUIDs joined by _).
@@ -10,9 +10,9 @@ import { config } from "./config.js";
 let _sessionMemoryToken = null;
 
 function makeUUID() {
-  return "xxxxxxxx-xxxx-4xxx-yxxx-xxxxxxxxxxxx".replace(/[xy]/g, (c) => {
+  return 'xxxxxxxx-xxxx-4xxx-yxxx-xxxxxxxxxxxx'.replace(/[xy]/g, (c) => {
     const r = (Math.random() * 16) | 0;
-    const v = c === "x" ? r : (r & 0x3) | 0x8;
+    const v = c === 'x' ? r : (r & 0x3) | 0x8;
     return v.toString(16);
   });
 }
@@ -36,26 +36,30 @@ export function resetMemoryToken() {
 // nếu chỉ check ở top-level, flag --insecure-tls set sau import sẽ không kịp.
 let _tlsWarned = false;
 export function applyInsecureTLS() {
-  if (process.env.NOOB_INSECURE_TLS !== "1") return;
-  process.env.NODE_TLS_REJECT_UNAUTHORIZED = "0";
+  if (process.env.NOOB_INSECURE_TLS !== '1') return;
+  process.env.NODE_TLS_REJECT_UNAUTHORIZED = '0';
   if (_tlsWarned) return;
   _tlsWarned = true;
   // Cảnh báo rõ ràng: flag này tắt verify TLS TOÀN PROCESS. MITM-able.
-  console.warn("\x1b[33m⚠ NOOB_INSECURE_TLS=1: TLS verification DISABLED for this process. MITM-vulnerable. Unset when done.\x1b[0m");
+  console.warn(
+    '\x1b[33m⚠ NOOB_INSECURE_TLS=1: TLS verification DISABLED for this process. MITM-vulnerable. Unset when done.\x1b[0m'
+  );
 }
-// Vẫn áp dụng ngay nếu env có sẵn từ shell (export NOOB_INSECURE_TLS=1).
-applyInsecureTLS();
+// KHÔNG gọi applyInsecureTLS() ở top-level: vì ESM hoist, lệnh này sẽ chạy
+// TRƯỚC khi bin/noob.js kịp parse `--insecure-tls` và set env. bin/noob.js đã
+// gọi applyInsecureTLS() sau khi parse argv — đó là điểm duy nhất.
+// Nếu env NOOB_INSECURE_TLS=1 đã có sẵn từ shell, bin/noob.js cũng sẽ áp dụng.
 
 function authHeaders() {
-  const h = { "Content-Type": "application/json" };
-  if (config.apiKey) h["Authorization"] = "Bearer " + config.apiKey;
+  const h = { 'Content-Type': 'application/json' };
+  if (config.apiKey) h['Authorization'] = 'Bearer ' + config.apiKey;
   return h;
 }
 
 class ApiError extends Error {
   constructor(message, { status, code, reset_at, plan, retryable, partial } = {}) {
     super(message);
-    this.name = "ApiError";
+    this.name = 'ApiError';
     this.status = status;
     this.code = code;
     this.reset_at = reset_at;
@@ -64,15 +68,15 @@ class ApiError extends Error {
     // nếu lỗi 4xx auth/bad-request (retry vô nghĩa). Tự suy ra khi không truyền.
     this.retryable = retryable ?? deriveRetryable({ status, code });
     // partial: phần text đã nhận được trước khi lỗi (cho phép caller continue).
-    this.partial = partial || "";
+    this.partial = partial || '';
   }
 }
 
 // Phân loại lỗi gateway: 5xx + 408/429 (không phải plan limit) + timeout/network
 // → retryable. 4xx khác (401 auth, 400 bad request, 429 plan_limit) → KHÔNG retry.
 function deriveRetryable({ status, code }) {
-  if (code === "timeout") return true;
-  if (code === "plan_limit") return false;
+  if (code === 'timeout') return true;
+  if (code === 'plan_limit') return false;
   if (!status) return true; // không có status = network drop / fetch throw → retry
   if (status >= 500) return true;
   if (status === 408 || status === 429) return true;
@@ -96,7 +100,13 @@ async function parseError(resp) {
 // Unlimited.surf & similar proxies inject web search results as XML/markdown blocks
 // into the SSE stream. These get appended as regular text and confuse the AI model
 // when seen in subsequent turns (it thinks they are prompt injection attempts).
-function cleanResponseText(text) {
+//
+// QUAN TRỌNG: chỉ chạy stripping markdown-heading khi mode === 'search'. Với
+// chat/merge, người dùng có thể đang viết về chính chủ đề "Web Search Results"
+// (vd nội dung SEO/content) — regex tham sẽ nuốt mất câu trả lời thật.
+// Các tag XML/bracket/plain-marker thì hiếm khi xuất hiện tự nhiên nên vẫn an
+// toàn để strip ở mọi mode.
+function cleanResponseText(text, mode = 'chat') {
   if (!text) return text;
   let cleaned = text;
   // XML/SGML style: <web_search_results>...</web_search_results>
@@ -108,8 +118,15 @@ function cleanResponseText(text) {
   // Plain text markers: web_search_results ... web_search_results_end
   cleaned = cleaned.replace(/web_search_results[\s\S]*?web_search_results_end/gi, '');
   cleaned = cleaned.replace(/web_search_summary[\s\S]*?web_search_summary_end/gi, '');
-  // Markdown headings: ## Web Search Results / ## Web Search Summary (with content until next heading)
-  cleaned = cleaned.replace(/^#{1,3}\s+Web\s+Search\s+(Results|Summary)\s*[\s\S]*?(?=^#{1,3}|\n# |$)/gim, '');
+  // Markdown headings: chỉ áp dụng cho search mode (nguồn duy nhất bị inject
+  // dạng heading bởi proxy). Lookahead bám sát: dừng tại heading kế tiếp HOẶC
+  // 2 dòng trống liên tiếp (đoạn văn mới) — tránh nuốt phần còn lại của câu trả lời.
+  if (mode === 'search') {
+    cleaned = cleaned.replace(
+      /^#{1,3}\s+Web\s+Search\s+(Results|Summary)\b[^\n]*\n[\s\S]*?(?=^#{1,3}\s|\n\n\n|$)/gim,
+      ''
+    );
+  }
   return cleaned.trim();
 }
 
@@ -137,58 +154,98 @@ function hasUnclosedToolBlock(text) {
  *
  * @returns {Promise<{text:string, reasoning:string}>}
  */
-export async function stream({ mode = "chat", message, model, system, conversation, effort, signal, onDelta, onReasoning, onStatus, maxContinues = Infinity }) {
-  const endpoint = mode === "search" ? "/api/search" : mode === "merge" ? "/api/merge" : "/api/chat";
+export async function stream({
+  mode = 'chat',
+  message,
+  model,
+  system,
+  conversation,
+  effort,
+  signal,
+  onDelta,
+  onReasoning,
+  onStatus,
+  maxContinues = Infinity,
+}) {
+  const endpoint =
+    mode === 'search' ? '/api/search' : mode === 'merge' ? '/api/merge' : '/api/chat';
 
-  let fullText = "";
-  let reasoning = "";
+  let fullText = '';
+  let reasoning = '';
   let prompt = message; // prompt gửi đi: lần đầu = nguyên bản, các lần sau = nối tiếp
-  let lastFinishReason = "stop"; // stop | truncated | tool_unclosed | empty | network_drop
+  let lastFinishReason = 'stop'; // stop | truncated | tool_unclosed | empty | network_drop
   let emptyStreak = 0; // số lần liên tiếp stream rỗng (chống loop vô tận khi upstream chết hẳn)
 
   for (let attempt = 0; ; attempt++) {
-    const r = await streamOnce({ endpoint, mode, message: prompt, model, system, conversation, effort, signal, onStatus, onDelta, onReasoning });
-    fullText = mode === "chat" ? fullText + r.text : r.text; // chat: ghép các đoạn nối tiếp; mode khác: thay thế
+    const r = await streamOnce({
+      endpoint,
+      mode,
+      message: prompt,
+      model,
+      system,
+      conversation,
+      effort,
+      signal,
+      onStatus,
+      onDelta,
+      onReasoning,
+    });
+    fullText = mode === 'chat' ? fullText + r.text : r.text; // chat: ghép các đoạn nối tiếp; mode khác: thay thế
     if (r.reasoning) reasoning = r.reasoning;
 
     // Phát hiện thêm: tool block mở mà chưa đóng → coi như bị cắt dù gateway báo done.
-    const toolUnclosed = mode === "chat" && hasUnclosedToolBlock(fullText);
+    const toolUnclosed = mode === 'chat' && hasUnclosedToolBlock(fullText);
     const truncated = r.truncated || toolUnclosed;
 
-    if (!truncated) { lastFinishReason = "stop"; break; }
-    if (mode !== "chat") { lastFinishReason = "truncated"; break; }
+    if (!truncated) {
+      lastFinishReason = 'stop';
+      break;
+    }
+    if (mode !== 'chat') {
+      lastFinishReason = 'truncated';
+      break;
+    }
 
     // Đếm chuỗi rỗng: nếu 3 lần liên tiếp model trả rỗng → upstream chết hẳn, dừng.
     if (!r.text.trim()) {
       emptyStreak++;
-      if (emptyStreak >= 3) { lastFinishReason = "empty"; break; }
+      if (emptyStreak >= 3) {
+        lastFinishReason = 'empty';
+        break;
+      }
     } else {
       emptyStreak = 0;
     }
 
     if (attempt >= maxContinues) {
-      lastFinishReason = toolUnclosed ? "tool_unclosed" : "truncated";
+      lastFinishReason = toolUnclosed ? 'tool_unclosed' : 'truncated';
       break;
     }
     prompt = continuationPrompt(message, fullText);
   }
 
-  return { text: cleanResponseText(fullText.trim()), reasoning: reasoning.trim(), finishReason: lastFinishReason };
+  return {
+    text: cleanResponseText(fullText.trim(), mode),
+    reasoning: reasoning.trim(),
+    finishReason: lastFinishReason,
+  };
 }
 
 // Dựng prompt "nối tiếp" khi câu trả lời bị cắt giữa chừng: gửi lại nguyên ngữ
 // cảnh gốc + phần model đã viết dở + yêu cầu viết TIẾP đúng chỗ dừng, không lặp.
 function continuationPrompt(message, partial) {
-  const bar = "=".repeat(60);
+  const bar = '='.repeat(60);
   return (
     message +
-    "\n\n" + bar +
-    "\n## ASSISTANT (bị ngắt giữa chừng — phần trả lời dưới đây CHƯA hoàn tất)\n" +
+    '\n\n' +
+    bar +
+    '\n## ASSISTANT (bị ngắt giữa chừng — phần trả lời dưới đây CHƯA hoàn tất)\n' +
     partial +
-    "\n\n" + bar +
-    "\n# SYSTEM: Phần trả lời ngay trên bị mạng/timeout cắt ngang trước khi xong. " +
-    "Hãy VIẾT TIẾP liền mạch từ ĐÚNG ký tự cuối cùng ở trên — KHÔNG lặp lại hay diễn đạt lại bất kỳ chữ nào đã hiện, KHÔNG mở đầu lại, KHÔNG thêm lời dẫn. " +
-    "Chỉ xuất phần CÒN LẠI. Nếu đang viết dở một khối tool thì hoàn tất đúng khối đó. Nếu thật ra đã xong, chỉ xuất một dấu cách rồi dừng."
+    '\n\n' +
+    bar +
+    '\n# SYSTEM: Phần trả lời ngay trên bị mạng/timeout cắt ngang trước khi xong. ' +
+    'Hãy VIẾT TIẾP liền mạch từ ĐÚNG ký tự cuối cùng ở trên — KHÔNG lặp lại hay diễn đạt lại bất kỳ chữ nào đã hiện, KHÔNG mở đầu lại, KHÔNG thêm lời dẫn. ' +
+    'Chỉ xuất phần CÒN LẠI. Nếu đang viết dở một khối tool thì hoàn tất đúng khối đó. Nếu thật ra đã xong, chỉ xuất một dấu cách rồi dừng.'
   );
 }
 
@@ -196,12 +253,24 @@ function continuationPrompt(message, partial) {
  * One network attempt of the stream. Returns this attempt's accumulated text +
  * a `truncated` flag telling the caller whether the reply was cut short.
  */
-async function streamOnce({ endpoint, mode, message, model, system, conversation, effort, signal, onStatus, onDelta, onReasoning }) {
+async function streamOnce({
+  endpoint,
+  mode,
+  message,
+  model,
+  system,
+  conversation,
+  effort,
+  signal,
+  onStatus,
+  onDelta,
+  onReasoning,
+}) {
   // chat body: gửi system + conversation riêng để gateway forward đúng tới upstream.
   // Worker gateway (handleChat) + upstream đều nhận shape này.
   let body;
-  if (mode === "search") body = { query: message };
-  else if (mode === "merge") body = { message };
+  if (mode === 'search') body = { query: message };
+  else if (mode === 'merge') body = { message };
   else {
     body = { message, model, remember: true, memoryToken: getMemoryToken() };
     if (system) body.customInstructions = system;
@@ -211,17 +280,17 @@ async function streamOnce({ endpoint, mode, message, model, system, conversation
 
   const ctrl = new AbortController();
   const onUserAbort = () => ctrl.abort();
-  signal?.addEventListener("abort", onUserAbort, { once: true });
+  signal?.addEventListener('abort', onUserAbort, { once: true });
 
-  let text = "";
-  let reasoning = "";
+  let text = '';
+  let reasoning = '';
   let sawDone = false; // thấy {done} = stream kết thúc tử tế (không bị cắt)
   let truncated = false; // gateway báo upstream bị cắt giữa chừng
 
   // Một dòng SSE → cập nhật text/reasoning. Tách ra để dùng lại khi flush dòng cuối.
   const processLine = (rawLine) => {
     const line = rawLine.trim();
-    if (!line.startsWith("data:")) return;
+    if (!line.startsWith('data:')) return;
     const data = line.slice(5).trim();
     if (!data) return;
     let p;
@@ -243,14 +312,14 @@ async function streamOnce({ endpoint, mode, message, model, system, conversation
     if (p.truncated) truncated = true;
     // Handle {finish: true, reason: "stop"} — unlimited.surf & some proxies
     // send this as completion signal instead of (or in addition to) {done: true}
-    if (p.finish === true && p.reason === "stop") sawDone = true;
+    if (p.finish === true && p.reason === 'stop') sawDone = true;
     if (p.done) sawDone = true;
     if (p.error) throw new ApiError(p.error, {});
   };
 
   try {
     const resp = await fetch(config.gatewayUrl + endpoint, {
-      method: "POST",
+      method: 'POST',
       headers: authHeaders(),
       body: JSON.stringify(body),
       signal: ctrl.signal,
@@ -259,13 +328,18 @@ async function streamOnce({ endpoint, mode, message, model, system, conversation
 
     const reader = resp.body.getReader();
     const decoder = new TextDecoder();
-    let buf = "";
+    let buf = '';
     while (true) {
       const { done, value } = await reader.read();
       if (done) break;
       buf += decoder.decode(value, { stream: true });
+      // Chuẩn hoá CRLF→LF: một số reverse proxy (Cloudflare, nginx config khác)
+      // gửi line endings \r\n. processLine() có trim() nên \r trailing tự rụng,
+      // nhưng nếu chunk biên giới rơi đúng giữa \r và \n thì vẫn ổn — đây là
+      // safety belt cho trường hợp \r đơn lẻ (rất hiếm nhưng có thấy thực tế).
+      buf = buf.replace(/\r\n/g, '\n').replace(/\r/g, '\n');
       let nl;
-      while ((nl = buf.indexOf("\n")) !== -1) {
+      while ((nl = buf.indexOf('\n')) !== -1) {
         processLine(buf.slice(0, nl));
         buf = buf.slice(nl + 1);
       }
@@ -275,23 +349,23 @@ async function streamOnce({ endpoint, mode, message, model, system, conversation
 
     // Chat: gateway gửi {done} khi xong sạch. Stream EOF mà chưa thấy {done} dù
     // đã có chữ → kết nối/edge rớt giữa chừng → coi như bị cắt để nối tiếp.
-    if (mode === "chat" && !sawDone && text) truncated = true;
+    if (mode === 'chat' && !sawDone && text) truncated = true;
 
     return { text, reasoning, truncated };
   } catch (err) {
     if (signal?.aborted) throw err; // người dùng bấm Ctrl+C → huỷ thật, không nối tiếp
     // Rớt mạng giữa chừng (không phải huỷ): với chat, nếu đã có
     // chữ thì trả phần đã nhận + cờ truncated để lớp trên nối tiếp.
-    if (mode === "chat" && text) return { text, reasoning, truncated: true };
+    if (mode === 'chat' && text) return { text, reasoning, truncated: true };
     throw err;
   } finally {
-    signal?.removeEventListener("abort", onUserAbort);
+    signal?.removeEventListener('abort', onUserAbort);
   }
 }
 
 /** Fetch the current key's quota/usage from the gateway (no request consumed). */
 export async function usage() {
-  const resp = await fetch(config.gatewayUrl + "/api/usage", { headers: authHeaders() });
+  const resp = await fetch(config.gatewayUrl + '/api/usage', { headers: authHeaders() });
   if (!resp.ok) throw await parseError(resp);
   return await resp.json();
 }

package/src/config.js

@@ -1,17 +1,17 @@
-import fs from "node:fs";
-import path from "node:path";
-import os from "node:os";
+import fs from 'node:fs';
+import path from 'node:path';
+import os from 'node:os';
 
 // Default gateway — the Noob Demon endpoint. The real upstream is hidden behind
 // it; the CLI only ever talks to this gateway.
-const DEFAULT_GATEWAY = "https://claude-code-proxy.lohieuky678.workers.dev";
+const DEFAULT_GATEWAY = 'https://claude-code-proxy.lohieuky678.workers.dev';
 
-const DIR = path.join(os.homedir(), ".noob");
-const FILE = path.join(DIR, "config.json");
+const DIR = path.join(os.homedir(), '.noob');
+const FILE = path.join(DIR, 'config.json');
 
 function read() {
   try {
-    return JSON.parse(fs.readFileSync(FILE, "utf8"));
+    return JSON.parse(fs.readFileSync(FILE, 'utf8'));
   } catch {
     return {};
   }
@@ -20,24 +20,24 @@ function read() {
 function write(cfg) {
   try {
     fs.mkdirSync(DIR, { recursive: true });
-    fs.writeFileSync(FILE, JSON.stringify(cfg, null, 2), "utf8");
+    fs.writeFileSync(FILE, JSON.stringify(cfg, null, 2), 'utf8');
     return true;
   } catch {
     return false;
   }
 }
 
-let cache = read();
+const cache = read();
 
 export const config = {
   get gatewayUrl() {
     return process.env.NOOB_API_BASE || cache.gatewayUrl || DEFAULT_GATEWAY;
   },
   get apiKey() {
-    return process.env.NOOB_API_KEY || cache.apiKey || "";
+    return process.env.NOOB_API_KEY || cache.apiKey || '';
   },
   get model() {
-    return cache.model || "";
+    return cache.model || '';
   },
   setKey(key) {
     cache.apiKey = key;