v1/services/discord.js

// @flow
import Service from './Service'
import superagent from 'superagent'
// import invariant from 'invariant'
import { type AppContext } from '../Roleypoly'
import {
  type Message,
  type Guild,
  type Role,
  type GuildMember,
  type Collection,
  Client
} from 'discord.js'

export type UserPartial = {
  id: string,
  username: string,
  discriminator: string,
  avatar: string
}

export type Permissions = {
  isAdmin: boolean,
  canManageRoles: boolean
}

type ChatCommandHandler = (message: Message, matches: string[], reply: (text: string) => Promise<Message>) => Promise<void>|void
type ChatCommand = {
  regex: RegExp,
  handler: ChatCommandHandler
}

class DiscordService extends Service {
  botToken: string = process.env.DISCORD_BOT_TOKEN || ''
  clientId: string = process.env.DISCORD_CLIENT_ID || ''
  clientSecret: string = process.env.DISCORD_CLIENT_SECRET || ''
  oauthCallback: string = process.env.OAUTH_AUTH_CALLBACK || ''
  isBot: boolean = process.env.IS_BOT === 'true'

  appUrl: string
  botCallback: string
  rootUsers: Set<string>
  client: Client
  cmds: ChatCommand[]
  constructor (ctx: AppContext) {
    super(ctx)
    this.appUrl = ctx.config.appUrl

    this.botCallback = `${this.appUrl}/api/oauth/bot/callback`
    this.rootUsers = new Set((process.env.ROOT_USERS || '').split(','))

    this.client = new Client()
    this.client.options.disableEveryone = true

    this._cmds()

    this.startBot()
  }

  ownGm (server: string) {
    return this.gm(server, this.client.user.id)
  }

  fakeGm ({ id = '0', nickname = '[none]', displayHexColor = '#ffffff' }: $Shape<{ id: string, nickname: string, displayHexColor: string }>) { // eslint-disable-line no-undef
    return { id, nickname, displayHexColor, __faked: true, roles: { has () { return false } } }
  }

  isRoot (id: string): boolean {
    return this.rootUsers.has(id)
  }

  async startBot () {
    await this.client.login(this.botToken)

    // not all roleypolys are bots.
    if (this.isBot) {
      this.log.info('this roleypoly is a bot')
      this.client.on('message', this.handleMessage.bind(this))
      this.client.on('guildCreate', this.handleJoin.bind(this))
    }

    for (let server of this.client.guilds.array()) {
      await this.ctx.server.ensure(server)
    }
  }

  getRelevantServers (userId: string): Collection<string, Guild> {
    return this.client.guilds.filter((g) => g.members.has(userId))
  }

  gm (serverId: string, userId: string): ?GuildMember {
    const s = this.client.guilds.get(serverId)
    if (s == null) {
      return null
    }
    return s.members.get(userId)
  }

  getRoles (server: string) {
    const s = this.client.guilds.get(server)
    if (s == null) {
      return null
    }
    return s.roles
  }

  getPermissions (gm: GuildMember): Permissions {
    if (this.isRoot(gm.id)) {
      return {
        isAdmin: true,
        canManageRoles: true
      }
    }

    return {
      isAdmin: gm.permissions.has('ADMINISTRATOR'),
      canManageRoles: gm.permissions.has('MANAGE_ROLES', true)
    }
  }

  safeRole (server: string, role: string) {
    const rl = this.getRoles(server)
    if (rl == null) {
      throw new Error(`safeRole can't see ${server}`)
    }
    const r: ?Role = rl.get(role)
    if (r == null) {
      throw new Error(`safeRole can't find ${role} in ${server}`)
    }

    return r.editable && !r.hasPermission('MANAGE_ROLES', false, true)
  }

  // oauth step 2 flow, grab the auth token via code
  async getAuthToken (code: string) {
    const url = 'https://discordapp.com/api/oauth2/token'
    try {
      const rsp =
        await superagent
          .post(url)
          .set('Content-Type', 'application/x-www-form-urlencoded')
          .send({
            client_id: this.clientId,
            client_secret: this.clientSecret,
            grant_type: 'authorization_code',
            code: code,
            redirect_uri: this.oauthCallback
          })

      return rsp.body
    } catch (e) {
      this.log.error('getAuthToken failed', e)
      throw e
    }
  }

