Merge pull request #43 from argon-chat/feature/environment

Feature/environment

Argon.Server.sln.DotSettings

@@ -86,6 +86,7 @@
 	<s:Boolean x:Key="/Default/InstalledDictionaries/InstalledDictionaries/=C_003A_005Cgit_005Chunspell_002Ddictionaries_005Cru_005FRU_005Cru_005FRU_002Edic/@EntryIndexedValue">True</s:Boolean>
 	<s:Boolean x:Key="/Default/UserDictionary/Words/=appsettings/@EntryIndexedValue">True</s:Boolean>
 	<s:Boolean x:Key="/Default/UserDictionary/Words/=Behaviours/@EntryIndexedValue">True</s:Boolean>
+	<s:Boolean x:Key="/Default/UserDictionary/Words/=dockerenv/@EntryIndexedValue">True</s:Boolean>
 	<s:Boolean x:Key="/Default/UserDictionary/Words/=hwnd/@EntryIndexedValue">True</s:Boolean>
 	<s:Boolean x:Key="/Default/UserDictionary/Words/=livekit/@EntryIndexedValue">True</s:Boolean>
 	<s:Boolean x:Key="/Default/UserDictionary/Words/=Rebalance/@EntryIndexedValue">True</s:Boolean>

src/Argon.Api/Argon.Api.csproj

@@ -1,4 +1,4 @@
-<Project Sdk="Microsoft.NET.Sdk.Web">
+<Project Sdk="Microsoft.NET.Sdk.Web">
 
     <PropertyGroup>
         <TargetFramework>net8.0</TargetFramework>

src/Argon.Api/Features/Env/EnvironmentExtensions.cs

@@ -0,0 +1,35 @@
+namespace Argon.Api.Features.Env;
+
+using static File;
+
+public static class EnvironmentExtensions
+{
+    public static bool IsKube(this IHostEnvironment env)
+        => env.Determine() == ArgonEnvironmentKind.Kubernetes;
+
+    public static bool IsDocker(this IHostEnvironment env)
+        => env.Determine() == ArgonEnvironmentKind.Docker;
+
+    public static bool IsClassicHost(this IHostEnvironment env)
+        => env.Determine() == ArgonEnvironmentKind.HostMachine;
+
+    public static bool IsManaged(this IHostEnvironment env)
+        => env.IsDocker() || env.IsKube();
+
+    public static ArgonEnvironmentKind Determine(this IHostEnvironment _)
+    {
+        if (Environment.GetEnvironmentVariable("KUBERNETES_SERVICE_HOST") != null)
+            return ArgonEnvironmentKind.Kubernetes;
+        if (Exists("/.dockerenv") || Directory.Exists("/proc/self/cgroup") &&
+            ReadAllText("/proc/self/cgroup").Contains("docker"))
+            return ArgonEnvironmentKind.Docker;
+        return ArgonEnvironmentKind.HostMachine;
+    }
+}
+
+public enum ArgonEnvironmentKind
+{
+    HostMachine,
+    Docker,
+    Kubernetes,
+}

src/Argon.Api/Features/Orleanse/BalanceRule.cs

@@ -2,6 +2,7 @@ namespace Argon.Api.Features;
 
 using System.Collections.Concurrent;
 using System.Globalization;
+using Env;
 using k8s;
 using Orleans.Placement.Repartitioning;
 using static Math;
@@ -13,7 +14,7 @@ public static IServiceCollection AddKubeResources(this WebApplicationBuilder bui
     {
         var services = builder.Services;
 
-        if (builder.Environment.IsProduction())
+        if (builder.Environment.IsKube())
         {
             var config = KubernetesClientConfiguration.InClusterConfig();
             services.AddSingleton(config);
@@ -49,7 +50,7 @@ public async ValueTask FetchAsync()
 
     private async Task<double> GetAvgCpu()
     {
-        if (!env.IsProduction())
+        if (!env.IsManaged())
             return 10;
         await using var
             scope = serviceProvider.CreateAsyncScope();

src/Argon.Api/Features/Orleanse/OrleansExtension.cs

@@ -1,6 +1,7 @@
 namespace Argon.Api.Features;
 
 using Contracts;
+using Env;
 using Extensions;
 using Orleans.Clustering.Kubernetes;
 using Orleans.Configuration;
@@ -32,10 +33,10 @@ public static WebApplicationBuilder AddOrleans(this WebApplicationBuilder builde
                .AddPersistentStreams(IArgonEvent.ProviderId, NatsAdapterFactory.Create, options => { }).UseDashboard(o => o.Port = 22832);
         #pragma warning restore ORLEANSEXP001
 
-            if (builder.Environment.IsDevelopment())
-                siloBuilder.UseLocalhostClustering();
-            else
+            if (builder.Environment.IsKube())
                 siloBuilder.UseKubeMembership();
+            else
+                siloBuilder.UseLocalhostClustering();
         });
 
         return builder;

src/Argon.Api/Program.cs

@@ -6,6 +6,7 @@
 using Argon.Api.Extensions;
 using Argon.Api.Features;
 using Argon.Api.Features.EmailForms;
+using Argon.Api.Features.Env;
 using Argon.Api.Features.Jwt;
 using Argon.Api.Features.Otp;
 using Argon.Api.Grains.Interfaces;
@@ -25,7 +26,7 @@
 builder.AddNatsJetStream();
 builder.AddNpgsqlDbContext<ApplicationDbContext>("DefaultConnection");
 builder.Services.AddSingleton<IPasswordHashingService, PasswordHashingService>();
-if (!builder.Environment.IsProduction())
+if (!builder.Environment.IsManaged())
 {
     builder.AddJwt();
     builder.Services.AddControllers().AddNewtonsoftJson();
@@ -46,7 +47,7 @@
 builder.Services.AddAutoMapper(typeof(User).Assembly); // TODO
 var app = builder.Build();
 
-if (!builder.Environment.IsProduction())
+if (!builder.Environment.IsManaged())
 {
     app.UseWebSockets();
     app.MapRpcWebSocketServer();