rl-rockcli 0.0.7 → 0.0.8

package/utils/configManager.js

@@ -1,503 +0,0 @@
-const fs = require('fs-extra');
-const os = require('os');
-const path = require('path');
-const logger = require('./logger');
-
-class ConfigManager {
-  constructor() {
-    // 旧配置文件路径（用于向后兼容）
-    this.oldConfigDir = path.join(os.homedir(), '.rock-cli');
-    this.oldConfigPath = path.join(this.oldConfigDir, 'settings.json');
-    
-    // 新配置文件路径（统一配置）
-    this.newConfigDir = path.join(os.homedir(), '.rock');
-    this.newConfigPath = path.join(this.newConfigDir, 'settings.json');
-    
-    // 默认使用新配置路径
-    this.configPath = this.newConfigPath;
-    this.configDir = this.newConfigDir;
-    
-    this.ensureConfigDir();
-  }
-
-  /**
-   * 确保配置目录存在
-   */
-  ensureConfigDir() {
-    if (!fs.existsSync(this.configDir)) {
-      fs.mkdirpSync(this.configDir, { recursive: true });
-      logger.info(`Created config directory: ${this.configDir}`);
-    }
-  }
-
-  /**
-   * 读取配置 - 支持新旧格式
-   */
-  readConfig() {
-    // 优先读取新配置文件
-    if (fs.existsSync(this.newConfigPath)) {
-      try {
-        const configData = fs.readJsonSync(this.newConfigPath);
-        return configData;
-      } catch (error) {
-        logger.error(`Failed to read new config file: ${error.message}`);
-      }
-    }
-    
-    // 如果新配置文件不存在，尝试读取旧配置文件
-    if (fs.existsSync(this.oldConfigPath)) {
-      try {
-        const configData = fs.readJsonSync(this.oldConfigPath);
-        // 将旧格式转换为新格式
-        const newFormatConfig = this.convertOldToNewFormat(configData);
-        // 保存到新位置以实现迁移
-        this.saveToNewConfig(newFormatConfig);
-        return newFormatConfig;
-      } catch (error) {
-        logger.error(`Failed to read old config file: ${error.message}`);
-      }
-    }
-    
-    return {};
-  }
-
-  /**
-   * 将旧配置格式转换为新格式
-   */
-  convertOldToNewFormat(oldConfig) {
-    const newConfig = { sandbox: {}, test: {} };
-    
-    // 将旧配置映射到新格式的 sandbox 部分
-    for (const [key, value] of Object.entries(oldConfig)) {
-      if (key === 'base_url' || key === 'api_key' || key === 'cluster' || 
-          key === 'user_id' || key === 'experiment_id') {
-        newConfig.sandbox[key] = value;
-      } else if (key === 'private_token' || key === 'whale_apikey' || 
-                 key === 'code_private_token' || key === 'whale_api_key') {
-        newConfig.test[key] = value;
-      } else {
-        // 其他配置项也放入 sandbox（作为默认）
-        newConfig.sandbox[key] = value;
-      }
-    }
-    
-    return newConfig;
-  }
-
-  /**
-   * 保存到新配置文件
-   */
-  saveToNewConfig(config) {
-    try {
-      fs.writeJsonSync(this.newConfigPath, config, { spaces: 2 });
-      logger.info(`Config saved to: ${this.newConfigPath}`);
-      return true;
-    } catch (error) {
-      logger.error(`Failed to write new config file: ${error.message}`);
-      return false;
-    }
-  }
-
-  /**
-   * 写入配置
-   */
-  writeConfig(config) {
-    return this.saveToNewConfig(config);
-  }
-
-  /**
-   * 获取特定配置项（支持旧格式兼容）
-   */
-  getConfig(key) {
-    const config = this.readConfig();
-    
-    // 如果配置已经是新格式（有 sandbox 和 test 部分）
-    if (config.sandbox || config.test) {
-      // 检查 test 部分
-      if (config.test && config.test.hasOwnProperty(key)) {
-        return config.test[key];
-      }
-      // 检查 sandbox 部分
-      if (config.sandbox && config.sandbox.hasOwnProperty(key)) {
-        return config.sandbox[key];
-      }
-    }
-    
-    // 否则按旧格式处理
-    return config[key];
-  }
-
-  /**
-   * 设置特定配置项（支持新格式）
-   */
-  setConfig(key, value) {
-    const config = this.readConfig();
-    
-    // 检查是否是测试相关的配置键
-    const testKeys = ['private_token', 'whale_apikey', 'code_private_token', 'whale_api_key'];
-    if (testKeys.includes(key)) {
-      if (!config.test) config.test = {};
-      config.test[key] = value;
-    } else {
-      // 其他配置默认放在 sandbox 部分
-      if (!config.sandbox) config.sandbox = {};
-      config.sandbox[key] = value;
-    }
-    
-    return this.writeConfig(config);
-  }
-
-  /**
-   * 删除特定配置项
-   */
-  deleteConfig(key) {
-    const config = this.readConfig();
-    
-    if (config.test && config.test.hasOwnProperty(key)) {
-      delete config.test[key];
-    } else if (config.sandbox && config.sandbox.hasOwnProperty(key)) {
-      delete config.sandbox[key];
-    } else {
-      delete config[key];
-    }
-    
-    return this.writeConfig(config);
-  }
-
-  /**
-   * 设置沙箱配置
-   */
-  setSandboxConfig(key, value) {
-    const config = this.readConfig();
-    if (!config.sandbox) config.sandbox = {};
-    config.sandbox[key] = value;
-    return this.writeConfig(config);
-  }
-
-  /**
-   * 获取沙箱配置
-   */
-  getSandboxConfig(key) {
-    const config = this.readConfig();
-    if (config.sandbox && config.sandbox.hasOwnProperty(key)) {
-      return config.sandbox[key];
-    }
-    return undefined;
-  }
-
-  /**
-   * 获取 UI 配置
-   */
-  getUIConfig(key, defaultValue = undefined) {
-    const config = this.readConfig();
-    if (config.ui && Object.prototype.hasOwnProperty.call(config.ui, key)) {
-      return config.ui[key];
-    }
-    return defaultValue;
-  }
-
-  /**
-   * 设置 UI 配置
-   */
-  setUIConfig(key, value) {
-    const config = this.readConfig();
-    if (!config.ui) config.ui = {};
-    config.ui[key] = value;
-    return this.writeConfig(config);
-  }
-
-  /**
-   * 设置测试配置
-   */
-  setTestConfig(key, value) {
-    const config = this.readConfig();
-    if (!config.test) config.test = {};
-    config.test[key] = value;
-    return this.writeConfig(config);
-  }
-
-  /**
-   * 获取测试配置
-   */
-  getTestConfig(key) {
-    const config = this.readConfig();
-    if (config.test && config.test.hasOwnProperty(key)) {
-      return config.test[key];
-    }
-    return undefined;
-  }
-
-  /**
-   * 获取配置路径
-   */
-  getConfigPath() {
-    return this.configPath;
-  }
-
-  /**
-   * Get API key with priority: ROCK_API_KEY env > CLI arg > config file
-   * @param {Object} argv - Command line arguments
-   * @returns {string|null} - API key or null if not found
-   */
-  getApiKeyWithPriority(argv = {}) {
-    // Priority 1: Environment variable
-    if (process.env.ROCK_API_KEY) {
-      return process.env.ROCK_API_KEY;
-    }
-
-    // Priority 2: CLI argument
-    if (argv.apiKey) {
-      return argv.apiKey;
-    }
-
-    // Priority 3: Config file
-    const config = this.readConfig();
-    if (config?.sandbox?.api_key) {
-      return config.sandbox.api_key;
-    }
-
-    return null;
-  }
-
-  /**
-   * Get cluster with priority: ROCK_CLUSTER env > CLI arg > config file > default
-   * @param {Object} argv - Command line arguments
-   * @returns {string} - Cluster or default 'zb'
-   */
-  getClusterWithPriority(argv = {}) {
-    // Priority 1: Environment variable
-    if (process.env.ROCK_CLUSTER) {
-      return process.env.ROCK_CLUSTER;
-    }
-
-    // Priority 2: CLI argument
-    if (argv.cluster) {
-      return argv.cluster;
-    }
-
-    // Priority 3: Config file
-    const config = this.readConfig();
-    if (config?.sandbox?.cluster) {
-      return config.sandbox.cluster;
-    }
-
-    // Default value
-    return 'zb';
-  }
-
-  /**
-   * Get user ID with priority: ROCK_USER_ID env > CLI arg > config file
-   * @param {Object} argv - Command line arguments
-   * @returns {string|null} - User ID or null if not found
-   */
-  getUserIdWithPriority(argv = {}) {
-    // Priority 1: Environment variable
-    if (process.env.ROCK_USER_ID) {
-      return process.env.ROCK_USER_ID;
-    }
-
-    // Priority 2: CLI argument
-    if (argv.userId) {
-      return argv.userId;
-    }
-
-    // Priority 3: Config file
-    const config = this.readConfig();
-    if (config?.sandbox?.user_id) {
-      return config.sandbox.user_id;
-    }
-
-    return null;
-  }
-
-  /**
-   * Get experiment ID with priority: ROCK_EXPERIMENT_ID env > CLI arg > config file
-   * @param {Object} argv - Command line arguments
-   * @returns {string|null} - Experiment ID or null if not found
-   */
-  getExperimentIdWithPriority(argv = {}) {
-    // Priority 1: Environment variable
-    if (process.env.ROCK_EXPERIMENT_ID) {
-      return process.env.ROCK_EXPERIMENT_ID;
-    }
-
-    // Priority 2: CLI argument
-    if (argv.experimentId) {
-      return argv.experimentId;
-    }
-
-    // Priority 3: Config file
-    const config = this.readConfig();
-    if (config?.sandbox?.experiment_id) {
-      return config.sandbox.experiment_id;
-    }
-
-    return null;
-  }
-
-  /**
-   * 获取脱敏后的配置（用于显示）
-   */
-  getMaskedConfig() {
-    const config = this.readConfig();
-    
-    // 如果是新格式，对各部分分别脱敏
-    if (config.sandbox || config.test) {
-      const maskedConfig = {};
-      if (config.sandbox) {
-        maskedConfig.sandbox = {};
-        for (const [key, value] of Object.entries(config.sandbox)) {
-          if (typeof value === 'string' && (key.toLowerCase().includes('token') || key.toLowerCase().includes('key') || key.toLowerCase().includes('secret'))) {
-            maskedConfig.sandbox[key] = this.maskValue(value);
-          } else {
-            maskedConfig.sandbox[key] = value;
-          }
-        }
-      }
-      if (config.test) {
-        maskedConfig.test = {};
-        for (const [key, value] of Object.entries(config.test)) {
-          if (typeof value === 'string' && (key.toLowerCase().includes('token') || key.toLowerCase().includes('key') || key.toLowerCase().includes('secret'))) {
-            maskedConfig.test[key] = this.maskValue(value);
-          } else {
-            maskedConfig.test[key] = value;
-          }
-        }
-      }
-      return maskedConfig;
-    } else {
-      // 旧格式处理
-      const maskedConfig = {};
-      for (const [key, value] of Object.entries(config)) {
-        if (typeof value === 'string' && (key.toLowerCase().includes('token') || key.toLowerCase().includes('key'))) {
-          maskedConfig[key] = this.maskValue(value);
-        } else {
-          maskedConfig[key] = value;
-        }
-      }
-      return maskedConfig;
-    }
-  }
-
-  /**
-   * 脱敏处理
-   */
-  maskValue(value) {
-    if (typeof value !== 'string') {
-      return value;
-    }
-
-    if (value.length <= 8) {
-      return '*'.repeat(Math.max(value.length, 4));
-    }
-
-    // 保留前2位和后2位，中间用星号代替
-    return value.substring(0, 2) + '*'.repeat(value.length - 4) + value.substring(value.length - 2);
-  }
-
-  /**
-   * 获取插件配置
-   * @param {string} pluginName - 插件名称
-   * @returns {Object} - 插件配置对象
-   */
-  getPluginConfig(pluginName) {
-    const config = this.readConfig();
-    const plugins = config.plugins || [];
-
-    // 旧格式：数组
-    if (Array.isArray(plugins)) {
-      return {}; // 旧格式不支持配置
-    }
-
-    // 新格式：对象
-    const pluginConfig = plugins[pluginName];
-    return pluginConfig?.config || {};
-  }
-
-  /**
-   * 获取所有插件配置（支持新旧格式转换）
-   * @returns {Object} - 插件配置对象
-   */
-  getPlugins() {
-    const config = this.readConfig();
-    const plugins = config.plugins || [];
-
-    // 旧格式：转换为新格式
-    if (Array.isArray(plugins)) {
-      return plugins.reduce((acc, path, index) => {
-        const name = `plugin-${index}`;
-        acc[name] = {
-          path: path,
-          config: {},
-          enabled: true
-        };
-        return acc;
-      }, {});
-    }
-
-    // 新格式：直接返回
-    return plugins;
-  }
-
-  /**
-   * 检查插件是否启用
-   * @param {string} pluginName - 插件名称
-   * @returns {boolean} - 是否启用
-   */
-  isPluginEnabled(pluginName) {
-    const plugins = this.getPlugins();
-    const pluginConfig = plugins[pluginName];
-    
-    // 插件不存在，返回 false
-    if (!pluginConfig) {
-      return false;
-    }
-    
-    // enabled 字段不存在，默认启用
-    if (pluginConfig.enabled === undefined) {
-      return true;
-    }
-    
-    // 返回 enabled 的值
-    return pluginConfig.enabled === true;
-  }
-
-  /**
-   * 设置插件配置（兼容旧格式）
-   * @param {Array<string>} plugins - 插件路径数组（旧格式）
-   * @returns {boolean}
-   */
-  setPlugins(plugins) {
-    const config = this.readConfig();
-    config.plugins = plugins;
-    return this.writeConfig(config);
-  }
-
-  /**
-   * 获取全局配置（带优先级合并）
-   * @param {Object} argv - 命令行参数对象
-   * @returns {Object} - 合并后的全局配置
-   */
-  getGlobalConfigWithPriority(argv = {}) {
-    const config = this.readConfig();
-    const sandboxConfig = config.sandbox || {};
-
-    // 实现优先级合并逻辑
-    const mergedConfig = {
-      ...config,
-      sandbox: {
-        ...sandboxConfig,
-        // 使用优先级函数获取所有参数
-        api_key: this.getApiKeyWithPriority(argv),
-        base_url: argv.baseUrl || sandboxConfig.base_url,
-        cluster: this.getClusterWithPriority(argv),
-        user_id: this.getUserIdWithPriority(argv),
-        experiment_id: this.getExperimentIdWithPriority(argv)
-      }
-    };
-
-    return mergedConfig;
-  }
-}
-
-module.exports = new ConfigManager();

