ToolJet/server/plugins/datasources/slack/index.ts

import { Injectable } from '@nestjs/common';
import { QueryError } from 'src/modules/data_sources/query.error';
import { QueryResult } from 'src/modules/data_sources/query_result.type';
import { QueryService } from 'src/modules/data_sources/query_service.interface';
const got = require('got');

@Injectable()
export default class SlackQueryService implements QueryService {
  authUrl(): string {
    const clientId = process.env.SLACK_CLIENT_ID;
    const tooljetHost = process.env.TOOLJET_HOST;
    return `https://slack.com/oauth/v2/authorize?response_type=code&client_id=${clientId}&redirect_uri=${tooljetHost}/oauth2/authorize`;
  }

  async accessDetailsFrom(authCode: string): Promise<object> {
    const accessTokenUrl = 'https://slack.com/api/oauth.v2.access';
    const clientId = process.env.SLACK_CLIENT_ID;
    const clientSecret = process.env.SLACK_CLIENT_SECRET;
    const tooljetHost = process.env.TOOLJET_HOST;
    const redirectUri = `${tooljetHost}/oauth2/authorize`;

    const body = `code=${authCode}&client_id=${clientId}&client_secret=${clientSecret}&redirect_uri=${redirectUri}`;

    const response = await got(accessTokenUrl, {
      method: 'post',
      body,
      headers: { 'Content-Type': 'application/x-www-form-urlencoded' },
    });

    const result = JSON.parse(response.body);

    if (response.statusCode !== 200) {
      throw Error('could not connect to Slack');
    }

    const authDetails = [];

    if (result['access_token']) {
      authDetails.push(['access_token', result['access_token']]);
    }

    if (result['refresh_token']) {
      authDetails.push(['refresh_token', result['refresh_token']]);
    }

    return authDetails;
  }

  authHeader(token: string): object {
    return { Authorization: `Bearer ${token}` };
  }

  async run(sourceOptions: any, queryOptions: any, dataSourceId: string): Promise<QueryResult> {
    let result = {};
    let response = null;
    const operation = queryOptions.operation;
    const accessToken = sourceOptions['access_token'];

    try {
      switch (operation) {
        case 'list_users':
          response = await got('https://slack.com/api/users.list', {
            method: 'get',
            headers: this.authHeader(accessToken),
          });

          result = JSON.parse(response.body);
          break;

        case 'send_message': {
          const body = {
            channel: queryOptions['channel'],
            text: queryOptions['message'],
            as_user: queryOptions['sendAsUser'],
          };

          response = await got('https://slack.com/api/chat.postMessage', {
            method: 'post',
            json: body,
            headers: this.authHeader(accessToken),
          });

          result = JSON.parse(response.body);
          break;
        }
      }
    } catch (error) {
      throw new QueryError('Query could not be completed', error.message, {});
    }

    return {
      status: 'ok',
      data: result,
    };
  }
}