225 lines
11 KiB
C#
225 lines
11 KiB
C#
using System.Net.Http.Headers;
|
|
using System.Text;
|
|
using System.Text.Json;
|
|
using System.Text.Json.Serialization;
|
|
using Microsoft.Extensions.Logging;
|
|
using MareSynchronos.WebAPI.SignalR;
|
|
using MareSynchronos.Services.AutoDetect;
|
|
|
|
namespace MareSynchronos.WebAPI.AutoDetect;
|
|
|
|
public class DiscoveryApiClient
|
|
{
|
|
private readonly ILogger<DiscoveryApiClient> _logger;
|
|
private readonly TokenProvider _tokenProvider;
|
|
private readonly DiscoveryConfigProvider _configProvider;
|
|
private readonly HttpClient _httpClient = new();
|
|
private static readonly JsonSerializerOptions JsonOpt = new() { PropertyNameCaseInsensitive = true };
|
|
|
|
public DiscoveryApiClient(ILogger<DiscoveryApiClient> logger, TokenProvider tokenProvider, DiscoveryConfigProvider configProvider)
|
|
{
|
|
_logger = logger;
|
|
_tokenProvider = tokenProvider;
|
|
_configProvider = configProvider;
|
|
_httpClient.Timeout = TimeSpan.FromSeconds(30);
|
|
}
|
|
|
|
public async Task<List<ServerMatch>> QueryAsync(string endpoint, IEnumerable<string> hashes, CancellationToken ct)
|
|
{
|
|
try
|
|
{
|
|
var token = await _tokenProvider.GetOrUpdateToken(ct).ConfigureAwait(false);
|
|
if (string.IsNullOrEmpty(token)) return [];
|
|
var distinctHashes = hashes.Distinct(StringComparer.Ordinal).ToArray();
|
|
using var req = new HttpRequestMessage(HttpMethod.Post, endpoint);
|
|
req.Headers.Authorization = new AuthenticationHeaderValue("Bearer", token);
|
|
var body = JsonSerializer.Serialize(new
|
|
{
|
|
hashes = distinctHashes,
|
|
salt = _configProvider.SaltB64
|
|
});
|
|
req.Content = new StringContent(body, Encoding.UTF8, "application/json");
|
|
var resp = await _httpClient.SendAsync(req, ct).ConfigureAwait(false);
|
|
if (resp.StatusCode == System.Net.HttpStatusCode.Unauthorized)
|
|
{
|
|
var token2 = await _tokenProvider.ForceRefreshToken(ct).ConfigureAwait(false);
|
|
if (string.IsNullOrEmpty(token2)) return [];
|
|
using var req2 = new HttpRequestMessage(HttpMethod.Post, endpoint);
|
|
req2.Headers.Authorization = new AuthenticationHeaderValue("Bearer", token2);
|
|
var body2 = JsonSerializer.Serialize(new
|
|
{
|
|
hashes = distinctHashes,
|
|
salt = _configProvider.SaltB64
|
|
});
|
|
req2.Content = new StringContent(body2, Encoding.UTF8, "application/json");
|
|
resp = await _httpClient.SendAsync(req2, ct).ConfigureAwait(false);
|
|
}
|
|
resp.EnsureSuccessStatusCode();
|
|
var json = await resp.Content.ReadAsStringAsync(ct).ConfigureAwait(false);
|
|
var result = JsonSerializer.Deserialize<List<ServerMatch>>(json, JsonOpt) ?? [];
|
|
return result;
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
_logger.LogWarning(ex, "Discovery query failed");
|
|
return [];
|
|
}
|
|
}
|
|
|
|
public async Task<bool> SendRequestAsync(string endpoint, string? token, string? targetUid, string? displayName, CancellationToken ct)
|
|
{
|
|
try
|
|
{
|
|
if (string.IsNullOrEmpty(token) && string.IsNullOrEmpty(targetUid))
|
|
{
|
|
_logger.LogWarning("Discovery request aborted: no token or targetUid provided");
|
|
return false;
|
|
}
|
|
|
|
var jwt = await _tokenProvider.GetOrUpdateToken(ct).ConfigureAwait(false);
|
|
if (string.IsNullOrEmpty(jwt)) return false;
|
|
using var req = new HttpRequestMessage(HttpMethod.Post, endpoint);
|
|
req.Headers.Authorization = new AuthenticationHeaderValue("Bearer", jwt);
|
|
var body = JsonSerializer.Serialize(new RequestPayload(token, targetUid, displayName));
|
|
req.Content = new StringContent(body, Encoding.UTF8, "application/json");
|
|
var resp = await _httpClient.SendAsync(req, ct).ConfigureAwait(false);
|
|
if (resp.StatusCode == System.Net.HttpStatusCode.Unauthorized)
|
|
{
|
|
var jwt2 = await _tokenProvider.ForceRefreshToken(ct).ConfigureAwait(false);
|
|
if (string.IsNullOrEmpty(jwt2)) return false;
|
|
using var req2 = new HttpRequestMessage(HttpMethod.Post, endpoint);
|
|
req2.Headers.Authorization = new AuthenticationHeaderValue("Bearer", jwt2);
|
|
var body2 = JsonSerializer.Serialize(new RequestPayload(token, targetUid, displayName));
|
|
req2.Content = new StringContent(body2, Encoding.UTF8, "application/json");
|
|
resp = await _httpClient.SendAsync(req2, ct).ConfigureAwait(false);
|
|
}
|
|
if (!resp.IsSuccessStatusCode)
|
|
{
|
|
string txt = string.Empty;
|
|
try { txt = await resp.Content.ReadAsStringAsync(ct).ConfigureAwait(false); } catch { }
|
|
_logger.LogWarning("Discovery request failed: {code} {reason} {body}", (int)resp.StatusCode, resp.ReasonPhrase, txt);
|
|
return false;
|
|
}
|
|
return true;
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
_logger.LogWarning(ex, "Discovery send request failed");
|
|
return false;
|
|
}
|
|
}
|
|
|
|
private sealed record RequestPayload(
|
|
[property: JsonPropertyName("token"), JsonIgnore(Condition = JsonIgnoreCondition.WhenWritingNull)]
|
|
string? Token,
|
|
[property: JsonPropertyName("targetUid"), JsonIgnore(Condition = JsonIgnoreCondition.WhenWritingNull)]
|
|
string? TargetUid,
|
|
[property: JsonPropertyName("displayName"), JsonIgnore(Condition = JsonIgnoreCondition.WhenWritingNull)]
|
|
string? DisplayName);
|
|
|
|
public async Task<bool> PublishAsync(string endpoint, IEnumerable<string> hashes, string? displayName, CancellationToken ct, bool allowRequests = true)
|
|
{
|
|
try
|
|
{
|
|
var jwt = await _tokenProvider.GetOrUpdateToken(ct).ConfigureAwait(false);
|
|
if (string.IsNullOrEmpty(jwt)) return false;
|
|
using var req = new HttpRequestMessage(HttpMethod.Post, endpoint);
|
|
req.Headers.Authorization = new AuthenticationHeaderValue("Bearer", jwt);
|
|
var bodyObj = new
|
|
{
|
|
hashes = hashes.Distinct(StringComparer.Ordinal).ToArray(),
|
|
displayName,
|
|
salt = _configProvider.SaltB64,
|
|
allowRequests
|
|
};
|
|
var body = JsonSerializer.Serialize(bodyObj);
|
|
req.Content = new StringContent(body, Encoding.UTF8, "application/json");
|
|
var resp = await _httpClient.SendAsync(req, ct).ConfigureAwait(false);
|
|
if (resp.StatusCode == System.Net.HttpStatusCode.Unauthorized)
|
|
{
|
|
var jwt2 = await _tokenProvider.ForceRefreshToken(ct).ConfigureAwait(false);
|
|
if (string.IsNullOrEmpty(jwt2)) return false;
|
|
using var req2 = new HttpRequestMessage(HttpMethod.Post, endpoint);
|
|
req2.Headers.Authorization = new AuthenticationHeaderValue("Bearer", jwt2);
|
|
var body2 = JsonSerializer.Serialize(bodyObj);
|
|
req2.Content = new StringContent(body2, Encoding.UTF8, "application/json");
|
|
resp = await _httpClient.SendAsync(req2, ct).ConfigureAwait(false);
|
|
}
|
|
return resp.IsSuccessStatusCode;
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
_logger.LogWarning(ex, "Discovery publish failed");
|
|
return false;
|
|
}
|
|
}
|
|
|
|
public async Task<bool> SendAcceptAsync(string endpoint, string targetUid, string? displayName, CancellationToken ct)
|
|
{
|
|
try
|
|
{
|
|
var jwt = await _tokenProvider.GetOrUpdateToken(ct).ConfigureAwait(false);
|
|
if (string.IsNullOrEmpty(jwt)) return false;
|
|
using var req = new HttpRequestMessage(HttpMethod.Post, endpoint);
|
|
req.Headers.Authorization = new AuthenticationHeaderValue("Bearer", jwt);
|
|
var bodyObj = new { targetUid, displayName };
|
|
var body = JsonSerializer.Serialize(bodyObj);
|
|
req.Content = new StringContent(body, Encoding.UTF8, "application/json");
|
|
var resp = await _httpClient.SendAsync(req, ct).ConfigureAwait(false);
|
|
if (resp.StatusCode == System.Net.HttpStatusCode.Unauthorized)
|
|
{
|
|
var jwt2 = await _tokenProvider.ForceRefreshToken(ct).ConfigureAwait(false);
|
|
if (string.IsNullOrEmpty(jwt2)) return false;
|
|
using var req2 = new HttpRequestMessage(HttpMethod.Post, endpoint);
|
|
req2.Headers.Authorization = new AuthenticationHeaderValue("Bearer", jwt2);
|
|
var body2 = JsonSerializer.Serialize(bodyObj);
|
|
req2.Content = new StringContent(body2, Encoding.UTF8, "application/json");
|
|
resp = await _httpClient.SendAsync(req2, ct).ConfigureAwait(false);
|
|
}
|
|
return resp.IsSuccessStatusCode;
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
_logger.LogWarning(ex, "Discovery accept notify failed");
|
|
return false;
|
|
}
|
|
}
|
|
public async Task DisableAsync(string endpoint, CancellationToken ct)
|
|
{
|
|
try
|
|
{
|
|
var jwt = await _tokenProvider.GetOrUpdateToken(ct).ConfigureAwait(false);
|
|
if (string.IsNullOrEmpty(jwt)) return;
|
|
using var req = new HttpRequestMessage(HttpMethod.Post, endpoint);
|
|
req.Headers.Authorization = new AuthenticationHeaderValue("Bearer", jwt);
|
|
var resp = await _httpClient.SendAsync(req, ct).ConfigureAwait(false);
|
|
if (resp.StatusCode == System.Net.HttpStatusCode.Unauthorized)
|
|
{
|
|
var jwt2 = await _tokenProvider.ForceRefreshToken(ct).ConfigureAwait(false);
|
|
if (string.IsNullOrEmpty(jwt2)) return;
|
|
using var req2 = new HttpRequestMessage(HttpMethod.Post, endpoint);
|
|
req2.Headers.Authorization = new AuthenticationHeaderValue("Bearer", jwt2);
|
|
resp = await _httpClient.SendAsync(req2, ct).ConfigureAwait(false);
|
|
}
|
|
if (!resp.IsSuccessStatusCode)
|
|
{
|
|
string txt = string.Empty;
|
|
try { txt = await resp.Content.ReadAsStringAsync(ct).ConfigureAwait(false); } catch { }
|
|
_logger.LogWarning("Discovery disable failed: {code} {reason} {body}", (int)resp.StatusCode, resp.ReasonPhrase, txt);
|
|
}
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
_logger.LogWarning(ex, "Discovery disable failed");
|
|
}
|
|
}
|
|
}
|
|
|
|
public sealed class ServerMatch
|
|
{
|
|
public string Hash { get; set; } = string.Empty;
|
|
public string? Token { get; set; }
|
|
public string? Uid { get; set; }
|
|
public string? DisplayName { get; set; }
|
|
}
|