import asyncio import datetime import json import os import subprocess import unicodedata from collections import namedtuple from datetime import timedelta from math import floor from random import choice, randint import discord from discord.ext import commands from pytz import exceptions, timezone # forgot to import this (randint) and ended up looking mentally unstable # troll literally pointed out atleast 4 things I did wrong in 3 lines of code # from PIL.Image import core as Image # import image as Image # from PIL import Image, ImageFilter # from selenium import webdriver # from selenium.webdriver.chrome.options import Options # def set_viewport_size(driver, width, height): # window_size = driver.execute_script( # """ # return [window.outerWidth - window.innerWidth + arguments[0], # window.outerHeight - window.innerHeight + arguments[1]]; # """, # width, # height, # ) # driver.set_window_size(*window_size) async def reportStuff(self, ctx, message): channel = self.bot.get_channel( int(self.bot.config[str(ctx.message.guild.id)]["report_channel"]) ) embed = discord.Embed( title=f"Report from {ctx.message.author}", description=f"{message}", color=ctx.message.author.color, timestamp=ctx.message.created_at, ) await channel.send(embed=embed) await ctx.author.send("Report has been submitted") # def save_leaderboard(): # DRIVER = "/usr/lib/chromium-browser/chromedriver" # chrome_options = Options() # chrome_options.add_argument("--disable-dev-shm-usage") # chrome_options.add_argument("--headless") # chrome_options.add_argument("--no-sandbox") # chrome_options.add_argument("--disable-gpu") # # chrome_options.binary_location = "" # driver = webdriver.Chrome(DRIVER, chrome_options=chrome_options) # set_viewport_size(driver, 1000, 1100) # driver.get("https://aninternettroll.github.io/mcbeVerifierLeaderboard/") # screenshot = driver.find_element_by_id("table").screenshot("leaderboard.png") # driver.quit() # # transparency time # img = Image.open("leaderboard.png") # img = img.convert("RGB") # pallette = Image.open("palette.png") # pallette = pallette.convert("P") # img = img.quantize(colors=256, method=3, kmeans=0, palette=pallette) # img = img.convert("RGBA") # datas = img.getdata() # # newData = [] # for item in datas: # if item[0] == 255 and item[1] == 255 and item[2] == 255: # newData.append((255, 255, 255, 0)) # else: # newData.append(item) # # img.putdata(newData) # """ # img = img.filter(ImageFilter.SHARPEN) # img = img.filter(ImageFilter.SHARPEN) # img = img.filter(ImageFilter.SHARPEN) # """ # # height, width = img.size # # img = img.resize((height*10,width*10), resample=Image.BOX) # img.save("leaderboard.png", "PNG") class Utils(commands.Cog): def __init__(self, bot): self.bot = bot self.tries = 1 self.pins = [] self.block_stats = [] with open("./required_block_states.json") as file: required_block_states = json.load(file) for block, states in required_block_states["minecraft"].items(): for s in states: self.block_stats.append(block) @commands.command( description="Pong!", help="Tells the ping of the bot to the discord servers", brief="Tells the ping", ) async def ping(self, ctx): await ctx.send(f"Pong! {round(self.bot.latency*1000)}ms") @commands.cooldown(1, 2, commands.BucketType.user) @commands.command() async def findblock(self, ctx): """Simulate the UGBC experience""" if ctx.message.channel.id != int( self.bot.config[str(ctx.message.guild.id)]["bot_channel"] ): await ctx.message.delete() ctx.command.reset_cooldown(ctx) return await ctx.send( f"{discord.utils.escape_mentions(ctx.message.author.display_name)} -> your block is a {choice(self.block_stats)}." ) @commands.cooldown(1, 25, commands.BucketType.guild) @commands.command() async def findseed(self, ctx): """Test your luck""" if ctx.message.channel.id != int( self.bot.config[str(ctx.message.guild.id)]["bot_channel"] ): await ctx.message.delete() ctx.command.reset_cooldown(ctx) return # Don't ask rigged_findseed = { 280428276810383370: 12, # Thomas's User ID 199070670221475842: -1, # Kai's User ID # What a fraud! They didn't use their real name. 615658069132836865: -12, } if ctx.author.id in rigged_findseed: total_eyes = rigged_findseed[ctx.author.id] else: total_eyes = sum(1 for i in range(12) if randint(1, 10) == 1) await ctx.send( f"{discord.utils.escape_mentions(ctx.message.author.display_name)} -> your seed is a {total_eyes} eye" ) @commands.command() async def findsleep(self, ctx): """Test your sleep""" if ctx.message.channel.id != int( self.bot.config[str(ctx.message.guild.id)]["bot_channel"] ): await ctx.message.delete() return # DON'T ASK! lessSleepMsg = ( "gn, insomniac!", "counting sheep didn't work? try counting chloroform vials!", "try a glass of water", "some decaf coffee might do the trick!", ) moreSleepMsg = ( "waaakeee uuuppp!", "are they dead or asleep? I can't tell.", "wake up, muffin head", "psst... coffeeee \\:D", ) # Optional TODO: Create non-normal distribution sleepHrs = randint(0, 24) # Add extra comment based on number of sleepHrs if sleepHrs == 0: await ctx.send( f"{discord.utils.escape_mentions(ctx.message.author.display_name)} -> your sleep is 0 hours long - nice try :D" ) elif sleepHrs <= 5: if sleepHrs == 1: s = "" else: s = "s" await ctx.send( f"{discord.utils.escape_mentions(ctx.message.author.display_name)} -> your sleep is {sleepHrs} hour{s} long - {lessSleepMsg[randint(0, len(lessSleepMsg) - 1)]}" ) else: await ctx.send( f"{discord.utils.escape_mentions(ctx.message.author.display_name)} -> your sleep is {sleepHrs} hours long - {moreSleepMsg[randint(0, len(moreSleepMsg) - 1)]}" ) @commands.Cog.listener() async def on_reaction_add(self, reaction, user): if ( reaction.emoji == "⭐" and not reaction.message.id in self.pins and reaction.count >= 5 ): self.pins.append(reaction.message.id) embed = discord.Embed( title="**New Starred Message**", description=reaction.message.content, colour=discord.Colour(0xB92C36), url=reaction.message.jump_url, timestamp=reaction.message.created_at, ) for attachement in reaction.message.attachments: if attachement.height: embed.set_image(url=attachement.url) embed.set_author( name=str(reaction.message.author), icon_url=reaction.message.author.avatar_url_as(format="png"), ) embed.set_footer(text=reaction.message.id) # embed.add_field(name="Stars", value=reaction.count) channel = self.bot.get_channel( int(self.bot.config[str(reaction.message.guild.id)]["pins_channel"]) ) await channel.send(embed=embed) @commands.Cog.listener() async def on_member_join(self, member): def check(msg): return msg.author == member and msg.type != discord.MessageType.new_member try: msg = await self.bot.wait_for("message", check=check, timeout=300) await msg.channel.send("<:PeepoPog:732172337956257872>") await msg.channel.send( f"{member.mention}\n" + "controller is allowed on mobile\n" + "keyboard and mouse is allowed on console\n" + "submit the run as the physical device you played on" ) except asyncio.TimeoutError: pass @commands.Cog.listener() async def on_message(self, message): if not message.guild: return if message.author.bot: return if ( "<@!280428276810383370>" in message.content or "<@280428276810383370>" in message.content ): await message.channel.send("<:MangoPing:760286455238361129>") badWords = ("fair", "ⓕⓐⓘⓡ") count = 0 year = datetime.date.today().year CoolKids = namedtuple( "CoolKid", [ "name", "user", "bday", ], ) coolKids = ( CoolKids( name="Cameron", user=self.bot.get_user(468262902969663488), bday=datetime.date(year, 10, 8), ), CoolKids( name="Indy", user=self.bot.get_user(274923326890311691), bday=datetime.date(year, 9, 10), ), CoolKids( name="Kai", user=self.bot.get_user(199070670221475842), bday=datetime.date(year, 11, 26), ), CoolKids( name="Luca", user=self.bot.get_user(99457716614885376), bday=datetime.date(year, 11, 5), ), CoolKids( name="Max", user=self.bot.get_user(543958509243596800), bday=datetime.date(year, 11, 10), ), CoolKids( name="Daniel", user=self.bot.get_user(264121998173536256), bday=datetime.date(year, 7, 6), ), # Mistaken CoolKids( name="Jake", user=self.bot.get_user(400344183333847060), bday=datetime.date(year, 11, 10), ), CoolKids( name="Faith", user=self.bot.get_user(744383381843738633), bday=datetime.date(year, 11, 5), ), # Prakxo CoolKids( name="Samanta", user=self.bot.get_user(226312219787264000), bday=datetime.date(year, 6, 24), ), CoolKids( name="Scott", user=self.bot.get_user(223937483774230528), bday=datetime.date(year, 6, 23), ), CoolKids( name="Kyra", user=self.bot.get_user(329538915805691905), bday=datetime.date(year, 6, 24), ), CoolKids( name="Thomas", user=self.bot.get_user(280428276810383370), bday=datetime.date(year, 9, 29), ), # Zyemlus CoolKids( name="Samantha", user=self.bot.get_user(536071288859656193), bday=datetime.date(year, 8, 23), ), CoolKids( name="Landen", user=self.bot.get_user(654025117025828885), bday=datetime.date(year, 8, 24), ), CoolKids( name="Oceanlight", user=self.bot.get_user(615658069132836865), bday=datetime.date(year, 4, 9), ), # Shadowfi CoolKids( name="Daniel", user=self.bot.get_user(586664256217415681), bday=datetime.date(year, 11, 14), ), CoolKids( name="Sawyer", user=self.bot.get_user(404873210597867541), bday=datetime.date(year, 9, 12), ), CoolKids( name="Marco", user=self.bot.get_user(299668599650648065), bday=datetime.date(year, 11, 12), ), CoolKids( name="Tyler", user=self.bot.get_user(619322461749641246), bday=datetime.date(year, 3, 4), ), CoolKids( name="Avery", user=self.bot.get_user(538824552906489874), bday=datetime.date(year, 10, 16), ), CoolKids( name="Matthew", user=self.bot.get_user(525365553842356225), bday=datetime.date(year, 1, 2), ), CoolKids( name="Samuel", user=self.bot.get_user(615235547920597150), bday=datetime.date(year, 12, 13), ), ) for kid in coolKids: if datetime.date.today() == kid.bday: try: for i in range(self.tries): await kid.user.send( f"Happy Birthday {kid.name}! You're a boomer now! <:mangopog:730683234039365722>" ) self.tries = 1 except: self.tries += 1 # *Don't* ask if message.author.id == 289721817516605440: # An ultra special dude text = ( unicodedata.normalize("NFD", message.content) .encode("ascii", "ignore") .decode("utf-8") ) if "women" in text.lower() or "woman" in text.lower(): await message.channel.send("<:shut:808843657167765546>") if message.reference is not None and len(message.content) == 0: reply = await message.channel.fetch_message(message.reference.message_id) if reply.is_system(): def check(m): return ( m.author == message.author and message.reference is not None and len(message.content) == 0 ) try: msg2 = await self.bot.wait_for("message", check=check) except asyncio.TimeoutError: return reply2 = await msg2.channel.fetch_message(msg2.reference.message_id) if not reply2.is_system(): return muted_role = message.guild.get_role( int(self.bot.config[str(message.guild.id)]["mute_role"]) ) await message.author.add_roles(muted_role, reason="spam") await message.channel.send( "{0.mention} has been muted for *spam*".format(message.author) ) await asyncio.sleep(300) await message.author.remove_roles(muted_role, reason="time's up ") for word in badWords: if word in message.content.lower().replace(" ", ""): # get fair object with open("fair.json", "r") as f: fair = json.load(f) # if fairer's ID in fair.json userId = str(message.author.id) if userId in fair: # TODO: use timezones (get this time based on timezones to fair.json - default to GMT) tz = fair[userId]["timezone"] today = str(datetime.datetime.now(timezone(tz)).date()) yesterday = str( datetime.datetime.now(timezone(tz)).date() - timedelta(1) ) # if date in json != current date date = fair[userId]["date"] if date != today: # increment fair day fairDay = fair[userId]["day"] + 1 fairStreak = fair[userId]["streak"] # if the user faired yesterday if yesterday == date: fairStreak = fair[userId]["streak"] + 1 else: fairStreak = 1 await message.channel.send( "streak lost. <:sad:716629485449117708>" ) # only send && update if user is fairing for the first time today fair[userId] = { "day": fairDay, "streak": fairStreak, "date": today, "timezone": tz, } fairInfo = f"day {fair[userId]['day']}, streak {fair[userId]['streak']}" await message.channel.send(fairInfo) # new user - not in fair.json else: # default to GMT tz = "Europe/London" today = str(datetime.datetime.now(timezone(tz)).date()) fair[userId] = { "day": 1, "streak": 1, "date": today, "timezone": tz, } fairInfo = ( f"day {fair[userId]['day']}, streak {fair[userId]['streak']}" ) await message.channel.send(fairInfo) # overwrite with new fair object with open("fair.json", "w") as f: json.dump(fair, f, indent=4) count += 1 fairMsg = "Fair " * count try: await message.channel.send(fairMsg) except UnboundLocalError: pass # 24 hour cooldown # should probably be longer - we can't have these kids cheating! @commands.cooldown(1, 86400, commands.BucketType.user) @commands.command() async def timezone(self, ctx, timeZone): """set timezone for fair days/streaks""" # get fair object with open("fair.json", "r") as f: fair = json.load(f) # if this user has faired before userId = str(ctx.author.id) if userId not in fair: # new user await ctx.send("try saying 'fair' first") return try: # let users timezone = input timezone (string version so as to please json) # use timezone() simply to see if it's valid tz = str(timezone(timeZone)) except exceptions.UnknownTimeZoneError: await ctx.send( "That's not a valid timezone. You can look them up at https://kevinnovak.github.io/Time-Zone-Picker/" ) return # set user's timezone to (verified) input zone fair[userId]["timezone"] = tz # overwrite with new fair object with open("fair.json", "w") as f: json.dump(fair, f, indent=4) await ctx.send( f"{discord.utils.escape_mentions(ctx.message.author.display_name)} your timezone has been set to {timeZone}" ) @commands.cooldown(2, 60, commands.BucketType.user) @commands.command() async def fboard(self, ctx, requested: str = "streak"): """Show fair leaderboard (day or streak).""" # Get fair object. with open("fair.json", "r") as f: fair = json.load(f) # Detect people who didn't maintain their streak. for user in fair: tz = fair[user]["timezone"] try: currentdate = datetime.datetime.now(timezone(tz)).date() except: # Ocean put in a test timezone and it messes with stuff continue if fair[user]["date"] != str(currentdate) and fair[user]["date"] != str( currentdate - timedelta(1) ): fair[user]["streak"] = 1 fair[user]["date"] = str(currentdate) try: # It seems like people don't like spam pings # await ctx.send( # self.bot.get_user(int(user)).mention # + ": You lost your streak! <:sad:716629485449117708>" # ) pass except: # User left the server so the bot can't find them continue with open("fair.json", "w") as f: json.dump(fair, f, indent=4) leaderboard = [] # String the requested info with its respective user and store it in our list if requested.lower() == "day": for user in fair: leaderboard.append(str(fair[user]["day"]) + " " + user) flag = "Day" elif requested.lower() == "streak": for user in fair: leaderboard.append(str(fair[user]["streak"]) + " " + user) flag = "Streak" else: await ctx.send('Request the leaderboard by "day" or "streak".') return # Sort the "info + user" strings as ints. leaderboard.sort(reverse=True, key=lambda pair: int(pair.split(" ")[0])) text = "" embed = discord.Embed( title="Fair Leaderboard", color=7853978, timestamp=ctx.message.created_at ) addNewField = True callerid = ctx.message.author.id # Get the top 10 entries and extract the info and user from each string for i in range(10): entry = leaderboard[i].split(" ")[0] userstr = leaderboard[i].split(" ")[1] # If caller is in top 10, then we highlight their entry if str(callerid) == userstr: text += f"**{i+1}. {self.bot.get_user(int(userstr))}: {entry}** \n" addNewField = False else: text += f"{i+1}. {self.bot.get_user(int(userstr))}: {entry} \n" if not str(callerid) in fair: addNewField = False # If caller is not in top 10, we add a new field with their place on the leaderboard and another one with number of people that have the same value (tied) if addNewField: if flag == "Day": value = fair[str(callerid)]["day"] elif flag == "Streak": value = fair[str(callerid)]["streak"] # Binary searches for caller in leaderboard. First we find leftmost element that has [value] in it. left = 0 right = len(leaderboard) while left < right: middle = floor((left + right) / 2) if int(leaderboard[middle].split(" ")[0]) > value: left = middle + 1 else: right = middle leftIndex = left # Now we find the rightmost one. left = 0 right = len(leaderboard) while left < right: middle = floor((left + right) / 2) if int(leaderboard[middle].split(" ")[0]) < value: right = middle else: left = middle + 1 rightIndex = right cutlist = leaderboard[leftIndex:right] cutlist.sort(key=lambda pair: int(pair.split(" ")[1])) # Finally, we find the caller id index left = 0 right = len(cutlist) - 1 while left <= right: middle = floor((left + right) / 2) if int(cutlist[middle].split(" ")[1]) < callerid: left = middle + 1 elif int(cutlist[middle].split(" ")[1]) > callerid: right = middle - 1 else: callerIndex = middle break entry = cutlist[callerIndex].split(" ")[0] userstr = cutlist[callerIndex].split(" ")[1] # If caller is the 11th place on the leaderboard, highlight their entry. if callerIndex + leftIndex == 10: text += f"**11. {self.bot.get_user(int(userstr))}: {entry}** \n" else: text += f"\n{callerIndex + leftIndex + 1}. {self.bot.get_user(int(userstr))}: {entry} \nRange of positions with value {entry}: {leftIndex+1} - {rightIndex}\n" embed.add_field(name=flag, value=text, inline=False) await ctx.send(embed=embed) @commands.cooldown(1, 60, commands.BucketType.member) @commands.command() async def report(self, ctx, *, message=None): """Send a message to the super mods about anything""" if ctx.message.guild != None: await ctx.message.delete() if message == None: await ctx.message.author.send( "Please type a report to report (hehe, sounds funny)" ) else: await reportStuff(self, ctx, message) @commands.cooldown(1, 20, commands.BucketType.guild) @commands.command() async def leaderboard(self, ctx): """Leaderboard of the people that matter""" async with ctx.typing(): # try: # lbFunc = functools.partial(save_leaderboard) # await self.bot.loop.run_in_executor(None, lbFunc) # await ctx.send(file=discord.File("leaderboard.png")) # except: await ctx.send("https://aninternettroll.github.io/mcbeVerifierLeaderboard/") @commands.cooldown(1, 60, commands.BucketType.guild) @commands.command() async def someone(self, ctx): """Discord's mistake""" await ctx.send(choice(ctx.guild.members).mention) @commands.command() async def roll(self, ctx, pool): """Roll the dice""" await ctx.send(f"You rolled a {randint(0, int(pool))}") @commands.command(aliases=["commands", "allcommands"]) async def listcommands(self, ctx): """List all custom commands""" with open("custom_commands.json", "r") as f: commands = json.load(f) output = ", ".join([*commands]) await ctx.send(f"```List of custom commands:\n{output}```") @commands.command(aliases=["calc"]) async def math(self, ctx, *, eqn: str): if '"' in eqn or "print" in eqn: return try: # Allow for proper absolute value notation pipes = eqn.count("|") eqn = eqn.replace("|", "abs(", pipes // 2).replace("|", ")", pipes // 2) with open("bc_input.bc", "w") as f: f.write(eqn) result = subprocess.check_output("utils/math", shell=True) os.remove("bc_input.bc") await ctx.send(result.decode("utf-8").replace("\\\n", "").strip()) except subprocess.CalledProcessError as err: print(err) await ctx.send("Something went wrong") @commands.command() async def retime(self, ctx, start_sec, end_sec, frames=0, framerate=30): """Retimes a run using the start/end timestamps, leftover frames, and framerate""" if start_sec.count(":") == 2: start_sec = sum( x * int(t) for x, t in zip([3600, 60, 1], start_sec.split(":")) ) elif start_sec.count(":") == 1: start_sec = sum(x * int(t) for x, t in zip([60, 1], start_sec.split(":"))) else: start_sec = int(start_sec) if end_sec.count(":") == 2: end_sec = sum(x * int(t) for x, t in zip([3600, 60, 1], end_sec.split(":"))) elif end_sec.count(":") == 1: end_sec = sum(x * int(t) for x, t in zip([60, 1], end_sec.split(":"))) else: end_sec = int(end_sec) await ctx.send( str( timedelta( seconds=end_sec - start_sec + round((int(frames) / int(framerate)), 3) ) ).replace("000", "") ) async def setup(bot): await bot.add_cog(Utils(bot))