@xbongbong/xbbcli 2.1.1 → 2.4.0

package/xbb-config.js

@@ -0,0 +1,146 @@
+import fs from 'node:fs';
+import os from 'node:os';
+import path from 'node:path';
+import { spawnSync } from 'node:child_process';
+
+const CONFIG_DIR = path.join(os.homedir(), '.xbbcli');
+const CONFIG_FILE = path.join(CONFIG_DIR, 'config.env');
+const ENV_FILE = path.join(CONFIG_DIR, 'env.sh');
+
+// config.env 字段名 -> 对应环境变量名（环境变量模式只支持单公司）
+export const ENV_VAR_NAMES = Object.freeze({
+  corpid: 'XBB_CORPID',
+  token: 'XBB_TOKEN',
+  baseurl: 'XBB_BASEURL',
+  userId: 'XBB_USERID',
+  corpName: 'XBB_CORPNAME',
+  userName: 'XBB_USERNAME',
+});
+
+function trim(value) {
+  return String(value ?? '').trim();
+}
+
+function isTruthy(value) {
+  return ['1', 'true', 'yes', 'on'].includes(trim(value).toLowerCase());
+}
+
+// 读取环境中非空的 XBB_* 变量，返回与 config.env 单条配置同构的对象
+export function readEnvConfig() {
+  const config = {};
+  for (const [field, name] of Object.entries(ENV_VAR_NAMES)) {
+    const value = trim(process.env[name]);
+    if (value) config[field] = value;
+  }
+  return config;
+}
+
+export function hasEnvConfig(config = readEnvConfig()) {
+  return Boolean(trim(config.corpid) || trim(config.token));
+}
+
+// XBB_ENV_ONLY=1 时强制只使用环境变量，忽略 config.env
+export function isEnvOnly() {
+  return isTruthy(process.env.XBB_ENV_ONLY);
+}
+
+// 与旧版每个命令内联的 readConfig 行为保持一致：优先返回 enable=true 的公司
+function readFileConfig() {
+  try {
+    const parsed = JSON.parse(fs.readFileSync(CONFIG_FILE, 'utf8'));
+    const companies = Array.isArray(parsed)
+      ? parsed
+      : Array.isArray(parsed?.companies)
+        ? parsed.companies
+        : null;
+    if (companies) {
+      return companies.find((item) => item && item.enable) || {};
+    }
+    return parsed || {};
+  } catch {
+    return {};
+  }
+}
+
+// 配置文件优先：有启用公司就用配置，否则回落到环境变量
+export function readActiveConfig() {
+  const envConfig = readEnvConfig();
+  if (isEnvOnly()) {
+    return envConfig;
+  }
+  const fileConfig = readFileConfig();
+  if (Object.keys(fileConfig).length > 0) {
+    return fileConfig;
+  }
+  return envConfig;
+}
+
+// 当前生效来源是否为环境变量
+export function isEnvActive() {
+  if (isEnvOnly()) {
+    return hasEnvConfig();
+  }
+  return Object.keys(readFileConfig()).length === 0 && hasEnvConfig();
+}
+
+export function getEnvVarNames() {
+  return Object.values(ENV_VAR_NAMES);
+}
+
+// token-set 成功后把凭证写入环境变量：Windows 用 setx 写用户级变量，其他平台写 ~/.xbbcli/env.sh
+export function persistEnvVars(values) {
+  const entries = Object.entries(ENV_VAR_NAMES)
+    .map(([field, name]) => [name, trim(values?.[field])])
+    .filter(([, value]) => value !== '');
+  if (!entries.length) {
+    return { stored: 'none', file: '', failed: [] };
+  }
+  if (process.platform === 'win32') {
+    const failed = [];
+    for (const [name, value] of entries) {
+      const result = spawnSync('setx', [name, value], { encoding: 'utf8', windowsHide: true });
+      if (result.error || result.status !== 0) failed.push(name);
+    }
+    if (failed.length) {
+      throw new Error(`写入用户环境变量失败：${failed.join(', ')}`);
+    }
+    return { stored: 'user', file: '', failed: [] };
+  }
+  const lines = entries.map(([name, value]) => `export ${name}='${value.replaceAll("'", "'\\''")}'`);
+  fs.mkdirSync(CONFIG_DIR, { recursive: true });
+  fs.writeFileSync(ENV_FILE, `${lines.join('\n')}\n`, { encoding: 'utf8', mode: 0o600 });
+  return { stored: 'file', file: ENV_FILE, failed: [] };
+}
+
+// auth-logout --env：清除 XBB_* 环境变量（Windows 删除用户级变量，其他平台清理 env.sh 里的 export 行）
+export function clearEnvVars() {
+  const names = getEnvVarNames();
+  for (const name of names) {
+    delete process.env[name];
+  }
+  if (process.platform === 'win32') {
+    const failed = [];
+    let deleted = 0;
+    for (const name of names) {
+      const result = spawnSync('reg', ['delete', 'HKCU\\Environment', '/f', '/v', name], { encoding: 'utf8', windowsHide: true });
+      // 变量原本就不存在时 reg 返回非 0，这里只把进程无法启动视为失败
+      if (result.error) failed.push(name);
+      else if (result.status === 0) deleted += 1;
+    }
+    if (failed.length) {
+      throw new Error(`清除用户环境变量失败：${failed.join(', ')}`);
+    }
+    return { stored: deleted ? 'user' : 'none', file: '' };
+  }
+  if (!fs.existsSync(ENV_FILE)) {
+    return { stored: 'none', file: ENV_FILE };
+  }
+  const pattern = new RegExp('^\\s*export\\s+(' + names.join('|') + ')(\\s|=)');
+  const kept = fs.readFileSync(ENV_FILE, 'utf8').split(/\r?\n/).filter((line) => line.trim() !== '' && !pattern.test(line));
+  if (!kept.length) {
+    fs.rmSync(ENV_FILE, { force: true });
+    return { stored: 'removed', file: ENV_FILE };
+  }
+  fs.writeFileSync(ENV_FILE, `${kept.join('\n')}\n`, { encoding: 'utf8', mode: 0o600 });
+  return { stored: 'cleared', file: ENV_FILE };
+}

