Method connectWs

test/onebot11-api-test/core/ApiClient.ts:162–287

* 建立 WebSocket 连接 * @returns Promise that resolves when connected

()

   * @returns Promise that resolves when connected
   */
  private async connectWs(): Promise<void> {
    // 如果已经连接或正在连接，直接返回
    if (this.ws && this.ws.readyState === WebSocket.OPEN) {
      return;
    }

    if (this.wsConnecting) {
      // 等待连接完成
      return new Promise((resolve, reject) => {
        const checkInterval = setInterval(() => {
          if (!this.wsConnecting) {
            clearInterval(checkInterval);
            if (this.ws && this.ws.readyState === WebSocket.OPEN) {
              resolve();
            } else {
              reject(new NetworkError('WebSocket connection failed'));
            }
          }
        }, 100);

        // 10秒超时
        setTimeout(() => {
          clearInterval(checkInterval);
          reject(new TimeoutError('WebSocket connection timeout', 10000));
        }, 10000);
      });
    }

    this.wsConnecting = true;

    return new Promise((resolve, reject) => {
      try {
        // 将 http:// 或 https:// 转换为 ws:// 或 wss://
        let wsUrl = this.config.host;
        if (wsUrl.startsWith('http://')) {
          wsUrl = wsUrl.replace('http://', 'ws://');
        } else if (wsUrl.startsWith('https://')) {
          wsUrl = wsUrl.replace('https://', 'wss://');
        } else if (!wsUrl.startsWith('ws://') && !wsUrl.startsWith('wss://')) {
          wsUrl = 'ws://' + wsUrl;
        }

        // 创建 WebSocket 连接
        const headers: any = {};
        if (this.config.apiKey) {
          headers['Authorization'] = `Bearer ${this.config.apiKey}`;
        }

        this.ws = new WebSocket(wsUrl, { headers });

        // 连接成功
        this.ws.on('open', () => {
          this.wsConnecting = false;
          this.wsReconnectAttempts = 0;
          if (process.env.NODE_ENV !== 'test') {
            console.log(`WebSocket connected to ${wsUrl}`);
          }
          resolve();