@xrengine/server-core/src/user/discord-bot-auth/discord-bot-auth.class.ts

Shared components for XREngine server

import { errors } from '@feathersjs/errors'
import { Paginated, Params, ServiceMethods } from '@feathersjs/feathers'
import axios from 'axios'
import { SequelizeServiceOptions } from 'feathers-sequelize/types'

import { IdentityProviderInterface } from '@xrengine/common/src/dbmodels/IdentityProvider'

import { Application } from '../../../declarations'
import logger from '../../ServerLogger'

export class DicscordBotAuth<T = any> implements Partial<ServiceMethods<T>> {
  app: Application
  docs: any
  options: Partial<SequelizeServiceOptions>

  constructor(options: Partial<SequelizeServiceOptions>, app: Application) {
    this.options = options
    this.app = app
  }

  async find(params?: Params): Promise<any> {
    const url = `https://discord.com/api/users/@me`
    try {
      const authResponse = await axios.get(url, {
        headers: {
          Authorization: `Bot ${params!.query!.bot_token}`
        }
      })
      const resData = authResponse.data
      if (!resData?.bot) throw new Error('The authenticated Discord user is not a bot')
      const token = `discord:::${resData.id}`
      const ipResult = (await this.app.service('identity-provider').find({
        query: {
          token: token,
          type: 'discord'
        }
      })) as Paginated<IdentityProviderInterface>
      if (ipResult.total > 0) {
        return this.app.service('user').get(ipResult.data[0].userId)
      } else {
        const ipCreation = await this.app.service('identity-provider').create(
          {
            token: token,
            type: 'discord'
          },
          {
            bot: true
          }
        )
        return this.app.service('user').get(ipCreation.userId)
      }
    } catch (err) {
      logger.error(err)
      if (errors[err.response.status]) {
        throw new errors[err.response.status]()
      } else {
        throw new Error(err)
      }
    }
  }
}