kiku-agent 1.0.3 → 1.0.4

package/README.md

@@ -48,7 +48,7 @@ The same package also installs the stdio MCP server command:
 kiku-mcp
 ```
 
-For a local checkout before the package is published, use `npm pack` followed by `npm install -g ./kiku-agent-1.0.3.tgz`.
+For a local checkout before the package is published, use `npm pack` followed by `npm install -g ./kiku-agent-1.0.4.tgz`.
 
 ### ✨ Features
 - **LM Studio Integration**: Connects seamlessly to LM Studio's OpenAI-compatible server at `http://127.0.0.1:1234/v1` (default model `google/gemma-4-e4b`, with auto-detection of loaded models).

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "kiku-agent",
-  "version": "1.0.3",
+  "version": "1.0.4",
   "description": "Kiku AI terminal agent and MCP server for Ju Jingyi knowledge, media, and career data.",
   "type": "module",
   "bin": {

package/server/agent/cli.js

@@ -467,7 +467,7 @@ class InteractiveTuiPrompt {
     const mcpCount = this.agent?.mcpManager?.servers?.size || 0;
     const mcpLabel = mcpCount > 0 ? `${C.emerald}● ${mcpCount} MCP${C.reset}` : `${C.slate}○ 0 MCP${C.reset}`;
 
-    const headerLeft = `  ${C.bold}${C.sakura}◆ KIKU AGENT${C.reset} ${C.slate}v1.0.3${C.reset}`;
+    const headerLeft = `  ${C.bold}${C.sakura}◆ KIKU AGENT${C.reset} ${C.slate}v1.0.4${C.reset}`;
     const headerRight = `${modelLabel}  ${C.slate}│${C.reset}  ${mcpLabel}  ${C.slate}│${C.reset}  ${C.roseGlow}◆ Sakura${C.reset}  `;
     const safeCols = Math.max(1, termCols - 1);
     const headerPad = Math.max(1, safeCols - getStringDisplayWidth(headerLeft) - getStringDisplayWidth(headerRight));

package/server/agent/commands.js

@@ -1460,7 +1460,7 @@ export const handlers = {
     printSolidPanel({
       title: 'Version',
       lines: [
-        `${C.gold}Version:${C.reset} ${C.white}1.0.3${C.reset}`,
+        `${C.gold}Version:${C.reset} ${C.white}1.0.4${C.reset}`,
         `${C.gold}Node:${C.reset}    ${C.white}${process.version}${C.reset}`,
         `${C.gold}Platform:${C.reset} ${C.white}${process.platform} ${process.arch}${C.reset}`,
       ],

package/server/agent/mcpClient.js

@@ -8,7 +8,7 @@ import { KIKU_DATA_DIR } from './appPaths.js';
 
 export class McpClientManager {
   constructor(options = {}) {
-    this.clientInfo = options.clientInfo || { name: 'kiku-agent', version: '1.0.3' };
+    this.clientInfo = options.clientInfo || { name: 'kiku-agent', version: '1.0.4' };
     this.autoLoad = options.autoLoad !== false;
     this.servers = new Map();
     this.toolRegistry = new Map();

package/server/agent/ui.js

@@ -476,7 +476,7 @@ export function getWelcomeHeroLines() {
   return [
     '',
     `  ${C.sakura}██╗  ██╗██╗██╗  ██╗██╗   ██╗     █████╗ ██╗${C.reset}   ${C.bold}${C.white}Kiku AI Terminal Agent${C.reset}`,
-    `  ${C.sakura}██║ ██╔╝██║██║ ██╔╝██║   ██║    ██╔══██╗██║${C.reset}   ${C.gold}v1.0.3${C.reset} ${C.slate}• Intelligent Fan Portal${C.reset}`,
+    `  ${C.sakura}██║ ██╔╝██║██║ ██╔╝██║   ██║    ██╔══██╗██║${C.reset}   ${C.gold}v1.0.4${C.reset} ${C.slate}• Intelligent Fan Portal${C.reset}`,
     `  ${C.roseGlow}█████═╝ ██║█████═╝ ██║   ██║    ███████║██║${C.reset}   ${C.cyberCyan}SSOT • DAG • Dynamic Tools${C.reset}`,
     `  ${C.roseGlow}██╔═██╗ ██║██╔═██╗ ██║   ██║    ██╔══██╗██║${C.reset}   ${C.emerald}LM Studio / Cloud Providers${C.reset}`,
     `  ${C.sakura}██║ ╚██╗██║██║ ╚██╗╚██████╔╝    ██║  ██║██║${C.reset}`,

package/server/agent/youtubeAuth.js

@@ -2,41 +2,64 @@ import fs from 'fs';
 import path from 'path';
 import { spawn, execSync } from 'child_process';
 import { fileURLToPath } from 'url';
+import { KIKU_DATA_DIR } from './appPaths.js';
 
 const __dirname = path.dirname(fileURLToPath(import.meta.url));
 const ENV_PATH = path.resolve(__dirname, '..', '..', '.env');
+const AUTH_CONFIG_PATH = path.join(KIKU_DATA_DIR, 'yt_auth.json');
+const SECURE_COOKIE_PATH = path.join(KIKU_DATA_DIR, 'cookies.txt');
 const YTDLP_CMD = process.env.KIKU_YTDLP || 'yt-dlp';
 
-const BROWSERS = ['chrome', 'edge', 'brave', 'firefox'];
-
 
+const BROWSERS = ['firefox', 'chrome', 'edge', 'brave'];
 
 function findBrowserExe(browser) {
   const isWin = process.platform === 'win32';
   const probes = {
-    chrome: { cmd: isWin ? 'chrome' : 'google-chrome', paths: [
-      'C:\\Program Files\\Google\\Chrome\\Application\\chrome.exe',
-      (process.env.LOCALAPPDATA || '') + '\\Google\\Chrome\\Application\\chrome.exe',
-      'C:\\Program Files (x86)\\Google\\Chrome\\Application\\chrome.exe'
-    ]},
-    edge: { cmd: isWin ? 'msedge' : 'microsoft-edge', paths: [
-      'C:\\Program Files (x86)\\Microsoft\\Edge\\Application\\msedge.exe',
-      'C:\\Program Files\\Microsoft\\Edge\\Application\\msedge.exe'
-    ]},
-    brave: { cmd: isWin ? 'brave' : 'brave-browser', paths: [
-      'C:\\Program Files\\BraveSoftware\\Brave-Browser\\Application\\brave.exe',
-      (process.env.LOCALAPPDATA || '') + '\\BraveSoftware\\Brave-Browser\\Application\\brave.exe'
-    ]},
-    firefox: { cmd: isWin ? 'firefox' : 'firefox', paths: [
-      'C:\\Program Files\\Mozilla Firefox\\firefox.exe',
-      (process.env.LOCALAPPDATA || '') + '\\Mozilla Firefox\\firefox.exe'
-    ]}
+    firefox: {
+      cmd: isWin ? 'firefox' : 'firefox',
+      paths: [
+        'C:\\Program Files\\Mozilla Firefox\\firefox.exe',
+        'C:\\Program Files (x86)\\Mozilla Firefox\\firefox.exe',
+        (process.env.LOCALAPPDATA || '') + '\\Mozilla Firefox\\firefox.exe'
+      ]
+    },
+    chrome: {
+      cmd: isWin ? 'chrome' : 'google-chrome',
+      paths: [
+        'C:\\Program Files\\Google\\Chrome\\Application\\chrome.exe',
+        (process.env.LOCALAPPDATA || '') + '\\Google\\Chrome\\Application\\chrome.exe',
+        'C:\\Program Files (x86)\\Google\\Chrome\\Application\\chrome.exe'
+      ]
+    },
+    edge: {
+      cmd: isWin ? 'msedge' : 'microsoft-edge',
+      paths: [
+        'C:\\Program Files (x86)\\Microsoft\\Edge\\Application\\msedge.exe',
+        'C:\\Program Files\\Microsoft\\Edge\\Application\\msedge.exe',
+        (process.env.LOCALAPPDATA || '') + '\\Microsoft\\Edge\\Application\\msedge.exe'
+      ]
+    },
+    brave: {
+      cmd: isWin ? 'brave' : 'brave-browser',
+      paths: [
+        'C:\\Program Files\\BraveSoftware\\Brave-Browser\\Application\\brave.exe',
+        (process.env.LOCALAPPDATA || '') + '\\BraveSoftware\\Brave-Browser\\Application\\brave.exe'
+      ]
+    }
   };
+
   const p = probes[browser];
   if (!p) return null;
-  try { execSync(isWin ? `where ${p.cmd}` : `which ${p.cmd}`, { stdio: 'ignore', windowsHide: true }); return p.cmd; } catch {}
+  try {
+    execSync(isWin ? `where ${p.cmd}` : `which ${p.cmd}`, { stdio: 'ignore', windowsHide: true });
+    return p.cmd;
+  } catch {}
+
   if (isWin) {
-    for (const fp of p.paths) { if (fs.existsSync(fp)) return fp; }
+    for (const fp of p.paths) {
+      if (fp && fs.existsSync(fp)) return fp;
+    }
   }
   return null;
 }
@@ -63,9 +86,35 @@ function openBrowser(url, browser) {
   }
 }
 
+function readAuthConfig() {
+  try {
+    if (fs.existsSync(AUTH_CONFIG_PATH)) {
+      return JSON.parse(fs.readFileSync(AUTH_CONFIG_PATH, 'utf8')) || {};
+    }
+  } catch {}
+  return {};
+}
 
+function writeAuthConfig(updates) {
+  try {
+    fs.mkdirSync(KIKU_DATA_DIR, { recursive: true });
+    const current = readAuthConfig();
+    const merged = { ...current, ...updates };
+    fs.writeFileSync(AUTH_CONFIG_PATH, JSON.stringify(merged, null, 2), 'utf8');
+  } catch {}
+}
 
 function loadEnv() {
+
+  const userConfig = readAuthConfig();
+  if (userConfig.KIKU_YT_BROWSER && process.env.KIKU_YT_BROWSER === undefined) {
+    process.env.KIKU_YT_BROWSER = userConfig.KIKU_YT_BROWSER;
+  }
+  if (userConfig.KIKU_YT_COOKIES && process.env.KIKU_YT_COOKIES === undefined) {
+    process.env.KIKU_YT_COOKIES = userConfig.KIKU_YT_COOKIES;
+  }
+
+
   try {
     if (!fs.existsSync(ENV_PATH)) return;
     const text = fs.readFileSync(ENV_PATH, 'utf8');
@@ -73,7 +122,7 @@ function loadEnv() {
       const m = line.match(/^\s*([\w.-]+)\s*=\s*(.*)\s*$/);
       if (!m) continue;
       const key = m[1];
-      let val = m[2].replace(/^["']|["']$/g, '');
+      const val = m[2].replace(/^["']|["']$/g, '');
       if (process.env[key] === undefined) process.env[key] = val;
     }
   } catch {}
@@ -92,24 +141,32 @@ function readEnvMap() {
   return map;
 }
 
-
 function writeEnv(updates) {
-  const map = readEnvMap();
-  Object.assign(map, updates);
-  const out = Object.entries(map).map(([k, v]) => `${k}=${v}`).join('\n') + '\n';
-  fs.writeFileSync(ENV_PATH, out, 'utf8');
-}
 
+  for (const [k, v] of Object.entries(updates)) {
+    process.env[k] = v;
+  }
+  writeAuthConfig(updates);
+
+ 
+  try {
+    const map = readEnvMap();
+    Object.assign(map, updates);
+    const out = Object.entries(map).map(([k, v]) => `${k}=${v}`).join('\n') + '\n';
+    fs.writeFileSync(ENV_PATH, out, 'utf8');
+  } catch {}
+}
 
 function hasSession(browser) {
   return new Promise((resolve) => {
-    const timeoutMs = 12000;
+    const timeoutMs = 8000;
     const args = [
       '--cookies-from-browser', browser,
       '--print', '%(title)s',
       '--no-warnings',
+      '--no-playlist',
       '--no-check-certificates',
-      'ytsearch1:Ju Jingyi'
+      'https://www.youtube.com/watch?v=dQw4w9WgXcQ'
     ];
     const p = spawn(YTDLP_CMD, args, {
       windowsHide: true,
@@ -128,51 +185,71 @@ function hasSession(browser) {
       }
     }, timeoutMs);
 
-    p.stdout.on('data', d => { out += d.toString(); });
-    p.stderr.on('data', d => { err += d.toString(); });
-    p.on('error', () => { if (!resolved) { resolved = true; clearTimeout(timer); resolve(false); } });
+    p.stdout?.on('data', d => { out += d.toString(); });
+    p.stderr?.on('data', d => { err += d.toString(); });
+    p.on('error', () => {
+      if (!resolved) {
+        resolved = true;
+        clearTimeout(timer);
+        resolve(false);
+      }
+    });
     p.on('close', (code) => {
       if (resolved) return;
       resolved = true;
       clearTimeout(timer);
       const title = out.trim();
-      const blocked = /403|Forbidden|Sign in|confirm you'?re not a bot|Login Required|Private Video/i.test(err);
+      const blocked = /403|Forbidden|Sign in|confirm you'?re not a bot|Login Required|Private Video|database is locked/i.test(err);
       resolve(code === 0 && title.length > 0 && !blocked);
     });
   });
 }
 
-
-
 let _ensured = null;
 let _lastFailTime = 0;
-const RETRY_COOLDOWN_MS = 30000; 
-
+let _inFlightPromise = null;
+const RETRY_COOLDOWN_MS = 30000;
 
 export async function ensureYouTubeCookies(opts = {}) {
- 
   if (_ensured && _ensured.ok) return _ensured;
   if (_ensured && !_ensured.ok && (Date.now() - _lastFailTime) < RETRY_COOLDOWN_MS && !opts.force) {
     return _ensured;
   }
 
+ 
+  if (_inFlightPromise) {
+    return _inFlightPromise;
+  }
+
+  _inFlightPromise = (async () => {
+    try {
+      return await _ensureYouTubeCookiesImpl(opts);
+    } finally {
+      _inFlightPromise = null;
+    }
+  })();
+
+  return _inFlightPromise;
+}
+
+async function _ensureYouTubeCookiesImpl(opts = {}) {
   loadEnv();
 
-  
-  if (process.env.KIKU_YT_COOKIES && fs.existsSync(process.env.KIKU_YT_COOKIES)) {
-    _ensured = { ok: true, source: 'file', cookiesPath: process.env.KIKU_YT_COOKIES };
+ 
+  const cookiesCandidate = process.env.KIKU_YT_COOKIES || (fs.existsSync(SECURE_COOKIE_PATH) ? SECURE_COOKIE_PATH : null);
+  if (cookiesCandidate && fs.existsSync(cookiesCandidate)) {
+    _ensured = { ok: true, source: 'file', cookiesPath: cookiesCandidate };
     return _ensured;
   }
 
  
   const available = detectBrowsers();
   if (!available.length) {
-    _ensured = { ok: false, reason: 'no-browser', detail: 'No supported browser found (Chrome, Edge, Brave, Firefox)' };
+    _ensured = { ok: false, reason: 'no-browser', detail: 'No supported browser found (Firefox, Chrome, Edge, Brave)' };
     _lastFailTime = Date.now();
     return _ensured;
   }
 
- 
   const preferred = process.env.KIKU_YT_BROWSER;
   const candidates = preferred && available.includes(preferred)
     ? [preferred, ...available.filter(b => b !== preferred)]
@@ -180,9 +257,7 @@ export async function ensureYouTubeCookies(opts = {}) {
 
   for (const b of candidates) {
     if (await hasSession(b)) {
-    
       writeEnv({ KIKU_YT_BROWSER: b });
-      process.env.KIKU_YT_BROWSER = b;
       _ensured = { ok: true, source: 'existing-session', browser: b };
       return _ensured;
     }
@@ -195,7 +270,7 @@ export async function ensureYouTubeCookies(opts = {}) {
     return _ensured;
   }
 
-  
+
   const loginBrowser = candidates[0];
   const loginUrl = 'https://accounts.google.com/ServiceLogin?service=youtube&continue=https://www.youtube.com/';
   openBrowser(loginUrl, loginBrowser);
@@ -208,21 +283,24 @@ export async function ensureYouTubeCookies(opts = {}) {
   await new Promise(r => setTimeout(r, 2000));
 
   const deadline = Date.now() + (opts.timeout || 120000);
-  const pollInterval = 3000;
+  const pollInterval = 4000;
   let attempt = 0;
 
   while (Date.now() < deadline) {
     attempt++;
     const remaining = Math.round((deadline - Date.now()) / 1000);
-    process.stdout.write(`\r     Checking... (attempt ${attempt}, ${remaining}s remaining) `);
+    if (!opts.silent && process.stdout.isTTY) {
+      process.stdout.write(`\r     Checking session... (attempt ${attempt}, ${remaining}s remaining) `);
+    }
 
     for (const b of available) {
       if (await hasSession(b)) {
         writeEnv({ KIKU_YT_BROWSER: b });
-        process.env.KIKU_YT_BROWSER = b;
-        process.stdout.write('\r' + ' '.repeat(60) + '\r');
+        if (process.stdout.isTTY) {
+          process.stdout.write('\r' + ' '.repeat(65) + '\r');
+        }
         console.log(`  [OK] YouTube session detected in ${b}.`);
-        console.log(`     Saved KIKU_YT_BROWSER=${b} to .env\n`);
+        console.log(`     Saved preference (KIKU_YT_BROWSER=${b})\n`);
         _ensured = { ok: true, source: 'new-session', browser: b };
         return _ensured;
       }
@@ -231,14 +309,15 @@ export async function ensureYouTubeCookies(opts = {}) {
     await new Promise(r => setTimeout(r, pollInterval));
   }
 
-  process.stdout.write('\r' + ' '.repeat(60) + '\r');
+  if (process.stdout.isTTY) {
+    process.stdout.write('\r' + ' '.repeat(65) + '\r');
+  }
   console.log(`  [ERROR] Login timed out. You can retry by running the /login command.\n`);
   _ensured = { ok: false, reason: 'login-timeout' };
   _lastFailTime = Date.now();
   return _ensured;
 }
 
-
 export async function exportCookiesToFile() {
   loadEnv();
 
@@ -249,22 +328,22 @@ export async function exportCookiesToFile() {
     : available;
 
   for (const b of candidates) {
-    const cookiesPath = path.resolve(__dirname, '..', '..', '.kiku_yt_cookies.txt');
     try {
+      fs.mkdirSync(KIKU_DATA_DIR, { recursive: true });
       const args = [
         '--cookies-from-browser', b,
-        '--cookies', cookiesPath,
+        '--cookies', SECURE_COOKIE_PATH,
         '--no-warnings',
         '--skip-download',
-        'ytsearch1:__export_test__'
+        'https://www.youtube.com/watch?v=dQw4w9WgXcQ'
       ];
       await new Promise((resolve, reject) => {
         const p = spawn(YTDLP_CMD, args, { windowsHide: true, stdio: ['ignore', 'ignore', 'pipe'] });
         let err = '';
-        p.stderr.on('data', d => { err += d.toString(); });
+        p.stderr?.on('data', d => { err += d.toString(); });
         p.on('error', reject);
         p.on('close', code => {
-          if (code === 0 && fs.existsSync(cookiesPath)) {
+          if (code === 0 && fs.existsSync(SECURE_COOKIE_PATH)) {
             resolve();
           } else {
             reject(new Error(err || `yt-dlp exited ${code}`));
@@ -272,9 +351,8 @@ export async function exportCookiesToFile() {
         });
       });
 
-      writeEnv({ KIKU_YT_COOKIES: cookiesPath });
-      process.env.KIKU_YT_COOKIES = cookiesPath;
-      return { ok: true, browser: b, cookiesPath };
+      writeEnv({ KIKU_YT_COOKIES: SECURE_COOKIE_PATH });
+      return { ok: true, browser: b, cookiesPath: SECURE_COOKIE_PATH };
     } catch {}
   }
 
@@ -284,16 +362,17 @@ export async function exportCookiesToFile() {
 export function resetCookieCheck() {
   _ensured = null;
   _lastFailTime = 0;
+  _inFlightPromise = null;
 }
 
 export function getCookieStatus() {
   return {
     ensured: _ensured,
     envBrowser: process.env.KIKU_YT_BROWSER || null,
-    envCookies: process.env.KIKU_YT_COOKIES || null,
-    envPath: ENV_PATH
+    envCookies: process.env.KIKU_YT_COOKIES || (fs.existsSync(SECURE_COOKIE_PATH) ? SECURE_COOKIE_PATH : null),
+    envPath: ENV_PATH,
+    authConfigPath: AUTH_CONFIG_PATH
   };
 }
 
-
 loadEnv();