PoC Discord webhook support

ArmaForces · Nov 17, 2024 · 74b077f · 74b077f
1 parent 2c08d47
commit 74b077f
Show file tree

Hide file tree

Showing 12 changed files with 153 additions and 12 deletions.
diff --git a/ArmaForces.Arma.Server.Tests/Helpers/TestSettings.cs b/ArmaForces.Arma.Server.Tests/Helpers/TestSettings.cs
@@ -19,6 +19,7 @@ public class TestSettings : ISettings
         public string ServerExecutableName { get; set; } = "arma3.exe";
         public string? SteamUser { get; set; } = "TEST_USER";
         public string? SteamPassword { get; set; } = "TEST_PASSWORD";
+        public string? WebhookUrl { get; set; }
 
         public Result LoadSettings() => throw new System.NotImplementedException();
 

diff --git a/ArmaForces.Arma.Server/Config/ISettings.cs b/ArmaForces.Arma.Server/Config/ISettings.cs
@@ -68,6 +68,11 @@ public interface ISettings {
         /// TODO: Handle no steam password
         string? SteamPassword { get; }
 
+        /// <summary>
+        /// URL for sending webhook notifications.
+        /// </summary>
+        string? WebhookUrl { get; }
+
         /// <summary>
         /// Loads settings from configuration.
         /// </summary>

diff --git a/ArmaForces.Arma.Server/Config/Settings.cs b/ArmaForces.Arma.Server/Config/Settings.cs
@@ -28,6 +28,7 @@ public class Settings : ISettings
         public string ServerExecutableName { get; set; } = "arma3server_x64.exe";
         public string? SteamUser { get; set; }
         public string? SteamPassword { get; set; }
+        public string? WebhookUrl { get; set; }
 
         private readonly IFileSystem _fileSystem;
         private readonly IRegistryReader _registryReader;
@@ -64,7 +65,8 @@ public Result LoadSettings()
                 .Tap(ObtainApiMissionsBaseUrl)
                 .Tap(ObtainApiModsetsBaseUrl)
                 .Tap(ObtainSteamUserName)
-                .Tap(ObtainSteamPassword);
+                .Tap(ObtainSteamPassword)
+                .Tap(ObtainWebhookUrl);
         }
 
         public Result ReloadSettings()
@@ -86,6 +88,7 @@ public async Task<Result> ReloadSettings(ISettings settings)
             ServerExecutableName = settings.ServerExecutableName;
             SteamUser = settings.SteamUser;
             SteamPassword = settings.SteamPassword;
+            WebhookUrl = settings.WebhookUrl;
 
             var json = JsonSerializer.Serialize(this, JsonOptions.Default);
             await _fileSystem.File.WriteAllTextAsync(SettingsJsonPath, json);
@@ -113,6 +116,8 @@ private void ObtainModsDirectory()
         private void ObtainSteamUserName() => SteamUser ??= _config["steamUserName"];
 
         private void ObtainSteamPassword() => SteamPassword ??= _config["steamPassword"];
+
+        private void ObtainWebhookUrl() => WebhookUrl ??= _config["webhookUrl"];
 
         private Result GetServerPath()
         {

diff --git a/ArmaForces.ArmaServerManager/Common/HttpClientBase.cs b/ArmaForces.ArmaServerManager/Common/HttpClientBase.cs
@@ -1,4 +1,5 @@
 using System.Net.Http;
+using System.Text;
 using System.Text.Json;
 using System.Threading.Tasks;
 using ArmaForces.Arma.Server.Constants;
@@ -19,21 +20,38 @@ protected async Task<Result<T>> HttpGetAsync<T>(string requestUrl)
         {
             var httpResponseMessage = await _httpClient.GetAsync(requestUrl);
 
+            var responseBody = await httpResponseMessage.Content.ReadAsStringAsync();
+
             if (httpResponseMessage.IsSuccessStatusCode)
             {
-                var responseBody = await httpResponseMessage.Content.ReadAsStringAsync();
                 return JsonSerializer.Deserialize<T>(responseBody, JsonOptions.Default) ??
                        Result.Failure<T>($"Failed to deserialize response: {responseBody}");
             }
-            else
-            {
-                var responseBody = await httpResponseMessage.Content.ReadAsStringAsync();
-                var error = string.IsNullOrWhiteSpace(responseBody)
-                    ? httpResponseMessage.ReasonPhrase
-                    : responseBody;
+
+            var error = string.IsNullOrWhiteSpace(responseBody)
+                ? httpResponseMessage.ReasonPhrase
+                : responseBody;
 
-                return Result.Failure<T>(error);
+            return Result.Failure<T>(error);
+        }
+
+        protected async Task<Result> HttpPostAsync<T>(string? requestUrl, T content)
+        {
+            var stringContent = new StringContent(JsonSerializer.Serialize(content, JsonSerializerOptions.Web), Encoding.UTF8, "application/json");
+            var httpResponseMessage = await _httpClient.PostAsync(requestUrl, stringContent);
+
+            var responseBody = await httpResponseMessage.Content.ReadAsStringAsync();
+
+            if (httpResponseMessage.IsSuccessStatusCode)
+            {
+                return Result.Success();
             }
+
+            var error = string.IsNullOrWhiteSpace(responseBody)
+                ? httpResponseMessage.ReasonPhrase
+                : responseBody;
+
+            return Result.Failure(error);
         }
     }
 }
