const logger = require('../utils/logger');
const db = require('../db');
const { processGuestMessages } = require('../routes/chat');

/**
 * Сервис для работы с сессиями пользователей
 */
class SessionService {
  /**
   * Сохраняет сессию, обрабатывая ошибки и логируя результат
   * @param {Object} session - Объект сессии Express
   * @returns {Promise<boolean>} - Успешно ли сохранена сессия
   */
  async saveSession(session) {
    try {
      // Логируем содержимое сессии перед сохранением
      logger.info('[SessionService] Saving session data:', session);

      return new Promise((resolve, reject) => {
        session.save((err) => {
          if (err) {
            logger.error('Error saving session:', err);
            reject(err);
          } else {
            logger.info('Session saved successfully');
            resolve(true);
          }
        });
      });
    } catch (error) {
      logger.error(`[saveSession] Error:`, error);
      throw error;
    }
  }

  /**
   * Связывает гостевые сообщения с пользователем после аутентификации
   * @param {Object} session - Объект сессии Express
   * @param {number} userId - ID пользователя
   * @returns {Promise<Object>} - Результат операции
   */
  async linkGuestMessages(session, userId) {
    try {
      logger.info(
        `[linkGuestMessages] Starting for user ${userId} with guestId=${session.guestId}, previousGuestId=${session.previousGuestId}`
      );

      // Инициализируем массив обработанных гостевых ID, если его нет
      if (!session.processedGuestIds) {
        session.processedGuestIds = [];
      }

      // Получаем все гостевые ID для текущего пользователя из новой таблицы
      const guestIdsResult = await db.getQuery()(
        'SELECT guest_id FROM guest_user_mapping WHERE user_id = $1',
        [userId]
      );
      const userGuestIds = guestIdsResult.rows.map((row) => row.guest_id);

      // Собираем все гостевые ID, которые нужно обработать
      const guestIdsToProcess = new Set();

      // Добавляем текущий гостевой ID
      if (session.guestId && !session.processedGuestIds.includes(session.guestId)) {
        guestIdsToProcess.add(session.guestId);

        // Записываем связь с пользователем в новую таблицу
        await db.getQuery()(
          'INSERT INTO guest_user_mapping (user_id, guest_id) VALUES ($1, $2) ON CONFLICT (guest_id) DO UPDATE SET user_id = $1',
          [userId, session.guestId]
        );
      }

      // Добавляем предыдущий гостевой ID
      if (session.previousGuestId && !session.processedGuestIds.includes(session.previousGuestId)) {
        guestIdsToProcess.add(session.previousGuestId);

        // Записываем связь с пользователем в новую таблицу
        await db.getQuery()(
          'INSERT INTO guest_user_mapping (user_id, guest_id) VALUES ($1, $2) ON CONFLICT (guest_id) DO UPDATE SET user_id = $1',
          [userId, session.previousGuestId]
        );
      }

      // Добавляем все гостевые ID пользователя из таблицы
      for (const guestId of userGuestIds) {
        if (!session.processedGuestIds.includes(guestId)) {
          guestIdsToProcess.add(guestId);
        }
      }

      // Обрабатываем все собранные гостевые ID
      for (const guestId of guestIdsToProcess) {
        await this.processGuestMessagesWrapper(userId, guestId);
        session.processedGuestIds.push(guestId);

        // Помечаем guestId как обработанный в базе данных
        await db.getQuery()(
          'UPDATE guest_user_mapping SET processed = true WHERE guest_id = $1',
          [guestId]
        );
      }

      // Сохраняем сессию
      await this.saveSession(session);

      return { success: true };
    } catch (error) {
      logger.error('[linkGuestMessages] Error:', error);
      return { success: false, error: error.message };
    }
  }

  /**
   * Обертка для функции processGuestMessages
   * @param {number} userId - ID пользователя
   * @param {string} guestId - ID гостя
   * @returns {Promise<Object>} - Результат операции
   */
  async processGuestMessagesWrapper(userId, guestId) {
    try {
      logger.info(
        `[processGuestMessagesWrapper] Processing messages: userId=${userId}, guestId=${guestId}`
      );
      return await processGuestMessages(userId, guestId);
    } catch (error) {
      logger.error(`[processGuestMessagesWrapper] Error: ${error.message}`, error);
      throw error;
    }
  }

  /**
   * Получает сессию из хранилища по ID
   * @param {string} sessionId - ID сессии
   * @returns {Promise<Object|null>} - Объект сессии или null
   */
  async getSession(sessionId) {
    try {
      // Реализация будет зависеть от используемого хранилища сессий
      // Этот метод будет полезен, если нужно получить сессию не из текущего запроса
      return null; // Временная заглушка
    } catch (error) {
      logger.error(`[getSession] Error getting session ${sessionId}:`, error);
      return null;
    }
  }

