@chongqingliangyunzhijing/ssquant-cli 0.0.0-stage → 0.2.0

package/src/http.js

@@ -0,0 +1,217 @@
+/**
+ * HTTP 客户端。
+ *
+ * 三件事必须做对：
+ *  ① access 过期自动用 refresh 换，选手无感
+ *  ② 服务端错误信封原样透出 code / message / suggestion —— Agent 靠 suggestion 自我修正
+ *  ③ 限流按 retryAfter 等待，**不立即重试**
+ */
+import {
+  apiUrl, clientId, loadCredentials, saveCredentials, withCredentialLock,
+} from './config.js';
+import { EXIT_AUTH, EXIT_NET } from './output.js';
+import { CLIENT_VERSION } from './version.js';
+
+export class ApiError extends Error {
+  constructor(code, message, { suggestion, requestId, data, exitCode } = {}) {
+    super(message);
+    this.code = code;
+    this.suggestion = suggestion;
+    this.requestId = requestId;
+    this.data = data || {};
+    this.exitCode = exitCode;
+  }
+}
+
+export class AuthError extends ApiError {
+  constructor(message = '未登录或凭证已失效，请执行 ssquant login') {
+    super('login_required', message, { exitCode: EXIT_AUTH });
+  }
+}
+
+function creds() {
+  const c = loadCredentials();
+  if (!c || !c.accessToken) throw new AuthError();
+  return c;
+}
+
+async function rawFetch(url, {
+  method = 'GET', body, token, form = false, timeout = 30000,
+} = {}) {
+  const ctrl = new AbortController();
+  const timer = setTimeout(() => ctrl.abort(), timeout);
+  let res;
+  try {
+    const headers = {
+      Accept: 'application/json',
+      'X-Client-Source': 'cli',
+      'X-Client-Version': CLIENT_VERSION,
+      ...(token ? { Authorization: `Bearer ${token}` } : {}),
+    };
+    let requestBody;
+    if (body !== undefined) {
+      if (form) {
+        headers['Content-Type'] = 'application/x-www-form-urlencoded';
+        requestBody = new URLSearchParams(body).toString();
+      } else {
+        headers['Content-Type'] = 'application/json';
+        requestBody = JSON.stringify(body);
+      }
+    }
+    res = await fetch(url, {
+      method,
+      headers,
+      body: requestBody,
+      signal: ctrl.signal,
+    });
+  } catch (e) {
+    if (e.name === 'AbortError') {
+      throw new ApiError('timeout', `请求超时（${timeout / 1000}s）：${url}`, {
+        suggestion: '检查网络，或稍后重试', exitCode: EXIT_NET,
+      });
+    }
+    throw new ApiError('network_error', `无法连接服务：${e.message}`, {
+      suggestion: '检查网络与 SSQUANT_API_BASE 是否正确', exitCode: EXIT_NET,
+    });
+  } finally {
+    clearTimeout(timer);
+  }
+
+  const text = await res.text();
+  let json = null;
+  try {
+    json = text ? JSON.parse(text) : null;
+  } catch {
+    throw new ApiError('bad_response',
+      `服务返回了非 JSON 内容（HTTP ${res.status}）：${text.slice(0, 200)}`,
+      { exitCode: EXIT_NET });
+  }
+  return { res, json };
+}
+
+/** 服务端信封 → 结果或抛错 */
+function unwrap({ res, json }) {
+  if (res.ok && json && json.code === '0') {
+    return { data: json.data, meta: json.meta, requestId: json.request_id };
+  }
+  const code = (json && json.code) || `http_${res.status}`;
+  const message = (json && json.message) || `HTTP ${res.status}`;
+  const data = (json && json.data) || {};
+  const exitCode = res.status === 401 ? EXIT_AUTH
+    : res.status >= 500 ? EXIT_NET : undefined;
+  throw new ApiError(code, message, {
+    suggestion: data.suggestion,
+    requestId: json && json.request_id,
+    data,
+    exitCode,
+  });
+}
+
+let refreshPromise = null;
+
+async function refreshTokenLocked(c) {
+  const body = {
+    grant_type: 'refresh_token',
+    refresh_token: c.refreshToken,
+    client_id: c.clientId || clientId(),
+  };
+  let out = await rawFetch(apiUrl('/oauth/token'), {
+    method: 'POST', form: true, body,
+  });
+  if (out.res.status === 409 && out.json?.code === 'refresh_in_progress') {
+    await new Promise((resolve) => setTimeout(resolve, 250));
+    out = await rawFetch(apiUrl('/oauth/token'), {
+      method: 'POST', form: true, body,
+    });
+  }
+  const { res, json } = out;
+  if (!res.ok || !json || json.code !== '0') {
+    if (json?.code === 'refresh_in_progress') {
+      throw new ApiError(json.code, json.message || '登录凭证正在刷新，请稍后重试', {
+        suggestion: json.data?.suggestion,
+        requestId: json.request_id,
+        data: json.data,
+      });
+    }
+    throw new AuthError('登录已过期，请重新执行 ssquant login');
+  }
+  const d = json.data;
+  const updated = {
+    ...c,
+    accessToken: d.access_token,
+    refreshToken: d.refresh_token || c.refreshToken,
+    expiresAt: Date.now() + (d.expires_in || 2592000) * 1000,
+    accountId: d.account_id || c.accountId,
+  };
+  saveCredentials(updated);
+  return updated;
+}
+
+async function refreshToken(c) {
+  if (refreshPromise) return refreshPromise;
+  refreshPromise = withCredentialLock(async () => {
+    // 另一个进程可能已经完成轮转，拿锁后必须读取最新凭据。
+    const latest = loadCredentials();
+    if (latest && latest.accountId === c.accountId && latest.accessToken && latest.refreshToken
+      && latest.refreshToken !== c.refreshToken
+      && (!latest.expiresAt || latest.expiresAt - Date.now() >= 60_000)) {
+      return latest;
+    }
+    if (latest && latest.accountId !== c.accountId) throw new AuthError('当前账户已切换，请重新发起操作');
+    return refreshTokenLocked(latest || c);
+  }).finally(() => {
+    refreshPromise = null;
+  });
+  return refreshPromise;
+}
+
+/**
+ * 带鉴权的请求。access 快过期（剩余 <60s）时先刷新，
+ * 拿到 401 再刷一次并重试一次 —— 服务端可能提前吊销。
+ */
+export async function request(path, { method = 'GET', body, retry = true } = {}) {
+  let c = creds();
+  if (c.expiresAt && c.expiresAt - Date.now() < 60_000 && c.refreshToken) {
+    c = await refreshToken(c);
+  }
+  const out = await rawFetch(apiUrl(path), { method, body, token: c.accessToken });
+  if (out.res.status === 401 && retry && c.refreshToken) {
+    c = await refreshToken(c);
+    const again = await rawFetch(apiUrl(path), { method, body, token: c.accessToken });
+    return unwrap(again);
+  }
+  return unwrap(out);
+}
+
+/** 不带鉴权（OAuth 授权码 / token 端点） */
+export async function requestAnon(path, { method = 'POST', body, form = false } = {}) {
+  return unwrap(await rawFetch(apiUrl(path), { method, body, form }));
+}
+
+/** 下载公开文本资源（skill 文档等），地址由调用方决定。 */
+export async function fetchText(url) {
+  const ctrl = new AbortController();
+  const timer = setTimeout(() => ctrl.abort(), 30000);
+  try {
+    const res = await fetch(url, { signal: ctrl.signal });
+    if (!res.ok) {
+      throw new ApiError('download_failed', `下载失败 HTTP ${res.status}：${url}`,
+        { exitCode: EXIT_NET });
+    }
+    return await res.text();
+  } catch (e) {
+    if (e instanceof ApiError) throw e;
+    throw new ApiError('network_error', `下载失败：${e.message}`, { exitCode: EXIT_NET });
+  } finally {
+    clearTimeout(timer);
+  }
+}
+
+export function qs(params) {
+  const p = new URLSearchParams();
+  for (const [k, v] of Object.entries(params || {})) {
+    if (v !== undefined && v !== null && v !== '') p.append(k, String(v));
+  }
+  const s = p.toString();
+  return s ? `?${s}` : '';
+}

