1314mc-helper 0.2.21

package/src/tools/codex.js

@@ -0,0 +1,154 @@
+import fs from 'fs-extra';
+import path from 'path';
+import os from 'os';
+import TOML from '@iarna/toml';
+import { getApiBaseForProtocol } from '../index.js';
+
+const DUOJIE_PROVIDER_ID = 'duojie';
+const DEFAULT_MODEL = 'gpt-5.4';
+
+/**
+ * 获取 Codex CLI 配置路径
+ */
+function getCodexConfigPaths() {
+  const home = os.homedir();
+  return {
+    configDir: path.join(home, '.codex'),
+    configFile: path.join(home, '.codex', 'config.toml'),
+  };
+}
+
+function getWireApi(model) {
+  return model.startsWith('gpt') ? 'responses' : 'chat';
+}
+
+async function readConfigToml(configFile) {
+  if (!(await fs.pathExists(configFile))) {
+    return {};
+  }
+
+  const content = await fs.readFile(configFile, 'utf-8');
+  if (!content.trim()) {
+    return {};
+  }
+
+  return TOML.parse(content);
+}
+
+/**
+ * 配置 Codex CLI
+ * 使用官方 config.toml 结构
+ */
+export async function configureCodex(apiKey) {
+  const paths = getCodexConfigPaths();
+
+  try {
+    await fs.ensureDir(paths.configDir);
+
+    let existingConfig = {};
+    try {
+      existingConfig = await readConfigToml(paths.configFile);
+    } catch {
+      existingConfig = {};
+    }
+
+    const wireApi = getWireApi(DEFAULT_MODEL);
+
+    const modelProviders = {
+      ...(existingConfig.model_providers || {}),
+      [DUOJIE_PROVIDER_ID]: {
+        ...(existingConfig.model_providers?.[DUOJIE_PROVIDER_ID] || {}),
+        name: '1314mc',
+        base_url: getApiBaseForProtocol(wireApi),
+        experimental_bearer_token: apiKey,
+        wire_api: wireApi,
+      },
+    };
+
+    const newConfig = {
+      ...existingConfig,
+      model: DEFAULT_MODEL,
+      model_provider: DUOJIE_PROVIDER_ID,
+      model_providers: modelProviders,
+    };
+
+    const tomlContent = `# Codex CLI Configuration\n# Generated by duojie-helper\n# Docs: https://developers.openai.com/codex/config-reference/\n\n${TOML.stringify(newConfig)}`;
+
+    await fs.writeFile(paths.configFile, tomlContent, 'utf-8');
+
+    return {
+      success: true,
+      message: `已配置 → ${paths.configFile}`,
+      configPath: paths.configFile,
+      hint: '已写入 config.toml（纯配置，无需额外环境变量）',
+    };
+  } catch (error) {
+    return {
+      success: false,
+      message: `配置失败: ${error.message}`,
+    };
+  }
+}
+
+/**
+ * 检查 Codex CLI 配置状态
+ */
+configureCodex.checkStatus = async function() {
+  const paths = getCodexConfigPaths();
+
+  if (await fs.pathExists(paths.configFile)) {
+    try {
+      const config = await readConfigToml(paths.configFile);
+      const provider = config.model_providers?.[DUOJIE_PROVIDER_ID];
+
+      if (provider?.base_url?.includes('duojie') || provider?.base_url?.includes('1314mc')) {
+        return { configured: true, message: '已配置 1314mc API' };
+      }
+
+      if (config.model_provider || Object.keys(config.model_providers || {}).length > 0) {
+        return { configured: true, message: '已配置（非 1314mc）' };
+      }
+    } catch {
+      return { configured: true, message: '配置文件存在但格式无效' };
+    }
+  }
+
+  return { configured: false };
+};
+
+/**
+ * 撤销 Codex CLI 配置
+ */
+configureCodex.revoke = async function() {
+  const paths = getCodexConfigPaths();
+
+  if (!(await fs.pathExists(paths.configFile))) {
+    return;
+  }
+
+  try {
+    const config = await readConfigToml(paths.configFile);
+
+    if (config.model_providers?.[DUOJIE_PROVIDER_ID]) {
+      delete config.model_providers[DUOJIE_PROVIDER_ID];
+      if (Object.keys(config.model_providers).length === 0) {
+        delete config.model_providers;
+      }
+    }
+
+    if (config.model_provider === DUOJIE_PROVIDER_ID) {
+      delete config.model_provider;
+    }
+
+    if (config.model === DEFAULT_MODEL) {
+      delete config.model;
+    }
+
+    const tomlContent = TOML.stringify(config);
+    await fs.writeFile(paths.configFile, tomlContent ? `${tomlContent}` : '', 'utf-8');
+  } catch {
+    // ignore
+  }
+};
+
+export default configureCodex;