  /**
   * Удаляет сессию
   * @param {Object} session - Объект сессии Express
   * @returns {Promise<boolean>} - Успешно ли удалена сессия
   */
  async destroySession(session) {
    try {
      return new Promise((resolve, reject) => {
        session.destroy((err) => {
          if (err) {
            logger.error('Error destroying session:', err);
            reject(err);
          } else {
            logger.info('Session destroyed successfully');
            resolve(true);
          }
        });
      });
    } catch (error) {
      logger.error(`[destroySession] Error:`, error);
      throw error;
    }
  }

  /**
   * Восстанавливает сессию из базы данных по ID
   * @param {string} sessionId - ID сессии
   * @returns {Promise<object|null>} - Данные сессии или null
   */
  async getSessionData(sessionId) {
    try {
      if (!sessionId) {
        logger.warn('[SessionService] Cannot restore session without sessionId');
        return null;
      }

      logger.info(`[SessionService] Attempting to retrieve session ${sessionId}`);

      const result = await db.getQuery()(
        'SELECT sess FROM session WHERE sid = $1',
        [sessionId]
      );

      if (result.rows.length === 0) {
        logger.info(`[SessionService] No session found with ID ${sessionId}`);
        return null;
      }

      const sessionData = result.rows[0].sess;
      logger.info(`[SessionService] Retrieved session data for ${sessionId}`);

      return sessionData;
    } catch (error) {
      logger.error(`[SessionService] Error retrieving session ${sessionId}:`, error);
      return null;
    }
  }

  /**
   * Обновляет данные аутентификации в сессии
   * @param {object} session - Объект сессии
   * @param {object} authData - Данные аутентификации
   * @returns {Promise<boolean>} - Результат операции
   */
  async updateAuthData(session, authData) {
    try {
      if (!session || !authData) {
        logger.warn('[SessionService] Missing parameters for updateAuthData');
        return false;
      }

      const { userId, authType, isAdmin, ...otherData } = authData;

      if (!userId || !authType) {
        logger.warn('[SessionService] Missing userId or authType in authData');
        return false;
      }

      // Обновляем основные поля аутентификации
      session.userId = userId;
      session.authType = authType;
      session.authenticated = true;

      if (isAdmin !== undefined) {
        session.isAdmin = isAdmin;
      }

      // Обновляем дополнительные данные в зависимости от типа аутентификации
      if (authType === 'wallet' && otherData.address) {
        session.address = otherData.address.toLowerCase();
      } else if (authType === 'email' && otherData.email) {
        session.email = otherData.email.toLowerCase();
      } else if (authType === 'telegram') {
        if (otherData.telegramId) session.telegramId = otherData.telegramId;
        if (otherData.telegramUsername) session.telegramUsername = otherData.telegramUsername;
        if (otherData.telegramFirstName) session.telegramFirstName = otherData.telegramFirstName;
      }

      // Сохраняем гостевые ID, если они предоставлены и не были ранее в сессии
      if (otherData.guestId && !session.guestId) {
        session.guestId = otherData.guestId;
      }

      if (otherData.previousGuestId && !session.previousGuestId) {
        session.previousGuestId = otherData.previousGuestId;
      }

      // Сохраняем обновленную сессию
      return await this.saveSession(session, 'updateAuthData');
    } catch (error) {
      logger.error('[SessionService] Error updating auth data:', error);
      return false;
    }
  }

  /**
   * Очищает данные аутентификации в сессии
   * @param {object} session - Объект сессии
   * @returns {Promise<boolean>} - Результат операции
   */
  async logout(session) {
    try {
      if (!session) {
        logger.warn('[SessionService] Cannot logout null session');
        return false;
      }

      // Сохраняем гостевые ID перед очисткой
      const guestId = session.guestId;

      // Удаляем данные аутентификации
      delete session.userId;
      delete session.authenticated;
      delete session.authType;
      delete session.isAdmin;
      delete session.address;
      delete session.email;
      delete session.telegramId;
      delete session.telegramUsername;
      delete session.telegramFirstName;

      // Восстанавливаем гостевой ID для продолжения работы
      if (guestId) {
        session.guestId = guestId;
      }

      // Сохраняем обновленную сессию
      return await this.saveSession(session, 'logout');
    } catch (error) {
      logger.error('[SessionService] Error during logout:', error);
      return false;
    }
  }
}

const sessionService = new SessionService();
module.exports = sessionService;
