bot.py

import sys

import requests
from twitchio import Message
from twitchio.ext import commands

import api as safegif


def delete_chat_message(client_id: str, access_token: str, broadcaster_id: str, moderator_id: str, message_id: str):
    return requests.delete(
        f'https://api.twitch.tv/helix/moderation/chat?broadcaster_id={broadcaster_id}&moderator_id={moderator_id}&message_id={message_id}',
        headers={
            'Client-ID': client_id,
            'Authorization': f'Bearer {access_token}'
        })


def send_whisper(client_id: str, access_token: str, from_user_id: str, to_user_id: str, message: str):
    return requests.post(f'https://api.twitch.tv/helix/whispers?from_user_id={from_user_id}&to_user_id={to_user_id}',
                         headers={
                             'Client-ID': client_id,
                             'Authorization': f'Bearer {access_token}'
                         }, json={'message': message})


class Bot(commands.Bot):
    def __init__(self, client_id: str, access_token: str, initial_channels: list[str], whispers_on_deletion: bool,
                 whisper_text: str = None):
        super().__init__(token=access_token, prefix='safegifbot', initial_channels=initial_channels)
        self.client_id = client_id
        self.access_token = access_token
        self.whispers_on_deletion = whispers_on_deletion
        self.whisper_text = whisper_text

    async def event_ready(self):
        # We are logged in and ready to chat and use commands...
        print(f'Logged in as {self.nick}')
        print(f'User id is {self.user_id}')

    async def event_message(self, message: Message) -> None:
        emotes_str = message.tags.get('emotes')
        if emotes_str is None:
            emotes_str = ''
        if emotes_str.strip() == '':
            return
        emotes = [emote.split(":")[0] for emote in emotes_str.split('/')]
        print(emotes_str)
        print(emotes)
        for emote in emotes:
            if safegif.process_gif(f'https://static-cdn.jtvnw.net/emoticons/v2/{emote}/default/dark/3.0'):
                print(f'https://static-cdn.jtvnw.net/emoticons/v2/{emote}/default/dark/3.0 - True')
                response = requests.get('https://id.twitch.tv/oauth2/validate', headers={
                    'Authorization': f'OAuth {self.access_token}'
                })
                if response.status_code == 200:
                    response = delete_chat_message(
                        client_id=self.client_id,
                        access_token=self.access_token,
                        broadcaster_id=(await message.channel.user()).id,
                        moderator_id=self.user_id,
                        message_id=message.id
                    )
                    if response.status_code == 204:
                        print('Successfully deleted message')
                        if self.whisper_text is not None and self.whispers_on_deletion:
                            response = send_whisper(
                                client_id=self.client_id,
                                access_token=self.access_token,
                                from_user_id=self.user_id,
                                to_user_id=(await message.author.user()).id,
                                message=self.whisper_text
                            )
                            if response.status_code == 204:
                                print(f'Successfully sent or silently dropped whisper message to {message.author.name}')
                            else:
                                print(f'Error sending Whisper: {response.status_code}: {response.text}')
                            break
                else:
                    print('Failed to delete the message. Exiting...')
                    sys.exit()
                break
            else:
                print(f'https://static-cdn.jtvnw.net/emoticons/v2/{emote}/default/dark/3.0 - False')