/**
 * Copyright (c) 2024-2025 Тарабанов Александр Викторович
 * All rights reserved.
 * 
 * This software is proprietary and confidential.
 * Unauthorized copying, modification, or distribution is prohibited.
 * 
 * For licensing inquiries: info@hb3-accelerator.com
 * Website: https://hb3-accelerator.com
 * GitHub: https://github.com/VC-HB3-Accelerator
 */

const WebSocket = require('ws');
const tokenBalanceService = require('./services/tokenBalanceService');
const deploymentTracker = require('./utils/deploymentTracker');
const deploymentWebSocketService = require('./services/deploymentWebSocketService');

let wss = null;
// Храним клиентов по userId для персонализированных уведомлений
const wsClients = new Map(); // userId -> Set of WebSocket connections

// Кэш для отслеживания изменений тегов
const tagsChangeCache = new Map();
const TAGS_CACHE_TTL = 5000; // 5 секунд

// Дебаунс для broadcastTagsUpdate
let tagsUpdateTimeout = null;
const TAGS_UPDATE_DEBOUNCE = 100; // 100ms

function initWSS(server) {
  wss = new WebSocket.Server({ server, path: '/ws' });
  console.log('🔌 [WebSocket] Сервер инициализирован на пути /ws');
  
  // Инициализируем deploymentWebSocketService с WebSocket сервером после создания wss
  deploymentWebSocketService.initialize(server, wss);
  
  // Подключаем deployment tracker к WebSocket
  deploymentTracker.on('deployment_updated', (data) => {
    broadcastDeploymentUpdate(data);
  });
  
  // Дополнительная инициализация deploymentWebSocketService после создания wss
  console.log('[wsHub] Инициализируем deploymentWebSocketService с wss:', !!wss);
  deploymentWebSocketService.setWebSocketServer(wss);
  console.log('[wsHub] deploymentWebSocketService инициализирован');
  
  wss.on('connection', (ws, req) => {
    console.log('🔌 [WebSocket] Новое подключение');
    console.log('🔌 [WebSocket] IP клиента:', req.socket.remoteAddress);
    console.log('🔌 [WebSocket] User-Agent:', req.headers['user-agent']);
    console.log('🔌 [WebSocket] Origin:', req.headers.origin);
    
    // Добавляем клиента в общий список
    if (!wsClients.has('anonymous')) {
      wsClients.set('anonymous', new Set());
    }
    wsClients.get('anonymous').add(ws);
    
    // Обработка сообщений от клиента
    ws.on('message', (message) => {
      try {
        const data = JSON.parse(message);
        // console.log('📨 [WebSocket] Получено сообщение:', data);
        
        if (data.type === 'auth' && data.userId) {
          // Аутентификация пользователя
          authenticateUser(ws, data.userId);
        }
        
        if (data.type === 'ping') {
          // Отправляем pong ответ
          ws.send(JSON.stringify({ 
            type: 'pong', 
            timestamp: data.timestamp 
          }));
        }

        if (data.type === 'ollama_ready') {
          // Уведомление о готовности Ollama - запускаем инициализацию ботов
          console.log('🚀 [WebSocket] Получено уведомление о готовности Ollama!');
          handleOllamaReady();
        }
        
        if (data.type === 'request_token_balances' && data.address) {
          // Запрос балансов токенов
          handleTokenBalancesRequest(ws, data.address, data.userId);
        }
        
        // Обработка сообщений для деплоя модулей
        if (data.type === 'subscribe' && data.dleAddress) {
          // Подписка на деплой для конкретного DLE
          deploymentWebSocketService.subscribeToDeployment(ws, data.dleAddress);
        }
        
        if (data.type === 'unsubscribe' && data.dleAddress) {
          // Отписка от деплоя
          deploymentWebSocketService.unsubscribeFromDeployment(ws, data.dleAddress);
        }
        
        // Обработка сообщений деплоя через deploymentWebSocketService
        if (data.type === 'deployment_log' || 
            data.type === 'deployment_started' || 
            data.type === 'deployment_status' ||
            data.type === 'deployment_finished' ||
            data.type === 'deployment_update') {
          // Эти сообщения обрабатываются в deploymentWebSocketService
          // Просто логируем для отладки
          console.log(`[WebSocket] Получено сообщение деплоя: ${data.type}`);
          console.log(`[WebSocket] Данные:`, JSON.stringify(data, null, 2));
        }
      } catch (error) {
        // console.error('❌ [WebSocket] Ошибка парсинга сообщения:', error);
      }
    });
    
    ws.on('close', (code, reason) => {
      // console.log('🔌 [WebSocket] Соединение закрыто', { code, reason: reason.toString() });
      // Удаляем клиента из всех списков
      for (const [userId, clients] of wsClients.entries()) {
        clients.delete(ws);
        if (clients.size === 0) {
          wsClients.delete(userId);
        }
      }
      
      // Удаляем клиента из deploymentWebSocketService
      deploymentWebSocketService.removeClient(ws);
    });
    
    ws.on('error', (error) => {
      // console.error('❌ [WebSocket] Ошибка соединения:', error.message);
    });
  });
  
  // console.log('🚀 [WebSocket] Сервер запущен на /ws');
}

