using System.Text.Json.Nodes;
using LibMatrix.HomeserverEmulator.Services;
using LibMatrix.Responses;
using Microsoft.AspNetCore.Mvc;

namespace LibMatrix.HomeserverEmulator.Controllers;

[ApiController]
[Route("/_matrix/client/{version}/")]
public class AuthController(ILogger<AuthController> logger, UserStore userStore, TokenService tokenService, HseConfiguration config) : ControllerBase {
    [HttpPost("login")]
    public async Task<LoginResponse> Login(LoginRequest request) {
        if (!request.Identifier.User.StartsWith('@'))
            request.Identifier.User = $"@{request.Identifier.User}:{tokenService.GenerateServerName(HttpContext)}";
        if (request.Identifier.User.EndsWith("localhost"))
            request.Identifier.User = request.Identifier.User.Replace("localhost", tokenService.GenerateServerName(HttpContext));

        var user = await userStore.GetUserById(request.Identifier.User);
        if (user is null) {
            user = await userStore.CreateUser(request.Identifier.User);
        }

        return user.Login();
    }

    [HttpGet("login")]
    public async Task<LoginFlowsResponse> GetLoginFlows() {
        return new LoginFlowsResponse {
            Flows = ((string[]) [
                "m.login.password",
                "m.login.recaptcha",
                "m.login.sso",
                "m.login.email.identity",
                "m.login.msisdn",
                "m.login.dummy",
                "m.login.registration_token",
            ]).Select(x => new LoginFlowsResponse.LoginFlow { Type = x }).ToList()
        };
    }

    [HttpPost("logout")]
    public async Task<object> Logout() {
        var token = tokenService.GetAccessToken(HttpContext);
        var user = await userStore.GetUserByToken(token);
        if (user == null)
            throw new MatrixException() {
                ErrorCode = "M_UNKNOWN_TOKEN",
                Error = "No such user"
            };

        if (!user.AccessTokens.ContainsKey(token))
            throw new MatrixException() {
                ErrorCode = MatrixException.ErrorCodes.M_NOT_FOUND,
                Error = "Token not found"
            };

        user.AccessTokens.Remove(token);
        return new { };
    }

    [HttpPost("register")]
    public async Task<object> Register(JsonObject request, [FromQuery] string kind = "user") {
        if (kind == "guest") {
            var user = await userStore.CreateUser(Random.Shared.NextInt64(long.MaxValue).ToString(), kind: "guest");
            return user.Login();
        }

        if (request.Count == 0) {
            return new {
                session = Guid.NewGuid().ToString(),
                flows = new {
                    stages = new[] {
                        "m.login.dummy",
                    }
                }
            };
        }

        if (request.ContainsKey("password")) {
            var parts = request["username"].ToString().Split(':');
            var localpart = parts[0].TrimStart('@');
            var user = await userStore.CreateUser($"@{localpart}:{config.ServerName}");
            var login = user.Login();

            if (request.ContainsKey("initial_device_display_name"))
                user.AccessTokens[login.AccessToken].DeviceName = request["initial_device_display_name"]!.ToString();

            return login;
        }

        return new { };
    }

    [HttpGet("register/available")]
    public async Task<object> IsUsernameAvailable([FromQuery] string username) {
        return new {
            available = await userStore.GetUserById($"@{username}:{config.ServerName}") is null
        };
    }
    
    // [HttpPost("account/deactivate")]
    // public async Task<object> DeactivateAccount() {
    //     var token = tokenService.GetAccessToken(HttpContext);
    //     var user = await userStore.GetUserByToken(token);
    //     if (user == null)
    //         throw new MatrixException() {
    //             ErrorCode = "M_UNKNOWN_TOKEN",
    //             Error = "No such user"
    //         };
    //
    //     
    //     return new { };
    // }
    
    #region 3PID
    
    [HttpGet("account/3pid")]
    public async Task<object> Get3pid() {
        var token = tokenService.GetAccessToken(HttpContext);
        var user = await userStore.GetUserByToken(token);
        if (user == null)
            throw new MatrixException() {
                ErrorCode = "M_UNKNOWN_TOKEN",
                Error = "No such user"
            };

        return new {
            threepids = (object[])[]
        };
    }
    
    #endregion
}

public class LoginFlowsResponse {
    public required List<LoginFlow> Flows { get; set; }

    public class LoginFlow {
        public required string Type { get; set; }
    }
}