package/src/tools/continue.js

@@ -0,0 +1,149 @@
+import fs from 'fs-extra';
+import path from 'path';
+import os from 'os';
+import { getModels, getApiBaseForProtocol } from '../index.js';
+
+/**
+ * 获取 Continue 配置路径
+ */
+function getContinueConfigPaths() {
+  const home = os.homedir();
+  return {
+    configDir: path.join(home, '.continue'),
+    configFile: path.join(home, '.continue', 'config.json'),
+  };
+}
+
+/**
+ * 配置 Continue
+ * 自动配置所有可用模型
+ */
+export async function configureContinue(apiKey) {
+  const paths = getContinueConfigPaths();
+  
+  try {
+    await fs.ensureDir(paths.configDir);
+
+    let config = {};
+    if (await fs.pathExists(paths.configFile)) {
+      try {
+        config = await fs.readJson(paths.configFile);
+      } catch {
+        config = {};
+      }
+    }
+
+    // 获取可用模型列表
+    const models = getModels();
+    const duojieModels = [];
+
+    const openaiProtocols = new Set([
+      'openai', 'openai-response', 'openai-responses',
+      'openai-completion', 'openai-completions', 'responses', 'chat',
+    ]);
+
+    function providerForApi(api) {
+      return openaiProtocols.has(`${api || ''}`.toLowerCase()) ? 'openai' : 'anthropic';
+    }
+
+    function protocolLabel(api) {
+      return providerForApi(api) === 'openai' ? 'OpenAI' : 'Anthropic';
+    }
+
+    // 处理所有模型（遍历 apis 数组，双协议生成两条记录）
+    const allCategories = [
+      ...(models.claude || []),
+      ...(models.gpt || []),
+      ...(models.gemini || []),
+      ...(models.other || []),
+    ];
+
+    for (const m of allCategories) {
+      const apis = m.apis || ['anthropic-messages'];
+      for (const api of apis) {
+        const provider = providerForApi(api);
+        duojieModels.push({
+          title: `1314mc ${m.name} (${protocolLabel(api)})`,
+          provider: provider,
+          model: m.id,
+          apiKey: apiKey,
+          apiBase: getApiBaseForProtocol(api),
+          completionOptions: {
+            maxTokens: 16384,
+          },
+        });
+      }
+    }
+
+    // 把默认模型尽量设为 gpt-5.4（如果存在），否则保持当前顺序
+    const gpt54 = duojieModels.find(m => m.model === 'gpt-5.4');
+    if (gpt54) {
+      const others = duojieModels.filter(m => m !== gpt54);
+      duojieModels.length = 0;
+      duojieModels.push(gpt54, ...others);
+    }
+
+    // 合并模型配置 - 移除旧的 Duojie 配置
+    const existingModels = config.models || [];
+    const filteredModels = existingModels.filter(m => 
+      !m.title?.startsWith('1314mc')
+    );
+
+    const newConfig = {
+      ...config,
+      models: [...duojieModels, ...filteredModels],
+      _mc1314: {
+        configured: true,
+        configuredAt: new Date().toISOString(),
+      },
+    };
+
+    await fs.writeJson(paths.configFile, newConfig, { spaces: 2 });
+
+    return {
+      success: true,
+      message: `已配置 ${duojieModels.length} 个模型 → ${paths.configFile}`,
+      configPath: paths.configFile,
+      hint: '重启 VS Code/JetBrains 使配置生效',
+    };
+  } catch (error) {
+    return {
+      success: false,
+      message: `配置失败: ${error.message}`,
+    };
+  }
+}
+
+configureContinue.checkStatus = async function() {
+  const paths = getContinueConfigPaths();
+  
+  if (await fs.pathExists(paths.configFile)) {
+    try {
+      const config = await fs.readJson(paths.configFile);
+      if (config._mc1314?.configured) {
+        return { configured: true, message: '已配置 1314mc API' };
+      } else if (config.models?.length > 0) {
+        return { configured: true, message: '已配置（非 1314mc）' };
+      }
+    } catch {}
+  }
+  return { configured: false };
+};
+
+configureContinue.revoke = async function() {
+  const paths = getContinueConfigPaths();
+  
+  if (await fs.pathExists(paths.configFile)) {
+    try {
+      const config = await fs.readJson(paths.configFile);
+      // 移除 Duojie 模型
+      config.models = (config.models || []).filter(m => 
+        !m.title?.startsWith('1314mc')
+      );
+      delete config._mc1314;
+      await fs.writeJson(paths.configFile, config, { spaces: 2 });
+    } catch {}
+  }
+};
+
+export default configureContinue;