function authenticateUser(ws, userId) {
  // console.log(`🔐 [WebSocket] Аутентификация пользователя ${userId}`);
  
  // Удаляем из анонимных
  if (wsClients.has('anonymous')) {
    wsClients.get('anonymous').delete(ws);
  }
  
  // Добавляем в список пользователя
  if (!wsClients.has(userId.toString())) {
    wsClients.set(userId.toString(), new Set());
  }
  wsClients.get(userId.toString()).add(ws);
  
  // Отправляем подтверждение
  ws.send(JSON.stringify({ 
    type: 'auth-success', 
    userId: userId 
  }));
}

function broadcastContactsUpdate() {
  // console.log('📢 [WebSocket] Отправка обновления контактов всем клиентам');
  for (const [userId, clients] of wsClients.entries()) {
    for (const ws of clients) {
      if (ws.readyState === WebSocket.OPEN) {
        ws.send(JSON.stringify({ type: 'contacts-updated' }));
      }
    }
  }
}

function broadcastMessagesUpdate() {
  // console.log('📢 [WebSocket] Отправка обновления сообщений всем клиентам');
  for (const [userId, clients] of wsClients.entries()) {
    for (const ws of clients) {
      if (ws.readyState === WebSocket.OPEN) {
        ws.send(JSON.stringify({ type: 'messages-updated' }));
      }
    }
  }
}

function broadcastChatMessage(message, targetUserId = null) {
  // console.log(`📢 [WebSocket] Отправка сообщения чата`, { 
  //   messageId: message.id, 
  //   targetUserId 
  // });
  
  if (targetUserId) {
    // Отправляем конкретному пользователю
    const userClients = wsClients.get(targetUserId.toString());
    if (userClients) {
      for (const ws of userClients) {
        if (ws.readyState === WebSocket.OPEN) {
          ws.send(JSON.stringify({ 
            type: 'chat-message', 
            message 
          }));
        }
      }
    }
  } else {
    // Отправляем всем
    for (const [userId, clients] of wsClients.entries()) {
      for (const ws of clients) {
        if (ws.readyState === WebSocket.OPEN) {
          ws.send(JSON.stringify({ 
            type: 'chat-message', 
            message 
          }));
        }
      }
    }
  }
}

function broadcastConversationUpdate(conversationId, targetUserId = null) {
  // console.log(`📢 [WebSocket] Отправка обновления диалога`, { 
  //   conversationId, 
  //   targetUserId 
  // });
  
  const payload = { 
    type: 'conversation-updated', 
    conversationId 
  };
  
  if (targetUserId) {
    // Отправляем конкретному пользователю
    const userClients = wsClients.get(targetUserId.toString());
    if (userClients) {
      for (const ws of userClients) {
        if (ws.readyState === WebSocket.OPEN) {
          ws.send(JSON.stringify(payload));
        }
      }
    }
  } else {
    // Отправляем всем
    for (const [userId, clients] of wsClients.entries()) {
      for (const ws of clients) {
        if (ws.readyState === WebSocket.OPEN) {
          ws.send(JSON.stringify(payload));
        }
      }
    }
  }
}

