rl-rockcli 0.0.2 → 0.0.4

package/commands/sandbox/core/replay/request.js

@@ -1,109 +0,0 @@
-/**
- * 请求分类常量
- */
-const RequestCategory = {
-  STARTUP: 'startup',                  // start_async
-  STARTUP_POLLING: 'startup_polling',  // get_status after start
-  SESSION: 'session',                  // create_session, close_session
-  NOHUP_COMMAND: 'nohup_command',      // nohup xxx &
-  PROCESS_CHECK: 'process_check',      // kill -0 {pid}
-  LOG_QUERY: 'log_query',              // head -c xxx file
-  UPLOAD: 'upload',                    // upload file
-  STOP: 'stop',                        // stop sandbox
-  OTHER: 'other'                       // other commands
-};
-
-/**
- * 处理动作常量
- */
-const RequestAction = {
-  EXECUTE: 'execute',                              // 执行
-  SKIP: 'skip',                                    // 跳过
-  MERGE: 'merge',                                  // 合并到等待逻辑
-  EXECUTE_WITH_PROCESS_WAIT: 'execute_with_process_wait'   // 执行 nohup 并等待进程结束
-};
-
-/**
- * 单个回放请求
- */
-class ReplayRequest {
-  constructor(raw, index) {
-    this.index = index;
-    this.method = raw.method;
-    this.uri = raw.uri;
-    this.requestBody = raw.requestBody || {};
-    this.headers = raw.headers || {};
-    this.responseTime = raw.responseTime ? new Date(raw.responseTime) : null;
-
-    // 分析后填充的字段
-    this.category = null;       // 请求分类
-    this.action = null;         // 处理动作
-    this.skipReason = null;     // 跳过原因
-  }
-
-  /**
-   * 获取命令（如果是 run_in_session 请求）
-   */
-  getCommand() {
-    return this.requestBody?.command || '';
-  }
-
-  /**
-   * 获取简短描述
-   */
-  getShortDescription(maxLen = 60) {
-    const command = this.getCommand();
-    if (command) {
-      if (command.length > maxLen) {
-        return command.substring(0, maxLen - 3) + '...';
-      }
-      return command;
-    }
-    return this.uri.split('?')[0].split('/').pop();
-  }
-}
-
-/**
- * 回放计划
- */
-class ReplayPlan {
-  constructor() {
-    this.originalCount = 0;       // 原始请求数
-    this.requests = [];           // 待执行的请求列表
-    this.skippedRequests = [];    // 被跳过的请求列表
-    this.mergedRequests = [];     // 被合并的请求列表
-
-    // 统计信息
-    this.stats = {
-      startup: 0,
-      startupPolling: 0,
-      session: 0,
-      nohupCommand: 0,
-      processCheck: 0,
-      logQuery: 0,
-      upload: 0,
-      stop: 0,
-      other: 0
-    };
-
-  }
-
-  get skippedCount() {
-    return this.skippedRequests.length;
-  }
-
-  get mergedCount() {
-    return this.mergedRequests.length;
-  }
-
-  get executeCount() {
-    return this.requests.length;
-  }
-}
-
-module.exports = {
-  RequestCategory,
-  RequestAction,
-  ReplayRequest,
-  ReplayPlan
-};

package/commands/sandbox/core/replay/worker.js

