refactor: use consts for name of feature flags

pollubnet · Jun 14, 2024 · 704ddbe · 704ddbe
1 parent c19a1a1
commit 704ddbe
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 2 deletions.
diff --git a/Server/Modules/Items/Services/ItemGenerator.cs b/Server/Modules/Items/Services/ItemGenerator.cs
@@ -1,6 +1,7 @@
 using Fracture.Server.Modules.AI.Services;
 using Fracture.Server.Modules.Items.Models;
 using Fracture.Server.Modules.Shared;
+using Fracture.Server.Modules.Shared.Configuration;
 using Microsoft.FeatureManagement;
 using System.Text.Json;
 
@@ -103,7 +104,7 @@ public async Task<Item> Generate()
 
             await _prefixes.AddPrefixes(item);
 
-            if (await _featureManager.IsEnabledAsync("UseAI"))
+            if (await _featureManager.IsEnabledAsync(FeatureFlags.USE_AI))
             {
                 item.History = await GenerateDescription(item);
             }

diff --git a/Server/Modules/Shared/Configuration/FeatureFlags.cs b/Server/Modules/Shared/Configuration/FeatureFlags.cs
@@ -0,0 +1,8 @@
+namespace Fracture.Server.Modules.Shared.Configuration
+{
+    public static class FeatureFlags
+    {
+        public const string CONFIG_SECTION = "FeatureFlags";
+        public const string USE_AI = "UseAI";
+    }
+}
diff --git a/Server/Program.cs b/Server/Program.cs
@@ -26,7 +26,9 @@
 builder.Services.AddScoped<IUsersRepository, UsersRepository>();
 builder.Services.AddScoped<IItemsRepository, ItemsRepository>();
 
-builder.Services.AddFeatureManagement(builder.Configuration.GetSection("FeatureFlags"));
+builder.Services.AddFeatureManagement(
+    builder.Configuration.GetSection(FeatureFlags.CONFIG_SECTION)
+);
 
 builder.Services
     .AddRazorComponents()