function broadcastTableUpdate(tableId) {
  // console.log('📢 [WebSocket] Отправка обновления таблицы', tableId);
  const payload = { type: 'table-updated', tableId };
    for (const [userId, clients] of wsClients.entries()) {
      for (const ws of clients) {
        if (ws.readyState === WebSocket.OPEN) {
          ws.send(JSON.stringify(payload));
      }
    }
  }
}

function broadcastTableRelationsUpdate(tableId, rowId, targetUserId = null) {
  // console.log(`📢 [WebSocket] Отправка обновления связей таблицы`, { 
  //   tableId, 
  //   rowId, 
  //   targetUserId 
  // });
  
  const payload = { 
    type: 'table-relations-updated', 
    tableId, 
    rowId 
  };
  
  if (targetUserId) {
    // Отправляем конкретному пользователю
    const userClients = wsClients.get(targetUserId.toString());
    if (userClients) {
      for (const ws of userClients) {
        if (ws.readyState === WebSocket.OPEN) {
          ws.send(JSON.stringify(payload));
        }
      }
    }
  } else {
    // Отправляем всем
    for (const [userId, clients] of wsClients.entries()) {
      for (const ws of clients) {
        if (ws.readyState === WebSocket.OPEN) {
          ws.send(JSON.stringify(payload));
        }
      }
    }
  }
}

function broadcastTagsUpdate(targetUserId = null, rowId = null) {
  // Дебаунс: отменяем предыдущий таймаут
  if (tagsUpdateTimeout) {
    clearTimeout(tagsUpdateTimeout);
  }
  
  // Устанавливаем новый таймаут
  tagsUpdateTimeout = setTimeout(() => {
    // console.log('🔔 [WebSocket] Отправляем уведомление об обновлении тегов', rowId ? `для строки ${rowId}` : '');
    const message = JSON.stringify({
      type: 'tags-updated',
      timestamp: Date.now(),
      rowId: rowId // Добавляем информацию о конкретной строке
    });
    
    let sentCount = 0;
    // Отправляем всем подключенным клиентам
    wss.clients.forEach((client) => {
      if (client.readyState === WebSocket.OPEN) {
        client.send(message);
        sentCount++;
      }
    });
    
    // console.log(`🔔 [WebSocket] Отправлено tags-updated ${sentCount} клиентам`);
  }, TAGS_UPDATE_DEBOUNCE);
}

function getConnectedUsers() {
  const users = [];
  for (const [userId, clients] of wsClients.entries()) {
    if (userId !== 'anonymous' && clients.size > 0) {
      users.push({
        userId: parseInt(userId),
        connections: clients.size
      });
    }
  }
  return users;
}

function getStats() {
  let totalConnections = 0;
  let anonymousConnections = 0;
  
  for (const [userId, clients] of wsClients.entries()) {
    if (userId === 'anonymous') {
      anonymousConnections = clients.size;
    }
    totalConnections += clients.size;
  }
  
  return {
    totalConnections,
    anonymousConnections,
    authenticatedUsers: getConnectedUsers(),
    totalUsers: wsClients.size - (wsClients.has('anonymous') ? 1 : 0)
  };
}

// Функция для отправки уведомлений о статусе AI
function broadcastAIStatus(status) {
  // console.log('📢 [WebSocket] Отправка статуса AI всем клиентам');
  for (const [userId, clients] of wsClients.entries()) {
    for (const ws of clients) {
      if (ws.readyState === WebSocket.OPEN) {
        ws.send(JSON.stringify({ 
          type: 'ai-status', 
          status 
        }));
      }
    }
  }
}

