@fivetu53/soul-chat 1.0.0

package/bin/api.js

@@ -0,0 +1,262 @@
+import { loadAuth, saveAuth, clearAuth, loadConfig } from './config.js';
+
+// Backend URL (优先级: 环境变量 > 配置文件 > 默认值)
+const config = loadConfig();
+const BASE_URL = process.env.SOUL_API_URL || config.apiUrl;
+
+// HTTP request with auth
+async function request(path, options = {}) {
+  const auth = loadAuth();
+  const headers = {
+    'Content-Type': 'application/json',
+    ...options.headers
+  };
+
+  if (auth?.accessToken) {
+    headers['Authorization'] = `Bearer ${auth.accessToken}`;
+  }
+
+  const response = await fetch(`${BASE_URL}${path}`, {
+    ...options,
+    headers
+  });
+
+  // Handle 401 - try refresh token
+  if (response.status === 401 && auth?.refreshToken) {
+    const refreshed = await refreshToken();
+    if (refreshed) {
+      // Retry with new token
+      const newAuth = loadAuth();
+      headers['Authorization'] = `Bearer ${newAuth.accessToken}`;
+      return fetch(`${BASE_URL}${path}`, { ...options, headers });
+    }
+  }
+
+  return response;
+}
+
+// Refresh access token
+async function refreshToken() {
+  const auth = loadAuth();
+  if (!auth?.refreshToken) return false;
+
+  try {
+    const response = await fetch(`${BASE_URL}/api/auth/refresh`, {
+      method: 'POST',
+      headers: { 'Content-Type': 'application/json' },
+      body: JSON.stringify({ refreshToken: auth.refreshToken })
+    });
+
+    if (response.ok) {
+      const data = await response.json();
+      saveAuth({
+        ...auth,
+        accessToken: data.accessToken
+      });
+      return true;
+    }
+  } catch (err) {
+    console.error('Refresh token failed:', err.message);
+  }
+
+  clearAuth();
+  return false;
+}
+
+// Auth APIs
+export async function sendVerificationCode(email, type = 'register') {
+  const response = await fetch(`${BASE_URL}/api/auth/send-code`, {
+    method: 'POST',
+    headers: { 'Content-Type': 'application/json' },
+    body: JSON.stringify({ email, type })
+  });
+
+  const data = await response.json();
+  if (!response.ok) {
+    throw new Error(data.error || '发送验证码失败');
+  }
+
+  return data;
+}
+
+export async function register(username, email, password, code) {
+  const response = await fetch(`${BASE_URL}/api/auth/register`, {
+    method: 'POST',
+    headers: { 'Content-Type': 'application/json' },
+    body: JSON.stringify({ username, email, password, code })
+  });
+
+  const data = await response.json();
+  if (!response.ok) {
+    throw new Error(data.error || 'Registration failed');
+  }
+
+  saveAuth({
+    accessToken: data.accessToken,
+    refreshToken: data.refreshToken,
+    user: data.user
+  });
+
+  return data;
+}
+
+export async function login(email, password) {
+  const response = await fetch(`${BASE_URL}/api/auth/login`, {
+    method: 'POST',
+    headers: { 'Content-Type': 'application/json' },
+    body: JSON.stringify({ email, password })
+  });
+
+  const data = await response.json();
+  if (!response.ok) {
+    throw new Error(data.error || 'Login failed');
+  }
+
+  saveAuth({
+    accessToken: data.accessToken,
+    refreshToken: data.refreshToken,
+    user: data.user
+  });
+
+  return data;
+}
+
+export async function logout() {
+  try {
+    await request('/api/auth/logout', { method: 'POST' });
+  } catch (err) {
+    // Ignore errors
+  }
+  clearAuth();
+}
+
+export async function updateProfile(username) {
+  const response = await request('/api/auth/profile', {
+    method: 'PUT',
+    body: JSON.stringify({ username })
+  });
+
+  const data = await response.json();
+  if (!response.ok) throw new Error(data.error);
+
+  // Update local auth with new tokens
+  const auth = loadAuth();
+  saveAuth({
+    ...auth,
+    accessToken: data.accessToken,
+    refreshToken: data.refreshToken,
+    user: data.user
+  });
+
+  return data;
+}
+
+export async function verifyToken() {
+  const auth = loadAuth();
+  if (!auth?.accessToken) return false;
+
+  try {
+    const response = await request('/api/auth/verify');
+    return response.ok;
+  } catch (err) {
+    return false;
+  }
+}
+
+// Character APIs
+export async function getCharacters() {
+  const response = await request('/api/characters');
+  const data = await response.json();
+  if (!response.ok) throw new Error(data.error);
+  return data.characters;
+}
+
+export async function createCharacter(character) {
+  const response = await request('/api/characters', {
+    method: 'POST',
+    body: JSON.stringify(character)
+  });
+  const data = await response.json();
+  if (!response.ok) throw new Error(data.error);
+  return data.character;
+}
+
+// Chat APIs
+export async function sendMessage(message, characterId, conversationId, image = null) {
+  const response = await request('/api/chat', {
+    method: 'POST',
+    body: JSON.stringify({ message, characterId, conversationId, image })
+  });
+
+  if (!response.ok) {
+    const data = await response.json();
+    throw new Error(data.error);
+  }
+
+  return response;
+}
+
+export async function getConversations() {
+  const response = await request('/api/chat/conversations');
+  const data = await response.json();
+  if (!response.ok) throw new Error(data.error);
+  return data.conversations;
+}
+
+export async function getMessages(conversationId) {
+  const response = await request(`/api/chat/conversations/${conversationId}/messages`);
+  const data = await response.json();
+  if (!response.ok) throw new Error(data.error);
+  return data.messages;
+}
+
+export async function clearConversation(conversationId) {
+  const response = await request(`/api/chat/conversations/${conversationId}`, {
+    method: 'DELETE'
+  });
+  const data = await response.json();
+  if (!response.ok) throw new Error(data.error);
+  return data;
+}
+
+// Memory APIs
+export async function getMemories(characterId) {
+  const response = await request(`/api/memories/${characterId}`);
+  const data = await response.json();
+  if (!response.ok) throw new Error(data.error);
+  return data.memories;
+}
+
+export async function searchMemories(characterId, query) {
+  const response = await request(`/api/memories/${characterId}/search?q=${encodeURIComponent(query)}`);
+  const data = await response.json();
+  if (!response.ok) throw new Error(data.error);
+  return data.memories;
+}
+
+export async function pinMemory(memoryId, pinned = true) {
+  const response = await request(`/api/memories/${memoryId}/pin`, {
+    method: 'POST',
+    body: JSON.stringify({ pinned })
+  });
+  return response.ok;
+}
+
+export async function deleteMemory(memoryId) {
+  const response = await request(`/api/memories/${memoryId}`, {
+    method: 'DELETE'
+  });
+  return response.ok;
+}
+
+// Check if logged in
+export function isLoggedIn() {
+  const auth = loadAuth();
+  return !!auth?.accessToken;
+}
+
+// Get current user
+export function getCurrentUser() {
+  const auth = loadAuth();
+  return auth?.user || null;
+}

