main.py

import os
import random
from asyncio import sleep
from datetime import timedelta
from re import search, sub

import discord
import pymongo
from discord import message
from discord.ext import commands, tasks
from discord.utils import get

from config import *
from util import *

# mongo database
client = pymongo.MongoClient(MONGODB_ATLAS_URI)
db = client['discord-bot']['discord-bot']

# Manage members permission
intents = discord.Intents.default()
intents.members = True

# Bot client
bot = commands.Bot(command_prefix='>', intents=intents)

# Refreshes the sheets' commands and triggers
def refreshBot():
    # Sets use of global variables
    global spreadsheet; global commandSheet; global triggerSheet

    spreadsheet, commandSheet, triggerSheet, isEmpty = refreshSheet()
    refreshCogs(bot)

    return isEmpty

# When the bot has finished loading after being launched
@bot.event
async def on_ready():
    print("\n [*] The bot is running.")

    await bot.change_presence(activity=discord.Activity(type=discord.ActivityType.listening, name=">help"))
    print("\n [*] The bot's status was successfully set.")

    periodicRefresh.start()
    print("\n [*] The periodic refresh task was successfully started.")

# Whenever a member joins the serverpp
@bot.event
async def on_member_join(member):
    print(f"\n [*] {member.display_name} has joined the server.")
    
    # Gets all roles that sould be added to the members
    roles = db.find_one({"description": "onMemberJoinRoles"})['roles']
    roles = list(filter(lambda r: r, map(lambda r: get(member.guild.roles, name=r), roles)))

    # If there are any roles to be added, add them to the member that just joined
    if roles: await member.add_roles(roles if len(roles) > 1 else roles[0])

    # Sends a welcome message at the WELCOME_CHANNEL
    channel = get(member.guild.text_channels, name=WELCOME_CHANNEL)
    response = await channel.send(f'{member.mention} Seja bem vindo(a), meu {random.choice(VOCATIVES)}!')
    print("   [**] The welcome message was successfully sent.")
    
    await reactToResponse(bot, response)

# Whenever a new message is sent to a channel the bot has access to
@bot.event
async def on_message(message):
    if message.author == bot.user: return

    currMessage = message.content
    currSheet = triggerSheet
    messageLog = "Trigger"
    messageName = "TRIGGER"
    messageAliases = "TRIGGER ALIASES"

    if message.content.startswith(">") :
        
        currSheet = commandSheet
        messageLog = "Command"
        messageName = "COMMAND NAME"
        messageAliases = "COMMAND ALIASES"
        currMessage = message.content[1:]
    # Checks for all triggers listed in the spreadsheet

    for element in currSheet:

        compareOptions = element[messageAliases]
        compareOptions = compareOptions.split('\n') if compareOptions != '' else []
        compareOptions.append(element[messageName])

        if currMessage and currMessage.lower() in filter(lambda e: e, compareOptions):
            print(f"\n [*] '{messageLog}': '{currMessage}', by {message.author.display_name}.")

            await reactToMessage(bot, message, [MESSAGE_EMOJI])

            # gets image attatchment
            img = getImage(element["RESPONSE IMAGE"])

            # activates text-to-speech if specified
            tts = 'True' if element['TTS'] == 'TRUE' else 'False'

            # If an image link was specified
            if img: 
                response = await message.channel.send(content=element["RESPONSE TEXT"], file=discord.File(img), tts=tts)
                os.remove(img) # Deletes the image from local directory

            else:
                response = await message.channel.send(content=element["RESPONSE TEXT"], tts=tts)

            print("   [**] The response was successfully sent.")

            await reactToResponse(bot, response)

            return

    await bot.process_commands(message)

# Bot's developer
@bot.command(brief='Desenvolvedor do bot e repositório no GitHub.', aliases=['créditos', 'creditos', 'dev'])
async def credits(ctx):
    await ctx.trigger_typing()

    print("\n [*] '>credits' command called.")

    await reactToMessage(bot, ctx.message, ['🤙', '🍉', '😁', '💜', '👋'])

    response = await ctx.reply("Esse bot foi desenvolvido pelo Flip em um momento de tédio. Obrigado pelo interesse <3 \nGitHub: https://github.com/lucasvianav. \nRepositório no GitHub: https://github.com/lucasvianav/discord-chatbot.")
    print("   [**] The response was successfully sent.")
    await reactToResponse(bot, response)

# Calls refreshSheet() (definition in config.py)
@bot.command(brief='Atualiza os comandos a partir da planilha.', aliases=['atualizar', 'update'])
async def refresh(ctx):
    await ctx.trigger_typing()

    print("\n [*] '>refresh' command called.")

    await reactToMessage(bot, ctx.message, ['🔝', '👌', '🆗'])

    isEmpty = refreshBot()

    if not isEmpty:
        print("   [**] The commands and triggers were successfully updated.")
        response = await ctx.send("Os comandos e triggers foram atualizados com sucesso.")
        print("   [**] The response was successfully sent.")
        await reactToResponse(bot, response)

    else:
        print("   [**] There are no commands nor triggers registered.")
        response = await ctx.send("Não há comandos nem triggers cadastrados.")
        print("   [**] The response was successfully sent.")
        await reactToResponse(bot, response, emojiList=['😢'])