// Blockchain уведомления
function broadcastProposalCreated(dleAddress, proposalId, txHash) {
  const message = JSON.stringify({
    type: 'proposal_created',
    data: {
      dleAddress: dleAddress,
      proposalId: proposalId,
      txHash: txHash
    },
    timestamp: Date.now()
  });
  
  broadcastToAllClients(message);
}

function broadcastProposalVoted(dleAddress, proposalId, support, txHash) {
  const message = JSON.stringify({
    type: 'proposal_voted',
    data: {
      dleAddress: dleAddress,
      proposalId: proposalId,
      support: support,
      txHash: txHash
    },
    timestamp: Date.now()
  });
  
  broadcastToAllClients(message);
}

function broadcastProposalExecuted(dleAddress, proposalId, txHash) {
  const message = JSON.stringify({
    type: 'proposal_executed',
    data: {
      dleAddress: dleAddress,
      proposalId: proposalId,
      txHash: txHash
    },
    timestamp: Date.now()
  });
  
  broadcastToAllClients(message);
}

function broadcastToAllClients(message) {
  wss.clients.forEach((client) => {
    if (client.readyState === WebSocket.OPEN) {
      client.send(message);
    }
  });
}

// Функции для уведомлений об изменениях токенов
function broadcastAuthTokenAdded(tokenData) {
  const message = JSON.stringify({
    type: 'auth_token_added',
    data: {
      token: tokenData,
      timestamp: Date.now()
    }
  });
  
  broadcastToAllClients(message);
}

function broadcastAuthTokenDeleted(tokenData) {
  const message = JSON.stringify({
    type: 'auth_token_deleted',
    data: {
      token: tokenData,
      timestamp: Date.now()
    }
  });
  
  broadcastToAllClients(message);
}

function broadcastAuthTokenUpdated(tokenData) {
  const message = JSON.stringify({
    type: 'auth_token_updated',
    data: {
      token: tokenData,
      timestamp: Date.now()
    }
  });
  
  broadcastToAllClients(message);
}

// Функции для балансов токенов
function broadcastTokenBalancesUpdate(userId, balances) {
  const message = JSON.stringify({
    type: 'token_balances_updated',
    data: {
      balances: balances,
      timestamp: Date.now()
    }
  });
  
  // Отправляем конкретному пользователю
  const userClients = wsClients.get(userId.toString());
  if (userClients) {
    for (const ws of userClients) {
      if (ws.readyState === WebSocket.OPEN) {
        ws.send(message);
      }
    }
  }
}

function broadcastTokenBalanceChanged(userId, tokenAddress, newBalance, network) {
  const message = JSON.stringify({
    type: 'token_balance_changed',
    data: {
      tokenAddress: tokenAddress,
      balance: newBalance,
      network: network,
      timestamp: Date.now()
    }
  });
  
  // Отправляем конкретному пользователю
  const userClients = wsClients.get(userId.toString());
  if (userClients) {
    for (const ws of userClients) {
      if (ws.readyState === WebSocket.OPEN) {
        ws.send(message);
      }
    }
  }
}

// Функции для деплоя
function broadcastDeploymentUpdate(data) {
  if (!wss) return;
  
  console.log(`📡 [WebSocket] broadcastDeploymentUpdate вызвана с данными:`, JSON.stringify(data, null, 2));
  
  const message = JSON.stringify({
    type: 'deployment_update',
    data: data
  });
  
  console.log(`📡 [WebSocket] Отправляем сообщение:`, message);
  
  // Отправляем всем подключенным клиентам
  wss.clients.forEach(client => {
    if (client.readyState === WebSocket.OPEN) {
      try {
        client.send(message);
      } catch (error) {
        console.error('[WebSocket] Ошибка при отправке deployment update:', error);
      }
    }
  });
  
  console.log(`📡 [WebSocket] Отправлено deployment update: deployment_update`);
}

// broadcastModulesUpdate удалена - используем deploymentWebSocketService.broadcastToDLE

