diff --git a/MxApiExtensions/Services/AuthenticatedHomeserverProviderService.cs b/MxApiExtensions/Services/AuthenticatedHomeserverProviderService.cs
new file mode 100644
index 0000000..96b0254
--- /dev/null
+++ b/MxApiExtensions/Services/AuthenticatedHomeserverProviderService.cs
@@ -0,0 +1,35 @@
+using LibMatrix.Homeservers;
+using LibMatrix.Services;
+
+namespace MxApiExtensions.Services;
+
+public class AuthenticatedHomeserverProviderService {
+ private readonly AuthenticationService _authenticationService;
+ private readonly HomeserverProviderService _homeserverProviderService;
+
+ public AuthenticatedHomeserverProviderService(AuthenticationService authenticationService, HomeserverProviderService homeserverProviderService) {
+ _authenticationService = authenticationService;
+ _homeserverProviderService = homeserverProviderService;
+ }
+
+ public async Task<AuthenticatedHomeserverGeneric> GetHomeserver() {
+ var token = _authenticationService.GetToken();
+ if (token == null) {
+ throw new MxApiMatrixException {
+ ErrorCode = "M_MISSING_TOKEN",
+ Error = "Missing access token"
+ };
+ }
+
+ var mxid = await _authenticationService.GetMxidFromToken(token);
+ if (mxid == "@anonymous:*") {
+ throw new MxApiMatrixException {
+ ErrorCode = "M_MISSING_TOKEN",
+ Error = "Missing access token"
+ };
+ }
+
+ var hsCanonical = string.Join(":", mxid.Split(':').Skip(1));
+ return await _homeserverProviderService.GetAuthenticatedWithToken(hsCanonical, token);
+ }
+}
diff --git a/MxApiExtensions/Services/AuthenticationService.cs b/MxApiExtensions/Services/AuthenticationService.cs
new file mode 100644
index 0000000..09a6e70
--- /dev/null
+++ b/MxApiExtensions/Services/AuthenticationService.cs
@@ -0,0 +1,121 @@
+using LibMatrix.Services;
+
+namespace MxApiExtensions.Services;
+
+public class AuthenticationService {
+ private readonly ILogger<AuthenticationService> _logger;
+ private readonly MxApiExtensionsConfiguration _config;
+ private readonly HomeserverProviderService _homeserverProviderService;
+ private readonly HttpRequest _request;
+
+ private static Dictionary<string, string> _tokenMap = new();
+
+ public AuthenticationService(ILogger<AuthenticationService> logger, MxApiExtensionsConfiguration config, IHttpContextAccessor request, HomeserverProviderService homeserverProviderService) {
+ _logger = logger;
+ _config = config;
+ _homeserverProviderService = homeserverProviderService;
+ _request = request.HttpContext!.Request;
+ }
+
+ internal string? GetToken(bool fail = true) {
+ string? token;
+ if (_request.Headers.TryGetValue("Authorization", out var tokens)) {
+ token = tokens.FirstOrDefault()?[7..];
+ }
+ else {
+ token = _request.Query["access_token"];
+ }
+
+ if (token == null && fail) {
+ throw new MxApiMatrixException {
+ ErrorCode = "M_MISSING_TOKEN",
+ Error = "Missing access token"
+ };
+ }
+
+ return token;
+ }
+
+ public async Task<string> GetMxidFromToken(string? token = null, bool fail = true) {
+ token ??= GetToken(fail);
+ if (token == null) {
+ if (fail) {
+ throw new MxApiMatrixException {
+ ErrorCode = "M_MISSING_TOKEN",
+ Error = "Missing access token"
+ };
+ }
+
+ return "@anonymous:*";
+ }
+
+ if(_tokenMap is not { Count: >0 } && File.Exists("token_map")) {
+ _tokenMap = (await File.ReadAllLinesAsync("token_map"))
+ .Select(l => l.Split('\t'))
+ .ToDictionary(l => l[0], l => l[1]);
+ }
+
+ if (_tokenMap.TryGetValue(token, out var mxid)) return mxid;
+
+ var lookupTasks = new Dictionary<string, Task<string?>>();
+ foreach (var homeserver in _config.AuthHomeservers) {
+ lookupTasks.Add(homeserver, GetMxidFromToken(token, homeserver));
+ await lookupTasks[homeserver].WaitAsync(TimeSpan.FromMilliseconds(250));
+ if(lookupTasks[homeserver].IsCompletedSuccessfully && !string.IsNullOrWhiteSpace(lookupTasks[homeserver].Result)) break;
+ }
+ await Task.WhenAll(lookupTasks.Values);
+
+ mxid = lookupTasks.Values.FirstOrDefault(x => x.Result != null)?.Result;
+ if(mxid is null) {
+ throw new MxApiMatrixException {
+ ErrorCode = "M_UNKNOWN_TOKEN",
+ Error = "Token not found on any configured homeservers: " + string.Join(", ", _config.AuthHomeservers)
+ };
+ }
+
+ // using var hc = new HttpClient();
+ // hc.DefaultRequestHeaders.Authorization = new AuthenticationHeaderValue("Bearer", token);
+ // var resp = hc.GetAsync($"{_config.Homeserver}/_matrix/client/v3/account/whoami").Result;
+ // if (!resp.IsSuccessStatusCode) {
+ // throw new MatrixException {
+ // ErrorCode = "M_UNKNOWN",
+ // Error = "[Rory&::MxSyncCache] Whoami request failed"
+ // };
+ // }
+ //
+ // if (resp.Content is null) {
+ // throw new MatrixException {
+ // ErrorCode = "M_UNKNOWN",
+ // Error = "No content in response"
+ // };
+ // }
+ //
+ // var json = (await JsonDocument.ParseAsync(await resp.Content.ReadAsStreamAsync())).RootElement;
+ // var mxid = json.GetProperty("user_id").GetString()!;
+ _logger.LogInformation("Got mxid {} from token {}", mxid, token);
+ await SaveMxidForToken(token, mxid);
+ return mxid;
+ }
+
+ private async Task<string?> GetMxidFromToken(string token, string hsDomain) {
+ _logger.LogInformation("Looking up mxid for token {} on {}", token, hsDomain);
+ var hs = await _homeserverProviderService.GetAuthenticatedWithToken(hsDomain, token);
+ try {
+ var res = hs.WhoAmI.UserId;
+ _logger.LogInformation("Got mxid {} for token {} on {}", res, token, hsDomain);
+ return res;
+ }
+ catch (MxApiMatrixException e) {
+ if (e.ErrorCode == "M_UNKNOWN_TOKEN") {
+ return null;
+ }
+
+ throw;
+ }
+ }
+
+ public async Task SaveMxidForToken(string token, string mxid) {
+ _tokenMap.Add(token, mxid);
+ await File.AppendAllLinesAsync("token_map", new[] { $"{token}\t{mxid}" });
+ }
+}
|