  async getUser (authToken?: string): Promise<UserPartial> {
    const url = 'https://discordapp.com/api/v6/users/@me'
    try {
      if (authToken == null || authToken === '') {
        throw new Error('not logged in')
      }

      const rsp =
        await superagent
          .get(url)
          .set('Authorization', `Bearer ${authToken}`)
      return rsp.body
    } catch (e) {
      this.log.error('getUser error', e)
      throw e
    }
  }

  // on sign out, we revoke the token we had.
  // async revokeAuthToken (code, state) {
  //   const url = 'https://discordapp.com/api/oauth2/revoke'
  //   try {
  //     const rsp =
  //       await superagent
  //         .post(url)
  //         .send({
  //           client_id: this.clientId,
  //           client_secret: this.clientSecret,
  //           grant_type: 'authorization_code',
  //           code: code,
  //           redirect_uri: this.oauthCallback
  //         })

  //     return rsp.body
  //   } catch (e) {
  //     this.log.error('getAuthToken failed', e)
  //     throw e
  //   }
  // }

  // returns oauth authorize url with IDENTIFY permission
  // we only need IDENTIFY because we only use it for matching IDs from the bot
  getAuthUrl (state: string): string {
    return `https://discordapp.com/oauth2/authorize?client_id=${this.clientId}&redirect_uri=${this.oauthCallback}&response_type=code&scope=identify&state=${state}`
  }

  // returns the bot join url with MANAGE_ROLES permission
  // MANAGE_ROLES is the only permission we really need.
  getBotJoinUrl (): string {
    return `https://discordapp.com/oauth2/authorize?client_id=${this.clientId}&scope=bot&permissions=268435456`
  }

  mentionResponse (message: Message) {
    message.channel.send(`🔰 Assign your roles here! <${this.appUrl}/s/${message.guild.id}>`, { disableEveryone: true })
  }

  _cmds () {
    const cmds = [
      {
        regex: /say (.*)/,
        handler (message, matches, r) {
          r(matches[0])
        }
      },
      {
        regex: /set username (.*)/,
        async handler (message, matches) {
          const { username } = this.client.user
          await this.client.user.setUsername(matches[0])
          message.channel.send(`Username changed from ${username} to ${matches[0]}`)
        }
      },
      {
        regex: /stats/,
        async handler (message, matches) {
          const t = [
            `**Stats** 📈`,
            '',
            `👩‍❤️‍👩 **Users Served:** ${this.client.guilds.reduce((acc, g) => acc + g.memberCount, 0)}`,
            `🔰 **Servers:** ${this.client.guilds.size}`,
            `💮 **Roles Seen:** ${this.client.guilds.reduce((acc, g) => acc + g.roles.size, 0)}`
          ]
          message.channel.send(t.join('\n'))
        }
      }
    ]
      // prefix regex with ^ for ease of code
      .map(({ regex, ...rest }) => ({ regex: new RegExp(`^${regex.source}`, regex.flags), ...rest }))

    this.cmds = cmds
    return cmds
  }

  async handleCommand (message: Message) {
    const cmd = message.content.replace(`<@${this.client.user.id}> `, '')
    this.log.debug(`got command from ${message.author.username}`, cmd)
    for (let { regex, handler } of this.cmds) {
      const match = regex.exec(cmd)
      if (match !== null) {
        this.log.debug('command accepted', { cmd, match })
        try {
          await handler.call(this, message, match.slice(1))
          return
        } catch (e) {
          this.log.error('command errored', { e, cmd, message })
          message.channel.send(`❌ **An error occured.** ${e}`)
          return
        }
      }
    }

    // nothing matched?
    this.mentionResponse(message)
  }

  handleMessage (message: Message) {
    if (message.author.bot || message.channel.type !== 'text') { // drop bot messages and dms
      return
    }

    if (message.mentions.users.has(this.client.user.id)) {
      if (this.rootUsers.has(message.author.id)) {
        this.handleCommand(message)
      } else {
        this.mentionResponse(message)
      }
    }
  }

  async handleJoin (guild: Guild) {
    await this.ctx.server.ensure(guild)
  }
}

module.exports = DiscordService