worker.js

export default {
	async fetch(request, env, ctx) {
	  if (request.headers.get("Upgrade") !== "websocket") {
		return new Response("Expected WebSocket connection", { status: 400 });
	  }
  
	  const url = new URL(request.url);
	  const pathAndQuery = url.pathname + url.search;
	  const targetUrl = `wss://generativelanguage.googleapis.com${pathAndQuery}`;
	  
	  console.log('Target URL:', targetUrl);
  
	  const [client, proxy] = new WebSocketPair();
	  proxy.accept();
  
	  // 用于存储在连接建立前收到的消息
	  let pendingMessages = [];
  
	  const connectPromise = new Promise((resolve, reject) => {
		const targetWebSocket = new WebSocket(targetUrl);
  
		console.log('Initial targetWebSocket readyState:', targetWebSocket.readyState);
  
		targetWebSocket.addEventListener("open", () => {
		  console.log('Connected to target server');
		  console.log('targetWebSocket readyState after open:', targetWebSocket.readyState);
		  
		  // 连接建立后，发送所有待处理的消息
		  console.log(`Processing ${pendingMessages.length} pending messages`);
		  for (const message of pendingMessages) {
			try {
			  targetWebSocket.send(message);
			  console.log('Sent pending message:', message.slice(0, 100));
			} catch (error) {
			  console.error('Error sending pending message:', error);
			}
		  }
		  pendingMessages = []; // 清空待处理消息队列
		  resolve(targetWebSocket);
		});
  
		proxy.addEventListener("message", async (event) => {
		  console.log('Received message from client:', {
			dataPreview: typeof event.data === 'string' ? event.data.slice(0, 200) : 'Binary data',
			dataType: typeof event.data,
			timestamp: new Date().toISOString()
		  });
		  
		  if (targetWebSocket.readyState === WebSocket.OPEN) {
			try {
			  targetWebSocket.send(event.data);
			  console.log('Successfully sent message to gemini');
			} catch (error) {
			  console.error('Error sending to gemini:', error);
			}
		  } else {
			// 如果连接还未建立，将消息加入待处理队列
			console.log('Connection not ready, queueing message');
			pendingMessages.push(event.data);
		  }
		});
  
		targetWebSocket.addEventListener("message", (event) => {
		  console.log('Received message from gemini:', {
			dataPreview: typeof event.data === 'string' ? event.data.slice(0, 200) : 'Binary data',
			dataType: typeof event.data,
			timestamp: new Date().toISOString()
		  });
		  
		  try {
			if (proxy.readyState === WebSocket.OPEN) {
			  proxy.send(event.data);
			  console.log('Successfully forwarded message to client');
			}
		  } catch (error) {
			console.error('Error forwarding to client:', error);
		  }
		});
  
		targetWebSocket.addEventListener("close", (event) => {
		  console.log('Gemini connection closed:', {
			code: event.code,
			reason: event.reason || 'No reason provided',
			wasClean: event.wasClean,
			timestamp: new Date().toISOString(),
			readyState: targetWebSocket.readyState
		  });
		  if (proxy.readyState === WebSocket.OPEN) {
			proxy.close(event.code, event.reason);
		  }
		});
  
		proxy.addEventListener("close", (event) => {
		  console.log('Client connection closed:', {
			code: event.code,
			reason: event.reason || 'No reason provided',
			wasClean: event.wasClean,
			timestamp: new Date().toISOString()
		  });
		  if (targetWebSocket.readyState === WebSocket.OPEN) {
			targetWebSocket.close(event.code, event.reason);
		  }
		});
  
		targetWebSocket.addEventListener("error", (error) => {
		  console.error('Gemini WebSocket error:', {
			error: error.message || 'Unknown error',
			timestamp: new Date().toISOString(),
			readyState: targetWebSocket.readyState
		  });
		});
	  });
  
	  ctx.waitUntil(connectPromise);
  
	  return new Response(null, {
		status: 101,
		webSocket: client,
	  });
	},
  };