import os import re import time from datetime import datetime import discord from discord.ext import commands from dotenv import load_dotenv import logscript from functions import config_entry, get_prefix, hit_endpoint, get_server_status from unicode import clipboard #print = logscript.logging.getLogger().critical load_dotenv() botToken = os.getenv('botToken') intents = discord.Intents.all() bot = commands.Bot(command_prefix=get_prefix, case_insensitive=True, description='TotalFreedom bot help command', intents=intents) bot.server_liaison = config_entry("server_liaison") bot.event_host = config_entry("event_host") bot.server_banned = config_entry("server_banned") bot.senior_admin = config_entry("senior_admin") bot.admin = config_entry("admin") bot.master_builder = config_entry("master_builder") bot.reports_channel_id = config_entry("reports_channel_id") bot.archived_reports_channel_id = config_entry("archived_reports_channel_id") bot.guild_id = config_entry("guild_id") bot.mentions_channel_id = config_entry("mentions_channel_id") bot.discord_admin = config_entry("discord_admin") bot.discord_mod = config_entry("discord_mod") bot.devs = config_entry("devs") bot.bot_logs_channel_id = config_entry("bot_logs_channel_id") bot.executive = config_entry("executive") bot.asst_exec = config_entry("asst_exec") bot.developer = config_entry("developer") bot.creative_designer = config_entry("creative_designer") bot.server_chat = config_entry("server_chat") bot.verification_role = config_entry("verification_role") bot.server_chat_2 = config_entry("server_chat_2") bot.guild_id = 769659653096472627 bot.smp_server_chat = config_entry("smp_server_chat") bot.smpstaff_role_id = config_entry("smpstaff_role_id") bot.smp_owner_id = config_entry("smp_owner_id") bot.gmod_server_chat = config_entry("gmod_server_chat") bot.gmodstaff_role_id = config_entry("gmodstaff_role_id") bot.gmod_owner_id = config_entry('gmod_owner_id') bot.auto_restart = False extensions = [ "commands.moderation", "commands.server_commands", "commands.help", "commands.miscellaneous", "events" ] if __name__ == '__main__': for extension in extensions: try: bot.load_extension(extension) print( f"[{str(datetime.utcnow().replace(microsecond=0))[11:]} INFO]: [Extensions] {extension} loaded " f"successfully") except Exception as e: print( f"[{str(datetime.utcnow().replace(microsecond=0))[11:]} INFO]: [Extensions] {extension} didn't load {e}" ) @bot.event async def on_message(message): if isinstance(message.channel, discord.channel.DMChannel): print(f'{message.author} DM: {message.content}') return elif message.guild and message.author is message.guild.me and message.channel.id == bot.reports_channel_id: await message.add_reaction(clipboard) elif message.type == discord.MessageType.new_member: if re.search('discord\.gg\/[a-zA-z0-9\-]{1,16}', message.author.name.lower()) or re.search( 'discordapp\.com\/invite\/[a-z0-9]+/ig', message.author.name.lower()): await message.author.ban(reason="Name is an invite link.") await message.delete() bypass_roles = [bot.discord_admin, bot.discord_mod] bypass = False if message.author != bot.user: for role in message.author.roles: if role.id in bypass_roles: bypass = True asked_for_ip = re.search("(((give *(me)?)|(wh?at'? *i?s?))( *the)?( *server)? *ip)|(ip\?)", message.content.lower()) if asked_for_ip is not None: print("THE IP IS GAY") em = discord.Embed( title='Server IP', colour=0xA84300, description='play.totalfreedom.me' ) await message.channel.send(embed=em) if bot.auto_restart: server_chats = [bot.server_chat, bot.server_chat_2] for server in range(1, 3): if message.channel.id == server_chats[server]: if not get_server_status(server): em = discord.Embed() em.title = 'Automatic restart' attempt = hit_endpoint('start', server) if attempt == "ERROR - There is an instance of the server already running. Make sure it is " \ "killed first and try again": hit_endpoint('stop', server) hit_endpoint('stop', server) hit_endpoint('start', server) em.colour = 0x00FF00 em.description = 'Server has been automatically restarted.' else: em.colour = 0x00FF00 em.description = 'Server has been automatically started.' await message.channel.send(embed=em) if 'Server has started' in message.content: # Telnet reconnect script try: bot.telnet_object.connect() except Exception as err: print(f'Failed to reconnect telnet: {err}') time.sleep(5) try: bot.telnet_object.connect() except Exception as fuckup: print( f'Second attempt failed to reconnect telnet: {fuckup}') try: bot.telnet_object_2.connect() except Exception as err: print(f'Failed to reconnect telnet 2: {err}') time.sleep(5) try: bot.telnet_object_2.connect() except Exception as fuckup: print( f'Second attempt failed to reconnect telnet 2: {fuckup}') if not bypass: if re.search('discord\.gg\/[a-zA-z0-9\-]{1,16}', message.content) or re.search( 'discordapp\.com\/invite\/[a-z0-9]+/ig', message.content): await message.delete() await message.channel.send(f"{message.author.mention} do not post invite links to other discord servers.") await bot.process_commands(message) bot.run(botToken)