package/src/output.js

@@ -0,0 +1,133 @@
+/**
+ * 输出与退出码。
+ *
+ * ★ --json 的信封是**对 Agent 的契约**，不能随意改：
+ *     成功 { ok:true,  data, meta }
+ *     失败 { ok:false, error:{ code, message, suggestion, requestId } }
+ *   Agent 只需判断 ok，失败时读 error.suggestion 就知道该怎么办。
+ *
+ * ★ 退出码让 shell 脚本能区分「该重试」和「该停手」：
+ *     0 成功   1 业务错误   2 认证失效（需重新 login）   3 网络/服务不可用
+ */
+
+export const EXIT_OK = 0;
+export const EXIT_BIZ = 1;
+export const EXIT_AUTH = 2;
+export const EXIT_NET = 3;
+
+let jsonMode = false;
+let updateNotice = null;
+
+export function setUpdateNotice(notice) {
+  updateNotice = notice;
+  if (!jsonMode && notice?.available) {
+    process.stderr.write(`更新提示：${notice.message}\n`);
+    if (notice.cli.updateAvailable) process.stderr.write(`  ${notice.cli.command}\n`);
+    for (const item of notice.skill.installations.filter(x => x.updateAvailable)) {
+      process.stderr.write(`  ${item.command}\n`);
+    }
+  }
+}
+
+function withUpdate(meta) {
+  return updateNotice ? { ...meta, update: updateNotice } : meta;
+}
+
+export function setJsonMode(on) {
+  jsonMode = !!on;
+}
+
+export function isJson() {
+  return jsonMode;
+}
+
+export function emit(data, meta) {
+  if (jsonMode) {
+    const body = { ok: true, data };
+    const combined = withUpdate(meta);
+    if (combined) body.meta = combined;
+    process.stdout.write(JSON.stringify(body) + '\n');
+  }
+  return data;
+}
+
+export function fail(error, exitCode = EXIT_BIZ) {
+  const payload = {
+    code: error.code || 'error',
+    message: error.message || String(error),
+  };
+  if (error.suggestion) payload.suggestion = error.suggestion;
+  if (error.requestId) payload.requestId = error.requestId;
+  if (error.data && Object.keys(error.data).length) payload.detail = error.data;
+
+  if (jsonMode) {
+    const body = { ok: false, error: payload };
+    if (updateNotice) body.meta = withUpdate();
+    process.stdout.write(JSON.stringify(body) + '\n');
+  } else {
+    process.stderr.write(`✗ ${payload.message}\n`);
+    if (payload.suggestion) process.stderr.write(`  建议：${payload.suggestion}\n`);
+    if (payload.requestId) process.stderr.write(`  请求号：${payload.requestId}\n`);
+  }
+  process.exitCode = exitCode;
+}
+
+/** 人读模式下打印一行行的 key: value */
+export function kv(pairs) {
+  if (jsonMode) return;
+  const width = Math.max(...pairs.map(([k]) => strWidth(k)));
+  for (const [k, v] of pairs) {
+    const pad = ' '.repeat(Math.max(0, width - strWidth(k)));
+    process.stdout.write(`  ${k}${pad}  ${v}\n`);
+  }
+}
+
+/** 人读模式下打印表格。中文按 2 列宽计算，否则对不齐 */
+export function table(headers, rows) {
+  if (jsonMode) return;
+  if (!rows.length) {
+    process.stdout.write('  （无数据）\n');
+    return;
+  }
+  const all = [headers, ...rows];
+  const widths = headers.map((_, i) =>
+    Math.max(...all.map((r) => strWidth(String(r[i] ?? '')))));
+  const line = (r) =>
+    '  ' + r.map((c, i) => {
+      const s = String(c ?? '');
+      return s + ' '.repeat(Math.max(0, widths[i] - strWidth(s)));
+    }).join('  ');
+  process.stdout.write(line(headers) + '\n');
+  process.stdout.write('  ' + widths.map((w) => '─'.repeat(w)).join('  ') + '\n');
+  for (const r of rows) process.stdout.write(line(r) + '\n');
+}
+
+export function say(msg) {
+  if (!jsonMode) process.stdout.write(msg + '\n');
+}
+
+/** 全角字符占 2 列 */
+function strWidth(s) {
+  let w = 0;
+  for (const ch of String(s)) {
+    w += /[ᄀ-ᅟ⺀-꓏가-힣豈-﫿︰-﹏＀-｠￠-￦]/.test(ch) ? 2 : 1;
+  }
+  return w;
+}
+
+export function num(v, digits = 2) {
+  if (v === null || v === undefined || v === '') return '-';
+  const n = Number(v);
+  if (Number.isNaN(n)) return String(v);
+  return n.toLocaleString('zh-CN', {
+    minimumFractionDigits: digits,
+    maximumFractionDigits: digits,
+  });
+}
+
+export function pct(v, digits = 2) {
+  if (v === null || v === undefined || v === '') return '-';
+  const n = Number(v);
+  if (Number.isNaN(n)) return String(v);
+  return `${(n * 100).toFixed(digits)}%`;
+}

package/src/version.js

@@ -0,0 +1,7 @@
+import { createRequire } from 'node:module';
+
+const require = createRequire(import.meta.url);
+const packageJson = require('../package.json');
+
+export const PACKAGE_NAME = '@chongqingliangyunzhijing/ssquant-cli';
+export const CLIENT_VERSION = packageJson.version;