Added a lot of utilities

1 files changed, 69 insertions, 0 deletions

diff --git a/MxApiExtensions/Controllers/LoginController.cs b/MxApiExtensions/Controllers/LoginController.cs
new file mode 100644

index 0000000..1a7970a
--- /dev/null
+++ b/MxApiExtensions/Controllers/LoginController.cs
@@ -0,0 +1,69 @@
+using System.Net.Http.Headers;
+using LibMatrix;
+using LibMatrix.Extensions;
+using LibMatrix.Responses;
+using LibMatrix.Services;
+using Microsoft.AspNetCore.Mvc;
+using MxApiExtensions.Services;
+
+namespace MxApiExtensions.Controllers;
+
+[ApiController]
+[Route("/")]
+public class LoginController : ControllerBase {
+    private readonly ILogger _logger;
+    private readonly HomeserverProviderService _hsProvider;
+    private readonly HomeserverResolverService _hsResolver;
+    private readonly AuthenticationService _auth;
+    private readonly MxApiExtensionsConfiguration _conf;
+
+    public LoginController(ILogger<LoginController> logger, HomeserverProviderService hsProvider, HomeserverResolverService hsResolver, AuthenticationService auth, MxApiExtensionsConfiguration conf) {
+        _logger = logger;
+        _hsProvider = hsProvider;
+        _hsResolver = hsResolver;
+        _auth = auth;
+        _conf = conf;
+    }
+
+    [HttpPost("/_matrix/client/{_}/login")]
+    public async Task Proxy([FromBody] LoginRequest request, string _) {
+        if (!request.Identifier.User.Contains("#")) {
+            Response.StatusCode = (int)StatusCodes.Status403Forbidden;
+            Response.ContentType = "application/json";
+            await Response.StartAsync();
+            await Response.WriteAsync(new MxApiMatrixException() {
+                ErrorCode = "M_FORBIDDEN",
+                Error = "[MxApiExtensions] Invalid username, must be of the form @user#domain:" + Request.Host.Value
+            }.GetAsJson() ?? "");
+            await Response.CompleteAsync();
+        }
+        var hsCanonical = request.Identifier.User.Split('#')[1].Split(':')[0];
+        request.Identifier.User = request.Identifier.User.Split(':')[0].Replace('#', ':');
+        if(!request.Identifier.User.StartsWith('@')) request.Identifier.User = '@' + request.Identifier.User;
+        var hs = await _hsResolver.ResolveHomeserverFromWellKnown(hsCanonical);
+        //var hs = await _hsProvider.Login(hsCanonical, mxid, request.Password);
+        var hsClient = new MatrixHttpClient { BaseAddress = new Uri(hs) };
+        //hsClient.DefaultRequestHeaders.Authorization = new AuthenticationHeaderValue("Bearer", hsClient.DefaultRequestHeaders.Authorization!.Parameter);
+        var resp = await hsClient.PostAsJsonAsync("/_matrix/client/r0/login", request);
+        var loginResp = await resp.Content.ReadAsStringAsync();
+        Response.StatusCode = (int)resp.StatusCode;
+        Response.ContentType = resp.Content.Headers.ContentType?.ToString() ?? "application/json";
+        await Response.StartAsync();
+        await Response.WriteAsync(loginResp);
+        await Response.CompleteAsync();
+        var token = (await resp.Content.ReadFromJsonAsync<LoginResponse>())!.AccessToken;
+        await _auth.SaveMxidForToken(token, request.Identifier.User);
+    }
+
+
+    [HttpGet("/_matrix/client/{_}/login")]
+    public async Task<object> Proxy(string? _) {
+        return new {
+            flows = new[] {
+                new {
+                    type = "m.login.password"
+                }
+            }
+        };
+    }
+}