@f2a/network 0.1.2 → 0.2.0

package/src/daemon/control-server.test.ts

@@ -1,216 +0,0 @@
-import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest';
-import { ControlServer } from './control-server.js';
-import { TokenManager } from '../core/token-manager.js';
-
-// Track mock server instances
-let lastMockServer: any = null;
-
-const TEST_TOKEN = 'test-token-12345';
-
-vi.mock('http', () => ({
-  createServer: vi.fn((handler) => {
-    lastMockServer = {
-      listen: vi.fn((port, callback) => {
-        if (callback) callback();
-        return { port };
-      }),
-      close: vi.fn((callback) => {
-        if (callback) callback();
-      }),
-      on: vi.fn(),
-      _handler: handler
-    };
-    return lastMockServer;
-  })
-}));
-
-// Mock TokenManager
-vi.mock('../core/token-manager', () => ({
-  TokenManager: vi.fn().mockImplementation(() => ({
-    getToken: vi.fn().mockReturnValue(TEST_TOKEN),
-    verifyToken: vi.fn((token) => token === TEST_TOKEN),
-    getTokenPath: vi.fn().mockReturnValue('/mock/path'),
-    logTokenUsage: vi.fn() // 添加审计日志方法
-  }))
-}));
-
-describe('ControlServer', () => {
-  let mockF2A: any;
-  let server: ControlServer;
-
-  beforeEach(() => {
-    vi.clearAllMocks();
-    lastMockServer = null;
-    
-    mockF2A = {
-      peerId: 'test-peer-id',
-      agentInfo: { displayName: 'Test Agent' },
-      getConnectedPeers: vi.fn().mockReturnValue([
-        { peerId: 'peer1', displayName: 'Peer 1' }
-      ]),
-      discoverAgents: vi.fn().mockResolvedValue([
-        { peerId: 'agent1', displayName: 'Agent 1' }
-      ])
-    };
-    
-    server = new ControlServer(mockF2A, 9001);
-  });
-
-  afterEach(() => {
-    server.stop();
-  });
-
-  describe('start/stop', () => {
-    it('should start server on specified port', async () => {
-      await server.start();
-      expect(lastMockServer).not.toBeNull();
-      expect(lastMockServer.listen).toHaveBeenCalledWith(9001, expect.any(Function));
-    });
-
-    it('should stop server gracefully', async () => {
-      await server.start();
-      server.stop();
-      expect(lastMockServer.close).toHaveBeenCalled();
-    });
-  });
-
-  describe('request handling', () => {
-    const createMockReq = (method: string, body?: object, headers?: Record<string, string>) => ({
-      method,
-      headers: {
-        'x-f2a-token': TEST_TOKEN,
-        ...headers
-      },
-      socket: { remoteAddress: '127.0.0.1' },
-      on: vi.fn((event, callback) => {
-        if (event === 'data' && body) {
-          callback(Buffer.from(JSON.stringify(body)));
-        }
-        if (event === 'end') {
-          callback();
-        }
-      })
-    });
-
-    const createMockRes = () => ({
-      writeHead: vi.fn(),
-      end: vi.fn(),
-      setHeader: vi.fn()
-    });
-
-    it('should handle OPTIONS request for CORS', async () => {
-      await server.start();
-      
-      const handler = lastMockServer._handler;
-      const req = createMockReq('OPTIONS');
-      const res = createMockRes();
-      
-      handler(req, res);
-      
-      expect(res.writeHead).toHaveBeenCalledWith(200);
-      expect(res.end).toHaveBeenCalled();
-    });
-
-    it('should reject non-POST methods', async () => {
-      await server.start();
-      
-      const handler = lastMockServer._handler;
-      const req = createMockReq('GET');
-      const res = createMockRes();
-      
-      handler(req, res);
-      
-      expect(res.writeHead).toHaveBeenCalledWith(405);
-    });
-
-    it('should handle status command', async () => {
-      await server.start();
-      
-      const handler = lastMockServer._handler;
-      const req = createMockReq('POST', { action: 'status' });
-      const res = createMockRes();
-      
-      handler(req, res);
-      
-      await new Promise(resolve => setTimeout(resolve, 10));
-      
-      expect(res.writeHead).toHaveBeenCalledWith(200);
-      const responseData = JSON.parse(res.end.mock.calls[0][0]);
-      expect(responseData.success).toBe(true);
-      expect(responseData.peerId).toBe('test-peer-id');
-    });
-
-    it('should handle peers command', async () => {
-      await server.start();
-      
-      const handler = lastMockServer._handler;
-      const req = createMockReq('POST', { action: 'peers' });
-      const res = createMockRes();
-      
-      handler(req, res);
-      
-      await new Promise(resolve => setTimeout(resolve, 10));
-      
-      expect(res.writeHead).toHaveBeenCalledWith(200);
-      const responseData = JSON.parse(res.end.mock.calls[0][0]);
-      expect(responseData.success).toBe(true);
-      expect(responseData.peers).toHaveLength(1);
-    });
-
-    it('should handle discover command', async () => {
-      await server.start();
-      
-      const handler = lastMockServer._handler;
-      const req = createMockReq('POST', { action: 'discover', capability: 'test' });
-      const res = createMockRes();
-      
-      handler(req, res);
-      
-      await new Promise(resolve => setTimeout(resolve, 10));
-      
-      expect(res.writeHead).toHaveBeenCalledWith(200);
-      const responseData = JSON.parse(res.end.mock.calls[0][0]);
-      expect(responseData.success).toBe(true);
-    });
-
-    it('should handle unknown commands', async () => {
-      await server.start();
-      
-      const handler = lastMockServer._handler;
-      const req = createMockReq('POST', { action: 'unknown' });
-      const res = createMockRes();
-      
-      handler(req, res);
-      
-      await new Promise(resolve => setTimeout(resolve, 10));
-      
-      expect(res.writeHead).toHaveBeenCalledWith(400);
-    });
-
-    it('should handle invalid JSON', async () => {
-      await server.start();
-      
-      const handler = lastMockServer._handler;
-      const req = {
-        method: 'POST',
-        headers: { 'x-f2a-token': TEST_TOKEN },
-        socket: { remoteAddress: '127.0.0.1' },
-        on: vi.fn((event, callback) => {
-          if (event === 'data') {
-            callback(Buffer.from('invalid json'));
-          }
-          if (event === 'end') {
-            callback();
-          }
-        })
-      };
-      const res = createMockRes();
-      
-      handler(req, res);
-      
-      await new Promise(resolve => setTimeout(resolve, 10));
-      
-      expect(res.writeHead).toHaveBeenCalledWith(400);
-    });
-  });
-});

