Load internal Controllers as well as public ones

TheEadie · Aug 3, 2023 · 7b3059d · 7b3059d
1 parent 729b053
commit 7b3059d
Show file tree

Hide file tree

Showing 3 changed files with 21 additions and 5 deletions.
diff --git a/src/hub/src/Worms.Gateway/InternalControllerProvider.cs b/src/hub/src/Worms.Gateway/InternalControllerProvider.cs
@@ -0,0 +1,14 @@
+using System.Reflection;
+using Microsoft.AspNetCore.Mvc;
+using Microsoft.AspNetCore.Mvc.Controllers;
+
+namespace Worms.Gateway;
+
+/// <summary>
+/// Overloads the default detection of controllers in ASP.NET Core to include controllers marked as internal
+/// rather than just public.
+/// </summary>
+internal sealed class InternalControllerProvider : ControllerFeatureProvider
+{
+    protected override bool IsController(TypeInfo typeInfo) => typeInfo.IsDefined(typeof(ApiControllerAttribute));
+}
diff --git a/src/hub/src/Worms.Gateway/Program.cs b/src/hub/src/Worms.Gateway/Program.cs
@@ -1,4 +1,5 @@
 using Microsoft.IdentityModel.Tokens;
+using Worms.Gateway;
 using Worms.Gateway.Announcers;
 using Worms.Gateway.Announcers.Slack;
 using Worms.Gateway.Database;
@@ -10,7 +11,8 @@
 builder.Logging.AddSimpleConsole(options => { options.SingleLine = true; });
 builder.Configuration.AddEnvironmentVariables("WORMS_");
 
-builder.Services.AddControllers();
+builder.Services.AddControllers()
+    .ConfigureApplicationPartManager(manager => { manager.FeatureProviders.Add(new InternalControllerProvider()); });
 builder.Services.AddApiVersioning();
 builder.Services.AddAuthentication()
     .AddJwtBearer(

diff --git a/src/hub/src/Worms.Gateway/Worms.Gateway.csproj b/src/hub/src/Worms.Gateway/Worms.Gateway.csproj
@@ -10,9 +10,9 @@
     <WarningsAsErrors>true</WarningsAsErrors>
   </PropertyGroup>
   <ItemGroup>
-    <PackageReference Include="Dapper" Version="2.0.143"/>
-    <PackageReference Include="Microsoft.AspNetCore.Authentication.JwtBearer" Version="7.0.9"/>
-    <PackageReference Include="Microsoft.AspNetCore.Mvc.Versioning" Version="5.1.0"/>
-    <PackageReference Include="Npgsql" Version="7.0.4"/>
+    <PackageReference Include="Dapper" Version="2.0.143" />
+    <PackageReference Include="Microsoft.AspNetCore.Authentication.JwtBearer" Version="7.0.9" />
+    <PackageReference Include="Microsoft.AspNetCore.Mvc.Versioning" Version="5.1.0" />
+    <PackageReference Include="Npgsql" Version="7.0.4" />
   </ItemGroup>
 </Project>