module.exports = { 
  initWSS, 
  broadcastContactsUpdate, 
  broadcastMessagesUpdate, 
  broadcastChatMessage,
  broadcastConversationUpdate,
  broadcastTableUpdate,
  broadcastTableRelationsUpdate,
  broadcastTagsUpdate,
  broadcastAIStatus,
  broadcastProposalCreated,
  broadcastProposalVoted,
  broadcastProposalExecuted,
  broadcastAuthTokenAdded,
  broadcastAuthTokenDeleted,
  broadcastAuthTokenUpdated,
  broadcastTokenBalancesUpdate,
  broadcastTokenBalanceChanged,
  broadcastDeploymentUpdate,
  getConnectedUsers,
  getStats
};

// Обработчик запроса балансов токенов
async function handleTokenBalancesRequest(ws, address, userId) {
  try {
    console.log(`[WebSocket] Запрос балансов для адреса: ${address}`);

    // Получаем балансы через отдельный сервис без зависимостей от wsHub
    const balances = await tokenBalanceService.getUserTokenBalances(address);
    
    console.log(`[WebSocket] Получены балансы для ${address}:`, balances);
    console.log(`[WebSocket] Количество токенов:`, balances?.length || 0);
    
    // Отправляем ответ клиенту
    const response = {
      type: 'token_balances_response',
      data: {
        address: address,
        balances: balances,
        timestamp: Date.now()
      }
    };
    
    console.log(`[WebSocket] Отправляем ответ:`, JSON.stringify(response, null, 2));
    ws.send(JSON.stringify(response));
  } catch (error) {
    console.error('[WebSocket] Ошибка при получении балансов:', error);
    
    // Определяем тип ошибки для лучшей диагностики
    let errorType = 'Неизвестная ошибка';
    const errorMessage = error.message || error.toString();
    
    if (errorMessage.includes('timeout') || errorMessage.includes('TIMEOUT')) {
      errorType = 'Таймаут соединения - возможно, нужен VPN';
    } else if (errorMessage.includes('ECONNREFUSED') || errorMessage.includes('ENOTFOUND')) {
      errorType = 'Не удается подключиться к RPC провайдеру';
    } else if (errorMessage.includes('TLS') || errorMessage.includes('socket disconnected')) {
      errorType = 'Проблема с TLS соединением - проверьте VPN';
    } else if (errorMessage.includes('NETWORK_ERROR')) {
      errorType = 'Ошибка сети - проверьте интернет-соединение';
    }
    
    // Отправляем ошибку клиенту
    const errorResponse = {
      type: 'token_balances_error',
      data: {
        address: address,
        error: errorType,
        errorDetails: errorMessage,
        timestamp: Date.now()
      }
    };
    
    console.log('[WebSocket] Отправляем ошибку клиенту:', JSON.stringify(errorResponse, null, 2));
    ws.send(JSON.stringify(errorResponse));
  }
}

/**
 * Обработка уведомления о готовности Ollama
 */
async function handleOllamaReady() {
  try {
    console.log('✅ [WebSocket] Ollama готов к работе');
    // Уведомляем всех подключенных клиентов о готовности системы
    broadcastSystemReady();
  } catch (error) {
    console.error('❌ [WebSocket] Ошибка обработки Ollama ready:', error);
  }
}

/**
 * Уведомление всех клиентов о готовности системы
 */
function broadcastSystemReady() {
  const message = JSON.stringify({
    type: 'system_ready',
    data: {
      message: 'Все модели загружены! Система готова к работе.',
      timestamp: Date.now(),
      bots: ['web', 'telegram', 'email']
    }
  });
  
  // Отправляем всем подключенным клиентам
  wsClients.forEach((clientSet) => {
    clientSet.forEach((ws) => {
      if (ws.readyState === WebSocket.OPEN) {
        ws.send(message);
      }
    });
  });
  
  console.log('📢 [WebSocket] Уведомление о готовности системы отправлено всем клиентам');
}