protocol-proxy 1.0.0

package/server.js

@@ -0,0 +1,202 @@
+const express = require('express');
+const path = require('path');
+const cors = require('cors');
+const { exec } = require('child_process');
+const os = require('os');
+const configStore = require('./lib/config-store');
+const proxyManager = require('./lib/proxy-manager');
+
+const app = express();
+const PORT = process.env.ADMIN_PORT || 3000;
+
+function openBrowser(url) {
+  const platform = os.platform();
+  let command;
+  if (platform === 'win32') {
+    command = `start "" "${url}"`;
+  } else if (platform === 'darwin') {
+    command = `open "${url}"`;
+  } else {
+    command = `xdg-open "${url}"`;
+  }
+  exec(command, (err) => {
+    if (err) console.error('[Browser] 打开浏览器失败:', err.message);
+  });
+}
+
+app.use(cors());
+app.use(express.json());
+app.use(express.static(path.join(__dirname, 'public')));
+
+// ==================== 管理 API ====================
+
+// 获取所有代理配置
+app.get('/api/proxies', (req, res) => {
+  const proxies = configStore.getProxies().map(p => ({
+    ...p,
+    target: p.target ? {
+      ...p.target,
+      apiKey: p.target.apiKey ? '***' : '',
+    } : null,
+    running: proxyManager.isRunning(p.id),
+  }));
+  res.json(proxies);
+});
+
+// 获取单个代理配置（含敏感信息）
+app.get('/api/proxies/:id', (req, res) => {
+  const proxy = configStore.getProxyById(req.params.id);
+  if (!proxy) return res.status(404).json({ error: 'Proxy not found' });
+  res.json(proxy);
+});
+
+// 创建代理
+app.post('/api/proxies', async (req, res) => {
+  const { name, port, requireAuth, authToken, target } = req.body;
+
+  if (!name || !port || !target) {
+    return res.status(400).json({ error: 'name, port and target are required' });
+  }
+
+  const parsedPort = parseInt(port);
+
+  // 端口冲突校验
+  const existing = configStore.getProxies().find(p => p.port === parsedPort);
+  if (existing) {
+    return res.status(409).json({
+      error: `端口 ${parsedPort} 已被代理「${existing.name}」占用，请更换端口`,
+    });
+  }
+
+  const proxy = configStore.addProxy({
+    name,
+    port: parsedPort,
+    requireAuth: !!requireAuth,
+    authToken: authToken || null,
+    target,
+  });
+
+  try {
+    await proxyManager.startProxy(proxy);
+    res.status(201).json({ ...proxy, running: true });
+  } catch (err) {
+    // 启动失败，回滚已保存的配置
+    configStore.removeProxy(proxy.id);
+    res.status(500).json({ error: `代理启动失败: ${err.message}` });
+  }
+});
+
+// 更新代理
+app.put('/api/proxies/:id', async (req, res) => {
+  const { name, port, requireAuth, authToken, target } = req.body;
+  const existing = configStore.getProxyById(req.params.id);
+  if (!existing) return res.status(404).json({ error: 'Proxy not found' });
+
+  const updates = {};
+  if (name !== undefined) updates.name = name;
+  if (port !== undefined) updates.port = parseInt(port);
+  if (requireAuth !== undefined) updates.requireAuth = !!requireAuth;
+  if (authToken !== undefined) updates.authToken = authToken || null;
+  if (target !== undefined) updates.target = target;
+
+  // 端口变更时校验冲突
+  const needRestart = updates.port !== undefined && updates.port !== existing.port;
+  if (needRestart) {
+    const conflict = configStore.getProxies().find(p => p.id !== req.params.id && p.port === updates.port);
+    if (conflict) {
+      return res.status(409).json({
+        error: `端口 ${updates.port} 已被代理「${conflict.name}」占用，请更换端口`,
+      });
+    }
+  }
+
+  const updated = configStore.updateProxy(req.params.id, updates);
+
+  if (needRestart) {
+    try {
+      await proxyManager.restartProxy(updated);
+    } catch (err) {
+      return res.status(500).json({ error: `代理重启失败: ${err.message}` });
+    }
+  } else {
+    proxyManager.updateProxyConfig(updated);
+  }
+
+  res.json({ ...updated, running: proxyManager.isRunning(updated.id) });
+});
+
+// 删除代理
+app.delete('/api/proxies/:id', async (req, res) => {
+  const existing = configStore.getProxyById(req.params.id);
+  if (!existing) return res.status(404).json({ error: 'Proxy not found' });
+
+  await proxyManager.stopProxy(req.params.id);
+  configStore.removeProxy(req.params.id);
+  res.json({ success: true });
+});
+
+// 启动/停止代理
+app.post('/api/proxies/:id/start', async (req, res) => {
+  const proxy = configStore.getProxyById(req.params.id);
+  if (!proxy) return res.status(404).json({ error: 'Proxy not found' });
+
+  try {
+    await proxyManager.startProxy(proxy);
+    res.json({ success: true, running: true });
+  } catch (err) {
+    res.status(500).json({ error: 'Failed to start proxy', message: err.message });
+  }
+});
+
+app.post('/api/proxies/:id/stop', async (req, res) => {
+  await proxyManager.stopProxy(req.params.id);
+  res.json({ success: true, running: false });
+});
+
+// 获取运行状态
+app.get('/api/status', (req, res) => {
+  res.json({
+    running: proxyManager.getRunningPorts(),
+    total: configStore.getProxies().length,
+  });
+});
+
+// 前端首页
+app.get('/', (req, res) => {
+  res.sendFile(path.join(__dirname, 'public', 'index.html'));
+});
+
+// ==================== 启动 ====================
+
+async function init() {
+  // 启动所有已配置的代理
+  const proxies = configStore.getProxies();
+  for (const proxy of proxies) {
+    try {
+      await proxyManager.startProxy(proxy);
+    } catch (err) {
+      console.error(`[Init] Failed to start proxy ${proxy.name}:`, err.message);
+    }
+  }
+
+  app.listen(PORT, () => {
+    const adminUrl = `http://localhost:${PORT}`;
+    console.log(`[Admin] Management server running on ${adminUrl}`);
+    console.log(`[Admin] ${proxies.length} proxy config(s) loaded`);
+    openBrowser(adminUrl);
+  });
+}
+
+// 优雅关闭
+process.on('SIGINT', async () => {
+  console.log('\nShutting down...');
+  await proxyManager.stopAll();
+  process.exit(0);
+});
+
+process.on('SIGTERM', async () => {
+  await proxyManager.stopAll();
+  process.exit(0);
+});
+
+init();