diff --git a/ArmaForces.ArmaServerManager/Features/Webhooks/DTOs/DiscordWebhook.cs b/ArmaForces.ArmaServerManager/Features/Webhooks/DTOs/DiscordWebhook.cs
@@ -0,0 +1,11 @@
+using System.Text.Json.Serialization;
+
+namespace ArmaForces.ArmaServerManager.Features.Webhooks.DTOs;
+
+internal record DiscordWebhook
+{
+    [JsonPropertyName("username")]
+    public string UserName { get; init; }
+
+    public string Content { get; init; }
+}
diff --git a/ArmaForces.ArmaServerManager/Features/Webhooks/IWebhookClient.cs b/ArmaForces.ArmaServerManager/Features/Webhooks/IWebhookClient.cs
@@ -0,0 +1,8 @@
+using System.Threading.Tasks;
+
+namespace ArmaForces.ArmaServerManager.Features.Webhooks;
+
+public interface IWebhookClient
+{
+    Task Send(string content);
+}
diff --git a/ArmaForces.ArmaServerManager/Features/Webhooks/WebhookClient.cs b/ArmaForces.ArmaServerManager/Features/Webhooks/WebhookClient.cs
@@ -0,0 +1,22 @@
+using System.Net.Http;
+using System.Threading.Tasks;
+using ArmaForces.ArmaServerManager.Common;
+using ArmaForces.ArmaServerManager.Features.Webhooks.DTOs;
+
+namespace ArmaForces.ArmaServerManager.Features.Webhooks;
+
+internal class WebhookClient : HttpClientBase, IWebhookClient
+{
+    public WebhookClient(HttpClient httpClient) : base(httpClient)
+    {
+    }
+
+    public async Task Send(string content)
+    {
+        await HttpPostAsync(requestUrl: null, content: new DiscordWebhook
+        {
+            UserName = "Arma Server",
+            Content = content
+        });
+    }
+}
diff --git a/ArmaForces.ArmaServerManager/Features/Webhooks/WebhookServiceCollectionExtensions.cs b/ArmaForces.ArmaServerManager/Features/Webhooks/WebhookServiceCollectionExtensions.cs
@@ -0,0 +1,26 @@
+using System;
+using System.Net.Http;
+using ArmaForces.Arma.Server.Config;
+using Microsoft.Extensions.DependencyInjection;
+
+namespace ArmaForces.ArmaServerManager.Features.Webhooks;
+
+internal static class WebhookServiceCollectionExtensions
+{
+    public static IServiceCollection AddWebhookClient(this IServiceCollection services)
+        => services.AddHttpClientForWebhookClient();
+
+    private static IServiceCollection AddHttpClientForWebhookClient(this IServiceCollection services)
+    {
+        services
+            .AddHttpClient<IWebhookClient, WebhookClient>()
+            .ConfigureHttpClient(SetBaseAddress());
+
+        return services;
+    }
+
+    private static Action<IServiceProvider, HttpClient> SetBaseAddress()
+        => (services, client) => client.BaseAddress = new Uri(
+            services.GetRequiredService<ISettings>().WebhookUrl ??
+            throw new NotSupportedException("Missions API is required, provide valid url address."));
+}
diff --git a/ArmaForces.ArmaServerManager/Services/IWebhookService.cs b/ArmaForces.ArmaServerManager/Services/IWebhookService.cs
@@ -0,0 +1,8 @@
+using System.Threading.Tasks;
+
+namespace ArmaForces.ArmaServerManager.Services;
+
+public interface IWebhookService
+{
+    Task AnnounceStart();
+}
diff --git a/ArmaForces.ArmaServerManager/Services/StartupService.cs b/ArmaForces.ArmaServerManager/Services/StartupService.cs
@@ -13,23 +13,25 @@ namespace ArmaForces.ArmaServerManager.Services
     public class StartupService : IHostedService
     {
         private readonly IWebHostEnvironment _webHostEnvironment;
+        private readonly IWebhookService _webhookService;
 
         /// <inheritdoc cref="StartupService"/>
-        public StartupService(IWebHostEnvironment webHostEnvironment)
+        public StartupService(IWebHostEnvironment webHostEnvironment, IWebhookService webhookService)
         {
             _webHostEnvironment = webHostEnvironment;
+            _webhookService = webhookService;
         }
 
         /// <inheritdoc />
-        public Task StartAsync(CancellationToken cancellationToken)
+        public async Task StartAsync(CancellationToken cancellationToken)
         {
             if (_webHostEnvironment.IsProduction())
             {
                 RecurringJob.AddOrUpdate<MaintenanceService>(x => x.PerformMaintenance(CancellationToken.None),
                     Cron.Daily(4));
             }
 
-            return Task.CompletedTask;
+            await _webhookService.AnnounceStart();
         }
 
         /// <inheritdoc />

diff --git a/ArmaForces.ArmaServerManager/Services/WebhookService.cs b/ArmaForces.ArmaServerManager/Services/WebhookService.cs
@@ -0,0 +1,30 @@
+using System.Threading.Tasks;
+using ArmaForces.Arma.Server.Config;
+using ArmaForces.ArmaServerManager.Features.Webhooks;
+using Microsoft.Extensions.Logging;
+
+namespace ArmaForces.ArmaServerManager.Services;
+
+public class WebhookService : IWebhookService
+{
+    private readonly IWebhookClient _webhookClient;
+    private readonly ILogger<WebhookService> _logger;
+    private readonly string? _webhookUrl;
+
+    public WebhookService(IWebhookClient webhookClient, ISettings settings, ILogger<WebhookService> logger)
+    {
+        _webhookClient = webhookClient;
+        _logger = logger;
+        _webhookUrl = settings.WebhookUrl;
+    }
+
+    public async Task AnnounceStart()
+    {
+        _logger.LogInformation("Announce start");
+        if (_webhookUrl == null) return;
+        _logger.LogInformation("Webhook url: {WebhookUrl}", _webhookUrl);
+
+        await Task.Delay(10000);
+        await _webhookClient.Send("Manager started");
+    }
+}
diff --git a/ArmaForces.ArmaServerManager/Startup.cs b/ArmaForces.ArmaServerManager/Startup.cs
@@ -12,6 +12,7 @@
 using ArmaForces.ArmaServerManager.Features.Servers;
 using ArmaForces.ArmaServerManager.Features.Servers.Providers;
 using ArmaForces.ArmaServerManager.Features.Status;
+using ArmaForces.ArmaServerManager.Features.Webhooks;
 using ArmaForces.ArmaServerManager.Infrastructure.Authentication;
 using ArmaForces.ArmaServerManager.Infrastructure.Converters;
 using ArmaForces.ArmaServerManager.Infrastructure.Documentation;
@@ -127,6 +128,10 @@ public void ConfigureServices(IServiceCollection services)
             // Status
             .AddSingleton<IAppStatusStore, AppStatusStore>()
             .AddSingleton<IStatusProvider, StatusProvider>()
+
+            // Webhooks
+            .AddSingleton<IWebhookService, WebhookService>()
+            .AddWebhookClient()
 
             // Hangfire
             .AddSingleton<IJobsScheduler, JobsScheduler>()