* move stuff out into file transfer manager * obnoxious unsupported version text, adjustments to filetransfermanager * add back file upload transfer progress * restructure code * cleanup some more stuff I guess * downloadids by playername * individual anim/sound bs * fix migration stuff, finalize impl of individual sound/anim pause * fixes with logging stuff * move download manager to transient * rework dl ui first iteration * some refactoring and cleanup * more code cleanup * refactoring * switch to hostbuilder * some more rework I guess * more refactoring * clean up mediator calls and disposal * fun code cleanup * push error message when log level is set to anything but information in non-debug builds * remove notificationservice * move message to after login * add download bars to gameworld * fixes download progress bar * set gpose ui min and max size * remove unnecessary usings * adjustments to reconnection logic * add options to set visible/offline groups visibility * add impl of uploading display, transfer list in settings ui * attempt to fix issues with server selection * add back download status to compact ui * make dl bar fixed size based * some fixes for upload/download handling * adjust text from Syncing back to Uploading --------- Co-authored-by: rootdarkarchon <root.darkarchon@outlook.com> Co-authored-by: Stanley Dimant <stanley.dimant@varian.com>
350 lines
14 KiB
C#
350 lines
14 KiB
C#
using MareSynchronos.API.Routes;
|
|
using MareSynchronos.Utils;
|
|
using Microsoft.AspNetCore.SignalR.Client;
|
|
using Microsoft.Extensions.Logging;
|
|
using MareSynchronos.API.Dto;
|
|
using MareSynchronos.API.SignalR;
|
|
using Dalamud.Utility;
|
|
using System.Reflection;
|
|
using MareSynchronos.WebAPI.SignalR.Utils;
|
|
using MareSynchronos.WebAPI.SignalR;
|
|
using MareSynchronos.PlayerData.Pairs;
|
|
using MareSynchronos.Services.Mediator;
|
|
using MareSynchronos.Services.ServerConfiguration;
|
|
using MareSynchronos.Services;
|
|
|
|
namespace MareSynchronos.WebAPI;
|
|
|
|
public sealed partial class ApiController : DisposableMediatorSubscriberBase, IMareHubClient
|
|
{
|
|
public const string MainServer = "Lunae Crescere Incipientis (Central Server EU)";
|
|
public const string MainServiceUri = "wss://maresynchronos.com";
|
|
|
|
private readonly DalamudUtilService _dalamudUtil;
|
|
private readonly HubFactory _hubFactory;
|
|
private readonly PairManager _pairManager;
|
|
private readonly ServerConfigurationManager _serverManager;
|
|
private CancellationTokenSource _connectionCancellationTokenSource;
|
|
private ConnectionDto? _connectionDto;
|
|
private bool _doNotNotifyOnNextInfo = false;
|
|
private CancellationTokenSource? _healthCheckTokenSource = new();
|
|
private bool _initialized;
|
|
private HubConnection? _mareHub;
|
|
private ServerState _serverState;
|
|
|
|
public ApiController(ILogger<ApiController> logger, HubFactory hubFactory, DalamudUtilService dalamudUtil,
|
|
PairManager pairManager, ServerConfigurationManager serverManager, MareMediator mediator) : base(logger, mediator)
|
|
{
|
|
_hubFactory = hubFactory;
|
|
_dalamudUtil = dalamudUtil;
|
|
_pairManager = pairManager;
|
|
_serverManager = serverManager;
|
|
_connectionCancellationTokenSource = new CancellationTokenSource();
|
|
|
|
Mediator.Subscribe<DalamudLoginMessage>(this, (_) => DalamudUtilOnLogIn());
|
|
Mediator.Subscribe<DalamudLogoutMessage>(this, (_) => DalamudUtilOnLogOut());
|
|
Mediator.Subscribe<HubClosedMessage>(this, (msg) => MareHubOnClosed(msg.Exception));
|
|
Mediator.Subscribe<HubReconnectedMessage>(this, (msg) => _ = Task.Run(MareHubOnReconnected));
|
|
Mediator.Subscribe<HubReconnectingMessage>(this, (msg) => MareHubOnReconnecting(msg.Exception));
|
|
|
|
ServerState = ServerState.Offline;
|
|
|
|
if (_dalamudUtil.IsLoggedIn)
|
|
{
|
|
DalamudUtilOnLogIn();
|
|
}
|
|
}
|
|
|
|
public string AuthFailureMessage { get; private set; } = string.Empty;
|
|
public Version CurrentClientVersion => _connectionDto?.CurrentClientVersion ?? new Version(0, 0, 0);
|
|
public string DisplayName => _connectionDto?.User.AliasOrUID ?? string.Empty;
|
|
public bool IsConnected => ServerState == ServerState.Connected;
|
|
public bool IsCurrentVersion => (Assembly.GetExecutingAssembly().GetName().Version ?? new Version(0, 0, 0, 0)) >= (_connectionDto?.CurrentClientVersion ?? new Version(0, 0, 0, 0));
|
|
public int OnlineUsers => SystemInfoDto.OnlineUsers;
|
|
public bool ServerAlive => ServerState is ServerState.Connected or ServerState.RateLimited or ServerState.Unauthorized or ServerState.Disconnected;
|
|
public ServerInfo ServerInfo => _connectionDto?.ServerInfo ?? new ServerInfo();
|
|
|
|
public ServerState ServerState
|
|
{
|
|
get => _serverState;
|
|
private set
|
|
{
|
|
Logger.LogDebug("New ServerState: {value}, prev ServerState: {_serverState}", value, _serverState);
|
|
_serverState = value;
|
|
}
|
|
}
|
|
|
|
public SystemInfoDto SystemInfoDto { get; private set; } = new();
|
|
public string UID => _connectionDto?.User.UID ?? string.Empty;
|
|
|
|
public async Task<bool> CheckClientHealth()
|
|
{
|
|
return await _mareHub!.InvokeAsync<bool>(nameof(CheckClientHealth)).ConfigureAwait(false);
|
|
}
|
|
|
|
public async Task CreateConnections(bool forceGetToken = false)
|
|
{
|
|
Logger.LogDebug("CreateConnections called");
|
|
|
|
if (_serverManager.CurrentServer?.FullPause ?? true)
|
|
{
|
|
Logger.LogInformation("Not recreating Connection, paused");
|
|
_connectionDto = null;
|
|
await StopConnection(ServerState.Disconnected).ConfigureAwait(false);
|
|
_connectionCancellationTokenSource.Cancel();
|
|
return;
|
|
}
|
|
|
|
var secretKey = _serverManager.GetSecretKey();
|
|
if (secretKey.IsNullOrEmpty())
|
|
{
|
|
Logger.LogWarning("No secret key set for current character");
|
|
_connectionDto = null;
|
|
await StopConnection(ServerState.NoSecretKey).ConfigureAwait(false);
|
|
_connectionCancellationTokenSource.Cancel();
|
|
return;
|
|
}
|
|
|
|
await StopConnection(ServerState.Disconnected).ConfigureAwait(false);
|
|
|
|
Logger.LogInformation("Recreating Connection");
|
|
|
|
_connectionCancellationTokenSource.Cancel();
|
|
_connectionCancellationTokenSource = new CancellationTokenSource();
|
|
var token = _connectionCancellationTokenSource.Token;
|
|
while (ServerState is not ServerState.Connected && !token.IsCancellationRequested)
|
|
{
|
|
AuthFailureMessage = string.Empty;
|
|
|
|
await StopConnection(ServerState.Disconnected).ConfigureAwait(false);
|
|
ServerState = ServerState.Connecting;
|
|
|
|
try
|
|
{
|
|
Logger.LogDebug("Building connection");
|
|
|
|
if (_serverManager.GetToken() == null || forceGetToken)
|
|
{
|
|
Logger.LogDebug("Requesting new JWT");
|
|
using HttpClient httpClient = new();
|
|
var postUri = MareAuth.AuthFullPath(new Uri(_serverManager.CurrentApiUrl
|
|
.Replace("wss://", "https://", StringComparison.OrdinalIgnoreCase)
|
|
.Replace("ws://", "http://", StringComparison.OrdinalIgnoreCase)));
|
|
var auth = secretKey.GetHash256();
|
|
var result = await httpClient.PostAsync(postUri, new FormUrlEncodedContent(new[]
|
|
{
|
|
new KeyValuePair<string, string>("auth", auth),
|
|
new KeyValuePair<string, string>("charaIdent", _dalamudUtil.PlayerNameHashed),
|
|
})).ConfigureAwait(false);
|
|
AuthFailureMessage = await result.Content.ReadAsStringAsync().ConfigureAwait(false);
|
|
result.EnsureSuccessStatusCode();
|
|
_serverManager.SaveToken(await result.Content.ReadAsStringAsync().ConfigureAwait(false));
|
|
Logger.LogDebug("JWT Success");
|
|
}
|
|
|
|
while (!_dalamudUtil.IsPlayerPresent && !token.IsCancellationRequested)
|
|
{
|
|
Logger.LogDebug("Player not loaded in yet, waiting");
|
|
await Task.Delay(TimeSpan.FromSeconds(1), token).ConfigureAwait(false);
|
|
}
|
|
|
|
if (token.IsCancellationRequested) break;
|
|
|
|
_mareHub = _hubFactory.GetOrCreate();
|
|
|
|
await _mareHub.StartAsync(token).ConfigureAwait(false);
|
|
|
|
await InitializeData().ConfigureAwait(false);
|
|
|
|
_connectionDto = await GetConnectionDto().ConfigureAwait(false);
|
|
|
|
ServerState = ServerState.Connected;
|
|
|
|
if (_connectionDto.ServerVersion != IMareHub.ApiVersion)
|
|
{
|
|
await StopConnection(ServerState.VersionMisMatch).ConfigureAwait(false);
|
|
return;
|
|
}
|
|
}
|
|
catch (HttpRequestException ex)
|
|
{
|
|
Logger.LogWarning(ex, "HttpRequestException on Connection");
|
|
|
|
if (ex.StatusCode == System.Net.HttpStatusCode.Unauthorized)
|
|
{
|
|
await StopConnection(ServerState.Unauthorized).ConfigureAwait(false);
|
|
return;
|
|
}
|
|
|
|
ServerState = ServerState.Reconnecting;
|
|
Logger.LogInformation("Failed to establish connection, retrying");
|
|
await Task.Delay(TimeSpan.FromSeconds(new Random().Next(5, 20)), token).ConfigureAwait(false);
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
Logger.LogWarning(ex, "Exception on Connection");
|
|
|
|
Logger.LogInformation("Failed to establish connection, retrying");
|
|
await Task.Delay(TimeSpan.FromSeconds(new Random().Next(5, 20)), token).ConfigureAwait(false);
|
|
}
|
|
}
|
|
}
|
|
|
|
public async Task<ConnectionDto> GetConnectionDto()
|
|
{
|
|
var dto = await _mareHub!.InvokeAsync<ConnectionDto>(nameof(GetConnectionDto)).ConfigureAwait(false);
|
|
Mediator.Publish(new ConnectedMessage(dto));
|
|
return dto;
|
|
}
|
|
|
|
protected override void Dispose(bool disposing)
|
|
{
|
|
base.Dispose(disposing);
|
|
|
|
_healthCheckTokenSource?.Cancel();
|
|
Task.Run(async () => await StopConnection(ServerState.Disconnected).ConfigureAwait(false));
|
|
_connectionCancellationTokenSource?.Cancel();
|
|
}
|
|
|
|
private async Task ClientHealthCheck(CancellationToken ct)
|
|
{
|
|
while (!ct.IsCancellationRequested && _mareHub != null)
|
|
{
|
|
await Task.Delay(TimeSpan.FromSeconds(30), ct).ConfigureAwait(false);
|
|
_ = await CheckClientHealth().ConfigureAwait(false);
|
|
Logger.LogDebug("Checked Client Health State");
|
|
}
|
|
}
|
|
|
|
private void DalamudUtilOnLogIn()
|
|
{
|
|
Task.Run(() => CreateConnections(forceGetToken: true));
|
|
}
|
|
|
|
private void DalamudUtilOnLogOut()
|
|
{
|
|
Task.Run(async () => await StopConnection(ServerState.Disconnected).ConfigureAwait(false));
|
|
ServerState = ServerState.Offline;
|
|
}
|
|
|
|
private async Task InitializeData()
|
|
{
|
|
if (_mareHub == null) return;
|
|
|
|
Logger.LogDebug("Initializing data");
|
|
OnDownloadReady((guid) => Client_DownloadReady(guid));
|
|
OnReceiveServerMessage((sev, msg) => Client_ReceiveServerMessage(sev, msg));
|
|
OnUpdateSystemInfo((dto) => Client_UpdateSystemInfo(dto));
|
|
|
|
OnUserSendOffline((dto) => Client_UserSendOffline(dto));
|
|
OnUserAddClientPair((dto) => Client_UserAddClientPair(dto));
|
|
OnUserReceiveCharacterData((dto) => Client_UserReceiveCharacterData(dto));
|
|
OnUserRemoveClientPair(dto => Client_UserRemoveClientPair(dto));
|
|
OnUserSendOnline(dto => Client_UserSendOnline(dto));
|
|
OnUserUpdateOtherPairPermissions(dto => Client_UserUpdateOtherPairPermissions(dto));
|
|
OnUserUpdateSelfPairPermissions(dto => Client_UserUpdateSelfPairPermissions(dto));
|
|
OnUserReceiveUploadStatus(dto => Client_UserReceiveUploadStatus(dto));
|
|
|
|
OnGroupChangePermissions((dto) => Client_GroupChangePermissions(dto));
|
|
OnGroupDelete((dto) => Client_GroupDelete(dto));
|
|
OnGroupPairChangePermissions((dto) => Client_GroupPairChangePermissions(dto));
|
|
OnGroupPairChangeUserInfo((dto) => Client_GroupPairChangeUserInfo(dto));
|
|
OnGroupPairJoined((dto) => Client_GroupPairJoined(dto));
|
|
OnGroupPairLeft((dto) => Client_GroupPairLeft(dto));
|
|
OnGroupSendFullInfo((dto) => Client_GroupSendFullInfo(dto));
|
|
OnGroupSendInfo((dto) => Client_GroupSendInfo(dto));
|
|
|
|
foreach (var userPair in await UserGetPairedClients().ConfigureAwait(false))
|
|
{
|
|
Logger.LogDebug("Individual Pair: {userPair}", userPair);
|
|
_pairManager.AddUserPair(userPair, addToLastAddedUser: false);
|
|
}
|
|
foreach (var entry in await GroupsGetAll().ConfigureAwait(false))
|
|
{
|
|
Logger.LogDebug("Group: {entry}", entry);
|
|
_pairManager.AddGroup(entry);
|
|
}
|
|
foreach (var group in _pairManager.GroupPairs.Keys)
|
|
{
|
|
var users = await GroupsGetUsersInGroup(group).ConfigureAwait(false);
|
|
foreach (var user in users)
|
|
{
|
|
Logger.LogDebug("Group Pair: {user}", user);
|
|
_pairManager.AddGroupPair(user);
|
|
}
|
|
}
|
|
|
|
foreach (var entry in await UserGetOnlinePairs().ConfigureAwait(false))
|
|
{
|
|
_pairManager.MarkPairOnline(entry, sendNotif: false);
|
|
}
|
|
|
|
_healthCheckTokenSource?.Cancel();
|
|
_healthCheckTokenSource?.Dispose();
|
|
_healthCheckTokenSource = new CancellationTokenSource();
|
|
_ = ClientHealthCheck(_healthCheckTokenSource.Token);
|
|
|
|
_initialized = true;
|
|
}
|
|
|
|
private void MareHubOnClosed(Exception? arg)
|
|
{
|
|
_healthCheckTokenSource?.Cancel();
|
|
Mediator.Publish(new DisconnectedMessage());
|
|
_pairManager.ClearPairs();
|
|
ServerState = ServerState.Offline;
|
|
if (arg != null)
|
|
{
|
|
Logger.LogWarning(arg, "Connection closed");
|
|
}
|
|
else
|
|
{
|
|
Logger.LogInformation("Connection closed");
|
|
}
|
|
}
|
|
|
|
private async Task MareHubOnReconnected()
|
|
{
|
|
ServerState = ServerState.Connecting;
|
|
try
|
|
{
|
|
await InitializeData().ConfigureAwait(false);
|
|
_connectionDto = await GetConnectionDto().ConfigureAwait(false);
|
|
if (_connectionDto.ServerVersion != IMareHub.ApiVersion)
|
|
{
|
|
await StopConnection(ServerState.VersionMisMatch).ConfigureAwait(false);
|
|
return;
|
|
}
|
|
ServerState = ServerState.Connected;
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
Logger.LogCritical(ex, "Failure to obtain data after reconnection");
|
|
await StopConnection(ServerState.Disconnected).ConfigureAwait(false);
|
|
}
|
|
}
|
|
|
|
private void MareHubOnReconnecting(Exception? arg)
|
|
{
|
|
_doNotNotifyOnNextInfo = true;
|
|
_healthCheckTokenSource?.Cancel();
|
|
ServerState = ServerState.Reconnecting;
|
|
Logger.LogWarning(arg, "Connection closed... Reconnecting");
|
|
}
|
|
|
|
private async Task StopConnection(ServerState state)
|
|
{
|
|
ServerState = state;
|
|
|
|
if (_mareHub is not null)
|
|
{
|
|
_initialized = false;
|
|
_healthCheckTokenSource?.Cancel();
|
|
Logger.LogInformation("Stopping existing connection");
|
|
await _hubFactory.DisposeHubAsync().ConfigureAwait(false);
|
|
Mediator.Publish(new DisconnectedMessage());
|
|
_mareHub = null;
|
|
_connectionDto = null;
|
|
}
|
|
}
|
|
} |