package/src/tools/cursor.js

@@ -0,0 +1,99 @@
+import fs from 'fs-extra';
+import path from 'path';
+import os from 'os';
+import chalk from 'chalk';
+import { API_CONFIG } from '../index.js';
+
+/**
+ * 获取 Cursor 配置路径
+ */
+function getCursorConfigPaths() {
+  const home = os.homedir();
+  const platform = os.platform();
+  
+  let configDir;
+  if (platform === 'win32') {
+    configDir = path.join(process.env.APPDATA || home, 'Cursor', 'User');
+  } else if (platform === 'darwin') {
+    configDir = path.join(home, 'Library', 'Application Support', 'Cursor', 'User');
+  } else {
+    configDir = path.join(home, '.config', 'Cursor', 'User');
+  }
+  
+  return {
+    configDir,
+    settingsFile: path.join(configDir, 'settings.json'),
+  };
+}
+
+/**
+ * 配置 Cursor
+ * Cursor 需要通过 UI 手动配置自定义模型（需要高级会员）
+ */
+export async function configureCursor(apiKey) {
+  const baseUrl = API_CONFIG.baseUrl;
+  
+  // 生成配置说明
+  const instructions = `
+${chalk.yellow('Cursor 需要在编辑器中手动配置:')}
+
+${chalk.red('⚠️  注意: 需要 Cursor Pro 或更高级订阅才能使用自定义模型')}
+
+${chalk.cyan('步骤 1:')} 打开 Cursor，进入 Settings → Models
+${chalk.cyan('步骤 2:')} 点击 "Add Custom Model"
+${chalk.cyan('步骤 3:')} 填入以下配置:
+
+  ${chalk.green('Protocol:')}              OpenAI
+  ${chalk.green('OpenAI API Key:')}        ${apiKey}
+  ${chalk.green('Override OpenAI Base URL:')} ${baseUrl}
+  ${chalk.green('Model Name:')}            输入模型名称（注意大小写）
+                          例如: gpt-5.4
+                          或: claude-sonnet-4-6
+
+${chalk.cyan('步骤 4:')} 点击保存，然后在模型列表中选择刚添加的模型
+
+${chalk.cyan('可选配置:')}
+  ${chalk.green('Max Tokens:')}            16384（推荐）
+
+${chalk.gray('提示: 如果遇到 "The model does not work with your current plan" 错误，')}
+${chalk.gray('      说明您的 Cursor 订阅不支持自定义模型')}
+`;
+
+  return {
+    success: true,
+    manual: true,
+    message: '需要手动配置（需 Cursor Pro）',
+    instructions: instructions,
+    configInfo: {
+      protocol: 'OpenAI',
+      baseUrl: baseUrl,
+      apiKey: apiKey,
+    }
+  };
+}
+
+/**
+ * 检查 Cursor 配置状态
+ */
+configureCursor.checkStatus = async function() {
+  const paths = getCursorConfigPaths();
+  
+  if (await fs.pathExists(paths.configDir)) {
+    return {
+      configured: true,
+      message: 'Cursor 已安装（配置状态需在 Cursor 中查看）',
+    };
+  }
+  
+  return { configured: false };
+};
+
+/**
+ * 撤销 Cursor 配置
+ */
+configureCursor.revoke = async function() {
+  // Cursor 自定义模型配置在内部数据库，无法通过文件操作撤销
+  console.log(chalk.yellow('Cursor 配置需要在 Cursor Settings → Models 中手动删除'));
+};
+
+export default configureCursor;