package/src/daemon/control-server.ts

@@ -1,292 +0,0 @@
-/**
- * HTTP 控制服务器
- * 接收 CLI 命令 - P2P 版本
- */
-
-import { createServer, Server, IncomingMessage, ServerResponse } from 'http';
-import { F2A } from '../core/f2a.js';
-import { TokenManager } from '../core/token-manager.js';
-import { Logger } from '../utils/logger.js';
-import { RateLimiter } from '../utils/rate-limiter.js';
-
-export interface ControlServerOptions {
-  port: number;
-  token?: string;
-  /** 允许的 CORS 来源列表，默认为 ['http://localhost'] */
-  allowedOrigins?: string[];
-}
-
-/** 默认允许的 CORS 来源 */
-const DEFAULT_ALLOWED_ORIGINS = ['http://localhost'];
-
-export class ControlServer {
-  private server?: Server;
-  private f2a: F2A;
-  private port: number;
-  private tokenManager: TokenManager;
-  private logger: Logger;
-  private rateLimiter: RateLimiter;
-  private allowedOrigins: string[];
-
-  constructor(f2a: F2A, port: number, tokenManager?: TokenManager, options?: ControlServerOptions) {
-    this.f2a = f2a;
-    this.port = port;
-    this.tokenManager = tokenManager || new TokenManager();
-    this.logger = new Logger({ component: 'ControlServer' });
-    // 速率限制: 每分钟最多 60 个请求
-    this.rateLimiter = new RateLimiter({ maxRequests: 60, windowMs: 60000 });
-    // CORS 配置：优先使用传入的 allowedOrigins，否则使用默认值
-    this.allowedOrigins = options?.allowedOrigins ?? DEFAULT_ALLOWED_ORIGINS;
-  }
-
-  /**
-   * 启动控制服务器
-   */
-  start(): Promise<void> {
-    return new Promise((resolve, reject) => {
-      this.server = createServer((req, res) => {
-        this.handleRequest(req, res);
-      });
-
-      this.server.on('error', reject);
-
-      this.server.listen(this.port, () => {
-        this.logger.info('Listening', { port: this.port });
-        resolve();
-      });
-    });
-  }
-
-  /**
-   * 停止控制服务器
-   */
-  stop(): void {
-    if (this.server) {
-      this.server.close();
-      this.server = undefined;
-    }
-    // 清理速率限制器资源
-    this.rateLimiter.stop();
-    this.logger.info('Stopped');
-  }
-
-  /**
-   * 从 Authorization header 提取 Bearer token
-   */
-  private extractBearerToken(authHeader: string | undefined): string | undefined {
-    if (!authHeader) return undefined;
-    const match = authHeader.match(/^Bearer\s+(.+)$/i);
-    return match ? match[1] : undefined;
-  }
-
-  /**
-   * 处理请求
-   */
-  private handleRequest(req: IncomingMessage, res: ServerResponse): void {
-// 设置 CORS - 使用配置的允许来源
-    const origin = req.headers.origin;
-    const allowOrigin = origin && this.allowedOrigins.includes(origin) 
-      ? origin 
-      : this.allowedOrigins[0];
-    
-    res.setHeader('Access-Control-Allow-Origin', allowOrigin);
-    res.setHeader('Access-Control-Allow-Methods', 'GET, POST, OPTIONS');
-    res.setHeader('Access-Control-Allow-Headers', 'Content-Type, X-F2A-Token');
-
-    if (req.method === 'OPTIONS') {
-      res.writeHead(200);
-      res.end();
-      return;
-    }
-
-    // 健康检查端点 (不需要认证)
-    if (req.method === 'GET' && req.url === '/health') {
-      res.writeHead(200);
-      res.end(JSON.stringify({ status: 'ok', peerId: this.f2a.peerId }));
-      return;
-    }
-
-    // GET /status - 获取状态 (需要认证)
-    if (req.method === 'GET' && req.url === '/status') {
-      const clientIp = req.socket.remoteAddress || 'unknown';
-      if (!this.rateLimiter.allowRequest(clientIp)) {
-        res.writeHead(429);
-        res.end(JSON.stringify({ success: false, error: 'Too many requests' }));
-        return;
-      }
-      // 支持 X-F2A-Token 或 Authorization: Bearer xxx
-      const token = req.headers['x-f2a-token'] as string | undefined 
-        || this.extractBearerToken(req.headers.authorization);
-      if (!this.tokenManager.verifyToken(token)) {
-        res.writeHead(401);
-        res.end(JSON.stringify({ success: false, error: 'Unauthorized' }));
-        return;
-      }
-      res.writeHead(200);
-      res.end(JSON.stringify({
-        success: true,
-        peerId: this.f2a.peerId,
-        multiaddrs: this.f2a.agentInfo.multiaddrs || []
-      }));
-      return;
-    }
-
-    // GET /peers - 获取已知的 Peers (需要认证)
-    if (req.method === 'GET' && req.url === '/peers') {
-      const clientIp = req.socket.remoteAddress || 'unknown';
-      if (!this.rateLimiter.allowRequest(clientIp)) {
-        res.writeHead(429);
-        res.end(JSON.stringify({ success: false, error: 'Too many requests' }));
-        return;
-      }
-      // 支持 X-F2A-Token 或 Authorization: Bearer xxx
-      const token = req.headers['x-f2a-token'] as string | undefined 
-        || this.extractBearerToken(req.headers.authorization);
-      if (!this.tokenManager.verifyToken(token)) {
-        res.writeHead(401);
-        res.end(JSON.stringify({ success: false, error: 'Unauthorized' }));
-        return;
-      }
-      // 返回所有已知的节点（包括已断开但已发现的）
-      const peers = this.f2a.getAllPeers();
-      res.writeHead(200);
-      res.end(JSON.stringify(peers));
-      return;
-    }
-
-    if (req.method !== 'POST') {
-      res.writeHead(405);
-      res.end(JSON.stringify({ 
-        success: false, 
-        error: 'Method not allowed',
-        code: 'METHOD_NOT_ALLOWED'
-      }));
-      return;
-    }
-    const clientIp = req.socket.remoteAddress || 'unknown';
-    if (!this.rateLimiter.allowRequest(clientIp)) {
-      this.logger.warn('Rate limit exceeded', { clientIp });
-      res.writeHead(429);
-      res.end(JSON.stringify({
-        success: false,
-        error: 'Too many requests',
-        code: 'RATE_LIMIT_EXCEEDED'
-      }));
-      return;
-    }
-
-    // 验证 Token
-    const token = req.headers['x-f2a-token'] as string | undefined;
-    
-    if (!this.tokenManager.verifyToken(token)) {
-      // 记录失败的验证尝试
-      this.tokenManager.logTokenUsage({
-        ip: clientIp,
-        action: 'auth',
-        success: false
-      });
-      
-      this.logger.warn('Unauthorized request', { clientIp });
-      res.writeHead(401);
-      res.end(JSON.stringify({
-        success: false,
-        error: 'Unauthorized: Invalid or missing token',
-        code: 'UNAUTHORIZED'
-      }));
-      return;
-    }
-    
-    // 记录成功的验证
-    this.tokenManager.logTokenUsage({
-      ip: clientIp,
-      action: 'auth',
-      success: true
-    });
-
-    let body = '';
-    req.on('data', chunk => body += chunk);
-    req.on('end', () => {
-      this.processCommand(body, res);
-    });
-  }
-
-  /**
-   * 处理命令
-   */
-  private processCommand(body: string, res: ServerResponse): void {
-    try {
-      const command = JSON.parse(body);
-      
-      switch (command.action) {
-        case 'status':
-          this.handleStatus(res);
-          break;
-        case 'peers':
-          this.handlePeers(res);
-          break;
-        case 'discover':
-          this.handleDiscover(command.capability, res);
-          break;
-        default:
-          res.writeHead(400);
-          res.end(JSON.stringify({ 
-            success: false, 
-            error: 'Unknown action',
-            code: 'UNKNOWN_ACTION'
-          }));
-      }
-    } catch {
-      res.writeHead(400);
-      res.end(JSON.stringify({ 
-        success: false, 
-        error: 'Invalid JSON',
-        code: 'INVALID_JSON'
-      }));
-    }
-  }
-
-  /**
-   * 获取状态
-   */
-  private handleStatus(res: ServerResponse): void {
-    res.writeHead(200);
-    res.end(JSON.stringify({
-      success: true,
-      peerId: this.f2a.peerId,
-      agentInfo: this.f2a.agentInfo
-    }));
-  }
-
-  /**
-   * 获取已连接的 Peers
-   */
-  private handlePeers(res: ServerResponse): void {
-    const peers = this.f2a.getConnectedPeers();
-    res.writeHead(200);
-    res.end(JSON.stringify({
-      success: true,
-      peers
-    }));
-  }
-
-  /**
-   * 发现 Agents
-   */
-  private async handleDiscover(capability: string | undefined, res: ServerResponse): Promise<void> {
-    try {
-      const agents = await this.f2a.discoverAgents(capability);
-      res.writeHead(200);
-      res.end(JSON.stringify({
-        success: true,
-        agents
-      }));
-    } catch (error) {
-      res.writeHead(500);
-      res.end(JSON.stringify({
-        success: false,
-        error: error instanceof Error ? error.message : String(error),
-        code: 'DISCOVER_FAILED'
-      }));
-    }
-  }
-}

