cocone engineering

import json
import os
import requests


CURRENT_PROJECT_VERSION = "0.0.1"
SLACK_CONVERSATIONS_REPLIES_URL = "https://slack.com/api/conversations.replies" 
SLACK_CHAT_POST_MESSAGE_URL = "https://slack.com/api/chat.postMessage" 

SLACK_TOKEN = os.environ.get('SLACK_TOKEN')
DEEPL_KEY = os.environ.get('DEEPL_KEY')
DEEPL_URL = os.environ.get('DEEPL_URL')
DEBUG_MODE = os.environ.get('DEBUG_MODE')

#data mapping between reaction names and language codes
LANG_CODES = {
  'us': 'en-us', #英語アメリカ
  'gb': 'en-gb', #英語イギリス
  'cn': 'ZH', #中国語
  'de': 'de', #ドイツ語
  'fr': 'fr', #フランス語
  'it': 'it', #イタリア語
  'jp': 'ja', #日本語
  'es': 'es',#スペイン語
  'nl': 'nl', #オランダ語
  'pl': 'pl', #ポーランド語
  'ru': 'ru', #ロシア語
  'flag-ua': 'uk', #Ukrainian
  'kr': 'ko', #韓国語
};
                
def is_challenge(event):
    if 'body' in event:
        body = json.loads(event['body'])
        if ('type' in body) & (body['type'] == 'url_verification'):
            return body['challenge']

def process_message(event):
    if 'body' in event:
        body = json.loads(event['body'])
        event_body = body['event']
        if ('type' in event_body) & (event_body['type'] == 'reaction_added') & ('reaction' in event_body) & (event_body['reaction'] in LANG_CODES):
            event_reaction = event_body['reaction']
            event_item = event_body['item']
            event_channel = event_item['channel']
            event_ts = event_item['ts']
            
            header_message={
                "Authorization": "Bearer {}".format(SLACK_TOKEN)
            }
            
            payload  = {
                "channel" : event_channel,
                "ts" : event_ts
            }
            res = requests.get(SLACK_CONVERSATIONS_REPLIES_URL, headers=header_message, params=payload)
            
            t_data = res.json()
            original_message = t_data['messages'][0]['text']
            
            if(DEBUG_MODE):
                print("original_message : ", original_message)
                
            trans_lang = LANG_CODES[event_reaction]

            # パラメータの指定
            deepl_params = {
                        "auth_key" : DEEPL_KEY,
                        "text" : original_message,
                        "target_lang": trans_lang  # 翻訳後の言語
                    }
            
            deepl_response = requests.post(DEEPL_URL, data=deepl_params) # URIは有償版, 無償版で異なるため要注意 -> 個人用

            trans_data = deepl_response.json()
            translate_messages = trans_data['translations'][0]['text']

            if(DEBUG_MODE):
                print("translate_messages : ", translate_messages)
                
            #判別ロージック追加
            marge_translate_messages = ":"+event_body['reaction']+": "+trans_lang+"\n"+translate_messages
                
            for i in t_data['messages']:
                if ('bot_id' in i) & (i['text'] == marge_translate_messages):
                    return

            chatpayload  = {
                "channel" : event_channel,
                "text" : marge_translate_messages,
                "thread_ts" : event_ts
            }
            
            if(DEBUG_MODE):
                print("marge_translate_messages : ", marge_translate_messages)
            
            res_chat = requests.post(SLACK_CHAT_POST_MESSAGE_URL, headers=header_message, params=chatpayload)
            return res.json()
    return

def lambda_handler(event, context):
    # Return Slack challenge parameter
    clg = is_challenge(event = event)
    
    process_message(event = event)

    return {
        "statusCode": 200,
        "headers": {
            "Content-Type": "application/json"
        },
        "body": json.dumps({
            "challenge ": clg
        })
    }