@wendongfly/myhi 1.0.1

package/bin/myhi.js

@@ -0,0 +1,143 @@
+#!/usr/bin/env node
+import { spawn } from 'child_process';
+import { fileURLToPath } from 'url';
+import { dirname, join } from 'path';
+import { mkdirSync, openSync, readFileSync, writeFileSync, unlinkSync } from 'fs';
+import { homedir } from 'os';
+
+const __filename = fileURLToPath(import.meta.url);
+const __dirname = dirname(__filename);
+const cmd = process.argv[2];
+
+const configDir = join(homedir(), '.myhi');
+const pidFile = join(configDir, 'daemon.pid');
+const logFile = join(configDir, 'daemon.log');
+
+// ── 工具函数 ──────────────────────────────────────────────
+
+function readPid() {
+  try { return parseInt(readFileSync(pidFile, 'utf8').trim(), 10); }
+  catch { return null; }
+}
+
+function isRunning(pid) {
+  try { process.kill(pid, 0); return true; }
+  catch { return false; }
+}
+
+function stopDaemon() {
+  const pid = readPid();
+  if (!pid) { console.log('[myhi] 没有正在运行的后台进程'); return false; }
+  if (!isRunning(pid)) {
+    try { unlinkSync(pidFile); } catch {}
+    console.log('[myhi] 进程已不存在，已清理 PID 文件');
+    return false;
+  }
+  process.kill(pid, 'SIGTERM');
+  try { unlinkSync(pidFile); } catch {}
+  console.log(`[myhi] 已停止后台进程 (PID ${pid})`);
+  return true;
+}
+
+function startDaemon() {
+  mkdirSync(configDir, { recursive: true });
+  const out = openSync(logFile, 'a');
+  const child = spawn(process.execPath, [join(__dirname, '..', 'dist', 'index.js')], {
+    detached: true,
+    stdio: ['ignore', out, out],
+    cwd: process.cwd(),
+    env: { ...process.env },
+  });
+  writeFileSync(pidFile, String(child.pid));
+  child.unref();
+  console.log(`[myhi] 已在后台启动 (PID ${child.pid})`);
+  console.log(`[myhi] 日志: ${logFile}`);
+  console.log(`[myhi] 停止: myhi stop`);
+}
+
+// ── 命令分发 ──────────────────────────────────────────────
+
+if (cmd === 'attach') {
+  process.argv.splice(2, 1);
+  await import('../dist/attach.js');
+
+} else if (cmd === 'stop') {
+  stopDaemon();
+
+} else if (cmd === 'restart') {
+  stopDaemon();
+  await new Promise(r => setTimeout(r, 500));
+  startDaemon();
+  process.exit(0);
+
+} else if (cmd === 'status') {
+  const pid = readPid();
+  if (pid && isRunning(pid)) {
+    console.log(`[myhi] 运行中 (PID ${pid})`);
+  } else {
+    console.log('[myhi] 未运行');
+  }
+
+} else if (cmd === 'log' || cmd === 'logs') {
+  try {
+    const content = readFileSync(logFile, 'utf8');
+    const lines = content.split('\n');
+    const n = parseInt(process.argv[3], 10) || 50;
+    console.log(lines.slice(-n).join('\n'));
+  } catch {
+    console.log('[myhi] 日志文件不存在');
+  }
+
+} else if (cmd === 'help' || cmd === '--help' || cmd === '-h') {
+  console.log(`
+myhi v1.0.0 — 基于 Web 的终端共享工具
+通过局域网或 Tailscale 在手机/平板上控制电脑终端，无需中继服务器。
+
+服务器：
+  myhi                 前台启动服务器
+  myhi start           前台启动服务器
+  myhi -d, --daemon    后台启动服务器（日志写入 ~/.myhi/daemon.log）
+  myhi stop            停止后台服务器
+  myhi restart         重启后台服务器
+  myhi status          查看后台运行状态
+  myhi log [N]         查看后台日志（最后 N 行，默认 50）
+
+远程连接：
+  myhi attach          列出活跃会话，交互式选择后附加
+  myhi attach <id>     直接附加到指定会话
+  myhi attach --new    创建新会话并附加
+
+快捷键（attach 模式）：
+  Ctrl+]               分离终端（退出但不关闭会话）
+
+环境变量：
+  PORT                 监听端口（默认 3000）
+  HOST                 监听地址（默认 0.0.0.0）
+  HTTPS_PORT           HTTPS 端口（默认 3443）
+  MYHI_AUTO_ATTACH     启动时自动创建会话（默认 1，设 0 关闭）
+  MYHI_CWD             会话工作目录（默认当前目录）
+  MYHI_SERVER          attach 连接的服务器地址（默认 http://localhost:3000）
+
+配置目录：~/.myhi/
+  token                认证令牌（自动生成）
+  password             登录密码（自动生成，可手动修改）
+  roles.json           多用户角色配置（可选）
+  sessions.json        会话持久化数据
+  daemon.pid           后台进程 PID
+  daemon.log           后台运行日志
+
+示例：
+  myhi                          # 启动后扫描二维码即可在手机上操作
+  PORT=8080 myhi -d             # 8080 端口后台启动
+  MYHI_CWD=/project myhi        # 指定会话工作目录
+  myhi attach                   # 从另一台机器连接到会话
+`);
+
+} else if (cmd === '-d' || cmd === '--daemon' || cmd === 'daemon') {
+  startDaemon();
+  process.exit(0);
+
+} else {
+  // 默认 / 'start'：前台运行
+  await import('../dist/index.js');
+}

