import { BodyParams, Controller, Get, Inject, PlatformResponse, Post, Req, Res, UseBefore, } from '@tsed/common'; import axios from 'axios'; import { WebAPICallResult } from '@slack/web-api'; import { Installation } from '@slack/oauth'; import { markdownSectionBlock, GrowiCommand, parseSlashCommand, postEphemeralErrors, verifySlackRequest, generateWebClient, InvalidGrowiCommandError, requiredScopes, postWelcomeMessage, REQUEST_TIMEOUT_FOR_PTOG, } from '@growi/slack'; import { Relation } from '~/entities/relation'; import { SlackOauthReq } from '~/interfaces/slack-to-growi/slack-oauth-req'; import { InstallationRepository } from '~/repositories/installation'; import { RelationRepository } from '~/repositories/relation'; import { OrderRepository } from '~/repositories/order'; import { AddSigningSecretToReq } from '~/middlewares/slack-to-growi/add-signing-secret-to-req'; import { AuthorizeCommandMiddleware, AuthorizeInteractionMiddleware } from '~/middlewares/slack-to-growi/authorizer'; import { ExtractGrowiUriFromReq } from '~/middlewares/slack-to-growi/extract-growi-uri-from-req'; import { InstallerService } from '~/services/InstallerService'; import { SelectGrowiService } from '~/services/SelectGrowiService'; import { RegisterService } from '~/services/RegisterService'; import { RelationsService } from '~/services/RelationsService'; import { UnregisterService } from '~/services/UnregisterService'; import { InvalidUrlError } from '../models/errors'; import loggerFactory from '~/utils/logger'; import { JoinToConversationMiddleware } from '~/middlewares/slack-to-growi/join-to-conversation'; const logger = loggerFactory('slackbot-proxy:controllers:slack'); @Controller('/slack') export class SlackCtrl { @Inject() installerService: InstallerService; @Inject() installationRepository: InstallationRepository; @Inject() relationRepository: RelationRepository; @Inject() orderRepository: OrderRepository; @Inject() selectGrowiService: SelectGrowiService; @Inject() registerService: RegisterService; @Inject() relationsService: RelationsService; @Inject() unregisterService: UnregisterService; /** * Send command to specified GROWIs * @param growiCommand * @param relations * @param body * @returns */ private async sendCommand(growiCommand: GrowiCommand, relations: Relation[], body: any) { if (relations.length === 0) { throw new Error('relations must be set'); } const botToken = relations[0].installation?.data.bot?.token; // relations[0] should be exist const promises = relations.map((relation: Relation) => { // generate API URL const url = new URL('/_api/v3/slack-integration/proxied/commands', relation.growiUri); return axios.post(url.toString(), { ...body, growiCommand, }, { headers: { 'x-growi-ptog-tokens': relation.tokenPtoG, }, timeout: REQUEST_TIMEOUT_FOR_PTOG, }); }); // pickup PromiseRejectedResult only const results = await Promise.allSettled(promises); const rejectedResults: PromiseRejectedResult[] = results.filter((result): result is PromiseRejectedResult => result.status === 'rejected'); try { // eslint-disable-next-line @typescript-eslint/no-non-null-assertion return postEphemeralErrors(rejectedResults, body.channel_id, body.user_id, botToken!); } catch (err) { logger.error(err); } } @Post('/commands') @UseBefore(AddSigningSecretToReq, verifySlackRequest, AuthorizeCommandMiddleware, JoinToConversationMiddleware) async handleCommand(@Req() req: SlackOauthReq, @Res() res: Res): Promise { const { body, authorizeResult } = req; let growiCommand; try { growiCommand = parseSlashCommand(body); } catch (err) { if (err instanceof InvalidGrowiCommandError) { res.json({ blocks: [ markdownSectionBlock('*Command type is not specified.*'), markdownSectionBlock('Run `/growi help` to check the commands you can use.'), ], }); } logger.error(err.message); return; } // register if (growiCommand.growiCommandType === 'register') { return this.registerService.process(growiCommand, authorizeResult, body as {[key:string]:string}); } // unregister if (growiCommand.growiCommandType === 'unregister') { if (growiCommand.growiCommandArgs.length === 0) { return 'GROWI Urls is required.'; } if (!growiCommand.growiCommandArgs.every(v => v.match(/^(https?:\/\/)/))) { return 'GROWI Urls must be urls.'; } return this.unregisterService.process(growiCommand, authorizeResult, body as {[key:string]:string}); } const installationId = authorizeResult.enterpriseId || authorizeResult.teamId; // eslint-disable-next-line @typescript-eslint/no-non-null-assertion const installation = await this.installationRepository.findByTeamIdOrEnterpriseId(installationId!); const relations = await this.relationRepository.createQueryBuilder('relation') .where('relation.installationId = :id', { id: installation?.id }) .leftJoinAndSelect('relation.installation', 'installation') .getMany(); if (relations.length === 0) { return res.json({ blocks: [ markdownSectionBlock('*No relation found.*'), markdownSectionBlock('Run `/growi register` first.'), ], }); } // status if (growiCommand.growiCommandType === 'status') { return res.json({ blocks: [ markdownSectionBlock('*Found Relations to GROWI.*'), ...relations.map(relation => markdownSectionBlock(`GROWI url: ${relation.growiUri}`)), ], }); } // Send response immediately to avoid opelation_timeout error // See https://api.slack.com/apis/connections/events-api#the-events-api__responding-to-events res.json({ response_type: 'ephemeral', text: 'Processing your request ...', }); const baseDate = new Date(); const allowedRelationsForSingleUse:Relation[] = []; const allowedRelationsForBroadcastUse:Relation[] = []; const disallowedGrowiUrls: Set = new Set(); // check permission await Promise.all(relations.map(async(relation) => { const isSupportedForSingleUse = await this.relationsService.isSupportedGrowiCommandForSingleUse( relation, growiCommand.growiCommandType, baseDate, ); let isSupportedForBroadcastUse = false; if (!isSupportedForSingleUse) { isSupportedForBroadcastUse = await this.relationsService.isSupportedGrowiCommandForBroadcastUse( relation, growiCommand.growiCommandType, baseDate, ); } if (isSupportedForSingleUse) { allowedRelationsForSingleUse.push(relation); } else if (isSupportedForBroadcastUse) { allowedRelationsForBroadcastUse.push(relation); } else { disallowedGrowiUrls.add(relation.growiUri); } })); // when all of GROWI disallowed if (relations.length === disallowedGrowiUrls.size) { // eslint-disable-next-line @typescript-eslint/no-non-null-assertion const client = generateWebClient(authorizeResult.botToken!); const linkUrlList = Array.from(disallowedGrowiUrls).map((growiUrl) => { return '\n' + `• ${new URL('/admin/slack-integration', growiUrl).toString()}`; }); const growiDocsLink = 'https://docs.growi.org/en/admin-guide/upgrading/43x.html'; return client.chat.postEphemeral({ text: 'Error occured.', channel: body.channel_id, user: body.user_id, blocks: [ markdownSectionBlock('*None of GROWI permitted the command.*'), markdownSectionBlock(`*'${growiCommand.growiCommandType}'* command was not allowed.`), markdownSectionBlock( `To use this command, modify settings from following pages: ${linkUrlList}`, ), markdownSectionBlock( `Or, if your GROWI version is 4.3.0 or below, upgrade GROWI to use commands and permission settings: ${growiDocsLink}`, ), ], }); } // select GROWI if (allowedRelationsForSingleUse.length > 0) { body.growiUrisForSingleUse = allowedRelationsForSingleUse.map(v => v.growiUri); return this.selectGrowiService.process(growiCommand, authorizeResult, body); } // forward to GROWI server if (allowedRelationsForBroadcastUse.length > 0) { return this.sendCommand(growiCommand, allowedRelationsForBroadcastUse, body); } } @Post('/interactions') @UseBefore(AuthorizeInteractionMiddleware, ExtractGrowiUriFromReq) async handleInteraction(@Req() req: SlackOauthReq, @Res() res: Res): Promise { logger.info('receive interaction', req.authorizeResult); logger.debug('receive interaction', req.body); const { body, authorizeResult } = req; // pass if (body.ssl_check != null) { return; } const installationId = authorizeResult.enterpriseId || authorizeResult.teamId; // eslint-disable-next-line @typescript-eslint/no-non-null-assertion const installation = await this.installationRepository.findByTeamIdOrEnterpriseId(installationId!); const payload = JSON.parse(body.payload); const callBackId = payload?.view?.callback_id; // register if (callBackId === 'register') { try { await this.registerService.insertOrderRecord(installation, authorizeResult.botToken, payload); } catch (err) { if (err instanceof InvalidUrlError) { logger.info(err.message); return; } logger.error(err); } await this.registerService.notifyServerUriToSlack(authorizeResult.botToken, payload); return; } // unregister if (callBackId === 'unregister') { await this.unregisterService.unregister(installation, authorizeResult, payload); return; } // forward to GROWI server if (callBackId === 'select_growi') { // Send response immediately to avoid opelation_timeout error // See https://api.slack.com/apis/connections/events-api#the-events-api__responding-to-events res.send(); const selectedGrowiInformation = await this.selectGrowiService.handleSelectInteraction(installation, payload); return this.sendCommand(selectedGrowiInformation.growiCommand, [selectedGrowiInformation.relation], selectedGrowiInformation.sendCommandBody); } // Send response immediately to avoid opelation_timeout error // See https://api.slack.com/apis/connections/events-api#the-events-api__responding-to-events res.send(); /* * forward to GROWI server */ const relation = await this.relationRepository.findOne({ installation, growiUri: req.growiUri }); if (relation == null) { logger.error('*No relation found.*'); return; } try { // generate API URL const url = new URL('/_api/v3/slack-integration/proxied/interactions', req.growiUri); await axios.post(url.toString(), { ...body, }, { headers: { 'x-growi-ptog-tokens': relation.tokenPtoG, }, timeout: REQUEST_TIMEOUT_FOR_PTOG, }); } catch (err) { logger.error(err); } } @Post('/events') async handleEvent(@BodyParams() body:{[key:string]:string} /* , @Res() res: Res */): Promise { // eslint-disable-next-line max-len // see: https://api.slack.com/apis/connections/events-api#the-events-api__subscribing-to-event-types__events-api-request-urls__request-url-configuration--verification if (body.type === 'url_verification') { return body.challenge; } logger.info('receive event', body); return; } @Get('/oauth_redirect') async handleOauthRedirect(@Req() req: Req, @Res() serverRes: Res, @Res() platformRes: PlatformResponse): Promise { // create 'Add to Slack' url const addToSlackUrl = await this.installerService.installer.generateInstallUrl({ scopes: requiredScopes, }); const state = req.query.state; if (state == null || state === '') { return platformRes.status(400).render('install-failed.ejs', { url: addToSlackUrl }); } // promisify const installPromise = new Promise((resolve, reject) => { this.installerService.installer.handleCallback(req, serverRes, { success: async(installation, metadata) => { logger.info('Success to install', { installation, metadata }); resolve(installation); }, failure: async(error) => { reject(error); // go to catch block }, }); }); let httpStatus = 200; let httpBody; try { const installation = await installPromise; // check whether bot is not null if (installation.bot == null) { logger.warn('Success to install but something wrong. `installation.bot` is null.'); httpStatus = 500; httpBody = await platformRes.render('install-succeeded-but-has-problem.ejs', { reason: '`installation.bot` is null' }); } // MAIN PATH: everything is fine else { const appPageUrl = `https://slack.com/apps/${installation.appId}`; httpBody = await platformRes.render('install-succeeded.ejs', { appPageUrl }); // generate client const client = generateWebClient(installation.bot.token); const userId = installation.user.id; await Promise.all([ // post message postWelcomeMessage(client, userId), // publish home // TODO When Home tab show off, use bellow. // publishInitialHomeView(client, userId), ]); } } catch (error) { logger.error(error); httpStatus = 500; httpBody = await platformRes.status(400).render('install-failed.ejs', { url: addToSlackUrl }); } platformRes.status(httpStatus); return httpBody; } }