package/src/tools/droid.js

@@ -0,0 +1,253 @@
+import fs from 'fs-extra';
+import path from 'path';
+import os from 'os';
+import { API_CONFIG, getModels, getApiBaseForProtocol } from '../index.js';
+
+function normalizeBaseUrl(url) {
+  return `${url || ''}`.replace(/\/+$/, '');
+}
+
+/**
+ * 获取 Droid (Factory) 配置路径
+ */
+function getDroidConfigPaths() {
+  const home = os.homedir();
+  return {
+    configDir: path.join(home, '.factory'),
+    configFile: path.join(home, '.factory', 'config.json'),
+    settingsFile: path.join(home, '.factory', 'settings.json'),
+  };
+}
+
+const DUOJIE_MODEL_TAG = '[1314mc]';
+
+function isDuojieModel(m) {
+  return (
+    (m.base_url && (m.base_url.includes('duojie') || m.base_url.includes('1314mc'))) ||
+    (m.model_display_name && m.model_display_name.includes(DUOJIE_MODEL_TAG))
+  );
+}
+
+/**
+ * 生成带协议后缀的显示名称
+ * 双协议模型会生成两条记录，如 "GLM 5 Turbo [1314mc OpenAI]" 和 "GLM 5 Turbo [1314mc Anthropic]"
+ */
+function generateDisplayName(modelId, api) {
+  let name = modelId;
+  if (name.startsWith('claude-')) {
+    name = name.substring(7);
+  } else if (name.startsWith('gpt-')) {
+    name = 'GPT ' + name.substring(4);
+  } else if (name.startsWith('gemini-')) {
+    name = 'Gemini ' + name.substring(7);
+  } else if (name.startsWith('glm-')) {
+    name = 'GLM-' + name.substring(4);
+  }
+
+  name = name
+    .split('-')
+    .map(part => {
+      if (/^\d+$/.test(part)) return part;
+      return part.charAt(0).toUpperCase() + part.slice(1);
+    })
+    .join(' ');
+
+  name = name.replace(/(\d+)\s+(\d+)(?=\s|$)/g, '$1.$2');
+
+  const provider = providerForApi(api);
+  const suffix = provider === 'openai' ? 'OpenAI' : 'Anthropic';
+  return `${name} [1314mc ${suffix}]`;
+}
+
+/**
+ * 根据模型 ID 确定 provider
+ * @param {string} modelId - 模型 ID
+ * @param {string[]} endpoints - 支持的 endpoint 类型
+ * @returns {string} provider 名称
+ */
+/**
+ * 根据协议字符串判断 provider 类型
+ */
+function providerForApi(api) {
+  const openaiProtocols = new Set([
+    'openai', 'openai-response', 'openai-responses',
+    'openai-completion', 'openai-completions', 'responses', 'chat',
+  ]);
+  return openaiProtocols.has(`${api || ''}`.toLowerCase()) ? 'openai' : 'anthropic';
+}
+
+/**
+ * 配置 Droid (Factory)
+ * 自动配置所有可用模型到 custom_models
+ */
+export async function configureDroid(apiKey) {
+  const paths = getDroidConfigPaths();
+  
+  try {
+    // 确保目录存在
+    await fs.ensureDir(paths.configDir);
+
+    // 1. 读取现有配置（如果存在）
+    let existingConfig = {};
+    if (await fs.pathExists(paths.configFile)) {
+      try {
+        existingConfig = await fs.readJson(paths.configFile);
+      } catch {
+        existingConfig = {};
+      }
+    }
+
+    // 2. 获取模型列表
+    const models = getModels();
+    const rawModels = API_CONFIG.rawModels || [];
+    
+    // 创建模型 ID 到 endpoints 的映射
+    const endpointsMap = {};
+    for (const m of rawModels) {
+      endpointsMap[m.id] = m.supported_endpoint_types || [];
+    }
+
+    // 3. 构建 Duojie 的 custom_models
+    const duojieModels = [];
+    
+    // 处理所有模型类别
+    const allModels = [
+      ...(models.claude || []),
+      ...(models.gpt || []),
+      ...(models.gemini || []),
+      ...(models.other || []),
+    ];
+
+    // 默认模型优先放在列表前（便于用户在 UI 中选择）
+    const preferredId = 'gpt-5.4';
+    const preferred = allModels.find(m => m.id === preferredId) || null;
+    const orderedModels = preferred
+      ? [preferred, ...allModels.filter(m => m.id !== preferredId)]
+      : allModels;
+    
+    for (const m of orderedModels) {
+      const apis = m.apis || ['anthropic-messages'];
+      for (const api of apis) {
+        const provider = providerForApi(api);
+        duojieModels.push({
+          model_display_name: generateDisplayName(m.id, api),
+          model: m.id,
+          base_url: normalizeBaseUrl(getApiBaseForProtocol(api)),
+          api_key: apiKey,
+          provider: provider,
+          supports_vision: true,
+          max_tokens: 16384,
+        });
+      }
+    }
+
+    if (duojieModels.length === 0) {
+      return {
+        success: false,
+        message: '没有可用的模型',
+      };
+    }
+
+    // 4. 合并配置（保留用户其他自定义模型）
+    let existingCustomModels = existingConfig.custom_models || [];
+    
+    // 过滤掉已有的 Duojie 模型（通过 base_url 或 display_name 判断）
+    existingCustomModels = existingCustomModels.filter(m => {
+      return !isDuojieModel(m);
+    });
+
+    // 5. 合并新旧模型
+    const newConfig = {
+      ...existingConfig,
+      custom_models: [
+        ...existingCustomModels,
+        ...duojieModels,
+      ],
+    };
+
+    // 6. 写入配置文件
+    await fs.writeJson(paths.configFile, newConfig, { spaces: 2 });
+
+    // 7. 清空 settings.json 中的 customModels（如文件和字段存在）
+    if (await fs.pathExists(paths.settingsFile)) {
+      try {
+        const settings = await fs.readJson(paths.settingsFile);
+        if (Array.isArray(settings.customModels) && settings.customModels.length > 0) {
+          settings.customModels = [];
+          await fs.writeJson(paths.settingsFile, settings, { spaces: 2 });
+        }
+      } catch {
+        // 忽略，不影响主流程
+      }
+    }
+
+    return {
+      success: true,
+      message: `已配置 ${duojieModels.length} 个模型 → ${paths.configFile}`,
+      configPath: paths.configFile,
+      hint: '重启 Factory 使配置生效',
+    };
+  } catch (error) {
+    return {
+      success: false,
+      message: `配置失败: ${error.message}`,
+    };
+  }
+}
+
+/**
+ * 检查 Droid 配置状态
+ */
+configureDroid.checkStatus = async function() {
+  const paths = getDroidConfigPaths();
+  
+  if (await fs.pathExists(paths.configFile)) {
+    try {
+      const config = await fs.readJson(paths.configFile);
+      
+      // 检查是否有 Duojie 的自定义模型
+      const customModels = config.custom_models || [];
+      const duojieModels = customModels.filter(m => isDuojieModel(m));
+      
+      if (duojieModels.length > 0) {
+        return {
+          configured: true,
+          message: `已配置 ${duojieModels.length} 个 1314mc 模型`,
+        };
+      } else if (customModels.length > 0) {
+        return {
+          configured: true,
+          message: '已配置（非 1314mc）',
+        };
+      }
+    } catch {
+      // ignore
+    }
+  }
+  
+  return { configured: false };
+};
+
+/**
+ * 撤销 Droid 配置
+ */
+configureDroid.revoke = async function() {
+  const paths = getDroidConfigPaths();
+  
+  if (await fs.pathExists(paths.configFile)) {
+    try {
+      const config = await fs.readJson(paths.configFile);
+      
+      // 只删除 Duojie 的自定义模型
+      if (config.custom_models) {
+        config.custom_models = config.custom_models.filter(m => !isDuojieModel(m));
+      }
+      
+      await fs.writeJson(paths.configFile, config, { spaces: 2 });
+    } catch {
+      // ignore
+    }
+  }
+};
+
+export default configureDroid;