rol-websocket-channel 1.4.2 → 1.4.9

package/src/admin/methods/system.ts

@@ -1,10 +1,14 @@
-import { exec } from 'node:child_process';
+import { exec, execFile } from 'node:child_process';
 import fs from 'node:fs/promises';
 import path from 'node:path';
 import { promisify } from 'node:util';
+import { JsonRpcException, JSON_RPC_ERRORS } from '../jsonrpc.js';
 import type { JsonValue, MethodHandler, MethodContext } from '../types.js';
 
 const execAsync = promisify(exec);
+const execFileAsync = promisify(execFile);
+const UPDATE_COMMAND_TIMEOUT_MS = 10 * 60 * 1000;
+const UPDATE_COMMAND_MAX_BUFFER = 10 * 1024 * 1024;
 
 export const ping: MethodHandler = async (): Promise<JsonValue> => {
   return {
@@ -74,6 +78,33 @@ export const doctorFix: MethodHandler = async (_params, context: MethodContext):
   }
 };
 
+export const openclawUpdate: MethodHandler = async (_params, context: MethodContext): Promise<JsonValue> => {
+  const result = await runOpenClawCommand(['update'], context, 'openclawUpdate');
+
+  return {
+    ok: true,
+    action: 'openclawUpdate',
+    restartRecommended: true,
+    ...result
+  };
+};
+
+export const pluginSelfUpdate: MethodHandler = async (_params, context: MethodContext): Promise<JsonValue> => {
+  const result = await runOpenClawCommand(
+    ['plugins', 'install', 'clawhub:rol-websocket-channel'],
+    context,
+    'pluginSelfUpdate'
+  );
+
+  return {
+    ok: true,
+    action: 'pluginSelfUpdate',
+    plugin: 'clawhub:rol-websocket-channel',
+    restartRecommended: true,
+    ...result
+  };
+};
+
 export const logs: MethodHandler = async (params: any, context: MethodContext): Promise<JsonValue> => {
   try {
     // 根据实际情况可能需要调整，这里默认尝试 project root 或 user home 下的 .openclaw/logs
@@ -203,3 +234,100 @@ export const logs: MethodHandler = async (params: any, context: MethodContext):
     };
   }
 };
+
+async function runOpenClawCommand(
+  args: string[],
+  context: MethodContext,
+  action: string
+): Promise<{ command: string; args: string[]; cwd: string; stdout: string; stderr: string }> {
+  const command = process.env.OPENCLAW_BIN || 'openclaw';
+  const options = buildOpenClawExecOptions(context.openclawRoot, context.openclawRoot);
+  const openclawBin = process.env.OPENCLAW_BIN || '';
+  const openclawHome = options.env?.OPENCLAW_HOME || '';
+  const home = options.env?.HOME || '';
+
+  console.log(
+    `[system] exec start: action=${action}, command=${command}, args=${JSON.stringify(args)}, cwd=${options.cwd}, OPENCLAW_BIN=${openclawBin}, OPENCLAW_HOME=${openclawHome}, HOME=${home}`
+  );
+
+  try {
+    const { stdout, stderr } = await execFileAsync(command, args, {
+      ...options,
+      timeout: UPDATE_COMMAND_TIMEOUT_MS,
+      maxBuffer: UPDATE_COMMAND_MAX_BUFFER
+    });
+
+    console.log(
+      `[system] exec success: action=${action}, command=${command}, args=${JSON.stringify(args)}, stdoutLength=${stdout.length}, stderrLength=${stderr.length}`
+    );
+
+    return {
+      command,
+      args,
+      cwd: options.cwd,
+      stdout,
+      stderr
+    };
+  } catch (err: any) {
+    const stdout = typeof err?.stdout === 'string' ? err.stdout : '';
+    const stderr = typeof err?.stderr === 'string' ? err.stderr : '';
+    console.error(
+      `[system] exec failed: action=${action}, command=${command}, args=${JSON.stringify(args)}, cwd=${options.cwd}, OPENCLAW_BIN=${openclawBin}, OPENCLAW_HOME=${openclawHome}, HOME=${home}, stdout=${JSON.stringify(stdout)}, stderr=${JSON.stringify(stderr)}`
+    );
+
+    throw new JsonRpcException(
+      JSON_RPC_ERRORS.internalError,
+      `OpenClaw system command failed: ${err instanceof Error ? err.message : String(err)}`,
+      {
+        action,
+        command,
+        args,
+        cwd: options.cwd,
+        stdout,
+        stderr
+      }
+    );
+  }
+}
+
+function buildOpenClawExecOptions(
+  cwd: string,
+  openclawRoot?: string
+): { cwd: string; shell?: boolean; env: NodeJS.ProcessEnv } {
+  const env: NodeJS.ProcessEnv = { ...process.env };
+  const openclawHome = resolveOpenClawHomeForCli(openclawRoot);
+  if (openclawHome) {
+    env.OPENCLAW_HOME = openclawHome;
+  }
+
+  if (openclawRoot && path.basename(path.resolve(openclawRoot)) === '.openclaw') {
+    const home = path.dirname(path.resolve(openclawRoot));
+    env.HOME = home;
+    if (process.platform === 'win32') {
+      env.USERPROFILE = home;
+    }
+  }
+
+  if (process.platform === 'win32') {
+    return {
+      cwd,
+      shell: true,
+      env
+    };
+  }
+
+  return { cwd, env };
+}
+
+function resolveOpenClawHomeForCli(openclawRoot?: string): string | undefined {
+  if (!openclawRoot) {
+    return process.env.OPENCLAW_HOME;
+  }
+
+  const resolvedRoot = path.resolve(openclawRoot);
+  if (path.basename(resolvedRoot) === '.openclaw') {
+    return path.dirname(resolvedRoot);
+  }
+
+  return process.env.OPENCLAW_HOME;
+}