helpers_slack.py

import os
import json
from time import time

from slack_bolt import App
from slack_sdk.errors import SlackApiError

from main import DETAILS_BLOCK_ID
from main import USER_CANCEL_RETURN_CODE
from main import REPORT_FILE

SLACK_MESSAGE_SIZE_LIMIT = 3001


def init_app(slack_bot_token, approve_action_id, cancel_action_id):
    app = App(token=slack_bot_token)

    @app.action(approve_action_id)
    def approve_request(ack, respond, body):
        # Acknowledge action request
        ack()
        print(body)
        print('User pressed approve. update message and exit 0')
        username = body['user']['username']
        original_text = ''
        for block in body['message']['blocks']:
            if 'block_id' in block and block['block_id'] == DETAILS_BLOCK_ID:
                original_text = block['text']['text']
        respond(original_text + f'\n\nApproved by {username} 👍')
        print(f'Approved by user {username}. Exit with shell return code 0')
        gen_report(
            [body['user']['username']],
            [body['team']['domain']],
            body['channel']['name'],
            body['message']['text'],
            0
        )
        os._exit(0)

    # pylint and flake8 thinks that we are redefining the function but
    # we are actually not
    # pylint: disable=E0102
    @app.action(cancel_action_id)
    def approve_request(ack, client, body):  # noqa: F811
        # Acknowledge action request
        ack()
        print(body)
        header_text = ''
        for block in body['message']['blocks']:
            if 'type' in block and block['type'] == 'header':
                header_text = block['text']['text']
        print('User pressed cancel. Delete message and exit 1')
        client.chat_delete(channel=body['container']['channel_id'], ts=body['container']['message_ts'])
        app.client.chat_postEphemeral(
            channel=body['container']['channel_id'],
            user=body['user']['id'],
            text=f'{header_text} canceled by you and deleted to keep approval log nice and clean'
        )
        print(f'Canceled by user. Exit with shell return code {USER_CANCEL_RETURN_CODE}')
        gen_report(
            [body['user']['username']],
            [body['team']['domain']],
            body['channel']['name'],
            body['message']['text'],
            USER_CANCEL_RETURN_CODE
        )
        os._exit(USER_CANCEL_RETURN_CODE)

    @app.middleware
    def middleware_func(logger, body, next):
        logger.info(f"request body: {body}")
        next()

    return app


def gen_report(usernames, teams, channel, message, approval_code):
    report = {
        'timestamp': int(time()),
        'approval_code': approval_code,
        'usernames': usernames,
        'teams': teams,
        'channel': channel,
        'message': message,
    }
    with open(REPORT_FILE, "w") as outfile:
        json.dump(report, outfile)


def user_id_by_email(app, email):
    try:
        result = app.client.users_lookupByEmail(email=email)
        return result['user']['id']
    except SlackApiError as err:
        if err.response['error'] == 'users_not_found':
            return None

    return None


def is_message_longer_than_limit(message):
    return True if len(message) > SLACK_MESSAGE_SIZE_LIMIT else False


def truncate_message_if_needed(message, append_message_if_truncated):
    # Return message as is if its len is below the limit
    if not is_message_longer_than_limit(message):
        return message
    # Otherwise truncate and append the message
    print(f'have to truncate the message sinse its size is {len(message)} > {SLACK_MESSAGE_SIZE_LIMIT}')
    size_available = SLACK_MESSAGE_SIZE_LIMIT - len(append_message_if_truncated) - len('\n') - 1
    truncated_message = message[:size_available] + '\n' + append_message_if_truncated
    print(f'size of truncated message {len(truncated_message)}')
    return truncated_message