From c2613aab129c8d1d5aba3b7ed02609059a826c84 Mon Sep 17 00:00:00 2001
From: Rory& <root@rory.gay>
Date: Wed, 16 Jul 2025 19:00:05 +0200
Subject: Federation tuff

---
 .../Controllers/Spec/FederationKeysController.cs   | 42 ++++++++++++++++++++++
 1 file changed, 42 insertions(+)
 create mode 100644 Utilities/LibMatrix.FederationTest/Controllers/Spec/FederationKeysController.cs

(limited to 'Utilities/LibMatrix.FederationTest/Controllers/Spec/FederationKeysController.cs')

diff --git a/Utilities/LibMatrix.FederationTest/Controllers/Spec/FederationKeysController.cs b/Utilities/LibMatrix.FederationTest/Controllers/Spec/FederationKeysController.cs
new file mode 100644
index 0000000..6516415
--- /dev/null
+++ b/Utilities/LibMatrix.FederationTest/Controllers/Spec/FederationKeysController.cs
@@ -0,0 +1,42 @@
+using LibMatrix.Abstractions;
+using LibMatrix.Federation.Extensions;
+using LibMatrix.FederationTest.Services;
+using LibMatrix.Homeservers;
+using LibMatrix.Responses.Federation;
+using Microsoft.AspNetCore.Mvc;
+
+namespace LibMatrix.FederationTest.Controllers.Spec;
+
+[ApiController]
+[Route("_matrix/key/v2/")]
+public class FederationKeysController(FederationTestConfiguration config, FederationKeyStore keyStore) {
+    static FederationKeysController() {
+        Console.WriteLine("INFO | FederationKeysController initialized.");
+    }
+
+    private static SignedObject<ServerKeysResponse>? _cachedServerKeysResponse;
+    private static SemaphoreSlim _serverKeyCacheLock = new SemaphoreSlim(1, 1);
+
+    [HttpGet("server")]
+    public async Task<SignedObject<ServerKeysResponse>> GetServerKeys() {
+        await _serverKeyCacheLock.WaitAsync();
+        if (_cachedServerKeysResponse == null || _cachedServerKeysResponse.TypedContent.ValidUntil < DateTime.Now + TimeSpan.FromSeconds(30)) {
+            var keys = keyStore.GetCurrentSigningKey();
+            _cachedServerKeysResponse = new ServerKeysResponse() {
+                ValidUntil = DateTime.Now + TimeSpan.FromMinutes(1),
+                ServerName = config.ServerName,
+                OldVerifyKeys = [],
+                VerifyKeysById = new() {
+                    {
+                        new() { Algorithm = "ed25519", KeyId = "0" }, new ServerKeysResponse.CurrentVerifyKey() {
+                            Key = keys.publicKey.ToUnpaddedBase64(),
+                        }
+                    }
+                }
+            }.Sign(config.ServerName, new VersionedKeyId() { Algorithm = "ed25519", KeyId = "0" }, keys.privateKey);
+        }
+        _serverKeyCacheLock.Release();
+
+        return _cachedServerKeysResponse;
+    }
+}
\ No newline at end of file
-- 
cgit 1.5.1