Added feature flag checks #338 #1.1.0.rc3

kdb-bot/src/modules/boot_log/boot_log_on_ready_event.py

@@ -11,6 +11,7 @@ from bot_core.abc.message_service_abc import MessageServiceABC
 from bot_core.configuration.feature_flags_enum import FeatureFlagsEnum
 from bot_core.configuration.feature_flags_settings import FeatureFlagsSettings
 from bot_data.model.server_config import ServerConfig
+from bot_data.model.technician_config import TechnicianConfig
 
 
 class BootLogOnReadyEvent(OnReadyABC):
@@ -21,6 +22,7 @@ class BootLogOnReadyEvent(OnReadyABC):
         bot: DiscordBotServiceABC,
         message_service: MessageServiceABC,
         translate: TranslatePipe,
+        tech_config: TechnicianConfig,
     ):
         OnReadyABC.__init__(self)
         self._config = config
@@ -29,11 +31,17 @@ class BootLogOnReadyEvent(OnReadyABC):
         self._bot = bot
         self._message_service = message_service
         self._t = translate
+        self._tech_config = tech_config
 
         self._logger.info(__name__, f"Module {type(self)} loaded")
 
     async def on_ready(self):
         self._logger.debug(__name__, f"Module {type(self)} started")
+        if not FeatureFlagsSettings.get_flag_from_dict(
+            self._tech_config.feature_flags, FeatureFlagsEnum.boot_log_module
+        ):
+            return
+
         try:
             start_time = self._config.get_configuration("Bot_StartTime")
             init_time = round(

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

@@ -1,14 +1,18 @@
 import discord
+from cpl_core.configuration import ConfigurationABC
 from cpl_core.database.context import DatabaseContextABC
 from cpl_discord.container import Guild
 from cpl_discord.service import DiscordBotServiceABC
 from discord import Permissions, Colour
 
+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.abc.data_seeder_abc import DataSeederABC
 from bot_data.abc.server_repository_abc import ServerRepositoryABC
 from bot_data.model.level import Level
 from bot_data.model.server import Server
+from bot_data.model.server_config import ServerConfig
 from bot_data.service.level_repository_service import LevelRepositoryService
 from modules.level.configuration.default_level_settings import DefaultLevelSettings
 from modules.level.service.level_service import LevelService
@@ -17,6 +21,7 @@ from modules.level.service.level_service import LevelService
 class LevelSeeder(DataSeederABC):
     def __init__(
         self,
+        config: ConfigurationABC,
         logger: DatabaseLogger,
         levels: DefaultLevelSettings,
         level_repo: LevelRepositoryService,
@@ -27,6 +32,7 @@ class LevelSeeder(DataSeederABC):
     ):
         DataSeederABC.__init__(self)
 
+        self._config = config
         self._logger = logger
         self._levels = level_repo
         self._servers = servers
@@ -65,6 +71,10 @@ class LevelSeeder(DataSeederABC):
     async def seed(self):
         # create levels
         for guild in self._bot.guilds:
+            server_config: ServerConfig = self._config.get_configuration(f"ServerConfig_{guild.id}")
+            if not FeatureFlagsSettings.get_flag_from_dict(server_config.feature_flags, FeatureFlagsEnum.level_module):
+                continue
+
             created_default = False
             if guild.roles.where(lambda r: r.name == self._level_header).first_or_default() is None:
                 await guild.create_role(name=self._level_header)

kdb-bot/src/modules/permission/events/permission_on_member_update_event.py

@@ -1,17 +1,29 @@
 import discord
+from cpl_core.configuration import ConfigurationABC
 from cpl_core.logging import LoggerABC
 from cpl_discord.events import OnMemberUpdateABC
 
+from bot_core.configuration.feature_flags_enum import FeatureFlagsEnum
+from bot_core.configuration.feature_flags_settings import FeatureFlagsSettings
+from bot_data.model.server_config import ServerConfig
 from modules.permission.abc.permission_service_abc import PermissionServiceABC
 
 
 class PermissionOnMemberUpdateEvent(OnMemberUpdateABC):
-    def __init__(self, logger: LoggerABC, permission_service: PermissionServiceABC):
+    def __init__(self, config: ConfigurationABC, logger: LoggerABC, permission_service: PermissionServiceABC):
         OnMemberUpdateABC.__init__(self)
+        self._config = config
         self._logger = logger
         self._permission_service = permission_service
 
     async def on_member_update(self, before: discord.Member, after: discord.Member):
+        if before.guild is not None:
+            server_config: ServerConfig = self._config.get_configuration(f"ServerConfig_{before.guild.id}")
+            if not FeatureFlagsSettings.get_flag_from_dict(
+                server_config.feature_flags, FeatureFlagsEnum.permission_module
+            ):
+                return
+
         self._logger.debug(__name__, f"Module {type(self)} started")
 
         if before.roles != after.roles:

kdb-bot/src/modules/permission/events/permission_on_ready_event.py

@@ -1,15 +1,22 @@
 from cpl_core.logging import LoggerABC
 from cpl_discord.events import OnReadyABC
 
+from bot_core.configuration.feature_flags_enum import FeatureFlagsEnum
+from bot_core.configuration.feature_flags_settings import FeatureFlagsSettings
+from bot_data.model.technician_config import TechnicianConfig
 from modules.permission.abc.permission_service_abc import PermissionServiceABC
 
 
 class PermissionOnReadyEvent(OnReadyABC):
-    def __init__(self, logger: LoggerABC, permission_service: PermissionServiceABC):
+    def __init__(self, logger: LoggerABC, permission_service: PermissionServiceABC, tech_config: TechnicianConfig):
         OnReadyABC.__init__(self)
         self._logger = logger
         self._permission_service = permission_service
+        self._tech_config = tech_config
 
     async def on_ready(self):
-        self._logger.debug(__name__, f"Module {type(self)} started")
+        if not FeatureFlagsSettings.get_flag_from_dict(
+            self._tech_config.feature_flags, FeatureFlagsEnum.permission_module
+        ):
+            return
         self._permission_service.on_ready()

kdb-web/package.json

@@ -1,6 +1,6 @@
 {
     "name": "kdb-web",
-    "version": "1.1.0.rc2",
+    "version": "1.1.0.rc3",
     "scripts": {
         "ng": "ng",
         "update-version": "ts-node-esm update-version.ts",

kdb-web/src/app/models/auth/auth-user.dto.ts

@@ -25,3 +25,9 @@ export interface UserDTO {
   isAdmin: boolean;
   isModerator: boolean;
 }
+
+export enum MemberRoles {
+  Moderator = 0,
+  Admin = 1,
+  Technician = 2,
+}

kdb-web/src/app/modules/shared/guards/auth/auth.guard.ts

@@ -1,17 +1,20 @@
-import { Injectable } from '@angular/core';
+import { Injectable } from "@angular/core";
-import { ActivatedRouteSnapshot, CanActivate, Router, RouterStateSnapshot } from '@angular/router';
+import { ActivatedRouteSnapshot, CanActivate, Router, RouterStateSnapshot } from "@angular/router";
-import { catchError } from 'rxjs/operators';
+import { catchError } from "rxjs/operators";
-import { AuthService } from 'src/app/services/auth/auth.service';
+import { AuthService } from "src/app/services/auth/auth.service";
-import { ThemeService } from 'src/app/services/theme/theme.service';
+import { ThemeService } from "src/app/services/theme/theme.service";
+import { SidebarService } from "../../../../services/sidebar/sidebar.service";
+import { MemberRoles } from "../../../../models/auth/auth-user.dto";
 
 @Injectable({
-  providedIn: 'root'
+  providedIn: "root"
 })
 export class AuthGuard implements CanActivate {
   constructor(
     private router: Router,
     private authService: AuthService,
-    private themeService: ThemeService
+    private themeService: ThemeService,
+    private sidebarService: SidebarService
   ) {
   }
 
@@ -26,19 +29,41 @@ export class AuthGuard implements CanActivate {
       return false;
     }
 
-    const role = route.data['role'];
+    const role = route.data["role"];
-    if (role) {
+    const memberRole = route.data["memberRole"];
+    if (role !== undefined) {
       this.authService.hasUserPermission(role).then(async hasPermission => {
         let authUser = await this.authService.getLoggedInUser();
         let isTechnician = authUser?.users?.map(u => u.isTechnician).filter(u => u) ?? [];
 
         if (!hasPermission && !isTechnician) {
-          this.router.navigate(['/dashboard']);
+          this.router.navigate(["/dashboard"]);
           return false;
         }
         return true;
       });
     }
+
+    if (memberRole !== undefined) {
+      let userHasAccess = false;
+      let authUser = await this.authService.getLoggedInUser();
+      authUser?.users?.forEach(u => {
+        if (u.server === +(this.sidebarService.server$.value?.id ?? 0)) {
+          if (
+            memberRole === MemberRoles.Moderator && u.isModerator ||
+            memberRole === MemberRoles.Admin && u.isAdmin ||
+            memberRole === MemberRoles.Technician && u.isTechnician
+          ) {
+            userHasAccess = true;
+          }
+        }
+      });
+      if (!userHasAccess) {
+        this.router.navigate(["/dashboard"]);
+        return false;
+      }
+      return true;
+    }
     return true;
   }
 }

kdb-web/src/app/modules/view/server/server-routing.module.ts

@@ -3,15 +3,22 @@ import { RouterModule, Routes } from "@angular/router";
 import { ServerDashboardComponent } from "./server-dashboard/server-dashboard.component";
 import { ProfileComponent } from "./profile/profile.component";
 import { MembersComponent } from "./members/members.component";
+import { AuthGuard } from "../../shared/guards/auth/auth.guard";
+import { MemberRoles } from "../../../models/auth/auth-user.dto";
 
 const routes: Routes = [
   { path: "", component: ServerDashboardComponent },
-  { path: "members", component: MembersComponent },
+  { path: "members", component: MembersComponent, canActivate: [AuthGuard], data: { memberRole: MemberRoles.Moderator } },
   { path: "members/:memberId", component: ProfileComponent },
-  { path: "auto-roles", loadChildren: () => import("./auto-role/auto-role.module").then(m => m.AutoRoleModule) },
+  {
-  { path: "levels", loadChildren: () => import("./levels/levels.module").then(m => m.LevelsModule) },
+    path: "auto-roles",
-  { path: "achievements", loadChildren: () => import("./achievements/achievements.module").then(m => m.AchievementsModule) },
+    loadChildren: () => import("./auto-role/auto-role.module").then(m => m.AutoRoleModule),
-  { path: "config", loadChildren: () => import("./config/config.module").then(m => m.ConfigModule) }
+    canActivate: [AuthGuard],
+    data: { memberRole: MemberRoles.Moderator }
+  },
+  { path: "levels", loadChildren: () => import("./levels/levels.module").then(m => m.LevelsModule), canActivate: [AuthGuard], data: { memberRole: MemberRoles.Moderator } },
+  { path: "achievements", loadChildren: () => import("./achievements/achievements.module").then(m => m.AchievementsModule), data: { memberRole: MemberRoles.Moderator } },
+  { path: "config", loadChildren: () => import("./config/config.module").then(m => m.ConfigModule), data: { memberRole: MemberRoles.Admin } }
 ];
 
 @NgModule({

kdb-web/src/assets/version.json

@@ -2,6 +2,6 @@
     "WebVersion": {
         "Major": "1",
         "Minor": "1",
-        "Micro": "0.rc2"
+        "Micro": "0.rc3"
     }
 }