TimingsBot/cogs/timings.py


import discord

from discord.ext import commands

import requests

import yaml


TIMINGS_CHECK = None

YAML_ERROR = None

with open("cogs/timings_check.yml", 'r') as stream:

 try:

 TIMINGS_CHECK = yaml.safe_load(stream)

 except yaml.YAMLError as exc:

 print(exc)

 YAML_ERROR = exc


class Timings(commands.Cog):


 def __init__(self, bot):

 self.bot = bot

 self.TIMINGS_TITLE = "Timings Analysis"

 self.TIMINGS_TITLE_COLOR = 0x55ffff


 # Use @commands.Cog.listener() instead of event and use @commands.command() for commands


 async def analyze_timings(self, message):

 words = message.content.replace("\n", " ").split(" ")

 timings_url = ""

 embed_var = discord.Embed(title=self.TIMINGS_TITLE, color=self.TIMINGS_TITLE_COLOR)


 for word in words:

 if word.startswith("https://timings.") and "/?id=" in word:

 timings_url = word

 break

 if word.startswith("https://www.spigotmc.org/go/timings?url=") or word.startswith(

 "https://timings.spigotmc.org/?url="):

 embed_var.add_field(name="❌ Spigot",

 value="Upgrade to [Purpur](https://purpur.pl3x.net/downloads/#1.16.4).")

 embed_var.set_footer(text="Requested by " + message.author.name, icon_url=message.author.avatar_url)

 embed_var.url = timings_url

 await message.reply(embed=embed_var)

 return


 if timings_url == "":

 return

 if "#" in timings_url:

 timings_url = timings_url.split("#")[0]

 if "?id=" not in timings_url:

 return


 timings_host, timings_id = timings_url.split("?id=")

 timings_json = timings_host + "data.php?id=" + timings_id


 r = requests.get(timings_json).json()

 if r is None:

 embed_var.add_field(name="❌ Invalid report",

 value="Create a new timings report.")

 embed_var.set_footer(text="Requested by " + message.author.name, icon_url=message.author.avatar_url)

 embed_var.url = timings_url

 await message.reply(embed=embed_var)

 return


 embed_var.set_footer(text="Requested by " + message.author.name, icon_url=message.author.avatar_url)

 embed_var.url = timings_url

 unchecked = 0

 try:

 try:

 version = r["timingsMaster"]["version"].lower()

 if "version" in TIMINGS_CHECK:

 if TIMINGS_CHECK["version"] not in version:

 embed_var.add_field(name="❌ Legacy Build",

 value="Update to " + TIMINGS_CHECK["version"])

 if "servers" in TIMINGS_CHECK:

 for server in TIMINGS_CHECK["servers"]:

 if server["name"] in version:

 embed_var.add_field(**create_field(server))

 break

 except KeyError as key:

 print("Missing: " + str(key))

 unchecked += 1


 try:

 timing_cost = int(r["timingsMaster"]["system"]["timingcost"])

 if timing_cost > 300:

 embed_var.add_field(name="❌ Timingcost",

 value="Your timingcost is " + str(timing_cost) + ". Find a [better host](https://www.birdflop.com).")

 except KeyError as key:

 print("Missing: " + str(key))

 unchecked += 1


 try:

 jvm_version = r["timingsMaster"]["system"]["jvmversion"]

 if jvm_version.startswith("1.8.") or jvm_version.startswith("9.") or jvm_version.startswith("10."):

 embed_var.add_field(name="❌ Java Version",

 value="You are using Java " + jvm_version + ". Update to [Java 11](https://adoptopenjdk.net/installation.html).")

 except KeyError as key:

 print("Missing: " + str(key))

 unchecked += 1


 try:

 flags = r["timingsMaster"]["system"]["flags"]

 if "-XX:+UseZGC" in flags:

 jvm_version = r["timingsMaster"]["system"]["jvmversion"]

 java_version = jvm_version.split(".")[0]

 if int(java_version) < 14:

 embed_var.add_field(name="❌ Java " + java_version,

 value="If you are going to use ZGC, you should also use Java 14+.")

 elif "-Daikars.new.flags=true" in flags:

 if "-XX:+PerfDisableSharedMem" not in flags:

 embed_var.add_field(name="❌ Outdated Flags",

 value="Add `-XX:+PerfDisableSharedMem` to flags")

 if "XX:G1MixedGCCountTarget=4" not in flags:

 embed_var.add_field(name="❌ Outdated Flags",

 value="Add `-XX:G1MixedGCCountTarget=4` to flags")

 if "-Xmx" in flags:

 max_mem = 0

 flaglist = flags.split(" ")

 for flag in flaglist:

 if flag.startswith("-Xmx"):

 max_mem = flag.split("-Xmx")[1]

 max_mem = max_mem.replace("G", "000")

 max_mem = max_mem.replace("M", "")

 max_mem = max_mem.replace("g", "000")

 max_mem = max_mem.replace("m", "")

 if int(max_mem) < 5400:

 embed_var.add_field(name="❌ Low Memory",

 value="Allocate at least 6-10GB of ram to your server if you can afford it.")

 if "-Xms" in flags:

 min_mem = 0

 flaglist = flags.split(" ")

 for flag in flaglist:

 if flag.startswith("-Xms"):

 min_mem = flag.split("-Xms")[1]

 min_mem = min_mem.replace("G", "000")

 min_mem = min_mem.replace("M", "")

 min_mem = min_mem.replace("g", "000")

 min_mem = min_mem.replace("m", "")

 if min_mem != max_mem:

 embed_var.add_field(name="❌ Aikar's Flags",

 value="Your Xmx and Xms values should be equivalent when using Aikar's flags.")

 elif "-Dusing.aikars.flags=mcflags.emc.gs" in flags:

 embed_var.add_field(name="❌ Outdated Flags",

 value="Update [Aikar's flags](https://aikar.co/2018/07/02/tuning-the-jvm-g1gc-garbage-collector-flags-for-minecraft/).")

 else:

 embed_var.add_field(name="❌ Aikar's Flags",

 value="Use [Aikar's flags](https://aikar.co/2018/07/02/tuning-the-jvm-g1gc-garbage-collector-flags-for-minecraft/).")

 except KeyError as key:

 print("Missing: " + str(key))

 unchecked += 1


 try:

 cpu = int(r["timingsMaster"]["system"]["cpu"])

 if cpu == 1:

 embed_var.add_field(name="❌ Threads",

 value="You have only " + str(cpu) + " thread. Find a [better host](https://www.birdflop.com).")

 if cpu == 2:

 embed_var.add_field(name="❌ Threads",

 value="You have only " + str(cpu) + " threads. Find a [better host](https://www.birdflop.com).")

 except KeyError as key:

 print("Missing: " + str(key))

 unchecked += 1


 plugins = r["timingsMaster"]["plugins"] if "plugins" in r["timingsMaster"] else None

 server_properties = r["timingsMaster"]["config"]["server.properties"] if "server.properties" in r["timingsMaster"]["config"] else None

 bukkit = r["timingsMaster"]["config"]["bukkit"] if "bukkit" in r["timingsMaster"]["config"] else None

 spigot = r["timingsMaster"]["config"]["spigot"] if "spigot" in r["timingsMaster"]["config"] else None

 paper = r["timingsMaster"]["config"]["paper"] if "paper" in r["timingsMaster"]["config"] else None

 purpur = r["timingsMaster"]["config"]["purpur"] if "purpur" in r["timingsMaster"]["config"] else None

 if not YAML_ERROR:

 if "plugins" in TIMINGS_CHECK:

 for server_name in TIMINGS_CHECK["plugins"]:

 if server_name in r["timingsMaster"]["config"]:

 for plugin in plugins:

 for plugin_name in TIMINGS_CHECK["plugins"][server_name]:

 if plugin == plugin_name:

 stored_plugin = TIMINGS_CHECK["plugins"][server_name][plugin_name]

 if isinstance(stored_plugin, dict):

 stored_plugin["name"] = plugin_name

 embed_var.add_field(**create_field(stored_plugin))

 else:

 eval_field(embed_var, stored_plugin, plugin_name, unchecked, plugins, server_properties, bukkit, spigot, paper, purpur)

 if "songoda" in r["timingsMaster"]["plugins"][plugin]["authors"].casefold():

 if plugin == "EpicHeads":

 embed_var.add_field(name="❌ EpicHeads",

 value="This plugin was made by Songoda. Songoda resources are poorly developed and often cause problems. You should find an alternative such as [HeadsPlus](spigotmc.org/resources/headsplus-»-1-8-1-16-4.40265/) or [HeadDatabase](https://www.spigotmc.org/resources/head-database.14280/).")

 elif plugin == "UltimateStacker":

 embed_var.add_field(name="❌ UltimateStacker",

 value="Stacking plugins actually causes more lag. "

 "Remove UltimateStacker.")

 else:

 embed_var.add_field(name="❌ " + plugin,

 value="This plugin was made by Songoda. Songoda resources are poorly developed and often cause problems. You should find an alternative.")

 if "config" in TIMINGS_CHECK:

 for config_name in TIMINGS_CHECK["config"]:

 config = TIMINGS_CHECK["config"][config_name]

 for option_name in config:

 option = config[option_name]

 eval_field(embed_var, option, option_name, unchecked, plugins, server_properties, bukkit, spigot, paper, purpur)

 else:

 embed_var.add_field(name="Error loading YAML file",

 value=YAML_ERROR)


 except ValueError as bruh:

 print(bruh)

 embed_var.add_field(name="❌ Invalid Configuration",

 value="At least one of your configuration files had an invalid data type.")


 if len(embed_var.fields) == 0:

 embed_var.add_field(name="✅ All good",

 value="Analyzed with no issues")

 await message.reply(embed=embed_var)

 return


 issue_count = len(embed_var.fields)

 if issue_count >= 25:

 embed_var.insert_field_at(index=24, name="Plus " + str(issue_count - 24) + " more recommendations", value="Create a new timings report after resolving some of the above issues to see more.")

 if unchecked > 0:

 embed_var.description = "||" + str(unchecked) + " missing configuration optimizations.||"

 await message.reply(embed=embed_var)