package/bin/auth.js

@@ -0,0 +1,363 @@
+import readline from 'readline';
+import http from 'http';
+import crypto from 'crypto';
+import { login, verifyToken, isLoggedIn } from './api.js';
+import { saveAuth, loadConfig } from './config.js';
+
+// ANSI colors
+const colors = {
+  reset: '\x1b[0m',
+  bold: '\x1b[1m',
+  red: '\x1b[31m',
+  green: '\x1b[32m',
+  yellow: '\x1b[33m',
+  cyan: '\x1b[36m',
+  gray: '\x1b[90m',
+};
+
+const c = (color, text) => `${colors[color]}${text}${colors.reset}`;
+
+const clearScreen = () => process.stdout.write('\x1b[2J\x1b[H');
+
+function prompt(query) {
+  const rl = readline.createInterface({
+    input: process.stdin,
+    output: process.stdout
+  });
+
+  return new Promise((resolve) => {
+    rl.question(query, (answer) => {
+      rl.close();
+      resolve(answer);
+    });
+  });
+}
+
+function promptPassword(query) {
+  return new Promise((resolve) => {
+    const stdin = process.stdin;
+
+    process.stdout.write(query);
+
+    let password = '';
+    stdin.setRawMode(true);
+    stdin.resume();
+
+    const onData = (chunk) => {
+      const char = chunk.toString();
+      if (char === '\n' || char === '\r' || char === '\u0004') {
+        stdin.removeListener('data', onData);
+        stdin.setRawMode(false);
+        console.log();
+        resolve(password);
+      } else if (char === '\u0003') {
+        process.exit();
+      } else if (char === '\u007F' || char === '\b') {
+        if (password.length > 0) {
+          password = password.slice(0, -1);
+          process.stdout.write('\b \b');
+        }
+      } else if (char.charCodeAt(0) >= 32) {
+        password += char;
+        process.stdout.write('*');
+      }
+    };
+
+    stdin.on('data', onData);
+  });
+}
+
+// Check if already logged in
+export async function checkAuth() {
+  if (!isLoggedIn()) {
+    return false;
+  }
+
+  // Verify token is still valid
+  const valid = await verifyToken();
+  return valid;
+}
+
+// 认证界面 - 登录或注册
+export async function showAuthScreen() {
+  clearScreen();
+  console.log();
+  console.log(c('cyan', c('bold', '  +-----------------------------+')));
+  console.log(c('cyan', c('bold', '  |       Soul Chat 登录        |')));
+  console.log(c('cyan', c('bold', '  +-----------------------------+')));
+  console.log();
+
+  const menuItems = [
+    { label: '[1] 浏览器登录 (推荐)', value: 'browser' },
+    { label: '[2] 邮箱密码登录', value: 'login' },
+    { label: '[3] 注册新账号', value: 'register' },
+    { label: '[4] 退出', value: 'exit' },
+  ];
+
+  let selected = 0;
+
+  const draw = () => {
+    process.stdout.write('\x1b[6;0H'); // Move to line 6
+    menuItems.forEach((item, i) => {
+      if (i === selected) {
+        console.log(c('cyan', `  > ${item.label}  `));
+      } else {
+        console.log(`    ${item.label}  `);
+      }
+    });
+    console.log();
+    console.log(c('gray', '  方向键选择，Enter 确认'));
+    console.log(c('yellow', '  门户网站: https://soul-chat.jdctools.com.cn'));
+  };
+
+  draw();
+
+  const choice = await new Promise((resolve) => {
+    process.stdin.setRawMode(true);
+    process.stdin.resume();
+
+    const onKeyPress = (key) => {
+      if (key[0] === 27 && key[1] === 91) {
+        if (key[2] === 65 && selected > 0) {
+          selected--;
+          draw();
+        } else if (key[2] === 66 && selected < menuItems.length - 1) {
+          selected++;
+          draw();
+        }
+      } else if (key[0] === 13) {
+        process.stdin.setRawMode(false);
+        process.stdin.removeListener('data', onKeyPress);
+        resolve(menuItems[selected].value);
+      } else if (key[0] === 3) {
+        process.exit();
+      }
+    };
+
+    process.stdin.on('data', onKeyPress);
+  });
+
+  if (choice === 'exit') {
+    console.log();
+    console.log(c('gray', '  再见!'));
+    process.exit(0);
+  }
+
+  if (choice === 'browser') {
+    return await browserAuth();
+  } else if (choice === 'login') {
+    return await showLoginForm();
+  } else {
+    return await showRegisterForm();
+  }
+}
+
+// 浏览器认证
+async function browserAuth() {
+  clearScreen();
+  console.log();
+  console.log(c('cyan', c('bold', '  [*] 浏览器登录')));
+  console.log();
+
+  // 生成随机 state
+  const state = crypto.randomUUID();
+
+  // 查找可用端口
+  const port = await findAvailablePort(9876);
+  if (!port) {
+    console.log(c('red', '  无法启动本地服务器，请使用邮箱密码登录'));
+    await sleep(2000);
+    return await showAuthScreen();
+  }
+
+  console.log(c('gray', `  正在启动本地认证服务器 (端口 ${port})...`));
+
+  // 创建 Promise 来等待认证结果
+  const authResult = await new Promise((resolve) => {
+    let resolved = false;
+
+    // 创建本地 HTTP 服务器
+    const server = http.createServer((req, res) => {
+      const url = new URL(req.url, `http://localhost:${port}`);
+
+      if (url.pathname === '/callback') {
+        const receivedState = url.searchParams.get('state');
+        const accessToken = url.searchParams.get('accessToken');
+        const refreshToken = url.searchParams.get('refreshToken');
+        const userJson = url.searchParams.get('user');
+
+        // 验证 state
+        if (receivedState !== state) {
+          res.writeHead(400, { 'Content-Type': 'text/html; charset=utf-8' });
+          res.end('<html><body style="font-family:sans-serif;text-align:center;padding:50px;"><h1>认证失败</h1><p>无效的 state 参数</p></body></html>');
+          return;
+        }
+
+        if (!accessToken || !refreshToken || !userJson) {
+          res.writeHead(400, { 'Content-Type': 'text/html; charset=utf-8' });
+          res.end('<html><body style="font-family:sans-serif;text-align:center;padding:50px;"><h1>认证失败</h1><p>缺少必要参数</p></body></html>');
+          return;
+        }
+
+        try {
+          const user = JSON.parse(userJson);
+
+          // 保存认证信息
+          saveAuth({ accessToken, refreshToken, user });
+
+          // 返回成功页面
+          res.writeHead(200, { 'Content-Type': 'text/html; charset=utf-8' });
+          res.end(`
+            <html>
+            <body style="font-family:sans-serif;text-align:center;padding:50px;background:#1a1a2e;color:#fff;">
+              <div style="max-width:400px;margin:0 auto;">
+                <div style="font-size:60px;margin-bottom:20px;">✓</div>
+                <h1 style="color:#f5c842;">认证成功!</h1>
+                <p style="color:#888;">欢迎回来, ${user.username}</p>
+                <p style="color:#666;margin-top:30px;">你可以关闭此页面，返回 CLI 继续使用</p>
+              </div>
+            </body>
+            </html>
+          `);
+
+          resolved = true;
+          resolve({ success: true, user });
+
+          // 延迟关闭服务器
+          setTimeout(() => server.close(), 500);
+        } catch (err) {
+          res.writeHead(400, { 'Content-Type': 'text/html; charset=utf-8' });
+          res.end('<html><body style="font-family:sans-serif;text-align:center;padding:50px;"><h1>认证失败</h1><p>解析用户信息失败</p></body></html>');
+        }
+      } else {
+        res.writeHead(404);
+        res.end('Not Found');
+      }
+    });
+
+    server.listen(port, '127.0.0.1', () => {
+      // 构建认证 URL
+      const config = loadConfig();
+      const baseUrl = config.apiUrl || 'https://soul-chat.jdctools.com.cn';
+      const authUrl = `${baseUrl}/cli-auth?port=${port}&state=${state}`;
+
+      console.log(c('yellow', '  正在打开浏览器...'));
+      console.log();
+      console.log(c('gray', `  如果浏览器没有自动打开，请手动访问:`));
+      console.log(c('cyan', `  ${authUrl}`));
+      console.log();
+      console.log(c('gray', '  等待浏览器认证... (按 Ctrl+C 取消)'));
+
+      // 打开浏览器
+      import('child_process').then(({ exec }) => {
+        const cmd = process.platform === 'darwin' ? 'open' :
+                    process.platform === 'win32' ? 'start' : 'xdg-open';
+        exec(`${cmd} "${authUrl}"`);
+      });
+    });
+
+    // 超时处理 (2 分钟)
+    setTimeout(() => {
+      if (!resolved) {
+        server.close();
+        resolve({ success: false, error: '认证超时' });
+      }
+    }, 120000);
+
+    // 处理服务器错误
+    server.on('error', (err) => {
+      if (!resolved) {
+        resolve({ success: false, error: err.message });
+      }
+    });
+  });
+
+  console.log();
+
+  if (authResult.success) {
+    console.log(c('green', `  欢迎回来, ${authResult.user.username}!`));
+    await sleep(1000);
+    return authResult.user;
+  } else {
+    console.log(c('red', `  认证失败: ${authResult.error}`));
+    await sleep(2000);
+    return await showAuthScreen();
+  }
+}
+
+// 查找可用端口
+async function findAvailablePort(startPort) {
+  for (let port = startPort; port < startPort + 10; port++) {
+    const available = await checkPort(port);
+    if (available) return port;
+  }
+  return null;
+}
+
+function checkPort(port) {
+  return new Promise((resolve) => {
+    const server = http.createServer();
+    server.listen(port, '127.0.0.1');
+    server.on('listening', () => {
+      server.close();
+      resolve(true);
+    });
+    server.on('error', () => {
+      resolve(false);
+    });
+  });
+}
+
+// 登录表单
+async function showLoginForm() {
+  clearScreen();
+  console.log();
+  console.log(c('cyan', c('bold', '  [*] 邮箱密码登录')));
+  console.log();
+
+  const email = await prompt(c('gray', '  邮箱: '));
+  const password = await promptPassword(c('gray', '  密码: '));
+
+  if (!email || !password) {
+    console.log();
+    console.log(c('red', '  邮箱和密码不能为空'));
+    await sleep(1500);
+    return await showAuthScreen();
+  }
+
+  console.log();
+  console.log(c('yellow', '  登录中...'));
+
+  try {
+    const result = await login(email, password);
+    console.log(c('green', `  欢迎回来, ${result.user.username}!`));
+    await sleep(1000);
+    return result.user;
+  } catch (err) {
+    console.log(c('red', `  错误: ${err.message}`));
+    await sleep(1500);
+    return await showAuthScreen();
+  }
+}
+
+// 注册 - 打开浏览器
+async function showRegisterForm() {
+  const config = loadConfig();
+  const url = config.apiUrl || 'https://soul-chat.jdctools.com.cn';
+  console.log();
+  console.log(c('yellow', `  正在打开浏览器: ${url}/register`));
+
+  // 根据系统打开浏览器
+  const { exec } = await import('child_process');
+  const cmd = process.platform === 'darwin' ? 'open' :
+              process.platform === 'win32' ? 'start' : 'xdg-open';
+  exec(`${cmd} ${url}/register`);
+
+  console.log(c('green', '  请在浏览器中完成注册，然后返回此处登录'));
+  await sleep(2000);
+  return await showAuthScreen();
+}
+
+function sleep(ms) {
+  return new Promise(resolve => setTimeout(resolve, ms));
+}