// @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) => Promise|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 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 { 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 { 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