152 lines
5.3 KiB
JavaScript
Executable File
152 lines
5.3 KiB
JavaScript
Executable File
#!/usr/bin/env node
|
||
|
||
/**
|
||
* TaskPing PTY Relay 启动脚本
|
||
* 启动基于 node-pty 的邮件命令中继服务
|
||
*/
|
||
|
||
const { spawn } = require('child_process');
|
||
const path = require('path');
|
||
const fs = require('fs');
|
||
|
||
// 检查环境配置
|
||
function checkConfig() {
|
||
const envPath = path.join(__dirname, '.env');
|
||
|
||
if (!fs.existsSync(envPath)) {
|
||
console.error('❌ 错误: 未找到 .env 配置文件');
|
||
console.log('\n请先复制 .env.example 到 .env 并配置您的邮件信息:');
|
||
console.log(' cp .env.example .env');
|
||
console.log(' 然后编辑 .env 文件填入您的邮件配置\n');
|
||
process.exit(1);
|
||
}
|
||
|
||
// 加载环境变量
|
||
require('dotenv').config();
|
||
|
||
// 检查必需的配置
|
||
const required = ['IMAP_HOST', 'IMAP_USER', 'IMAP_PASS'];
|
||
const missing = required.filter(key => !process.env[key]);
|
||
|
||
if (missing.length > 0) {
|
||
console.error('❌ 错误: 缺少必需的环境变量:');
|
||
missing.forEach(key => console.log(` - ${key}`));
|
||
console.log('\n请编辑 .env 文件并填入所有必需的配置\n');
|
||
process.exit(1);
|
||
}
|
||
|
||
console.log('✅ 配置检查通过');
|
||
console.log(`📧 IMAP服务器: ${process.env.IMAP_HOST}`);
|
||
console.log(`👤 邮件账号: ${process.env.IMAP_USER}`);
|
||
console.log(`🔒 白名单发件人: ${process.env.ALLOWED_SENDERS || '(未设置，将接受所有邮件)'}`);
|
||
console.log(`💾 会话存储路径: ${process.env.SESSION_MAP_PATH || '(使用默认路径)'}`);
|
||
console.log('');
|
||
}
|
||
|
||
// 创建会话示例
|
||
function createExampleSession() {
|
||
const sessionMapPath = process.env.SESSION_MAP_PATH || path.join(__dirname, 'src/data/session-map.json');
|
||
const sessionDir = path.dirname(sessionMapPath);
|
||
|
||
// 确保目录存在
|
||
if (!fs.existsSync(sessionDir)) {
|
||
fs.mkdirSync(sessionDir, { recursive: true });
|
||
}
|
||
|
||
// 如果会话文件不存在，创建一个示例
|
||
if (!fs.existsSync(sessionMapPath)) {
|
||
const exampleToken = 'TEST123';
|
||
const exampleSession = {
|
||
[exampleToken]: {
|
||
type: 'pty',
|
||
createdAt: Math.floor(Date.now() / 1000),
|
||
expiresAt: Math.floor((Date.now() + 24 * 60 * 60 * 1000) / 1000), // 24小时后过期
|
||
cwd: process.cwd(),
|
||
description: '测试会话 - 发送邮件时主题包含 [TaskPing #TEST123]'
|
||
}
|
||
};
|
||
|
||
fs.writeFileSync(sessionMapPath, JSON.stringify(exampleSession, null, 2));
|
||
console.log(`📝 已创建示例会话文件: ${sessionMapPath}`);
|
||
console.log(`🔑 测试Token: ${exampleToken}`);
|
||
console.log(' 发送测试邮件时，主题中包含: [TaskPing #TEST123]');
|
||
console.log('');
|
||
}
|
||
}
|
||
|
||
// 启动服务
|
||
function startService() {
|
||
console.log('🚀 正在启动 TaskPing PTY Relay 服务...\n');
|
||
|
||
const relayPath = path.join(__dirname, 'src/relay/relay-pty.js');
|
||
|
||
// 使用 node 直接运行，这样可以看到完整的日志输出
|
||
const relay = spawn('node', [relayPath], {
|
||
stdio: 'inherit',
|
||
env: {
|
||
...process.env,
|
||
INJECTION_MODE: 'pty'
|
||
}
|
||
});
|
||
|
||
// 处理退出
|
||
process.on('SIGINT', () => {
|
||
console.log('\n⏹️ 正在停止服务...');
|
||
relay.kill('SIGINT');
|
||
process.exit(0);
|
||
});
|
||
|
||
relay.on('error', (error) => {
|
||
console.error('❌ 启动失败:', error.message);
|
||
process.exit(1);
|
||
});
|
||
|
||
relay.on('exit', (code, signal) => {
|
||
if (signal) {
|
||
console.log(`\n服务已停止 (信号: ${signal})`);
|
||
} else if (code !== 0) {
|
||
console.error(`\n服务异常退出 (代码: ${code})`);
|
||
process.exit(code);
|
||
}
|
||
});
|
||
}
|
||
|
||
// 显示使用说明
|
||
function showInstructions() {
|
||
console.log('📖 使用说明:');
|
||
console.log('1. 在 Claude Code 中执行任务时，会发送包含 Token 的提醒邮件');
|
||
console.log('2. 回复该邮件，内容为要执行的命令');
|
||
console.log('3. 支持的命令格式:');
|
||
console.log(' - 直接输入命令文本');
|
||
console.log(' - 使用 CMD: 前缀，如 "CMD: 继续"');
|
||
console.log(' - 使用代码块包裹，如:');
|
||
console.log(' ```');
|
||
console.log(' 你的命令');
|
||
console.log(' ```');
|
||
console.log('4. 系统会自动提取命令并注入到对应的 Claude Code 会话中');
|
||
console.log('\n⌨️ 按 Ctrl+C 停止服务\n');
|
||
console.log('━'.repeat(60) + '\n');
|
||
}
|
||
|
||
// 主函数
|
||
function main() {
|
||
console.log('╔══════════════════════════════════════════════════════════╗');
|
||
console.log('║ TaskPing PTY Relay Service ║');
|
||
console.log('║ 邮件命令中继服务 - 基于 node-pty 的 PTY 模式 ║');
|
||
console.log('╚══════════════════════════════════════════════════════════╝\n');
|
||
|
||
// 检查配置
|
||
checkConfig();
|
||
|
||
// 创建示例会话
|
||
createExampleSession();
|
||
|
||
// 显示使用说明
|
||
showInstructions();
|
||
|
||
// 启动服务
|
||
startService();
|
||
}
|
||
|
||
// 运行
|
||
main(); |