Basic test
1 files changed, 47 insertions, 0 deletions
diff --git a/ModAS.Server/Services/AuthenticatedHomeserverProviderService.cs b/ModAS.Server/Services/AuthenticatedHomeserverProviderService.cs
new file mode 100644
index 0000000..fa8e00a
--- /dev/null
+++ b/ModAS.Server/Services/AuthenticatedHomeserverProviderService.cs
@@ -0,0 +1,47 @@
+using ArcaneLibs.Extensions;
+using LibMatrix;
+using LibMatrix.Homeservers;
+using LibMatrix.Services;
+using MxApiExtensions.Services;
+
+namespace ModAS.Server.Services;
+
+public class AuthenticatedHomeserverProviderService(
+ AuthenticationService authenticationService,
+ HomeserverProviderService homeserverProviderService,
+ IHttpContextAccessor request,
+ ModASConfiguration config,
+ AppServiceRegistration asRegistration
+ ) {
+ public HttpContext? _context = request.HttpContext;
+ public Dictionary<string, AuthenticatedHomeserverGeneric> KnownUsers { get; set; } = new();
+
+ public async Task<AuthenticatedHomeserverGeneric> GetImpersonatedHomeserver(string mxid) {
+ if (KnownUsers.TryGetValue(mxid, out var homeserver)) return homeserver;
+ var hs = await homeserverProviderService.GetAuthenticatedWithToken(config.ServerName, asRegistration.AsToken, config.HomeserverUrl);
+ await hs.SetImpersonate(mxid);
+ KnownUsers.TryAdd(mxid, hs);
+ return hs;
+ }
+
+ public async Task<AuthenticatedHomeserverGeneric> GetHomeserver() {
+ var token = authenticationService.GetToken();
+ if (token == null) {
+ throw new MatrixException {
+ ErrorCode = "M_MISSING_TOKEN",
+ Error = "Missing access token"
+ };
+ }
+
+ var mxid = await authenticationService.GetMxidFromToken(token);
+ if (mxid == "@anonymous:*") {
+ throw new MatrixException {
+ ErrorCode = "M_MISSING_TOKEN",
+ Error = "Missing access token"
+ };
+ }
+
+ var hsCanonical = string.Join(":", mxid.Split(':').Skip(1));
+ return await homeserverProviderService.GetAuthenticatedWithToken(hsCanonical, token);
+ }
+}
\ No newline at end of file
|
