Added logic to handle default levels #25

kdb-bot/src/modules/database/database_extension.py

@@ -9,6 +9,7 @@ from bot_core.configuration.feature_flags_enum import FeatureFlagsEnum
 from bot_core.configuration.feature_flags_settings import FeatureFlagsSettings
 from bot_core.logging.database_logger import DatabaseLogger
 from bot_data.service.migration_service import MigrationService
+from bot_data.service.seeder_service import SeederService
 
 
 class DatabaseExtension(ApplicationExtensionABC):

kdb-bot/src/modules/database/database_on_ready_event.py

@@ -20,6 +20,7 @@ from bot_data.model.server import Server
 from bot_data.model.user import User
 from bot_data.model.user_joined_server import UserJoinedServer
 from bot_data.model.user_joined_voice_channel import UserJoinedVoiceChannel
+from bot_data.service.seeder_service import SeederService
 from bot_data.service.user_repository_service import ServerRepositoryABC
 from modules.base.configuration.base_server_settings import BaseServerSettings
 
@@ -30,6 +31,7 @@ class DatabaseOnReadyEvent(OnReadyABC):
         self,
         config: ConfigurationABC,
         logger: DatabaseLogger,
+        seeder: SeederService,
         bot: DiscordBotServiceABC,
         db_context: DatabaseContextABC,
         server_repo: ServerRepositoryABC,
@@ -43,6 +45,7 @@ class DatabaseOnReadyEvent(OnReadyABC):
         self._config = config
 
         self._logger = logger
+        self._seeder = seeder
         self._bot = bot
         self._db_context = db_context
         self._servers = server_repo
@@ -303,6 +306,8 @@ class DatabaseOnReadyEvent(OnReadyABC):
     async def on_ready(self):
         self._logger.debug(__name__, f'Module {type(self)} started')
 
+        await self._seeder.seed()
+
         self._check_known_users()
         self._check_servers()
         self._check_clients()

kdb-bot/src/modules/level/configuration/__init__.py

@@ -0,0 +1 @@
+# imports

kdb-bot/src/modules/level/configuration/level_settings.py

@@ -0,0 +1,33 @@
+import traceback
+
+from cpl_core.configuration.configuration_model_abc import ConfigurationModelABC
+from cpl_core.console import Console
+from cpl_query.extension import List
+
+from bot_data.model.level import Level
+
+
+class LevelSettings(ConfigurationModelABC):
+
+    def __init__(self):
+        ConfigurationModelABC.__init__(self)
+
+        self._levels = List(Level)
+
+    @property
+    def levels(self) -> List[Level]:
+        return self._levels
+
+    def from_dict(self, settings: dict):
+        try:
+            for level in settings:
+                self._levels.append(Level(
+                    level['Name'],
+                    level['Color'],
+                    int(level['MinXp']),
+                    int(level['Permissions']),
+                    None
+                ))
+        except Exception as e:
+            Console.error(f'[ ERROR ] [ {__name__} ]: Reading error in {type(self).__name__} settings')
+            Console.error(f'[ EXCEPTION ] [ {__name__} ]: {e} -> {traceback.format_exc()}')

kdb-bot/src/modules/level/default-level.json

@@ -0,0 +1,28 @@
+{
+  "Level": [
+    {
+      "Name": "Newbie",
+      "Color": "0x1abc9c",
+      "MinXp": 0,
+      "Permissions": 968552209984
+    },
+    {
+      "Name": "Keks",
+      "Color": "0x2ecc71",
+      "MinXp": 100,
+      "Permissions": 1002928856640
+    },
+    {
+      "Name": "Doppelkeks",
+      "Color": "0x3498db",
+      "MinXp": 200,
+      "Permissions": 1071849660224
+    },
+    {
+      "Name": "Auror",
+      "Color": "0xf1c40f",
+      "MinXp": 300,
+      "Permissions": 1089042120513
+    }
+  ]
+}

kdb-bot/src/modules/level/level_module.py

@@ -1,3 +1,5 @@
+import os
+
 from cpl_core.configuration import ConfigurationABC
 from cpl_core.dependency_injection import ServiceCollectionABC
 from cpl_core.environment import ApplicationEnvironmentABC
@@ -5,6 +7,7 @@ from cpl_discord.service.discord_collection_abc import DiscordCollectionABC
 
 from bot_core.abc.module_abc import ModuleABC
 from bot_core.configuration.feature_flags_enum import FeatureFlagsEnum
+from modules.level.level_seeder import LevelSeeder
 
 
 class LevelModule(ModuleABC):
@@ -13,7 +16,10 @@ class LevelModule(ModuleABC):
         ModuleABC.__init__(self, dc, FeatureFlagsEnum.level_module)
 
     def configure_configuration(self, config: ConfigurationABC, env: ApplicationEnvironmentABC):
-        pass
+        cwd = env.working_directory
+        env.set_working_directory(os.path.dirname(os.path.realpath(__file__)))
+        config.add_json_file(f'default-level.json', optional=False)
+        env.set_working_directory(cwd)
 
-    def configure_services(self, service: ServiceCollectionABC, env: ApplicationEnvironmentABC):
-        pass
+    def configure_services(self, services: ServiceCollectionABC, env: ApplicationEnvironmentABC):
+        services.add_transient(LevelSeeder)

kdb-bot/src/modules/level/level_seeder.py

@@ -0,0 +1,46 @@
+from cpl_discord.service import DiscordBotServiceABC
+from discord import Permissions, Colour
+
+from bot_core.logging.database_logger import DatabaseLogger
+from bot_data.abc.data_seeder_abc import DataSeederABC
+from bot_data.abc.server_repository_abc import ServerRepositoryABC
+from bot_data.service.level_repository_service import LevelRepositoryService
+from modules.level.configuration.level_settings import LevelSettings
+
+
+class LevelSeeder(DataSeederABC):
+
+    def __init__(self, logger: DatabaseLogger, levels: LevelSettings, level_repo: LevelRepositoryService, servers: ServerRepositoryABC, bot: DiscordBotServiceABC):
+        DataSeederABC.__init__(self)
+
+        self._logger = logger
+        self._default_levels = levels.levels.order_by_descending(lambda l: l.min_xp)
+        self._levels = level_repo
+        self._servers = servers
+        self._bot = bot
+
+    async def seed(self):
+        if self._levels.get_levels().count() > 0:
+            return
+
+        for guild in self._bot.guilds:
+            server = self._servers.find_server_by_discord_id(guild.id)
+            if server is None:
+                continue
+
+            levels = self._levels.find_levels_by_server_id(server.server_id)
+            if levels is not None and levels.count() > 0:
+                continue
+
+            for level in self._default_levels:
+                try:
+                    if guild.roles.where(lambda r: r.name == level.name).count() == 0:
+                        await guild.create_role(name=level.name, colour=Colour(int(level.color, 16)), hoist=False, mentionable=True, permissions=Permissions(level.permissions))
+
+                    self._logger.info(__name__, f'Created default level {level.name}')
+                    level.server = server
+                    self._levels.add_level(level)
+                except Exception as e:
+                    self._logger.error(__name__, f'Creating default role failed', e)
+
+        self._logger.debug(__name__, f'Seeded default levels')