Fix disabling SectionTree for ScriptManager

patrickdemooij9 · Jun 30, 2023 · 2607d55 · 2607d55
1 parent 96a7e16
commit 2607d55
Showing 1 changed file with 8 additions and 7 deletions.
diff --git a/src/SeoToolkit.Umbraco.ScriptManager.Core/Composers/ScriptManagerComposer.cs b/src/SeoToolkit.Umbraco.ScriptManager.Core/Composers/ScriptManagerComposer.cs
@@ -34,20 +34,21 @@ public void Compose(IUmbracoBuilder builder)
             builder.Services.AddSingleton(typeof(ISettingsService<ScriptManagerConfigModel>), typeof(ScriptManagerConfigurationService));
 
             var disabledModules = section?.Get<ScriptManagerAppSettingsModel>()?.DisabledModules ?? Array.Empty<string>();
-
             if (disabledModules.Contains(DisabledModuleConstant.All))
             {
                 builder.Components().Append<DisableModuleComponent>();
                 builder.Trees().RemoveTreeController(typeof(ScriptManagerTreeController));
                 return;
+            } else if (disabledModules.Contains(DisabledModuleConstant.Trees)) {
+                builder.Trees().RemoveTreeController(typeof(ScriptManagerTreeController));
+            } else {
+                builder.ScriptDefinitions()
+                       .Add<GoogleTagManagerDefinition>()
+                       .Add<GoogleAnalyticsDefinition>()
+                       .Add<HotjarDefinition>()
+                       .Add<CustomScriptDefinition>();
             }
 
-            builder.ScriptDefinitions()
-                .Add<GoogleTagManagerDefinition>()
-                .Add<GoogleAnalyticsDefinition>()
-                .Add<HotjarDefinition>()
-                .Add<CustomScriptDefinition>();
-
             builder.Components().Append<EnableModuleComponent>();
 
             builder.Services.AddScoped<IScriptRepository, ScriptRepository>();