package/dist/attach.js

@@ -0,0 +1,188 @@
+#!/usr/bin/env node
+/**
+ * myhi attach — 将本地终端连接到运行中的 PTY 会话
+ *
+ * 用法：
+ *   node src/attach.js          # 列出会话，然后选择
+ *   node src/attach.js <id>     # 直接附加到指定会话
+ */
+import { createRequire } from 'module';
+import { readFileSync } from 'fs';
+import { join } from 'path';
+import { homedir } from 'os';
+import { createInterface } from 'readline';
+
+const require = createRequire(import.meta.url);
+const { io }  = require('socket.io-client');
+
+const SERVER  = process.env.MYHI_SERVER || 'http://localhost:3000';
+const TOKEN   = readFileSync(join(homedir(), '.myhi', 'token'), 'utf8').trim();
+
+function cleanup(socket) {
+  try { process.stdin.setRawMode(false); } catch {}
+  process.stdin.pause();
+  socket.disconnect();
+}
+
+function attach(socket, sessionId) {
+  socket.emit('join', sessionId);
+
+  socket.on('joined', (session) => {
+    process.stderr.write(`\r\n[myhi] 已附加到 "${session.title}" (${sessionId})\r\n`);
+    process.stderr.write('[myhi] 按 Ctrl+] 分离\r\n\r\n');
+
+    // 本地终端自动获取控制权
+    socket.emit('take-control', { sessionId });
+
+    try { process.stdin.setRawMode(true); } catch {}
+    process.stdin.resume();
+    process.stdin.setEncoding('binary');
+
+    // 标准输入 → PTY
+    process.stdin.on('data', (data) => {
+      // Ctrl+] (0x1d) = 分离
+      if (data === '\x1d') {
+        process.stderr.write('\r\n[myhi] 已分离\r\n');
+        cleanup(socket);
+        process.exit(0);
+      }
+      socket.emit('input', data);
+    });
+
+    // PTY → 标准输出
+    socket.on('output', (data) => {
+      process.stdout.write(data, 'binary');
+    });
+
+    // 仅在本地窗口实际调整大小时同步尺寸，
+    // 避免触发 PTY 重绘导致输出重复
+    process.stdout.on('resize', () => {
+      socket.emit('resize', {
+        cols: process.stdout.columns || 80,
+        rows: process.stdout.rows    || 24,
+      });
+    });
+  });
+
+  // 控制权被拒绝
+  socket.on('control-denied', ({ reason }) => {
+    process.stderr.write(`\r\n[myhi] 获取控制权失败: ${reason}\r\n`);
+  });
+
+  // 控制权被其他用户获取
+  socket.on('control-changed', ({ holder, holderName }) => {
+    if (holder && holder !== socket.id) {
+      process.stderr.write(`\r\n[myhi] ${holderName || '其他用户'} 已获取控制权，当前为只读\r\n`);
+    } else if (!holder) {
+      process.stderr.write('\r\n[myhi] 控制权已释放\r\n');
+    }
+  });
+
+  socket.on('session-exit', ({ code }) => {
+    process.stderr.write(`\r\n[myhi] 会话已退出 (code ${code})\r\n`);
+    cleanup(socket);
+    process.exit(0);
+  });
+
+  socket.on('error', ({ message }) => {
+    process.stderr.write(`\r\n[myhi] 错误: ${message}\r\n`);
+    cleanup(socket);
+    process.exit(1);
+  });
+}
+
+async function pickSession(socket) {
+  return new Promise((resolve, reject) => {
+    socket.emit('list');
+    socket.once('sessions', (sessions) => {
+      const alive = sessions.filter(s => s.alive);
+      if (!alive.length) {
+        process.stderr.write('[myhi] 没有活跃的会话。\n');
+        cleanup(socket);
+        process.exit(0);
+      }
+
+      process.stdout.write('\n活跃会话:\n');
+      alive.forEach((s, i) => {
+        const viewers = s.viewers > 0 ? ` (${s.viewers} 人在线)` : '';
+        process.stdout.write(`  [${i + 1}] ${s.title}${viewers}  — ${s.id}\n`);
+      });
+      process.stdout.write('\n');
+
+      const rl = createInterface({ input: process.stdin, output: process.stdout });
+      rl.question('选择会话编号: ', (answer) => {
+        rl.close();
+        const idx = parseInt(answer, 10) - 1;
+        if (idx < 0 || idx >= alive.length) {
+          process.stderr.write('[myhi] 无效的选择。\n');
+          cleanup(socket);
+          process.exit(1);
+        }
+        resolve(alive[idx].id);
+      });
+    });
+  });
+}
+
+async function createAndAttach(socket, opts) {
+  return new Promise((resolve, reject) => {
+    socket.emit('create', opts, (res) => {
+      if (!res?.ok) {
+        process.stderr.write(`[myhi] 创建失败: ${res?.error || '未知错误'}\n`);
+        cleanup(socket);
+        process.exit(1);
+      }
+      process.stdout.write(`[myhi] 已创建会话 "${res.session.title}" — ${res.session.id}\n`);
+      resolve(res.session.id);
+    });
+  });
+}
+
+async function promptNew(socket) {
+  const rl = createInterface({ input: process.stdin, output: process.stdout });
+  const ask = (q) => new Promise((r) => rl.question(q, r));
+  const title   = (await ask('会话名称 [shell]: ')).trim() || 'shell';
+  const initCmd = (await ask('启动命令（可选）: ')).trim() || undefined;
+  rl.close();
+  return createAndAttach(socket, { title, initCmd });
+}
+
+// ── 主入口 ──────────────────────────────────────────────
+
+const socket = io(SERVER, {
+  transports: ['websocket'],
+  auth: { token: TOKEN },
+});
+
+socket.on('connect_error', (err) => {
+  process.stderr.write(`[myhi] 连接失败: ${err.message}\n`);
+  process.exit(1);
+});
+
+socket.on('connect', async () => {
+  const arg = process.argv[2];
+
+  if (arg === '--new') {
+    // --new [标题] [命令]  →  创建后附加
+    const title   = process.argv[3];
+    const initCmd = process.argv[4] || undefined;
+    const sessionId = title
+      ? await createAndAttach(socket, { title, initCmd })
+      : await promptNew(socket);
+    attach(socket, sessionId);
+  } else {
+    const sessionId = arg || await pickSession(socket);
+    attach(socket, sessionId);
+  }
+});
+
+// 退出时恢复终端状态
+for (const sig of ['SIGINT', 'SIGTERM']) {
+  process.on(sig, () => {
+    cleanup(socket);
+    process.exit(0);
+  });
+}
+process.on('exit', () => {
+  try { process.stdin.setRawMode(false); } catch {}
+});