package/xbb-token-store.js

@@ -0,0 +1,389 @@
+import fs from 'node:fs';
+import os from 'node:os';
+import path from 'node:path';
+import { getRegistry } from './xbb-registry.js';
+import { persistEnvVars } from './xbb-config.js';
+
+// token-set 与 auth-login 共用的凭证落盘逻辑：写 config.env、表单缓存、命令映射、部门/员工缓存、环境变量
+export const CONFIG_DIR = path.join(os.homedir(), '.xbbcli');
+export const CONFIG_FILE = path.join(CONFIG_DIR, 'config.env');
+export const FORMLIST_FILE_SUFFIX = '.formlist.json';
+export const COMMAND_MAP_FILE_SUFFIX = '.command-map.md';
+export const DEPARTMENT_USER_FILE_SUFFIX = '.department-user.json';
+export const READ_ALL_PAGE_SIZE = 200;
+export const READ_ALL_MAX_PAGES = 1000;
+
+export function resolveBaseUrl(corpid) {
+  if (corpid.startsWith('ding') || corpid.includes('$$ding')) {
+    return 'https://proapi.xbongbong.com';
+  }
+  return 'https://appapi.xbongbong.com';
+}
+
+export function normalizeArg(value) {
+  return String(value || '').trim();
+}
+
+export function getFormlistFile(corpid) {
+  return path.join(CONFIG_DIR, `${corpid}${FORMLIST_FILE_SUFFIX}`);
+}
+
+export function getCommandMapFile(corpid) {
+  return path.join(CONFIG_DIR, `${corpid}${COMMAND_MAP_FILE_SUFFIX}`);
+}
+
+export function getDepartmentUserFile(corpid) {
+  return path.join(CONFIG_DIR, `${corpid}${DEPARTMENT_USER_FILE_SUFFIX}`);
+}
+
+export async function runCommandJson(name, kwargs) {
+  const command = getRegistry().get(`xbb/${name}`);
+  if (!command) throw new Error(`找不到命令：${name}`);
+  const rows = await command.func(kwargs);
+  if (!Array.isArray(rows)) throw new Error(`${name} 返回结果不是数组`);
+  return rows;
+}
+
+async function getFormlistRows(corpid, saasMark) {
+  const rows = await runCommandJson('form-list', { saasMark: String(saasMark) });
+
+  if (!Array.isArray(rows)) {
+    throw new Error(`form-list 返回结果不是数组：saasMark=${saasMark}`);
+  }
+
+  const errorRow = rows.find((item) => item && item.code);
+  if (errorRow) {
+    throw new Error(`form-list 获取失败（saasMark=${saasMark}）：${errorRow.code} ${errorRow.msg || ''}`.trim());
+  }
+
+  return rows;
+}
+
+// 先按 resetToken=0 查询是否已存在个人 token，返回空串再按 resetToken=1 重新生成
+async function requestPersonalToken(corpid, token, userId, resetToken) {
+  const rows = await runCommandJson('token-generate', {
+    checkUserId: userId,
+    resetToken: String(resetToken),
+    token,
+    corpid,
+  });
+
+  if (!Array.isArray(rows)) {
+    throw new Error('token-generate 返回结果不是数组');
+  }
+
+  const errorRow = rows.find((item) => item && String(item.code ?? '') !== '' && Number(item.code) !== 1);
+  if (errorRow) {
+    throw new Error(`token-generate 获取个人 token 失败：${errorRow.code} ${errorRow.msg || ''}`.trim());
+  }
+
+  return String(rows[0]?.token || '').trim();
+}
+
+export async function getPersonalToken(corpid, token, userId) {
+  const existingToken = await requestPersonalToken(corpid, token, userId, 0);
+
+  if (existingToken) {
+    if (!existingToken.startsWith('user_')) {
+      throw new Error('token-generate 未返回有效的个人 token');
+    }
+    return existingToken;
+  }
+
+  const personalToken = await requestPersonalToken(corpid, token, userId, 1);
+  if (!personalToken.startsWith('user_')) {
+    throw new Error('token-generate 未返回有效的个人 token');
+  }
+
+  return personalToken;
+}
+
+async function writeFormlistFile(corpid) {
+  const customForms = await getFormlistRows(corpid, 2);
+  const systemForms = await getFormlistRows(corpid, 1);
+  const formlistFile = getFormlistFile(corpid);
+  const mergedRows = [...customForms, ...systemForms];
+
+  fs.writeFileSync(formlistFile, JSON.stringify(mergedRows, null, 2) + '\n', 'utf8');
+  return formlistFile;
+}
+
+async function fetchAllRows(name, pageSize) {
+  const rows = [];
+  for (let page = 1; page <= READ_ALL_MAX_PAGES; page += 1) {
+    const pageRows = await runCommandJson(name, { page: String(page), pageSize: String(pageSize) });
+    const errorRow = pageRows.find((item) => item && String(item.code ?? '') !== '');
+    if (errorRow) {
+      if (String(errorRow.code) === 'NO_DATA') {
+        return rows;
+      }
+      throw new Error(`${name} 获取失败（page=${page}）：${errorRow.code} ${errorRow.msg || ''}`.trim());
+    }
+    rows.push(...pageRows);
+  }
+  throw new Error(`${name} 分页超过 ${READ_ALL_MAX_PAGES} 页，未获取到完整数据`);
+}
+
+function parseDepartmentListField(value) {
+  if (Array.isArray(value)) {
+    return value;
+  }
+  const text = String(value ?? '').trim();
+  if (!text) {
+    return [];
+  }
+  try {
+    const parsed = JSON.parse(text);
+    return Array.isArray(parsed) ? parsed : text;
+  } catch {
+    return text;
+  }
+}
+
+function toDepartmentEntry(row) {
+  return {
+    id: row.id ?? '',
+    name: row.name ?? '',
+    parentId: row.parentId ?? '',
+    depIdRouter: row.depIdRouter ?? '',
+  };
+}
+
+function toUserEntry(row) {
+  return {
+    userId: row.userId ?? '',
+    name: row.name ?? '',
+    position: row.position ?? '',
+    jobnumber: row.jobnumber ?? '',
+    departmentList: parseDepartmentListField(row.departmentList),
+  };
+}
+
+async function writeDepartmentUserFile(corpid, userId) {
+  const departmentRows = await fetchAllRows('department-list', READ_ALL_PAGE_SIZE);
+  const userRows = await fetchAllRows('user-list', READ_ALL_PAGE_SIZE);
+  const payload = {
+    corpid,
+    updatedAt: new Date().toISOString(),
+    pageSize: READ_ALL_PAGE_SIZE,
+    departmentCount: departmentRows.length,
+    userCount: userRows.length,
+    departments: departmentRows.map(toDepartmentEntry),
+    users: userRows.map(toUserEntry),
+  };
+  const file = getDepartmentUserFile(corpid);
+  fs.mkdirSync(CONFIG_DIR, { recursive: true });
+  fs.writeFileSync(file, JSON.stringify(payload, null, 2) + '\n', 'utf8');
+  return { file, corpName: getCorpName(departmentRows), userName: getUserName(userRows, userId) };
+}
+
+function getCorpName(departmentRows) {
+  const root = departmentRows.find((row) => Number(row.id) === 1);
+  return root ? String(root.name || '').trim() : '';
+}
+
+function getUserName(userRows, userId) {
+  const target = String(userId || '').trim();
+  if (!target) {
+    return '';
+  }
+  const exact = userRows.find((row) => String(row.userId || '').trim() === target);
+  const matched = exact || userRows.find((row) => String(row.userId || '').trim().toLowerCase() === target.toLowerCase());
+  return matched ? String(matched.name || '').trim() : '';
+}
+
+export function writeCompanyProfile(corpid, profile) {
+  const fields = Object.entries(profile).filter(([, value]) => String(value || '').trim() !== '');
+  if (!fields.length) {
+    return;
+  }
+  const patch = Object.fromEntries(fields);
+  const companies = readCompanies().map((item) => (
+    String(item.corpid || '').trim() === corpid ? { ...item, ...patch } : item
+  ));
+  fs.writeFileSync(CONFIG_FILE, JSON.stringify(companies, null, 2) + '\n', 'utf8');
+}
+
+function normalizeBusinessType(value) {
+  return String(value).trim().replace(/[)）\]】,，;；:：。!?！？]+$/g, '');
+}
+
+function getCommandBusinessType(command) {
+  if (command.businessType !== undefined && command.businessType !== null) {
+    return normalizeBusinessType(command.businessType);
+  }
+  const description = String(command.description || '');
+  const match = description.match(/businessType\s*:\s*(\d+)/i);
+  return match ? normalizeBusinessType(match[1]) : '';
+}
+
+function escapeCell(value) {
+  return String(value ?? '').replace(/\|/g, '\\|').replace(/\r?\n/g, ' ');
+}
+
+function optionDescription(option) {
+  return `--${option.name}: ${option.help || ''}`;
+}
+
+function writeCommandMap(corpid) {
+  const formFile = getFormlistFile(corpid);
+  if (!fs.existsSync(formFile)) {
+    throw new Error(`Form cache not found: ${formFile}`);
+  }
+  const raw = fs.readFileSync(formFile, 'utf8').replace(/^\uFEFF/, '');
+  const data = JSON.parse(raw);
+  const forms = Array.isArray(data) ? data : data.data || [];
+
+  const commands = [...getRegistry().values()]
+    .filter((command) => command.site === 'xbb')
+    .map((command) => ({
+      ...command,
+      command_options: command.args || [],
+    }));
+
+  const lines = [
+    '# XBB 命令映射',
+    '',
+    `来源：xbb --help；表单缓存：${path.basename(formFile)}`,
+    '',
+    '| 命令 | 命令名称 | formId | businessType | 字段说明 |',
+    '| --- | --- | --- | --- | --- |',
+  ];
+
+  for (const command of commands) {
+    const commandBusinessType = getCommandBusinessType(command);
+    const matchedForms = forms.filter((form) => String(form.businessType ?? '').trim() === commandBusinessType);
+    const formIds = [...new Set(matchedForms.map((form) => form.formId).filter(Boolean))];
+    const businessTypes = [...new Set(matchedForms.map((form) => form.businessType).filter(Boolean))];
+    const options = (command.command_options || []).map(optionDescription).join('<br>');
+    lines.push(
+      `| ${escapeCell(command.name)} | ${escapeCell(command.description)} | ${escapeCell(formIds.join(', '))} | ${escapeCell(businessTypes.join(', ') || commandBusinessType)} | ${escapeCell(options)} |`,
+    );
+  }
+
+  const commandMapFile = getCommandMapFile(corpid);
+  fs.mkdirSync(CONFIG_DIR, { recursive: true });
+  fs.writeFileSync(commandMapFile, `${lines.join('\n')}\n`, 'utf8');
+  return commandMapFile;
+}
+
+export function createResult(status, message, corpid, baseurl, userId, formlistFile = '', commandMapFile = '', enable = '', companyCount = '', departmentUserFile = '', corpName = '', userName = '', envStored = '', envFile = '') {
+  return [{
+    status,
+    message,
+    configFile: CONFIG_FILE,
+    corpid,
+    baseurl,
+    userId,
+    enable,
+    companyCount,
+    formlistFile,
+    commandMapFile,
+    departmentUserFile,
+    corpName,
+    userName,
+    envStored,
+    envFile,
+  }];
+}
+
+export function readCompanies() {
+  try {
+    const parsed = JSON.parse(fs.readFileSync(CONFIG_FILE, 'utf8'));
+    return normalizeCompanies(parsed);
+  } catch {
+    return [];
+  }
+}
+
+function normalizeCompanies(parsed) {
+  if (Array.isArray(parsed)) {
+    return parsed.filter((item) => item && typeof item === 'object');
+  }
+  if (parsed && Array.isArray(parsed.companies)) {
+    return parsed.companies.filter((item) => item && typeof item === 'object');
+  }
+  if (parsed && parsed.corpid) {
+    const { corpid, token, baseurl, userId } = parsed;
+    return [{ corpid, token, baseurl, userId, enable: true }];
+  }
+  return [];
+}
+
+export function upsertCompany(companies, entry) {
+  const index = companies.findIndex((item) => String(item.corpid || '').trim() === entry.corpid);
+  const next = companies.map((item) => ({ ...item, enable: false }));
+  if (index === -1) {
+    next.push({ ...entry, enable: true });
+  } else {
+    next[index] = { ...next[index], ...entry, enable: true };
+  }
+  return next;
+}
+
+// token-set / auth-login 共用：保存一家公司的凭证并刷新本地缓存，返回与 token-set 相同结构的行
+export async function saveCompanyCredentials(kwargs) {
+  const corpid = normalizeArg(kwargs.corpid);
+  const token = normalizeArg(kwargs.token);
+  const userId = normalizeArg(kwargs.userId);
+
+  if (!corpid) {
+    return createResult('error', '缺少 --corpid', '', '', '');
+  }
+
+  if (!token) {
+    return createResult('error', '缺少 --token', corpid, '', '');
+  }
+
+  if (!userId) {
+    return createResult('error', '缺少 --userId', corpid, '', '');
+  }
+
+  let personalToken = token;
+  if (!token.startsWith('user_')) {
+    try {
+      personalToken = await getPersonalToken(corpid, token, userId);
+    } catch (error) {
+      return createResult('error', `生成个人 token 失败：${error.message}`, corpid, '', userId);
+    }
+  }
+
+  const baseurl = resolveBaseUrl(corpid);
+  const companies = upsertCompany(readCompanies(), { corpid, token: personalToken, baseurl, userId });
+
+  fs.mkdirSync(CONFIG_DIR, { recursive: true });
+  fs.writeFileSync(CONFIG_FILE, JSON.stringify(companies, null, 2) + '\n', 'utf8');
+
+  try {
+    const formlistFile = await writeFormlistFile(corpid);
+    const commandMapFile = writeCommandMap(corpid);
+    const { file: departmentUserFile, corpName, userName } = await writeDepartmentUserFile(corpid, userId);
+    writeCompanyProfile(corpid, { corpName, userName });
+    let envStored = 'skipped';
+    let envFile = '';
+    if (!kwargs.noEnv) {
+      try {
+        const envResult = persistEnvVars({ corpid, token: personalToken, baseurl, userId, corpName, userName });
+        envStored = envResult.stored;
+        envFile = envResult.file;
+      } catch (error) {
+        envStored = 'failed';
+        envFile = String(error.message || error);
+      }
+    }
+    return createResult('ok', '已保存 corpid、token、baseurl、userId，并同步表单模板缓存、命令映射文件与部门/员工缓存', corpid, baseurl, userId, formlistFile, commandMapFile, true, companies.length, departmentUserFile, corpName, userName, envStored, envFile);
+  } catch (error) {
+    return createResult(
+      'partial',
+      `已保存 corpid、token、baseurl、userId，但同步缓存失败：${error.message}`,
+      corpid,
+      baseurl,
+      userId,
+      getFormlistFile(corpid),
+      '',
+      true,
+      companies.length,
+      '',
+    );
+  }
+}

package/version.js

@@ -1,18 +0,0 @@
-import { cli, Strategy } from './xbb-registry.js';
-
-const VERSION = 'v1.6';
-
-cli({
-  site: 'xbb',
-  name: 'version',
-  description: '查看销帮帮插件版本号',
-  strategy: Strategy.PUBLIC,
-  access: 'read',
-  browser: false,
-  domain: 'proapi.xbongbong.com',
-  args: [],
-  columns: ['version'],
-  func: async function () {
-    return [{ version: VERSION }];
-  },
-});