@@ -1,166 +0,0 @@
-const path = require('path');
-const { SandboxClient, SandboxConfig } = require('../../../../sdks/sandbox');
-const { ReplayExecutor } = require('./executor');
-const { ReplayRequest, ReplayPlan } = require('./request');
-
-/**
- * 单个沙箱回放工作者
- * 封装独立的 SandboxClient 和 ReplayExecutor
- */
-class ReplayWorker {
-  /**
-   * @param {number} workerId - 工作者 ID（1-based）
-   * @param {SandboxConfig} config - 沙箱配置
-   * @param {ReplayPlan} plan - 回放计划（会被深拷贝）
-   * @param {Object} options - 回放选项
-   * @param {string} options.timing - 时间控制
-   * @param {number} options.interval - 固定间隔
-   * @param {string} options.logFile - 日志文件基础路径
-   */
-  constructor(workerId, config, plan, options = {}) {
-    this.workerId = workerId;
-    this.config = config;
-    this.plan = this.deepClonePlan(plan);
-    this.options = options;
-
-    // 独立的沙箱客户端和执行器
-    this.client = null;
-    this.executor = null;
-
-    // 执行状态
-    this.status = 'pending';  // pending | running | completed | failed
-    this.error = null;
-    this.results = null;
-    this.startTime = null;
-    this.endTime = null;
-  }
-
-  /**
-   * 深拷贝回放计划
-   * 确保每个 worker 有独立的请求对象（requestBody 可能被修改）
-   * @param {ReplayPlan} plan
-   * @returns {ReplayPlan}
-   */
-  deepClonePlan(plan) {
-    const cloned = new ReplayPlan();
-    cloned.originalCount = plan.originalCount;
-    cloned.stats = { ...plan.stats };
-    cloned.pidAssociations = new Map(plan.pidAssociations);
-
-    // 深拷贝请求列表
-    cloned.requests = plan.requests.map(req => this.cloneRequest(req));
-    cloned.skippedRequests = plan.skippedRequests.map(req => this.cloneRequest(req));
-    cloned.mergedRequests = plan.mergedRequests.map(req => this.cloneRequest(req));
-
-    return cloned;
-  }
-
-  /**
-   * 深拷贝单个请求
-   * @param {ReplayRequest} req
-   * @returns {ReplayRequest}
-   */
-  cloneRequest(req) {
-    const cloned = new ReplayRequest({
-      method: req.method,
-      uri: req.uri,
-      requestBody: JSON.parse(JSON.stringify(req.requestBody || {})),
-      headers: JSON.parse(JSON.stringify(req.headers || {})),
-      responseTime: req.responseTime
-    }, req.index);
-
-    cloned.category = req.category;
-    cloned.action = req.action;
-    cloned.originalPid = req.originalPid;
-    cloned.skipReason = req.skipReason;
-
-    return cloned;
-  }
-
-  /**
-   * 获取 worker 专属的日志文件名
-   * @returns {string}
-   */
-  getLogFileName() {
-    const baseLogFile = this.options.logFile || 'replay.log';
-    const ext = path.extname(baseLogFile);
-    const base = path.basename(baseLogFile, ext);
-    const dir = path.dirname(baseLogFile);
-    return path.join(dir, `${base}-worker${this.workerId}${ext}`);
-  }
-
-  /**
-   * 执行回放
-   * @returns {Promise<WorkerResult>}
-   */
-  async execute() {
-    this.status = 'running';
-    this.startTime = Date.now();
-
-    try {
-      // 创建独立的 SandboxClient
-      this.client = new SandboxClient(this.config);
-
-      // 创建独立的 ReplayExecutor
-      this.executor = new ReplayExecutor(this.client, {
-        timing: this.options.timing,
-        interval: this.options.interval,
-        quiet: true,  // 并发模式下强制静默，由 orchestrator 统一输出
-        logFile: this.getLogFileName(),
-        plan: this.plan,
-        workerId: this.workerId
-      });
-
-      // 执行回放
-      this.results = await this.executor.execute(this.plan);
-      this.status = 'completed';
-
-    } catch (error) {
-      this.status = 'failed';
-      this.error = error;
-      this.results = {
-        total: this.plan.requests.length,
-        success: 0,
-        failed: this.plan.requests.length,
-        skipped: 0
-      };
-    } finally {
-      this.endTime = Date.now();
-    }
-
-    return this.getWorkerResult();
-  }
-
-  /**
-   * 获取工作者执行结果
-   * @returns {WorkerResult}
-   */
-  getWorkerResult() {
-    return {
-      workerId: this.workerId,
-      status: this.status,
-      sandboxId: this.executor?.currentSandboxId || null,
-      results: this.results,
-      error: this.error?.message || null,
-      duration: this.endTime - this.startTime,
-      logFile: this.getLogFileName()
-    };
-  }
-
-  /**
-   * 获取当前进度
-   * @returns {Object}
-   */
-  getProgress() {
-    return {
-      workerId: this.workerId,
-      status: this.status,
-      sandboxId: this.executor?.currentSandboxId?.substring(0, 8) || null,
-      success: this.executor?.results?.success || 0,
-      failed: this.executor?.results?.failed || 0,
-      total: this.plan.requests.length
-    };
-  }
-}
-
-module.exports = { ReplayWorker };

