Upload code

2025-11-13 23:49:30 +03:00
parent 3e46a4cd6e
commit ef827ba08b
7 changed files with 270 additions and 0 deletions
--- a/src/Services/MicrosoftAuthService.cs
+++ b/src/Services/MicrosoftAuthService.cs
@@ -0,0 +1,121 @@
+using System.Text.Json;
+using Microsoft.Playwright;
+using ModeusSchedule.MSAuth.BrowserScripts;
+
+namespace ModeusSchedule.MSAuth.Services;
+
+public class MicrosoftAuthService(ILogger<MicrosoftAuthService> logger, IConfiguration configuration)
+{
+    private static bool _browsersEnsured;
+    private static readonly SemaphoreSlim EnsureLock = new(1, 1);
+    private static readonly SemaphoreSlim FetchLock = new(1, 1);
+
+    private string? _cachedToken;
+    private DateTime _cachedAtUtc;
+    private readonly TimeSpan _cacheTtl = TimeSpan.FromMinutes(20);
+
+    public bool HasFreshToken => _cachedToken != null && DateTime.UtcNow - _cachedAtUtc < _cacheTtl;
+
+    public async Task<string> GetJwtAsync(CancellationToken ct = default)
+    {
+        await EnsureBrowsersAsync();
+
+        // Если кэш актуален — вернуть сразу
+        if (HasFreshToken)
+            return _cachedToken!;
+
+        // Пытаемся единолично выполнить авторизацию
+        if (!await FetchLock.WaitAsync(0, ct))
+        {
+            // Если кто-то уже выполняет, а кэша нет — просим повторить позже (429)
+            throw new MicrosoftAuthInProgressException();
+        }
+
+        using var playwright = await Playwright.CreateAsync();
+        await using var browser = await playwright.Chromium.LaunchAsync(new BrowserTypeLaunchOptions
+        {
+            Headless = true
+        });
+        var context = await browser.NewContextAsync(new BrowserNewContextOptions
+        {
+            ViewportSize = null
+        });
+        var page = await context.NewPageAsync();
+
+        try
+        {
+            logger.LogInformation("Старт авторизации через Microsoft");
+            await MicrosoftLoginHelper.LoginMicrosoftAsync(page, configuration["MS_USERNAME"]!, configuration["MS_PASSWORD"]!, configuration["MODEUS_URL"]!);
+
+            var sessionStorageJson = await page.EvaluateAsync<string>("JSON.stringify(sessionStorage)");
+
+            string? idToken = ExtractIdToken(sessionStorageJson);
+            if (string.IsNullOrWhiteSpace(idToken))
+                throw new Exception("Не удалось извлечь id_token из sessionStorage");
+
+            // Сохраняем в кэш
+            _cachedToken = idToken;
+            _cachedAtUtc = DateTime.UtcNow;
+            return idToken;
+        }
+        finally
+        {
+            await context.CloseAsync();
+            await browser.CloseAsync();
+
+            if (FetchLock.CurrentCount == 0) FetchLock.Release();
+        }
+    }
+
+    private static string? ExtractIdToken(string sessionStorageJson)
+    {
+        try
+        {
+            var dict = JsonSerializer.Deserialize<Dictionary<string, object>>(sessionStorageJson);
+            if (dict is null) return null;
+
+            var oidcKey = dict.Keys.FirstOrDefault(k => k.StartsWith("oidc.user:"));
+            if (oidcKey is null) return null;
+
+            var oidcValueJson = dict[oidcKey].ToString();
+            if (string.IsNullOrWhiteSpace(oidcValueJson)) return null;
+
+            using var doc = JsonDocument.Parse(oidcValueJson);
+            if (doc.RootElement.TryGetProperty("id_token", out var tokenEl))
+                return tokenEl.GetString();
+        }
+        catch
+        {
+            // ignore and return null
+        }
+        return null;
+    }
+
+    private static async Task EnsureBrowsersAsync()
+    {
+        if (_browsersEnsured) return;
+        await EnsureLock.WaitAsync();
+        try
+        {
+            if (_browsersEnsured) return;
+            try
+            {
+                // Устанавливаем Chromium, если не установлен
+                Microsoft.Playwright.Program.Main(["install", "chromium"]);
+            }
+            catch
+            {
+                // Игнорируем, если установка уже произведена или нет прав — попробуем дальше запустить браузер
+            }
+            _browsersEnsured = true;
+        }
+        finally
+        {
+            EnsureLock.Release();
+        }
+    }
+}
+
+public class MicrosoftAuthInProgressException : Exception
+{
+}