package/src/daemon/index.test.ts

@@ -1,85 +0,0 @@
-import { describe, it, expect, vi, beforeEach } from 'vitest';
-import { F2ADaemon } from './index.js';
-
-// Mock dependencies
-vi.mock('../core/f2a', () => ({
-  F2A: {
-    create: vi.fn().mockResolvedValue({
-      start: vi.fn().mockResolvedValue({ success: true }),
-      stop: vi.fn(),
-      peerId: 'test-peer-id'
-    })
-  }
-}));
-
-vi.mock('./control-server', () => ({
-  ControlServer: vi.fn().mockImplementation(() => ({
-    start: vi.fn().mockResolvedValue(undefined),
-    stop: vi.fn()
-  }))
-}));
-
-describe('F2ADaemon', () => {
-  let daemon: F2ADaemon;
-
-  beforeEach(() => {
-    daemon = new F2ADaemon({
-      displayName: 'Test Daemon'
-    });
-  });
-
-  describe('lifecycle', () => {
-    it('should create daemon with options', () => {
-      expect(daemon).toBeDefined();
-      expect(daemon.isRunning()).toBe(false);
-    });
-
-    it('should start successfully', async () => {
-      await daemon.start();
-      expect(daemon.isRunning()).toBe(true);
-    });
-
-    it('should not start twice', async () => {
-      await daemon.start();
-      expect(daemon.isRunning()).toBe(true);
-
-      await expect(daemon.start()).rejects.toThrow('Daemon already running');
-    });
-
-    it('should stop gracefully', async () => {
-      await daemon.start();
-      await daemon.stop();
-      expect(daemon.isRunning()).toBe(false);
-    });
-
-    it('should handle stop before start', async () => {
-      await daemon.stop(); // Should not throw
-      expect(daemon.isRunning()).toBe(false);
-    });
-  });
-
-  describe('getters', () => {
-    it('should return F2A instance', async () => {
-      await daemon.start();
-      const f2a = daemon.getF2A();
-      expect(f2a).toBeDefined();
-    });
-
-    it('should return undefined when not running', () => {
-      const f2a = daemon.getF2A();
-      expect(f2a).toBeUndefined();
-    });
-  });
-
-  describe('options', () => {
-    it('should use default control port', () => {
-      const defaultDaemon = new F2ADaemon();
-      expect(defaultDaemon).toBeDefined();
-    });
-
-    it('should accept custom control port', () => {
-      const customDaemon = new F2ADaemon({ controlPort: 8080 });
-      expect(customDaemon).toBeDefined();
-    });
-  });
-});

