From db835755e01b13dcb8d33a91f57ae8f20b931c57 Mon Sep 17 00:00:00 2001
From: Rory& <root@rory.gay>
Date: Sun, 9 Mar 2025 17:24:34 +0100
Subject: Well known resolver work, synapse admin work

---
 .../WellKnownResolver/WellKnownResolverService.cs  | 91 ++++++++++++++++++++++
 1 file changed, 91 insertions(+)
 create mode 100644 LibMatrix/Services/WellKnownResolver/WellKnownResolverService.cs

(limited to 'LibMatrix/Services/WellKnownResolver/WellKnownResolverService.cs')

diff --git a/LibMatrix/Services/WellKnownResolver/WellKnownResolverService.cs b/LibMatrix/Services/WellKnownResolver/WellKnownResolverService.cs
new file mode 100644
index 0000000..4c78347
--- /dev/null
+++ b/LibMatrix/Services/WellKnownResolver/WellKnownResolverService.cs
@@ -0,0 +1,91 @@
+using System.Diagnostics;
+using System.Text.Json.Serialization;
+using LibMatrix.Extensions;
+using LibMatrix.Services.WellKnownResolver.WellKnownResolvers;
+using Microsoft.Extensions.Logging;
+using Microsoft.Extensions.Logging.Abstractions;
+
+namespace LibMatrix.Services.WellKnownResolver;
+
+public class WellKnownResolverService {
+    private readonly MatrixHttpClient _httpClient = new();
+
+    private readonly ILogger<WellKnownResolverService> _logger;
+    private readonly ClientWellKnownResolver _clientWellKnownResolver;
+    private readonly SupportWellKnownResolver _supportWellKnownResolver;
+    private readonly ServerWellKnownResolver _serverWellKnownResolver;
+    private readonly WellKnownResolverConfiguration _configuration;
+
+    public WellKnownResolverService(ILogger<WellKnownResolverService> logger, ClientWellKnownResolver clientWellKnownResolver, SupportWellKnownResolver supportWellKnownResolver,
+        WellKnownResolverConfiguration configuration, ServerWellKnownResolver serverWellKnownResolver) {
+        _logger = logger;
+        _clientWellKnownResolver = clientWellKnownResolver;
+        _supportWellKnownResolver = supportWellKnownResolver;
+        _configuration = configuration;
+        _serverWellKnownResolver = serverWellKnownResolver;
+        if (logger is NullLogger<WellKnownResolverService>) {
+            var stackFrame = new StackTrace(true).GetFrame(1);
+            Console.WriteLine(
+                $"WARN | Null logger provided to WellKnownResolverService!\n{stackFrame?.GetMethod()?.DeclaringType?.ToString() ?? "null"} at {stackFrame?.GetFileName() ?? "null"}:{stackFrame?.GetFileLineNumber().ToString() ?? "null"}");
+        }
+    }
+
+    public async Task<WellKnownRecords> TryResolveWellKnownRecords(string homeserver, bool includeClient = true, bool includeServer = true, bool includeSupport = true,
+        WellKnownResolverConfiguration? config = null) {
+        WellKnownRecords records = new();
+        _logger.LogDebug($"Resolving well-knowns for {homeserver}");
+        if (includeClient && await _clientWellKnownResolver.TryResolveWellKnown(homeserver, config ?? _configuration) is { } clientResult) {
+            records.ClientWellKnown = clientResult;
+        }
+        
+        if (includeServer && await _serverWellKnownResolver.TryResolveWellKnown(homeserver, config ?? _configuration) is { } serverResult) {
+            records.ServerWellKnown = serverResult;
+        }
+        
+        if (includeSupport && await _supportWellKnownResolver.TryResolveWellKnown(homeserver, config ?? _configuration) is { } supportResult) {
+            records.SupportWellKnown = supportResult;
+        }
+
+        return records;
+    }
+
+    public class WellKnownRecords {
+        public WellKnownResolutionResult<ClientWellKnown?>? ClientWellKnown { get; set; }
+        public WellKnownResolutionResult<ServerWellKnown?>? ServerWellKnown { get; set; }
+        public WellKnownResolutionResult<SupportWellKnown?>? SupportWellKnown { get; set; }
+    }
+
+    public class WellKnownResolutionResult<T> {
+        public WellKnownSource Source { get; set; }
+        public string? SourceUri { get; set; }
+        public T? Content { get; set; }
+        public List<WellKnownResolutionWarning> Warnings { get; set; } = [];
+    }
+
+    [JsonConverter(typeof(JsonStringEnumConverter))]
+    public enum WellKnownSource {
+        None,
+        Https,
+        Dns,
+        Http,
+        ManualCheck,
+        Search
+    }
+
+    public struct WellKnownResolutionWarning {
+        public WellKnownResolutionWarningType Type { get; set; }
+        public string Message { get; set; }
+        [JsonIgnore]
+        public Exception? Exception { get; set; }
+        public string? ExceptionMessage => Exception?.Message;
+
+        [JsonConverter(typeof(JsonStringEnumConverter))]
+        public enum WellKnownResolutionWarningType {
+            None,
+            Exception,
+            InvalidResponse,
+            Timeout,
+            SlowResponse
+        }
+    }
+}
\ No newline at end of file
-- 
cgit 1.5.1