package/utils/dependency-resolver.js

@@ -1,135 +0,0 @@
-const logger = require('./logger');
-
-/**
- * 依赖关系解析器
- * 用于解析插件加载顺序，检测循环依赖
- */
-class DependencyResolver {
-  /**
-   * 解析插件加载顺序
-   * @param {Array<{name: string, dependsOn?: string[]}>} plugins - 插件列表
-   * @returns {Array<string>} - 加载顺序（插件名称数组）
-   * @throws {Error} - 如果存在循环依赖
-   */
-  resolveLoadOrder(plugins) {
-    if (!plugins || plugins.length === 0) {
-      return [];
-    }
-
-    // 构建依赖图
-    const graph = {};
-    const allPlugins = new Set();
-
-    plugins.forEach(plugin => {
-      const { name, dependsOn = [] } = plugin;
-      allPlugins.add(name);
-
-      if (!graph[name]) {
-        graph[name] = [];
-      }
-
-      dependsOn.forEach(dep => {
-        graph[name].push(dep);
-        allPlugins.add(dep);
-      });
-    });
-
-    // 确保所有插件都在图中
-    allPlugins.forEach(name => {
-      if (!graph[name]) {
-        graph[name] = [];
-      }
-    });
-
-    // 检测环
-    if (this._hasCycle(graph)) {
-      throw new Error('Circular dependency detected in plugins');
-    }
-
-    // 拓扑排序
-    return this._topologicalSort(graph);
-  }
-
-  /**
-   * 检测依赖图中是否存在环
-   * @private
-   */
-  _hasCycle(graph) {
-    const visited = new Set();
-    const recursionStack = new Set();
-
-    const dfs = (node) => {
-      if (recursionStack.has(node)) {
-        return true; // 发现环
-      }
-      if (visited.has(node)) {
-        return false;
-      }
-
-      visited.add(node);
-      recursionStack.add(node);
-
-      const neighbors = graph[node] || [];
-      for (const neighbor of neighbors) {
-        if (dfs(neighbor)) {
-          return true;
-        }
-      }
-
-      recursionStack.delete(node);
-      return false;
-    };
-
-    for (const node of Object.keys(graph)) {
-      if (!visited.has(node)) {
-        if (dfs(node)) {
-          return true;
-        }
-      }
-    }
-
-    return false;
-  }
-
-  /**
-   * 拓扑排序
-   * @private
-   */
-  _topologicalSort(graph) {
-    const outDegree = {};
-    const queue = [];
-    const result = [];
-
-    // 计算出度（这个节点依赖多少个节点）
-    for (const node of Object.keys(graph)) {
-      outDegree[node] = (graph[node] || []).length;
-    }
-
-    // 将出度为 0 的节点（不依赖任何节点的）加入队列
-    for (const node of Object.keys(outDegree)) {
-      if (outDegree[node] === 0) {
-        queue.push(node);
-      }
-    }
-
-    // 拓扑排序
-    while (queue.length > 0) {
-      const node = queue.shift();
-      result.push(node);
-
-      // node 已经加载完成，减少依赖 node 的节点的出度
-      for (const dependent of Object.keys(graph)) {
-        if (graph[dependent].includes(node)) {
-          outDegree[dependent]--;
-          if (outDegree[dependent] === 0) {
-            queue.push(dependent);
-          }
-        }
-      }
-    }
-
-    return result;
-  }
-}
-
-module.exports = DependencyResolver;