package/src/daemon/index.ts

@@ -1,89 +0,0 @@
-/**
- * F2A Daemon
- * 后台服务主入口 - P2P 版本
- */
-
-import { F2A } from '../core/f2a.js';
-import { ControlServer } from './control-server.js';
-import { F2AOptions, WebhookConfig } from '../types/index.js';
-
-export interface DaemonOptions extends F2AOptions {
-  webhook?: WebhookConfig;
-  controlPort?: number;
-}
-
-export class F2ADaemon {
-  private options: DaemonOptions;
-  private f2a?: F2A;
-  private controlServer?: ControlServer;
-  private running: boolean = false;
-
-  constructor(options: DaemonOptions = {}) {
-    this.options = {
-      controlPort: 9001,
-      ...options
-    };
-  }
-
-  /**
-   * 启动 Daemon
-   */
-  async start(): Promise<void> {
-    if (this.running) {
-      throw new Error('Daemon already running');
-    }
-
-    console.log('[Daemon] Starting F2A Daemon...');
-
-    // 创建并启动 F2A
-    this.f2a = await F2A.create(this.options);
-    const result = await this.f2a.start();
-    
-    if (!result.success) {
-      const errorData = (result as { error: unknown }).error;
-      const errorMsg = typeof errorData === 'string' 
-        ? errorData 
-        : JSON.stringify(errorData);
-      throw new Error(`Failed to start F2A: ${errorMsg}`);
-    }
-
-    // 启动控制服务器
-    this.controlServer = new ControlServer(this.f2a, this.options.controlPort!);
-    await this.controlServer.start();
-
-    this.running = true;
-    console.log(`[Daemon] F2A Daemon started with peerId: ${this.f2a.peerId.slice(0, 16)}...`);
-  }
-
-  /**
-   * 停止 Daemon
-   */
-  async stop(): Promise<void> {
-    if (!this.running) return;
-
-    console.log('[Daemon] Stopping F2A Daemon...');
-
-    await this.controlServer?.stop();
-    await this.f2a?.stop();
-
-    this.running = false;
-    console.log('[Daemon] F2A Daemon stopped');
-  }
-
-  /**
-   * 获取 F2A 实例
-   */
-  getF2A(): F2A | undefined {
-    return this.f2a;
-  }
-
-  /**
-   * 是否运行中
-   */
-  isRunning(): boolean {
-    return this.running;
-  }
-}
-
-// 默认导出
-export default F2ADaemon;