@znmin/claude-remote-cli 0.3.1 → 0.4.0

package/src/local/processes.js

@@ -1,144 +1,144 @@
-import { spawn } from 'node:child_process';
-import { randomUUID } from 'node:crypto';
-import path from 'node:path';
-import { findShell, prepareCommand } from './shell.js';
-
-// Characters retained per stream; older output is dropped but offsets stay absolute.
-const LIMIT = 256 * 1024;
-const PREVIEW = 8000;
-const delay = ms => new Promise(resolve => setTimeout(resolve, ms));
-
-export class Processes {
-  constructor(workspace, env = process.env, { platform = process.platform, shell } = {}) {
-    this.workspace = workspace; this.env = env; this.platform = platform; this.jobs = new Map();
-    if (shell) this.shellChoice = Promise.resolve(shell);
-  }
-
-  shell() { return this.shellChoice ||= findShell(this.env, this.platform); }
-
-  job(id) {
-    const job = this.jobs.get(id);
-    if (!job) throw new Error('找不到本会话的进程。');
-    return job;
-  }
-
-  meta(id) {
-    const job = this.job(id);
-    return { job_id: id, pid: job.child.pid, running: !job.ended, exit_code: job.code ?? null,
-      signal: job.signal ?? null, timed_out: Boolean(job.timedOut), cwd: job.cwd };
-  }
-
-  // Without a stream: the last PREVIEW characters of stdout and stderr.
-  // With a stream: a page starting at an absolute character offset.
-  status(id, { stream, offset, limit = PREVIEW } = {}) {
-    const job = this.job(id);
-    const totals = { stdout_total: job.stdout.dropped + job.stdout.text.length,
-      stderr_total: job.stderr.dropped + job.stderr.text.length };
-    if (!stream) {
-      const truncated = job.stdout.dropped > 0 || job.stderr.dropped > 0 ||
-        job.stdout.text.length > PREVIEW || job.stderr.text.length > PREVIEW;
-      return { ...this.meta(id), stdout: job.stdout.text.slice(-PREVIEW), stderr: job.stderr.text.slice(-PREVIEW),
-        output_truncated: truncated, ...totals,
-        ...(truncated ? { hint: '仅返回末尾输出。用 process_status 的 stream、offset、limit 参数分页读取更早的内容。' } : {}) };
-    }
-    const buffer = job[stream];
-    const total = buffer.dropped + buffer.text.length;
-    const start = Math.max(offset ?? 0, buffer.dropped);
-    const text = buffer.text.slice(start - buffer.dropped, start - buffer.dropped + limit);
-    const next = start + text.length;
-    return { ...this.meta(id), stream, offset: start, text, total, available_from: buffer.dropped,
-      next_offset: next < total ? next : null,
-      ...(offset !== undefined && offset < buffer.dropped ? { note: `前 ${buffer.dropped} 个字符已丢弃，仅保留最近 ${LIMIT} 个字符。` } : {}) };
-  }
-
-  async run(command, cwd = '.', timeoutMs = 30000, background = false, signal) {
-    if ([...this.jobs.values()].filter(job => !job.ended).length >= 16) throw new Error('后台进程过多，请先停止部分进程。');
-    for (const [id, job] of this.jobs) if (this.jobs.size >= 64 && job.ended) this.jobs.delete(id);
-    const directory = await this.workspace.resolve(cwd);
-    const prepared = await prepareCommand(await this.shell(), command, directory, this.env);
-    const windows = this.platform === 'win32';
-    // POSIX: a new process group, stopped as a whole. Windows: no new console
-    // window; the process tree is stopped with taskkill instead.
-    let child;
-    try {
-      child = spawn(prepared.file, prepared.args, {
-        cwd: directory, env: prepared.env, detached: !windows, windowsHide: true, stdio: ['ignore', 'pipe', 'pipe']
-      });
-    } catch (error) { await prepared.cleanup(); throw error; }
-    const id = randomUUID();
-    const job = { child, cwd: directory, ended: false,
-      stdout: { text: '', dropped: 0 }, stderr: { text: '', dropped: 0 } };
-    this.jobs.set(id, job);
-    job.done = new Promise(resolve => {
-      child.stdout.setEncoding('utf8'); child.stderr.setEncoding('utf8');
-      for (const stream of ['stdout', 'stderr']) child[stream].on('data', chunk => {
-        const buffer = job[stream];
-        buffer.text += chunk;
-        if (buffer.text.length > LIMIT) {
-          const excess = buffer.text.length - LIMIT;
-          buffer.text = buffer.text.slice(excess);
-          buffer.dropped += excess;
-        }
-      });
-      child.once('error', error => { job.stderr.text += error.message; job.code = 127; job.ended = true; resolve(); });
-      child.once('close', (code, sig) => { job.code = code; job.signal = sig; job.ended = true; resolve(); });
-    }).finally(prepared.cleanup);
-    if (background) {
-      await Promise.race([job.done, delay(80)]);
-      return this.status(id);
-    }
-    const abort = () => this.stop(id).catch(() => {});
-    signal?.addEventListener('abort', abort, { once: true });
-    const timer = setTimeout(() => { job.timedOut = true; abort(); }, timeoutMs);
-    try {
-      if (signal?.aborted) abort();
-      await job.done;
-      return this.status(id);
-    } finally { clearTimeout(timer); signal?.removeEventListener('abort', abort); }
-  }
-
-  killGroup(job, signal) {
-    if (!job.child.pid) return;
-    try { process.kill(-job.child.pid, signal); }
-    catch (error) { if (error.code !== 'ESRCH') throw error; }
-  }
-
-  // Console programs ignore taskkill's polite close request, so the tree is
-  // stopped forcefully. It only reaches processes whose parent is still alive.
-  killTree(job) {
-    if (!job.child.pid) return Promise.resolve();
-    const taskkill = path.win32.join(this.env.SystemRoot || this.env.SYSTEMROOT || 'C:\\Windows', 'System32', 'taskkill.exe');
-    return new Promise(resolve => {
-      const killer = spawn(taskkill, ['/PID', String(job.child.pid), '/T', '/F'], { stdio: 'ignore', windowsHide: true });
-      killer.once('error', resolve); killer.once('close', resolve);
-    });
-  }
-
-  async stop(id) {
-    const job = this.job(id);
-    if (job.ended) return this.status(id);
-    if (this.platform === 'win32') {
-      await this.killTree(job);
-      await Promise.race([job.done, delay(1500)]);
-      if (!job.ended) job.child.kill();
-      await Promise.race([job.done, delay(1000)]);
-      return this.status(id);
-    }
-    // Kill the owned group even if its parent shell has already exited.
-    this.killGroup(job, 'SIGTERM');
-    await Promise.race([job.done, delay(1500)]);
-    this.killGroup(job, 'SIGKILL');
-    await Promise.race([job.done, delay(1000)]);
-    return this.status(id);
-  }
-
-  list() {
-    return [...this.jobs.keys()].map(id => {
-      const job = this.jobs.get(id);
-      return { ...this.meta(id), stdout_characters: job.stdout.dropped + job.stdout.text.length,
-        stderr_characters: job.stderr.dropped + job.stderr.text.length };
-    });
-  }
-
-  async close() { await Promise.allSettled([...this.jobs.keys()].map(id => this.stop(id))); }
-}
+import { spawn } from 'node:child_process';
+import { randomUUID } from 'node:crypto';
+import path from 'node:path';
+import { findShell, prepareCommand } from './shell.js';
+
+// Characters retained per stream; older output is dropped but offsets stay absolute.
+const LIMIT = 256 * 1024;
+const PREVIEW = 8000;
+const delay = ms => new Promise(resolve => setTimeout(resolve, ms));
+
+export class Processes {
+  constructor(workspace, env = process.env, { platform = process.platform, shell } = {}) {
+    this.workspace = workspace; this.env = env; this.platform = platform; this.jobs = new Map();
+    if (shell) this.shellChoice = Promise.resolve(shell);
+  }
+
+  shell() { return this.shellChoice ||= findShell(this.env, this.platform); }
+
+  job(id) {
+    const job = this.jobs.get(id);
+    if (!job) throw new Error('找不到本会话的进程。');
+    return job;
+  }
+
+  meta(id) {
+    const job = this.job(id);
+    return { job_id: id, pid: job.child.pid, running: !job.ended, exit_code: job.code ?? null,
+      signal: job.signal ?? null, timed_out: Boolean(job.timedOut), cwd: job.cwd };
+  }
+
+  // Without a stream: the last PREVIEW characters of stdout and stderr.
+  // With a stream: a page starting at an absolute character offset.
+  status(id, { stream, offset, limit = PREVIEW } = {}) {
+    const job = this.job(id);
+    const totals = { stdout_total: job.stdout.dropped + job.stdout.text.length,
+      stderr_total: job.stderr.dropped + job.stderr.text.length };
+    if (!stream) {
+      const truncated = job.stdout.dropped > 0 || job.stderr.dropped > 0 ||
+        job.stdout.text.length > PREVIEW || job.stderr.text.length > PREVIEW;
+      return { ...this.meta(id), stdout: job.stdout.text.slice(-PREVIEW), stderr: job.stderr.text.slice(-PREVIEW),
+        output_truncated: truncated, ...totals,
+        ...(truncated ? { hint: '仅返回末尾输出。用 process_status 的 stream、offset、limit 参数分页读取更早的内容。' } : {}) };
+    }
+    const buffer = job[stream];
+    const total = buffer.dropped + buffer.text.length;
+    const start = Math.max(offset ?? 0, buffer.dropped);
+    const text = buffer.text.slice(start - buffer.dropped, start - buffer.dropped + limit);
+    const next = start + text.length;
+    return { ...this.meta(id), stream, offset: start, text, total, available_from: buffer.dropped,
+      next_offset: next < total ? next : null,
+      ...(offset !== undefined && offset < buffer.dropped ? { note: `前 ${buffer.dropped} 个字符已丢弃，仅保留最近 ${LIMIT} 个字符。` } : {}) };
+  }
+
+  async run(command, cwd = '.', timeoutMs = 30000, background = false, signal) {
+    if ([...this.jobs.values()].filter(job => !job.ended).length >= 16) throw new Error('后台进程过多，请先停止部分进程。');
+    for (const [id, job] of this.jobs) if (this.jobs.size >= 64 && job.ended) this.jobs.delete(id);
+    const directory = await this.workspace.resolve(cwd);
+    const prepared = await prepareCommand(await this.shell(), command, directory, this.env);
+    const windows = this.platform === 'win32';
+    // POSIX: a new process group, stopped as a whole. Windows: no new console
+    // window; the process tree is stopped with taskkill instead.
+    let child;
+    try {
+      child = spawn(prepared.file, prepared.args, {
+        cwd: directory, env: prepared.env, detached: !windows, windowsHide: true, stdio: ['ignore', 'pipe', 'pipe']
+      });
+    } catch (error) { await prepared.cleanup(); throw error; }
+    const id = randomUUID();
+    const job = { child, cwd: directory, ended: false,
+      stdout: { text: '', dropped: 0 }, stderr: { text: '', dropped: 0 } };
+    this.jobs.set(id, job);
+    job.done = new Promise(resolve => {
+      child.stdout.setEncoding('utf8'); child.stderr.setEncoding('utf8');
+      for (const stream of ['stdout', 'stderr']) child[stream].on('data', chunk => {
+        const buffer = job[stream];
+        buffer.text += chunk;
+        if (buffer.text.length > LIMIT) {
+          const excess = buffer.text.length - LIMIT;
+          buffer.text = buffer.text.slice(excess);
+          buffer.dropped += excess;
+        }
+      });
+      child.once('error', error => { job.stderr.text += error.message; job.code = 127; job.ended = true; resolve(); });
+      child.once('close', (code, sig) => { job.code = code; job.signal = sig; job.ended = true; resolve(); });
+    }).finally(prepared.cleanup);
+    if (background) {
+      await Promise.race([job.done, delay(80)]);
+      return this.status(id);
+    }
+    const abort = () => this.stop(id).catch(() => {});
+    signal?.addEventListener('abort', abort, { once: true });
+    const timer = setTimeout(() => { job.timedOut = true; abort(); }, timeoutMs);
+    try {
+      if (signal?.aborted) abort();
+      await job.done;
+      return this.status(id);
+    } finally { clearTimeout(timer); signal?.removeEventListener('abort', abort); }
+  }
+
+  killGroup(job, signal) {
+    if (!job.child.pid) return;
+    try { process.kill(-job.child.pid, signal); }
+    catch (error) { if (error.code !== 'ESRCH') throw error; }
+  }
+
+  // Console programs ignore taskkill's polite close request, so the tree is
+  // stopped forcefully. It only reaches processes whose parent is still alive.
+  killTree(job) {
+    if (!job.child.pid) return Promise.resolve();
+    const taskkill = path.win32.join(this.env.SystemRoot || this.env.SYSTEMROOT || 'C:\\Windows', 'System32', 'taskkill.exe');
+    return new Promise(resolve => {
+      const killer = spawn(taskkill, ['/PID', String(job.child.pid), '/T', '/F'], { stdio: 'ignore', windowsHide: true });
+      killer.once('error', resolve); killer.once('close', resolve);
+    });
+  }
+
+  async stop(id) {
+    const job = this.job(id);
+    if (job.ended) return this.status(id);
+    if (this.platform === 'win32') {
+      await this.killTree(job);
+      await Promise.race([job.done, delay(1500)]);
+      if (!job.ended) job.child.kill();
+      await Promise.race([job.done, delay(1000)]);
+      return this.status(id);
+    }
+    // Kill the owned group even if its parent shell has already exited.
+    this.killGroup(job, 'SIGTERM');
+    await Promise.race([job.done, delay(1500)]);
+    this.killGroup(job, 'SIGKILL');
+    await Promise.race([job.done, delay(1000)]);
+    return this.status(id);
+  }
+
+  list() {
+    return [...this.jobs.keys()].map(id => {
+      const job = this.jobs.get(id);
+      return { ...this.meta(id), stdout_characters: job.stdout.dropped + job.stdout.text.length,
+        stderr_characters: job.stderr.dropped + job.stderr.text.length };
+    });
+  }
+
+  async close() { await Promise.allSettled([...this.jobs.keys()].map(id => this.stop(id))); }
+}