TotalFreedomBot/events.py at main · TotalFreedomMC/TotalFreedomBot · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
from datetime import datetime

import discord
from discord.ext import commands

import logscript
from functions import read_json, get_avatar, did_mention_other_user, removed_user_mentions, removed_role_mentions
from telnet import telnet
from unicode import confirm, clipboard, cancel

print = logscript.logging.getLogger().critical


class Events(commands.Cog):
    def __init__(self, bot):
        self.bot = bot

    @commands.Cog.listener()
    async def on_ready(self):
        config = read_json('config')
        telnet_ip = config['SERVER_IP']
        telnet_port = config['TELNET_PORT']
        telnet_username = config['TELNET_USERNAME']
        telnet_password = config['TELNET_PASSWORD']

        self.bot.reaction_roles = []
        self.bot.telnet_object = telnet(
            telnet_ip, telnet_port, telnet_username, telnet_password)
        try:
            self.bot.telnet_object.connect()
        except ConnectionError:
            print("Freedom-01 Telnet failed to connect")
        else:
            print(
                f'[{str(datetime.utcnow().replace(microsecond=0))[11:]} INFO]: [TELNET] Bot logged into Telnet as: {self.bot.telnet_object.username}')

        reaction_data = read_json('config')
        self.bot.reaction_roles = reaction_data['reaction_roles']

        self.bot.command_cache = {}
        print("command cache initalized")
        print(f'[{str(datetime.utcnow().replace(microsecond=0))[11:]} INFO]: [Client] {self.bot.user.name} is online.')
        game = discord.Game('play.totalfreedom.me')
        await self.bot.change_presence(status=discord.Status.online, activity=game)

        print(f'[{datetime.utcnow().replace(microsecond=0)} INFO]: [Blacklist] Current blacklist:')
        for user_id in self.bot.blacklisted_users:
            user = self.bot.get_user(user_id)
            if user:
                print(f'[{datetime.utcnow().replace(microsecond=0)} INFO]: [Blacklist] - {user.name}')
            else:
                print(f'[{datetime.utcnow().replace(microsecond=0)} INFO]: [Blacklist] - {user_id}')

        guild_count = len(self.bot.guilds)
        print(
            f'[{str(datetime.utcnow().replace(microsecond=0))[11:]} INFO]: [Guilds] bot currently in {guild_count} guilds.')
        for guild in self.bot.guilds:
            print(
                f'[{str(datetime.utcnow().replace(microsecond=0))[11:]} INFO]: [Guilds] Connected to guild: {guild.name}, Owner: {guild.owner}')

    @commands.Cog.listener()
    async def on_message_edit(self, before, after):
        if not isinstance(before.author, discord.Member):
            return
        if before.guild.id != self.bot.guild_id:
            return
        users = removed_user_mentions(before.mentions, after.mentions)
        roles = removed_role_mentions(
            before.role_mentions, after.role_mentions)
        if users:
            users = ", ".join([str(member) for member in users])
        if roles:
            roles = ", ".join([role.name for role in roles])
        if not users and not roles:
            return
        embed = discord.Embed(
            description="In {}".format(before.channel.mention))
        if users:
            embed.add_field(name="Users", value=users, inline=True)
        if roles:
            embed.add_field(name="Roles", value=roles, inline=True)
        embed.color = 0xFF0000
        embed.title = "Message Edit"
        embed.set_footer(text=str(before.author),
                         icon_url=get_avatar(before.author))
        channel = before.guild.get_channel(self.bot.mentions_channel_id)
        await channel.send(embed=embed)

    @commands.Cog.listener()
    async def on_message_delete(self, message):
        if not isinstance(message.author, discord.Member):
            return
        if message.guild.id != self.bot.guild_id:
            return
        users = None
        roles = None
        if did_mention_other_user(message.mentions, message.author):
            users = ", ".join([str(member) for member in message.mentions])
        if message.role_mentions:
            roles = ", ".join([role.name for role in message.role_mentions])
        if not users and not roles:
            return
        embed = discord.Embed(
            description="In {}".format(message.channel.mention))
        if users is not None:
            embed.add_field(name="Users", value=users, inline=True)
        if roles is not None:
            embed.add_field(name="Roles", value=roles, inline=True)
        embed.color = 0xFF0000
        embed.title = "Message Deletion"
        embed.set_footer(text=str(message.author),
                         icon_url=get_avatar(message.author))
        channel = message.guild.get_channel(self.bot.mentions_channel_id)
        await channel.send(embed=embed)

    @commands.Cog.listener()
    async def on_member_join(self, member):
        pass
        # await member.add_roles(self.bot.get_guild(self.bot.guild_id).get_role(self.bot.verification_role))

    @commands.Cog.listener()
    async def on_command_error(self, ctx, error):
        em = discord.Embed()
        em.title = 'Command Error'
        em.description = f'{error}'
        em.colour = 0xFF0000
        await ctx.send(embed=em)
        print(
            f'[{str(datetime.utcnow().replace(microsecond=0))[11:]} INFO]: [Commands] {ctx.author} (ID: {ctx.author.id}) failed running: {ctx.message.content} in guild: {ctx.guild.name}')

    @commands.Cog.listener()
    async def on_command_completion(self, ctx):
        print(
            f'[{str(datetime.utcnow().replace(microsecond=0))[11:]} INFO]: [Commands] {ctx.author} (ID: {ctx.author.id}) ran: {ctx.message.content} in guild: {ctx.guild.name}')
        bot_logs_channel = self.bot.get_channel(self.bot.bot_logs_channel_id)
        log = discord.Embed(
            title='Logging', description=f'{ctx.author} (ID: {ctx.author.id}) ran: {ctx.message.content}',
            colour=0xA84300)
        await bot_logs_channel.send(embed=log)

    @commands.Cog.listener()
    async def on_raw_reaction_add(self, payload):
        if payload.member == self.bot.user:
            pass
        else:
            for role_id, msg_id, emoji in self.bot.reaction_roles:
                if 'sixx' in payload.member.name.lower():
                    return
                if msg_id == payload.message_id and emoji == str(payload.emoji.name):
                    await payload.member.add_roles(self.bot.get_guild(payload.guild_id).get_role(role_id),
                                                   reason='reaction')
            if payload.channel_id == self.bot.reports_channel_id:
                guild = self.bot.get_guild(self.bot.guild_id)
                reports_channel = self.bot.get_channel(self.bot.reports_channel_id)
                report = await reports_channel.fetch_message(payload.message_id)
                if report.author == guild.me:
                    if payload.emoji.name == clipboard:
                        await report.add_reaction(confirm)
                        await report.add_reaction(cancel)
                    elif payload.emoji.name == cancel:
                        await report.clear_reactions()
                        await report.add_reaction(clipboard)
                    elif payload.emoji.name == confirm:
                        embed = report.embeds[0]
                        archived_reports_channel = self.bot.get_channel(
                            self.bot.archived_reports_channel_id)
                        await report.delete()
                        await archived_reports_channel.send("Handled by " + guild.get_member(payload.user_id).mention,
                                                            embed=embed)

    @commands.Cog.listener()
    async def on_raw_reaction_remove(self, payload):
        if payload.member == self.bot.user:
            pass
        else:
            for role_id, msg_id, emoji in self.bot.reaction_roles:
                if msg_id == payload.message_id and emoji == str(payload.emoji.name):
                    await self.bot.get_guild(payload.guild_id).get_member(payload.user_id).remove_roles(
                        self.bot.get_guild(payload.guild_id).get_role(role_id), reason='reaction')


def setup(bot):
    bot.add_cog(Events(bot))