const WebSocket = require('ws');
const { sendMessageAsync } = require('./controllers/chatController');
const webSocketClients = require('./utils/webSocketClients');
const jwt = require('jsonwebtoken');
const { User } = require('./models');

// WebSocketサーバーの設定
function setupWebSocketServer(wss) {
  wss.on('connection', async (ws, req) => {
    try {
      const token = req.url.split('?token=')[1]; // トークンをURLから取得
      if (!token) {
        ws.close(4000, 'Unauthorized');
        return;
      }

      // トークンを検証
      const decoded = jwt.verify(token, process.env.JWT_SECRET);
      const user = await User.findByPk(decoded.userId);

      if (!user) {
        ws.close(4000, 'Unauthorized');
        return;
      }

      // WebSocketにユーザーIDを紐付け
      ws.userId = user.id;
      webSocketClients.push(ws);

      // WebSocketが閉じられたときの処理
      ws.on('close', () => {
        console.log('WebSocket connection closed');
        const index = webSocketClients.indexOf(ws);
        if (index !== -1) {
          webSocketClients.splice(index, 1);
        }
      });

      // WebSocket経由でメッセージを受信したときの処理
      ws.on('message', async (messageData) => {
        try {
          // Bufferから文字列に変換してJSONパース
          const { userMessage, isotopeId } = JSON.parse(messageData.toString('utf8'));

          // メッセージ送信処理を呼び出す
          await sendMessageAsync(ws, userMessage, isotopeId);
        } catch (error) {
          console.error('Error processing message:', error);
        }
      });
    } catch (error) {
      console.error('WebSocket connection error:', error);
      ws.close(4000, 'Unauthorized');
    }
  });
}

module.exports = setupWebSocketServer;