package/commands/sandbox/core/session.js

@@ -1,346 +0,0 @@
-const { createInternalSandboxClient, SandboxConfig, getInternalSandboxConfig } = require('../../../sdks/sandbox');
-const logger = require('../../../utils/logger');
-const configManager = require('../../../utils/configManager');
-const { gracefulExit } = require('../../../utils/execution_logger');
-
-/**
- * 读取配置文件中的 sandbox 配置
- * @param {Object} [argv] - CLI 参数，用于解析 API key
- */
-function readSandboxConfig(argv = {}) {
-  const config = configManager.readConfig();
-  const sandboxConfig = config?.sandbox || {};
-  const internalDefaults = getInternalSandboxConfig();
-
-  return {
-    base_url: argv.baseUrl || sandboxConfig.base_url || internalDefaults.baseUrl,
-    cluster: argv.cluster || sandboxConfig.cluster || internalDefaults.cluster,
-    ...sandboxConfig,
-    api_key: configManager.getApiKeyWithPriority(argv),
-  };
-}
-
-// Session command module for yargs
-module.exports = {
-  command: 'session <subaction>',
-  describe: 'Session operations:\n' +
-    '  create  创建 session\n' +
-    '  run     在 session 中执行命令\n' +
-    '  close   关闭 session',
-  builder: (yargs) => {
-    return yargs
-      .positional('subaction', {
-        describe: 'Session action to perform',
-        type: 'string',
-        choices: ['create', 'run', 'close'],
-      })
-      .option('sandbox-id', {
-        alias: 'id',
-        describe: 'Sandbox ID (required)',
-        type: 'string',
-        demandOption: true,
-        group: 'Command Options:'
-      })
-      .option('session', {
-        describe: 'Session name',
-        type: 'string',
-        default: 'default',
-        group: 'Command Options:'
-      })
-      .option('command', {
-        describe: 'Command to execute (for run action)',
-        type: 'string',
-        group: 'Command Options:'
-      })
-      .option('startup-source', {
-        describe: 'Startup source files for session (comma-separated)',
-        type: 'string',
-        group: 'Command Options:'
-      })
-      .option('startup-timeout', {
-        describe: 'Startup timeout for session in seconds',
-        type: 'number',
-        group: 'Command Options:'
-      })
-      .option('session-type', {
-        describe: 'Session type (default: bash)',
-        type: 'string',
-        default: 'bash',
-        group: 'Command Options:'
-      })
-      .option('env', {
-        describe: 'Environment variables as JSON string',
-        type: 'string',
-        group: 'Command Options:'
-      })
-      .option('env-enable', {
-        describe: 'Enable environment variables from parent process',
-        type: 'boolean',
-        default: false,
-        group: 'Command Options:'
-      })
-      .option('timeout', {
-        describe: 'Command execution timeout in seconds',
-        type: 'number',
-        group: 'Command Options:'
-      })
-      .option('is-interactive', {
-        describe: 'Is interactive command',
-        type: 'boolean',
-        default: false,
-        group: 'Command Options:'
-      })
-      .option('expect', {
-        describe: 'Expected output patterns for interactive commands (comma-separated)',
-        type: 'string',
-        group: 'Command Options:'
-      })
-      .option('check', {
-        describe: 'Exit check mode: raise (default), silent, ignore',
-        type: 'string',
-        default: 'raise',
-        choices: ['raise', 'silent', 'ignore'],
-        group: 'Command Options:'
-      })
-      .option('cluster', {
-        describe: 'Cluster ID (sets X-Cluster header)',
-        type: 'string',
-        group: 'Command Options:'
-      })
-      .option('experiment-id', {
-        describe: 'Experiment ID (sets X-Experiment-Id header)',
-        type: 'string',
-        group: 'Command Options:'
-      })
-      .option('user-id', {
-        describe: 'User ID (sets X-User-Id header)',
-        type: 'string',
-        group: 'Command Options:'
-      })
-      .option('extra-header', {
-        alias: 'H',
-        type: 'array',
-        description: 'Extra HTTP headers in format "Key=Value". Can be used multiple times.',
-        group: 'Command Options:',
-        coerce: (headers) => {
-          const result = {};
-          if (headers) {
-            headers.forEach(header => {
-              if (header.includes('=')) {
-                const [key, value] = header.split('=', 2);
-                result[key.trim()] = value.trim();
-              } else {
-                logger.warn(`Invalid header format: ${header}. Expected format: 'Key=Value'`);
-              }
-            });
-          }
-          return result;
-        }
-      })
-      .example([
-        ['$0 sandbox session create --id <sandbox-id>', 'Create a session'],
-        ['$0 sandbox session run --id <sandbox-id> --command "ls -la"', 'Run command in session'],
-        ['$0 sandbox session run --id <sandbox-id> -- ls -la', 'Run command in session (alternative)'],
-        ['$0 sandbox session close --id <sandbox-id>', 'Close a session'],
-      ]);
-  },
-  handler: async (argv) => {
-    try {
-      switch (argv.subaction) {
-        case 'create':
-          await handleSessionCreate(argv);
-          break;
-        case 'run':
-          await handleSessionRun(argv);
-          break;
-        case 'close':
-          await handleSessionClose(argv);
-          break;
-        default:
-          console.error(`Unknown session action: ${argv.subaction}`);
-          gracefulExit(1);
-      }
-    } catch (error) {
-      logger.error(`Session action failed: ${error.message}`);
-      gracefulExit(1);
-    }
-  },
-};
-
-/**
- * Handle session create action
- */
-async function handleSessionCreate(argv) {
-  const sandboxConfig = readSandboxConfig(argv);
-  const baseUrl = argv.baseUrl || sandboxConfig.base_url;
-  const apiKey = argv.apiKey || sandboxConfig.api_key;
-  const cluster = argv.cluster || sandboxConfig.cluster;
-  const userId = argv.userId || sandboxConfig.user_id;
-  const experimentId = argv.experimentId || sandboxConfig.experiment_id;
-
-  if (!baseUrl) {
-    throw new Error('base-url is required (use --base-url or run "rock-cli login" first)');
-  }
-
-  const config = new SandboxConfig({
-    baseUrl: baseUrl,
-    xrlAuthorization: apiKey,
-    cluster: cluster,
-    userId: userId,
-    experimentId: experimentId,
-    extraHeaders: argv.extraHeader || {},
-  });
-
-  const client = createInternalSandboxClient(config, { requireImage: false });
-  client._sandboxId = argv.sandboxId;
-
-  const options = {
-    session: argv.session || 'default',
-    startupSource: argv.startupSource ? argv.startupSource.split(',') : [],
-    envEnable: argv.envEnable || false,
-    env: argv.env ? JSON.parse(argv.env) : null,
-  };
-
-  if (argv.startupTimeout) {
-    options.startupTimeout = argv.startupTimeout;
-  }
-
-  if (!argv.raw) {
-    console.log(`Creating session: ${options.session}...`);
-  }
-  const result = await client.createSession(options);
-
-  if (argv.raw) {
-    console.log(JSON.stringify({
-      output: result.output || '',
-      exit_code: result.exit_code !== undefined ? result.exit_code : 0,
-      success: true,
-    }));
-  } else {
-    console.log('✅ Session created successfully!');
-    if (result.output) {
-      console.log('Output:');
-      console.log(result.output);
-    }
-    if (result.exit_code !== undefined && result.exit_code !== null) {
-      console.log(`Exit code: ${result.exit_code}`);
-    }
-  }
-}
-
-/**
- * Handle session run action
- */
-async function handleSessionRun(argv) {
-  let command = argv.command;
-  if (!command) {
-    const separatorIndex = process.argv.indexOf('--');
-    if (separatorIndex !== -1 && separatorIndex < process.argv.length - 1) {
-      command = process.argv.slice(separatorIndex + 1).join(' ');
-    }
-  }
-
-  if (!command) {
-    throw new Error('command is required. Use --command "your command" or: rock-cli sandbox session run --id <id> -- your command');
-  }
-
-  const sandboxConfig = readSandboxConfig(argv);
-  const baseUrl = argv.baseUrl || sandboxConfig.base_url;
-  const apiKey = argv.apiKey || sandboxConfig.api_key;
-  const cluster = argv.cluster || sandboxConfig.cluster;
-  const userId = argv.userId || sandboxConfig.user_id;
-  const experimentId = argv.experimentId || sandboxConfig.experiment_id;
-
-  if (!baseUrl) {
-    throw new Error('base-url is required (use --base-url or run "rock-cli login" first)');
-  }
-
-  const config = new SandboxConfig({
-    baseUrl: baseUrl,
-    xrlAuthorization: apiKey,
-    cluster: cluster,
-    userId: userId,
-    experimentId: experimentId,
-    extraHeaders: argv.extraHeader || {},
-  });
-
-  const client = createInternalSandboxClient(config, { requireImage: false });
-  client._sandboxId = argv.sandboxId;
-
-  const action = {
-    session: argv.session || 'default',
-    command: command,
-    check: argv.check || 'raise',
-    timeout: argv.timeout ? argv.timeout * 1000 : null,
-    isInteractiveCommand: argv.isInteractive || false,
-  };
-
-  if (argv.expect) {
-    action.expect = argv.expect.split(',');
-  }
-
-  if (!argv.raw) {
-    console.log(`Running command in session ${action.session}: ${command}`);
-  }
-  const result = await client.runInSession(action);
-
-  if (argv.raw) {
-    console.log(JSON.stringify({
-      output: result.output || '',
-      exit_code: result.exit_code !== undefined ? result.exit_code : 0,
-      success: true,
-    }));
-  } else {
-    if (result.output) {
-      console.log('Output:');
-      console.log(result.output);
-    }
-
-    if (result.exit_code !== undefined && result.exit_code !== null) {
-      console.log(`Exit code: ${result.exit_code}`);
-    }
-  }
-
-  if (result.exit_code !== undefined && result.exit_code !== 0 && action.check === 'raise') {
-    gracefulExit(1);
-  }
-}
-
-/**
- * Handle session close action
- */
-async function handleSessionClose(argv) {
-  const sandboxConfig = readSandboxConfig(argv);
-  const baseUrl = argv.baseUrl || sandboxConfig.base_url;
-  const apiKey = argv.apiKey || sandboxConfig.api_key;
-  const cluster = argv.cluster || sandboxConfig.cluster;
-  const userId = argv.userId || sandboxConfig.user_id;
-  const experimentId = argv.experimentId || sandboxConfig.experiment_id;
-
-  if (!baseUrl) {
-    throw new Error('base-url is required (use --base-url or run "rock-cli login" first)');
-  }
-
-  const config = new SandboxConfig({
-    baseUrl: baseUrl,
-    xrlAuthorization: apiKey,
-    cluster: cluster,
-    userId: userId,
-    experimentId: experimentId,
-    extraHeaders: argv.extraHeader || {},
-  });
-
-  const client = createInternalSandboxClient(config, { requireImage: false });
-  client._sandboxId = argv.sandboxId;
-
-  const session = argv.session || 'default';
-
-  console.log(`Closing session: ${session}...`);
-  const result = await client.closeSession(session);
-
-  console.log('✅ Session closed successfully!');
-  if (result && result.output) {
-    console.log('Output:');
-    console.log(result.output);
-  }
-}

package/commands/sandbox/log-bridge.js

@@ -1,2 +0,0 @@
-// Compatibility wrapper - delegates to core implementation
-module.exports = require('./core/log-bridge');

package/commands/sandbox/ray.js

@@ -1,2 +0,0 @@
-// Compatibility wrapper - delegates to internal implementation
-module.exports = require('./internal/ray');