82 lines
2.5 KiB
C#
82 lines
2.5 KiB
C#
using System;
|
|
using System.IO;
|
|
using System.Linq;
|
|
using System.Net.Http;
|
|
using System.Text.Json;
|
|
using System.Text.Json.Nodes;
|
|
using FsTool.Tasks;
|
|
|
|
namespace FsTool.Helpers;
|
|
|
|
public static class AuthHelper
|
|
{
|
|
private static async Task<string?> LoadCsrfTokenAsync()
|
|
{
|
|
if (!File.Exists("csrf_token.json")) return null;
|
|
|
|
var json = await File.ReadAllTextAsync("csrf_token.json");
|
|
var tokenInfo = JsonSerializer.Deserialize<JsonObject>(json);
|
|
|
|
if (tokenInfo == null) return null;
|
|
|
|
var token = tokenInfo["token"]?.GetValue<string>();
|
|
var expiresAtString = tokenInfo["expiresAt"]?.GetValue<string>();
|
|
|
|
if (string.IsNullOrWhiteSpace(token) || string.IsNullOrWhiteSpace(expiresAtString)) return null;
|
|
|
|
if (DateTime.TryParse(expiresAtString, out var expiresAt))
|
|
{
|
|
if (DateTime.UtcNow < expiresAt.ToUniversalTime())
|
|
return token;
|
|
}
|
|
|
|
return null;
|
|
}
|
|
|
|
internal static async Task StoreCsrfTokenAsync(string? csrfToken, DateTime expiration)
|
|
{
|
|
if (string.IsNullOrWhiteSpace(csrfToken)) return;
|
|
|
|
var tokenInfo = new
|
|
{
|
|
token = csrfToken,
|
|
expiresAt = expiration
|
|
};
|
|
|
|
var json = JsonSerializer.Serialize(tokenInfo, new JsonSerializerOptions
|
|
{
|
|
WriteIndented = true
|
|
});
|
|
|
|
await File.WriteAllTextAsync("csrf_token.json", json);
|
|
}
|
|
|
|
public static async Task EnsureAuthenticationAsync(HttpClient httpClient)
|
|
{
|
|
// Check if header already contains a CSRF token
|
|
if (httpClient.DefaultRequestHeaders.TryGetValues("X-CSRF-Token", out var existingTokens))
|
|
{
|
|
var existingToken = existingTokens.FirstOrDefault();
|
|
if (!string.IsNullOrWhiteSpace(existingToken))
|
|
return;
|
|
}
|
|
|
|
// Try to load CSRF token from file
|
|
var csrfToken = await LoadCsrfTokenAsync();
|
|
csrfToken = null;
|
|
|
|
// If no valid token found, call login endpoint to get a new one
|
|
if (string.IsNullOrWhiteSpace(csrfToken))
|
|
csrfToken = await UserTasks.CallLoginEndpointAsync(httpClient);
|
|
|
|
// Set CSRF token in HTTP client headers
|
|
if (!string.IsNullOrWhiteSpace(csrfToken))
|
|
{
|
|
csrfToken = csrfToken.ReplaceLineEndings(string.Empty);
|
|
|
|
httpClient.DefaultRequestHeaders.Remove("X-CSRF-Token");
|
|
httpClient.DefaultRequestHeaders.Add("X-CSRF-Token", csrfToken);
|
|
}
|
|
}
|
|
}
|