def eval_field(embed_var, option, option_name, unchecked, plugins=None, server_properties=None, bukkit=None, spigot=None, paper=None, purpur=None):

 try:

 for option_data in option:

 add_to_field = True

 for expression in option_data["expressions"]:

 if ("server_properties" in expression and server_properties or

 "bukkit" in expression and bukkit or

 "spigot" in expression and spigot or

 "paper" in expression and paper or

 "purpur" in expression and purpur or

 "plugins" in expression and plugins):

 if not eval(expression):

 add_to_field = False

 break

 if add_to_field:

 """ f strings don't like newlines so we replace the newlines with placeholder text before we eval """

 option_data["value"] = eval('f"""' + option_data["value"].replace("\n", "\\|n\\") + '"""').replace("\\|n\\", "\n")

 embed_var.add_field(**create_field({**{"name": option_name}, **option_data}))

 break

 except KeyError as key:

 print("Missing: " + str(key))

 unchecked += 1


def create_field(option):

 field = {"name": option["name"],

 "value": option["value"]}

 if "prefix" in option:

 field["name"] = option["prefix"] + field["name"]

 if "suffix" in option:

 field["name"] = field["name"] + option["suffix"]

 if "inline" in option:

 field["inline"] = option["inline"]

 return field


def setup(bot):

 bot.add_cog(Timings(bot))