ai_game/server/models/user.js

const pool = require('../config/db');

const UserModel = {
  // 通过openid查找或创建用户
  async findOrCreate(openid, userInfo = {}) {
    const [existing] = await pool.query(
      'SELECT * FROM users WHERE openid = ?',
      [openid]
    );
    
    if (existing.length > 0) {
      return existing[0];
    }
    
    // 创建新用户
    const [result] = await pool.query(
      'INSERT INTO users (openid, nickname, avatar_url, gender) VALUES (?, ?, ?, ?)',
      [openid, userInfo.nickname || '', userInfo.avatarUrl || '', userInfo.gender || 0]
    );
    
    return {
      id: result.insertId,
      openid,
      nickname: userInfo.nickname || '',
      avatar_url: userInfo.avatarUrl || '',
      gender: userInfo.gender || 0
    };
  },

  // 更新用户信息
  async updateProfile(userId, userInfo) {
    await pool.query(
      'UPDATE users SET nickname = ?, avatar_url = ?, gender = ? WHERE id = ?',
      [userInfo.nickname, userInfo.avatarUrl, userInfo.gender, userId]
    );
  },

  // 获取用户进度
  async getProgress(userId, storyId = null) {
    let sql = `SELECT up.*, s.title as story_title, s.cover_url 
               FROM user_progress up 
               JOIN stories s ON up.story_id = s.id 
               WHERE up.user_id = ?`;
    const params = [userId];
    
    if (storyId) {
      sql += ' AND up.story_id = ?';
      params.push(storyId);
    }
    
    sql += ' ORDER BY up.updated_at DESC';
    const [rows] = await pool.query(sql, params);
    return storyId ? rows[0] : rows;
  },

  // 保存用户进度
  async saveProgress(userId, storyId, data) {
    const { currentNodeKey, isCompleted, endingReached } = data;
    
    await pool.query(
      `INSERT INTO user_progress (user_id, story_id, current_node_key, is_completed, ending_reached)
       VALUES (?, ?, ?, ?, ?)
       ON DUPLICATE KEY UPDATE 
         current_node_key = VALUES(current_node_key),
         is_completed = VALUES(is_completed),
         ending_reached = VALUES(ending_reached),
         play_count = play_count + 1`,
      [userId, storyId, currentNodeKey, isCompleted || false, endingReached || '']
    );
    
    // 如果完成了，记录结局
    if (isCompleted && endingReached) {
      await pool.query(
        `INSERT IGNORE INTO user_endings (user_id, story_id, ending_name) VALUES (?, ?, ?)`,
        [userId, storyId, endingReached]
      );
      
      // 更新用户统计
      await pool.query(
        'UPDATE users SET total_play_count = total_play_count + 1, total_endings = (SELECT COUNT(*) FROM user_endings WHERE user_id = ?) WHERE id = ?',
        [userId, userId]
      );
    }
  },

  // 点赞/取消点赞
  async toggleLike(userId, storyId, isLiked) {
    await pool.query(
      `INSERT INTO user_progress (user_id, story_id, is_liked)
       VALUES (?, ?, ?)
       ON DUPLICATE KEY UPDATE is_liked = ?`,
      [userId, storyId, isLiked, isLiked]
    );
  },

  // 收藏/取消收藏
  async toggleCollect(userId, storyId, isCollected) {
    await pool.query(
      `INSERT INTO user_progress (user_id, story_id, is_collected)
       VALUES (?, ?, ?)
       ON DUPLICATE KEY UPDATE is_collected = ?`,
      [userId, storyId, isCollected, isCollected]
    );
  },

  // 获取收藏列表
  async getCollections(userId) {
    const [rows] = await pool.query(
      `SELECT s.id, s.title, s.cover_url, s.description, s.category, s.play_count, s.like_count
       FROM user_progress up
       JOIN stories s ON up.story_id = s.id
       WHERE up.user_id = ? AND up.is_collected = 1
       ORDER BY up.updated_at DESC`,
      [userId]
    );
    return rows;
  },

  // 获取用户解锁的结局
  async getUnlockedEndings(userId, storyId = null) {
    let sql = 'SELECT * FROM user_endings WHERE user_id = ?';
    const params = [userId];
    
    if (storyId) {
      sql += ' AND story_id = ?';
      params.push(storyId);
    }
    
    const [rows] = await pool.query(sql, params);
    return rows;
  }
};

module.exports = UserModel;