task-svr/src/controllers/chest.controller.ts

import { ZError, SyncLocker, ZRedisClient, BaseController, ROLE_ANON, role, router } from 'zutils'
import { ActivityChest, ChestStatusEnum } from 'models/ActivityChest'
import { ActivityUser } from 'models/ActivityUser'
import { rankKey, rankLevel, updateRankScore } from 'services/rank.svr'
import { formatDate } from 'zutils/utils/date.util'
import { ScoreRecord } from 'models/ScoreRecord'
import { ChestRecord } from 'models/chain/ChestRecord'
import { generateNewChest } from 'services/game.svr'
import { SCORE_OPEN_CHEST } from 'common/Constants'
import { formatAddress } from 'zutils/utils/chain.util'

/**
 * 宝箱相关接口
 */
class BoxController extends BaseController {
  /**
   * 宝箱列表
   */
  @router('get /api/chest/list')
  async chestList(req) {
    const user = req.user
    const openRecord = await ChestRecord.find({ from: user.address.toLowerCase() })
    const chestIds = openRecord.map(record => record.chestId)
    const chestSet = new Set(chestIds)
    const chests = await ActivityChest.find({
      activity: user.activity,
      user: user.id,
      status: { $in: [0, 1, 2] },
    }).sort({
      level: -1,
    })
    const results = chests.map(chest => chest.toJson())
    for (let result of results) {
      if (chestSet.has(result.id)) {
        result.stat = 2
      }
    }
    return results
  }
  /**
   * 宝箱助力列表
   */
  @router('post /api/chest/enhance/list')
  async enhanceList(req) {
    const user = req.user
    let { chestId, chestid } = req.params
    chestid = chestId || chestid
    if (!chestid) {
      throw new ZError(11, 'chestId is required')
    }
    const chest = await ActivityChest.findById(chestid)
    if (!chest) {
      throw new ZError(12, 'chest not found')
    }
    if (chest.status === 0) {
      throw new ZError(13, 'chest is locked')
    }
    if (chest.status === 9) {
      throw new ZError(14, 'chest had been opened')
    }
    const result = []
    const users = await ActivityUser.find({ _id: { $in: chest.bonusUsers } })
    const userMap = new Map()
    const totalKey = rankKey(user.activity)
    users.forEach(u => userMap.set(u.id, u))
    for (let i = 0; i < chest.bonusUsers.length; i++) {
      const u = userMap.get(chest.bonusUsers[i])
      const totalScore = await new ZRedisClient().zscore(totalKey, u.id)
      const score = totalScore ? parseInt(totalScore + '') : 0
      result.push({
        nickname: u?.twitterName || u?.discordName || u?.address ? formatAddress(u.address) : 'unknown',
        level: rankLevel(score),
        score: score,
      })
    }
    return result
  }
  /**
   * 宝箱助力
   */
  @router('post /api/chest/enhance')
  async enhance(req) {
    new SyncLocker().checkLock(req)
    const { code } = req.params
    const user = req.user
    const uid = user.id

    // TODO:: 待规则确定后, 检查用户是否符合助力条件
    const chest = await ActivityChest.findOne({ shareCode: code, activity: user.activity })
    if (chest.bonusUsers.includes(uid)) {
      throw new ZError(10, 'user already enhanced')
    }
    if (chest.bonusUsers.length >= chest.maxBounsCount) {
      throw new ZError(12, 'enhanced times exceed')
    }
    if (chest.status === ChestStatusEnum.OPENED) {
      throw new ZError(13, 'chest already opened')
    }
    if (chest.status === ChestStatusEnum.LOCKED) {
      throw new ZError(14, 'chest is locked')
    }
    // 生产环境不能助力自己
    if (process.env.NODE_ENV === 'production' && chest.user === uid) {
      throw new ZError(15, 'can not enhance self')
    }
    const score = Math.floor(Math.random() * (chest.bounsCfg[1] - chest.bounsCfg[0] + 1) + chest.bounsCfg[0])
    chest.bonusUsers.push(uid)
    chest.bonusScores.push(score)
    chest.scoreBonus += score
    await chest.save()
    const chestsForUser = await ActivityChest.find({ user: uid, activity: user.activity })
    // 如果用户没有宝箱, 则说明用户是新用户, 生成一个宝箱
    if (chestsForUser.length === 0) {
      const newChest = generateNewChest(uid, user.activity, 1, ChestStatusEnum.LOCKED)
      await newChest.save()
      return {
        score: 0,
        chests: [newChest.toJson()],
      }
    } else {
      return {
        score: 0,
      }
    }
  }

  /**
   * 开启宝箱
   * 流程如下:
   * 1. 客户端组装json string: data:,{"p":"cf-20","op":"chest_open","id":"${chestId}"}
   * 2. 将字符串转成hex
   * 3. 将hex字符串作为data, 发起一笔至特定地址的0ETH交易
   * 4. 服务端监听到交易后, 解析input, 获取chestId
   * 5. 服务端调用此接口, 传入chestId, 完成开箱操作
   */
  @router('post /api/chest/open')
  async openChest(req) {
    new SyncLocker().checkLock(req)
    const user = req.user
    const { chestId } = req.params
    if (!chestId) {
      throw new ZError(11, 'chestId is required')
    }
    const openRecord = await ChestRecord.findOne({ from: user.address.toLowerCase(), chestId })
    if (!openRecord) {
      throw new ZError(12, 'onchain open record not found')
    }
    const chest = await ActivityChest.findById(chestId)
    if (!chest) {
      throw new ZError(12, 'chest not found')
    }
    if (chest.user !== user.id) {
      throw new ZError(13, 'chest not belong to user')
    }
    if (chest.status === ChestStatusEnum.OPENED) {
      throw new ZError(14, 'chest already opened')
    }
    chest.status = ChestStatusEnum.OPENED
    const score = chest.scoreInit + chest.scoreBonus
    const dateTag = formatDate(new Date())
    await updateRankScore({
      user: user.id,
      score: score,
      activity: user.activity,
      scoreType: SCORE_OPEN_CHEST,
      scoreParams: {
        date: dateTag,
        chestId: chest.id,
        level: chest.level,
      },
    })
    await chest.save()
    return { score }
  }

  /**
   * 宝箱开启记录
   */
  @router('get /api/chest/open/history')
  async openChestHistory(req) {
    const user = req.user
    const records = await ScoreRecord.find({ user: user.id, activity: user.activity, type: SCORE_OPEN_CHEST }).sort({
      createdAt: -1,
    })
    return records.map(record => {
      return {
        chest: record.data.chestId,
        score: record.score,
        level: record.data.level,
        // @ts-ignore
        time: record.createdAt.getTime(),
      }
    })
  }
}