GoldyBot.goldy

  1import os
  2import sys
  3from typing import List
  4import GoldyBot, devgoldyutils
  5import nextcord
  6import threading, _thread
  7import time
  8
  9MODULE_NAME = "GOLDY"
 10
 11class Goldy(object):
 12    """💛 Goldy herself. More precisely the main class to control the whole of the Bot."""
 13    def __init__(self):
 14        self.nextcord_thread:threading.Thread = None
 15
 16    def start(self):
 17        """
 18        Awakens Goldy Bot! 👀💡⏰
 19        ---------------
 20        ### ***``Example:``***
 21
 22        ```python
 23        import GoldyBot
 24
 25        goldy = GoldyBot.Goldy()
 26
 27        goldy.start()
 28        ```
 29        """
 30        GoldyBot.log("warn", f"[{MODULE_NAME}] Goldy Bot is awakening...")
 31
 32        file_setup() # Run file setup.
 33        
 34        # Fixes some weird as fu#k bug when stopping the bot on windows.
 35        if GoldyBot.system.platform.system() == 'Windows':
 36            GoldyBot.asyncio.set_event_loop_policy(GoldyBot.asyncio.WindowsSelectorEventLoopPolicy())
 37
 38        print("")
 39
 40        # Start goldy input console.
 41        input_thread = threading.Thread(target=input_loop)
 42        input_thread.setDaemon(True)
 43        input_thread.start()
 44
 45        # Start V4
 46        from . import bot
 47        self.nextcord_thread = threading.Thread(target=bot.start)
 48        self.nextcord_thread.setDaemon(True)
 49        self.nextcord_thread.start()
 50
 51        # Cache goldy class.
 52        GoldyBot.cache.main_cache_dict["goldy_class"] = self
 53
 54        try: self.nextcord_thread.join()
 55        except KeyboardInterrupt: pass # Stops KeyboardInterrupt traceback.
 56
 57    async def setup(self, client:nextcord.Client):
 58        """Notifies Goldy Bot that the client is ready and it can do it's setup."""
 59
 60        #  Run setup on all allowed guilds.
 61        #------------------------------
 62        for guild in client.guilds:
 63            goldy_bot_guild = GoldyBot.utility.guilds.guild.Guild(guild)
 64
 65            if goldy_bot_guild.is_allowed:
 66                await goldy_bot_guild.setup()
 67
 68        #  Check if the config files have been edited for the guilds.
 69        not_edited_config_guilds = []
 70        for guild_ in GoldyBot.cache.main_cache_dict["guilds"]:
 71            guild:GoldyBot.utility.guilds.guild.Guild = GoldyBot.cache.main_cache_dict["guilds"][guild_]["object"]
 72
 73            if not guild.has_config_been_edited:
 74                not_edited_config_guilds.append(guild.code_name)
 75
 76        if not not_edited_config_guilds == []:
 77            self.stop(reason=f"Guild configs MUST be edited! These guilds have not had their config's edited: {not_edited_config_guilds}")
 78
 79        GoldyBot.logging.log(f"[{MODULE_NAME}] Guilds Setup Done!")
 80
 81    def stop(self, reason="Unknown"):
 82        """Safely shutdowns Goldy Bot and stops her from performing anymore actions, incase you know, things get weird. 😳"""
 83
 84        GoldyBot.log("warn", f"[{MODULE_NAME}] Goldy is Shuting down...")
 85        GoldyBot.log("info", f"[{MODULE_NAME}] Here's the reason why I was requested to shutdown for >>> {reason}")
 86
 87        GoldyBot.cache.main_cache_dict["bot_stop"] = True
 88
 89        sys.exit(reason)
 90
 91def file_setup():
 92    """Makes sure all files and directories are setup and ready to go."""
 93    GoldyBot.log(f"[{MODULE_NAME}] Setup is running...")
 94
 95    # Directories
 96    for dir in GoldyBot.settings.directories_to_check:
 97        if os.path.exists(dir) == False: # If the directory doesn't exist create it.
 98            GoldyBot.files.File(dir)
 99
100    # Create goldy.json if it doesn't exist already.
101    if not "goldy.json" in os.listdir(GoldyBot.paths.CONFIG):
102        goldy_config = GoldyBot.files.File(GoldyBot.paths.GOLDY_CONFIG_JSON).write(GoldyBot.files.File(GoldyBot.paths.GOLDY_CONFIG_JSON_TEMPLATE).read())
103
104    GoldyBot.log("info_2", f"[{MODULE_NAME}] File Setup Done!")
105
106def input_loop():
107    goldy = Goldy()
108
109    time.sleep(7)
110
111    GoldyBot.logging.log()
112    GoldyBot.logging.log("info", "Available Console Commands -> [stop] [extensions]")
113    GoldyBot.logging.log()
114
115    try:
116        while True:
117            command = input("")
118
119            if command.lower() == "stop":
120                raise EOFError
121
122            if command.lower() == "extensions":
123                all_modules = GoldyBot.utility.misc.merge_dict(
124                    GoldyBot.cache.main_cache_dict["modules"], 
125                    GoldyBot.cache.main_cache_dict["internal_modules"]
126                )
127
128                print("")
129
130                for module in all_modules:
131                    for extension in all_modules[module]["extensions"]:
132                        print(devgoldyutils.Console().BLUE(extension))
133
134                print("")
135
136            time.sleep(0.1)
137            
138    except EOFError:
139        print("\n")
140        _thread.interrupt_main()
141        goldy.stop("Master commanded me to stop!")