screepsmod-mongo/lib/backend/api/index.js

[](https://standardjs.com) [](https://github.com/ScreepsMods/screepsmod

const crypto = require('crypto')
const express = require('express')
const bodyParser = require('body-parser')
const jsonResponse = require('q-json-response')
const auth = require('@screeps/backend/lib/game/api/auth')
const authlib = require('@screeps/backend/lib/authlib')
const _ = require('lodash')

module.exports = function (config) {
  const { common } = config
  const { db, env, pubsub } = common.storage
  const router = new express.Router()
  config.backend.on('expressPreConfig', (app) => app.use(router))
  router.use(bodyParser.json({
    limit: '8mb',
    verify (request, response, buf, encoding) {
      request.rawBody = buf.toString(encoding)
    }
  }))
  router.post('/api/game/add-object-intent', auth.tokenAuth, jsonResponse(async (request) => {
    await checkGame(request)
    if (request.body.name === 'activateSafeMode') {
      const gameTime = parseInt(await env.get('gameTime'))
      const count = await db['rooms.objects'].count({ $and: [{ type: 'controller' }, { user: '' + request.user._id }, { safeMode: { $gt: gameTime } }] })
      if (count > 0) throw new Error('safe mode active already')
    }
    await env.hmset(env.keys.ROOM_INTENTS + request.body.room, request.user._id.toString() + '.manual', JSON.stringify({ [request.body._id]: { [request.body.name]: request.body.intent } }))
    await db.rooms.update({ _id: request.body.room }, { $set: { active: true } })
  }))
  router.get('/api/user/messages/index', auth.tokenAuth, jsonResponse((request) => {
    return db['users.messages'].findEx({ user: request.user._id }, { sort: { date: -1 } })
      .then(data => {
        const messages = []
        data.forEach(message => {
          if (!messages.find(i => i._id === message.respondent)) {
            messages.push({ _id: message.respondent, message })
          }
        })
        return db.users.find({ _id: { $in: _.map(messages, '_id') } })
          .then(users => {
            users = users.map(i => _.pick(i, ['_id', 'username', 'badge']))
            return { messages, users: _.keyBy(users, '_id') }
          })
      })
  }))
  router.post('/api/user/messages/mark-read', auth.tokenAuth, jsonResponse((request) => {
    const _id = request.body.id; let message
    console.log(_id, request.user._id)
    return db['users.messages'].findOne({ _id, user: request.user._id, type: 'in' })
      .then((_message) => {
        if (!_message) {
          return Promise.reject(new Error('invalid id'))
        }
        message = _message
        return db['users.messages'].update({ _id }, { $set: { unread: false } })
      })
      .then((data) => {
        return Promise.all([
          pubsub.publish(`user:${message.user}/message:${message.respondent}`, JSON.stringify({
            message: {
              _id: request.body.id,
              unread: false
            }
          })),
          pubsub.publish(`user:${message.respondent}/message:${message.user}`, JSON.stringify({
            message: {
              _id: message.outMessage,
              unread: false
            }
          })),
          db['users.messages'].update({ _id: message.outMessage }, { $set: { unread: false } })
        ])
      })
  }))
  authlib.genToken = function (id) {
    const token = crypto.createHmac('sha1', 'hsdhweh342sdbj34e').update(new Date().getTime() + id).digest('hex')
    return env.setex(`auth_${token}`, 300, id).then(() => token)
  }

  authlib.checkToken = function (token, noConsume) {
    const authKey = `auth_${token}`

    return env.get(authKey)
      .then((data) => {
        if (!data) {
          return Promise.reject(false) // eslint-disable-line prefer-promise-reject-errors
        }
        if (!noConsume) {
          env.ttl(authKey)
            .then((ttl) => {
              if (ttl > 300) {
                env.expire(authKey, 300)
              }
            })
        }
        return db.users.findOne({ _id: data })
      })
      .then((user) => {
        if (!user) {
          return Promise.reject(false) // eslint-disable-line prefer-promise-reject-errors
        }
        env.set(env.keys.USER_ONLINE + user._id, Date.now())
        return user
      })
  }

  function checkGame (req) {
    return db.rooms.findOne({ _id: req.body.room })
      .then((room) => {
        if (!room) {
          return Promise.reject(new Error('invalid room'))
        }
        if (/^(W|E)/.test(req.body.room)) {
          if (room.status === 'out of borders' || (room.openTime && room.openTime > Date.now())) {
            return Promise.reject(new Error('out of borders'))
          }
          return true
        }
        return Promise.reject(new Error('not supported'))
      })
  }
}