2025-02-12 09:19:32 +03:00
281 changed files with 14312 additions and 98 deletions
--- a/Endpoint/Controllers/V1/AuthController.cs
+++ b/Endpoint/Controllers/V1/AuthController.cs
@ -22,7 +22,7 @@ using OAuthProvider = Mirea.Api.Security.Common.Domain.OAuthProvider;
 namespace Mirea.Api.Endpoint.Controllers.V1;
 [ApiVersion("1.0")]
-public class AuthController(IOptionsSnapshot<Admin> user, AuthService auth, PasswordHashService passwordService) : BaseController
+public class AuthController(IOptionsSnapshot<Admin> user, IOptionsSnapshot<GeneralConfig> generalConfig, AuthService auth, PasswordHashService passwordService, OAuthService oAuthService) : BaseController
 {
    private CookieOptionsParameters GetCookieParams() =>
        new()
@ -31,6 +31,95 @@ public class AuthController(IOptionsSnapshot<Admin> user, AuthService auth, Pass
            Path = UrlHelper.GetSubPathWithoutFirstApiName + "api"
        };
    private static string GenerateHtmlResponse(string title, string message, OAuthProvider? provider, bool isError = false)
    {
        string messageColor = isError ? "red" : "white";
        string script = "<script>setTimeout(()=>{if(window.opener){window.opener.postMessage(" +
                        "{success:" + !isError +
                        ",provider:'" + (provider == null ? "null" : (int)provider) +
                        "',providerName:'" + (provider == null ? "null" : Enum.GetName(provider.Value)) +
                        "',message:'" + message.Replace("'", "\\'") +
                        "'},'*');}window.close();}, 5000);</script>";
        return $"<!DOCTYPE html><html lang=ru><head><meta charset=UTF-8><meta content=\"width=device-width,initial-scale=1\"name=viewport><link href=\"https://fonts.googleapis.com/css2?family=Roboto:wght@400;700&display=swap\"rel=stylesheet><style>body{{background-color:#121212;color:#fff;font-family:Roboto,sans-serif;display:flex;justify-content:center;align-items:center;height:100vh;margin:0;text-align:center}}.container{{max-width:600px;padding:20px;border-radius:8px;background-color:#1e1e1e;box-shadow:0 4px 20px rgba(0,0,0,.5)}}h1{{font-size:24px;margin-bottom:20px}}p{{font-size:16px;color:{messageColor}}}</style><title>{title}</title></head><body><div class=container><h1>{title}</h1><p>{message}<p style=font-size:14px;color:silver;>Это информационная страница. Вы можете закрыть её.</div>{script}</body></html>";
    }
    [HttpGet("OAuth2")]
    [BadRequestResponse]
    [ProducesResponseType(typeof(string), StatusCodes.Status200OK)]
    [Produces("text/html")]
    [MaintenanceModeIgnore]
    public async Task<ContentResult> OAuth2([FromQuery] string code, [FromQuery] string state)
    {
        var userId = HttpContext.User.FindFirstValue(ClaimTypes.NameIdentifier);
        string title;
        string message;
        OAuthProvider provider;
        OAuthUser oAuthUser;
        try
        {
            (provider, oAuthUser) = await oAuthService.LoginOAuth(HttpContext, GetCookieParams(),
                HttpContext.GetApiUrl(Url.Action("OAuth2")!), code, state);
        }
        catch (Exception e)
        {
            title = "Произошла ошибка при общении с провайдером OAuth!";
            message = e.Message;
            return Content(GenerateHtmlResponse(title, message, null, true), "text/html");
        }
        var userEntity = user.Value;
        if (userId != null)
        {
            userEntity.OAuthProviders ??= new Dictionary<OAuthProvider, OAuthUser>();
            if (!userEntity.OAuthProviders.TryAdd(provider, oAuthUser))
            {
                title = "Ошибка связи аккаунта!";
                message = "Этот OAuth провайдер уже связан с вашей учетной записью. Пожалуйста, используйте другого провайдера или удалите связь с аккаунтом.";
                return Content(GenerateHtmlResponse(title, message, provider, true), "text/html");
            }
            userEntity.SaveSetting();
            title = "Учетная запись успешно связана.";
            message = "Вы успешно связали свою учетную запись с провайдером OAuth. Вы можете продолжить использовать приложение.";
            return Content(GenerateHtmlResponse(title, message, provider), "text/html");
        }
        if (userEntity.OAuthProviders != null &&
            userEntity.OAuthProviders.TryGetValue(provider, out var userOAuth) &&
            userOAuth.Id == oAuthUser.Id)
        {
            await auth.LoginOAuthAsync(GetCookieParams(), HttpContext, new User
            {
                Id = 1.ToString(),
                Username = userEntity.Username,
                Email = userEntity.Email,
                PasswordHash = userEntity.PasswordHash,
                Salt = userEntity.Salt,
                TwoFactorAuthenticator = userEntity.TwoFactorAuthenticator,
                SecondFactorToken = userEntity.Secret,
                OAuthProviders = userEntity.OAuthProviders
            }, provider);
            title = "Успешный вход в аккаунт.";
            message = "Вы успешно вошли в свою учетную запись. Добро пожаловать!";
            return Content(GenerateHtmlResponse(title, message, provider), "text/html");
        }
        title = "Вы успешно зарегистрированы.";
        message = "Процесс завершен. Вы можете закрыть эту страницу.";
        userEntity.Email = string.IsNullOrEmpty(oAuthUser.Email) ? string.Empty : oAuthUser.Email;
        userEntity.Username = string.IsNullOrEmpty(oAuthUser.Username) ? string.Empty : oAuthUser.Username;
        userEntity.OAuthProviders ??= [];
        userEntity.OAuthProviders.Add(provider, oAuthUser);
        userEntity.SaveSetting();
        return Content(GenerateHtmlResponse(title, message, provider), "text/html");
    }
    /// <summary>
    /// Initiates the OAuth2 authorization process for the selected provider.
    /// </summary>