# deletes all messages sent by the bot or that triggered it
# (does not delete "important" messages, like from >trackPresence and >openProjects commands)
@bot.command(
    aliases=['clean', 'limpar'],
    brief='Limpa o chat,',
    help='Exclui todas os comandos e mensagens do bot que foram enviadas nos últimos 10 minutos.'
)
async def clear(ctx):
    await ctx.trigger_typing()

    print(f'\n [*] \'>clear\' command called on the {ctx.channel.name} channel.')

    await reactToMessage(bot, ctx.message, ['⚰️'])
    
    timestamp = ctx.message.created_at - timedelta(minutes=10)
    
    def filterFunction(m):
        notPinned = not m.pinned 
        isMe = (m.author == bot.user)
        isImportant = search('`\[ABERTURA DE PROJETOS\]`\n\n', m.content) or search('`\[PRESENÇA DE REUNIÃO\]`\n\n', m.content) or search('`\[VOTAÇÃO\]`\n\n', m.content)
        # hasPrefix = search('^>\S.*$', m.content)
        didIReact = get(m.reactions, me=True)
        isHelp = m.content == '>help'

        return isHelp or (notPinned and not isImportant and (isMe or didIReact))

    deleted = await ctx.channel.purge(after=timestamp, check=filterFunction)

    response = await ctx.send(f'`{len(deleted)} mensagens foram excluídas.`', delete_after=20)

    await reactToResponse(bot, response)

# Sends a passed message to the specified channel
@bot.command(
    brief='Manda uma mensagem no canal escolhido.',
    help='Sintaxe: >send "$MENSAGEM" @ "$CANAL"\n\nEsse comando vai enviar a $MENSAGEM no $CANAL, em que $CANAL pode ser tanto o nome do canal desejado quanto sua mention.\n\nPor exemplo, o comando a seguir vai enviar "Oi! Tudo bem?" no canal "random":\n>send "Oi! Tudo bem?" @ "random".\n\nOBS: Tanto a mensagem quanto o canal desejado devem estar entre aspas e deve haver exatamente um espaço entre cada bloco do comando (mensagem, @ e canal). Caso você queira que sua mensagem contenha uma arroba, escreva dentro dela \@.\n\n(esse comando foi feito especialmente para o Júlio <3)',
    aliases=[]
)
async def send(ctx, *argv):
    await ctx.trigger_typing()

    print('\n [*] \'>send\' command called.')
    await reactToMessage(bot, ctx.message, ['🆗', '📢'])

    argv = " ".join(argv)
    argv = argv.split(' @ ')
    
    # checks if arguments are valid
    if len(argv) != 2 or not search('[^"]', argv[0]) or not search('[^"]', argv[1]):
        response = 'Os argumentos passados são inválidos. Para mais informações, envie ">help send".'

    else:
        message = sub('^"(.+)"$', '\g<1>', argv[0]).replace('\@', '@')
        channelRef = sub('^"(.+)"$', '\g<1>', argv[1]).replace('\@', '@')
        
        channel = get(ctx.guild.text_channels, name=channelRef) or get(ctx.guild.text_channels, mention=channelRef)
        
        # if the channel exists
        if channel:
            permissionsUser= channel.permissions_for(ctx.author)
            permissionMe = channel.permissions_for(ctx.guild.me)

            # checks permissions
            if not permissionMe.send_messages or not permissionsUser.read_messages:
                response = f'Infelizmente eu não tenho permissão para mandar/visualizar mensagens no {channel.mention}... *O que será que eles têm a esconder, hein?!*'

            # if all permissions are ok
            elif permissionsUser.send_messages and permissionsUser.read_messages:
                message = await channel.send(message)
                await reactToMessage(bot, message, ['🤠', '📢'])
        
                response = f'A mensagem foi enviada com sucesso no {channel.mention}.'
                
            else: 
                response = f'Infelizmente você não tem permissão para mandar mensagens {channel.mention}'

        # if the channel does not exist
        else:
            response = f'Infelizmente o canal passado (`{channelRef}`) não foi encontrado.'

    response = await ctx.reply(response)

    await reactToResponse(bot, response)

# Refreshes the sheets' commands and triggers every 15 minutes
@tasks.loop(minutes=15)
async def periodicRefresh():
    print("\n [*] Time for a periodic refresh.")

    isEmpty = refreshBot()

    print("   [**] The commands and triggers were successfully updated", end="")
    print(" - none registered.") if isEmpty else print(".")


if __name__ == '__main__':
    refreshCogs(bot, hasLoaded=False)
    bot.run(DISCORD_TOKEN)