 0e84d312dc
			
		
	
	0e84d312dc
	
	
	
		
			
			* Add DTR entry with number of visible pairs * Simplify DTR visible pair counter - Merge the new PairHandlerInvisibleMessage and PairHandlerVisibleDisposedMessage ; - Leave alone a bunch of messages that were actually unneccessary. * Match encodings with the rest of the codebase * Make the DTR entry opt-in and add explanations * More reliable publishing of PairHandlerVisible/Invisible * Simplify things for DTR * Rework async
		
			
				
	
	
		
			370 lines
		
	
	
		
			14 KiB
		
	
	
	
		
			C#
		
	
	
	
	
	
			
		
		
	
	
			370 lines
		
	
	
		
			14 KiB
		
	
	
	
		
			C#
		
	
	
	
	
	
| using Dalamud.ContextMenu;
 | |
| using Dalamud.Interface.Internal.Notifications;
 | |
| using MareSynchronos.API.Data;
 | |
| using MareSynchronos.API.Data.Comparer;
 | |
| using MareSynchronos.API.Data.Extensions;
 | |
| using MareSynchronos.API.Dto.Group;
 | |
| using MareSynchronos.API.Dto.User;
 | |
| using MareSynchronos.MareConfiguration;
 | |
| using MareSynchronos.PlayerData.Factories;
 | |
| using MareSynchronos.Services.Mediator;
 | |
| using Microsoft.Extensions.Logging;
 | |
| using System.Collections.Concurrent;
 | |
| 
 | |
| namespace MareSynchronos.PlayerData.Pairs;
 | |
| 
 | |
| public sealed class PairManager : DisposableMediatorSubscriberBase
 | |
| {
 | |
|     private readonly ConcurrentDictionary<UserData, Pair> _allClientPairs = new(UserDataComparer.Instance);
 | |
|     private readonly ConcurrentDictionary<GroupData, GroupFullInfoDto> _allGroups = new(GroupDataComparer.Instance);
 | |
|     private readonly MareConfigService _configurationService;
 | |
|     private readonly DalamudContextMenu _dalamudContextMenu;
 | |
|     private readonly PairFactory _pairFactory;
 | |
|     private Lazy<List<Pair>> _directPairsInternal;
 | |
|     private Lazy<Dictionary<GroupFullInfoDto, List<Pair>>> _groupPairsInternal;
 | |
| 
 | |
|     public PairManager(ILogger<PairManager> logger, PairFactory pairFactory,
 | |
|                 MareConfigService configurationService, MareMediator mediator,
 | |
|                 DalamudContextMenu dalamudContextMenu) : base(logger, mediator)
 | |
|     {
 | |
|         _pairFactory = pairFactory;
 | |
|         _configurationService = configurationService;
 | |
|         _dalamudContextMenu = dalamudContextMenu;
 | |
|         Mediator.Subscribe<DisconnectedMessage>(this, (_) => ClearPairs());
 | |
|         Mediator.Subscribe<CutsceneEndMessage>(this, (_) => ReapplyPairData());
 | |
|         _directPairsInternal = DirectPairsLazy();
 | |
|         _groupPairsInternal = GroupPairsLazy();
 | |
| 
 | |
|         _dalamudContextMenu.OnOpenGameObjectContextMenu += DalamudContextMenuOnOnOpenGameObjectContextMenu;
 | |
|     }
 | |
| 
 | |
|     public List<Pair> DirectPairs => _directPairsInternal.Value;
 | |
| 
 | |
|     public Dictionary<GroupFullInfoDto, List<Pair>> GroupPairs => _groupPairsInternal.Value;
 | |
| 
 | |
|     public Pair? LastAddedUser { get; internal set; }
 | |
| 
 | |
|     public void AddGroup(GroupFullInfoDto dto)
 | |
|     {
 | |
|         _allGroups[dto.Group] = dto;
 | |
|         RecreateLazy();
 | |
|     }
 | |
| 
 | |
|     public void AddGroupPair(GroupPairFullInfoDto dto)
 | |
|     {
 | |
|         if (!_allClientPairs.ContainsKey(dto.User)) _allClientPairs[dto.User] = _pairFactory.Create();
 | |
| 
 | |
|         var group = _allGroups[dto.Group];
 | |
|         _allClientPairs[dto.User].GroupPair[group] = dto;
 | |
|         RecreateLazy();
 | |
|     }
 | |
| 
 | |
|     public void AddUserPair(UserPairDto dto, bool addToLastAddedUser = true)
 | |
|     {
 | |
|         if (!_allClientPairs.ContainsKey(dto.User))
 | |
|         {
 | |
|             _allClientPairs[dto.User] = _pairFactory.Create();
 | |
|         }
 | |
|         else
 | |
|         {
 | |
|             addToLastAddedUser = false;
 | |
|         }
 | |
| 
 | |
|         _allClientPairs[dto.User].UserPair = dto;
 | |
|         if (addToLastAddedUser)
 | |
|             LastAddedUser = _allClientPairs[dto.User];
 | |
|         _allClientPairs[dto.User].ApplyLastReceivedData();
 | |
|         RecreateLazy();
 | |
|     }
 | |
| 
 | |
|     public void ClearPairs()
 | |
|     {
 | |
|         Logger.LogDebug("Clearing all Pairs");
 | |
|         DisposePairs();
 | |
|         _allClientPairs.Clear();
 | |
|         _allGroups.Clear();
 | |
|         RecreateLazy();
 | |
|     }
 | |
| 
 | |
|     public List<Pair> GetOnlineUserPairs() => _allClientPairs.Where(p => !string.IsNullOrEmpty(p.Value.GetPlayerNameHash())).Select(p => p.Value).ToList();
 | |
| 
 | |
|     public List<UserData> GetVisibleUsers() => _allClientPairs.Where(p => p.Value.IsVisible).Select(p => p.Key).ToList();
 | |
| 
 | |
|     public int GetVisibleUserCount() => _allClientPairs.Count(p => p.Value.IsVisible);
 | |
| 
 | |
|     public void MarkPairOffline(UserData user)
 | |
|     {
 | |
|         if (_allClientPairs.TryGetValue(user, out var pair))
 | |
|         {
 | |
|             Mediator.Publish(new ClearProfileDataMessage(pair.UserData));
 | |
|             pair.MarkOffline();
 | |
|             RecreateLazy();
 | |
|         }
 | |
|     }
 | |
| 
 | |
|     public void MarkPairOnline(OnlineUserIdentDto dto, bool sendNotif = true)
 | |
|     {
 | |
|         if (!_allClientPairs.ContainsKey(dto.User)) throw new InvalidOperationException("No user found for " + dto);
 | |
| 
 | |
|         Mediator.Publish(new ClearProfileDataMessage(dto.User));
 | |
| 
 | |
|         var pair = _allClientPairs[dto.User];
 | |
|         if (pair.HasCachedPlayer) return;
 | |
| 
 | |
|         if (sendNotif && _configurationService.Current.ShowOnlineNotifications
 | |
|             && (_configurationService.Current.ShowOnlineNotificationsOnlyForIndividualPairs && pair.UserPair != null
 | |
|             || !_configurationService.Current.ShowOnlineNotificationsOnlyForIndividualPairs)
 | |
|             && (_configurationService.Current.ShowOnlineNotificationsOnlyForNamedPairs && !string.IsNullOrEmpty(pair.GetNote())
 | |
|             || !_configurationService.Current.ShowOnlineNotificationsOnlyForNamedPairs))
 | |
|         {
 | |
|             string? note = pair.GetNote();
 | |
|             var msg = !string.IsNullOrEmpty(note)
 | |
|                 ? $"{note} ({pair.UserData.AliasOrUID}) is now online"
 | |
|                 : $"{pair.UserData.AliasOrUID} is now online";
 | |
|             Mediator.Publish(new NotificationMessage("User online", msg, NotificationType.Info, 5000));
 | |
|         }
 | |
| 
 | |
|         pair.CreateCachedPlayer(dto);
 | |
|         RecreateLazy();
 | |
|     }
 | |
| 
 | |
|     public void ReceiveCharaData(OnlineUserCharaDataDto dto)
 | |
|     {
 | |
|         if (!_allClientPairs.ContainsKey(dto.User)) throw new InvalidOperationException("No user found for " + dto.User);
 | |
| 
 | |
|         _allClientPairs[dto.User].ApplyData(dto);
 | |
|     }
 | |
| 
 | |
|     public void RemoveGroup(GroupData data)
 | |
|     {
 | |
|         _allGroups.TryRemove(data, out _);
 | |
|         foreach (var item in _allClientPairs.ToList())
 | |
|         {
 | |
|             foreach (var grpPair in item.Value.GroupPair.Select(k => k.Key).Where(grpPair => GroupDataComparer.Instance.Equals(grpPair.Group, data)).ToList())
 | |
|             {
 | |
|                 _allClientPairs[item.Key].GroupPair.Remove(grpPair);
 | |
|             }
 | |
| 
 | |
|             if (!_allClientPairs[item.Key].HasAnyConnection() && _allClientPairs.TryRemove(item.Key, out var pair))
 | |
|             {
 | |
|                 pair.MarkOffline();
 | |
|             }
 | |
|         }
 | |
|         RecreateLazy();
 | |
|     }
 | |
| 
 | |
|     public void RemoveGroupPair(GroupPairDto dto)
 | |
|     {
 | |
|         if (_allClientPairs.TryGetValue(dto.User, out var pair))
 | |
|         {
 | |
|             var group = _allGroups[dto.Group];
 | |
|             pair.GroupPair.Remove(group);
 | |
| 
 | |
|             if (!pair.HasAnyConnection())
 | |
|             {
 | |
|                 pair.MarkOffline();
 | |
|                 _allClientPairs.TryRemove(dto.User, out _);
 | |
|             }
 | |
| 
 | |
|             RecreateLazy();
 | |
|         }
 | |
|     }
 | |
| 
 | |
|     public void RemoveUserPair(UserDto dto)
 | |
|     {
 | |
|         if (_allClientPairs.TryGetValue(dto.User, out var pair))
 | |
|         {
 | |
|             pair.UserPair = null;
 | |
|             if (!pair.HasAnyConnection())
 | |
|             {
 | |
|                 pair.MarkOffline();
 | |
|                 _allClientPairs.TryRemove(dto.User, out _);
 | |
|             }
 | |
|             else
 | |
|             {
 | |
|                 pair.ApplyLastReceivedData();
 | |
|             }
 | |
| 
 | |
|             RecreateLazy();
 | |
|         }
 | |
|     }
 | |
| 
 | |
|     public void SetGroupInfo(GroupInfoDto dto)
 | |
|     {
 | |
|         _allGroups[dto.Group].Group = dto.Group;
 | |
|         _allGroups[dto.Group].Owner = dto.Owner;
 | |
|         _allGroups[dto.Group].GroupPermissions = dto.GroupPermissions;
 | |
|         RecreateLazy();
 | |
|     }
 | |
| 
 | |
|     public void UpdatePairPermissions(UserPermissionsDto dto)
 | |
|     {
 | |
|         if (!_allClientPairs.TryGetValue(dto.User, out var pair))
 | |
|         {
 | |
|             throw new InvalidOperationException("No such pair for " + dto);
 | |
|         }
 | |
| 
 | |
|         if (pair.UserPair == null) throw new InvalidOperationException("No direct pair for " + dto);
 | |
| 
 | |
|         if (pair.UserPair.OtherPermissions.IsPaused() != dto.Permissions.IsPaused()
 | |
|             || pair.UserPair.OtherPermissions.IsPaired() != dto.Permissions.IsPaired())
 | |
|         {
 | |
|             Mediator.Publish(new ClearProfileDataMessage(dto.User));
 | |
|         }
 | |
| 
 | |
|         pair.UserPair.OtherPermissions = dto.Permissions;
 | |
| 
 | |
|         Logger.LogTrace("Paired: {synced}, Paused: {paused}, Anims: {anims}, Sounds: {sounds}, VFX: {vfx}",
 | |
|             pair.UserPair.OwnPermissions.IsPaired(), pair.UserPair.OwnPermissions.IsPaused(), pair.UserPair.OwnPermissions.IsDisableAnimations(), pair.UserPair.OwnPermissions.IsDisableSounds(),
 | |
|             pair.UserPair.OwnPermissions.IsDisableVFX());
 | |
|         pair.ApplyLastReceivedData();
 | |
|     }
 | |
| 
 | |
|     public void UpdateSelfPairPermissions(UserPermissionsDto dto)
 | |
|     {
 | |
|         if (!_allClientPairs.TryGetValue(dto.User, out var pair))
 | |
|         {
 | |
|             throw new InvalidOperationException("No such pair for " + dto);
 | |
|         }
 | |
| 
 | |
|         if (pair.UserPair == null) throw new InvalidOperationException("No direct pair for " + dto);
 | |
| 
 | |
|         if (pair.UserPair.OwnPermissions.IsPaused() != dto.Permissions.IsPaused()
 | |
|             || pair.UserPair.OwnPermissions.IsPaired() != dto.Permissions.IsPaired())
 | |
|         {
 | |
|             Mediator.Publish(new ClearProfileDataMessage(dto.User));
 | |
|         }
 | |
| 
 | |
|         pair.UserPair.OwnPermissions = dto.Permissions;
 | |
| 
 | |
|         Logger.LogTrace("Paired: {synced}, Paused: {paused}, Anims: {anims}, Sounds: {sounds}, VFX: {vfx}",
 | |
|             pair.UserPair.OwnPermissions.IsPaired(), pair.UserPair.OwnPermissions.IsPaused(), pair.UserPair.OwnPermissions.IsDisableAnimations(), pair.UserPair.OwnPermissions.IsDisableSounds(),
 | |
|             pair.UserPair.OwnPermissions.IsDisableVFX());
 | |
| 
 | |
|         pair.ApplyLastReceivedData();
 | |
|     }
 | |
| 
 | |
|     internal void ReceiveUploadStatus(UserDto dto)
 | |
|     {
 | |
|         if (_allClientPairs.TryGetValue(dto.User, out var existingPair) && existingPair.IsVisible)
 | |
|         {
 | |
|             existingPair.SetIsUploading();
 | |
|         }
 | |
|     }
 | |
| 
 | |
|     internal void SetGroupPairStatusInfo(GroupPairUserInfoDto dto)
 | |
|     {
 | |
|         var group = _allGroups[dto.Group];
 | |
|         _allClientPairs[dto.User].GroupPair[group].GroupPairStatusInfo = dto.GroupUserInfo;
 | |
|         RecreateLazy();
 | |
|     }
 | |
| 
 | |
|     internal void SetGroupPairUserPermissions(GroupPairUserPermissionDto dto)
 | |
|     {
 | |
|         var group = _allGroups[dto.Group];
 | |
|         var prevPermissions = _allClientPairs[dto.User].GroupPair[group].GroupUserPermissions;
 | |
|         _allClientPairs[dto.User].GroupPair[group].GroupUserPermissions = dto.GroupPairPermissions;
 | |
|         if (prevPermissions.IsDisableAnimations() != dto.GroupPairPermissions.IsDisableAnimations()
 | |
|             || prevPermissions.IsDisableSounds() != dto.GroupPairPermissions.IsDisableSounds()
 | |
|             || prevPermissions.IsDisableVFX() != dto.GroupPairPermissions.IsDisableVFX())
 | |
|         {
 | |
|             _allClientPairs[dto.User].ApplyLastReceivedData();
 | |
|         }
 | |
|         RecreateLazy();
 | |
|     }
 | |
| 
 | |
|     internal void SetGroupPermissions(GroupPermissionDto dto)
 | |
|     {
 | |
|         var prevPermissions = _allGroups[dto.Group].GroupPermissions;
 | |
|         _allGroups[dto.Group].GroupPermissions = dto.Permissions;
 | |
|         if (prevPermissions.IsDisableAnimations() != dto.Permissions.IsDisableAnimations()
 | |
|             || prevPermissions.IsDisableSounds() != dto.Permissions.IsDisableSounds()
 | |
|             || prevPermissions.IsDisableVFX() != dto.Permissions.IsDisableVFX())
 | |
|         {
 | |
|             RecreateLazy();
 | |
|             var group = _allGroups[dto.Group];
 | |
|             GroupPairs[group].ForEach(p => p.ApplyLastReceivedData());
 | |
|         }
 | |
|         RecreateLazy();
 | |
|     }
 | |
| 
 | |
|     internal void SetGroupStatusInfo(GroupPairUserInfoDto dto)
 | |
|     {
 | |
|         _allGroups[dto.Group].GroupUserInfo = dto.GroupUserInfo;
 | |
|     }
 | |
| 
 | |
|     internal void SetGroupUserPermissions(GroupPairUserPermissionDto dto)
 | |
|     {
 | |
|         var prevPermissions = _allGroups[dto.Group].GroupUserPermissions;
 | |
|         _allGroups[dto.Group].GroupUserPermissions = dto.GroupPairPermissions;
 | |
|         if (prevPermissions.IsDisableAnimations() != dto.GroupPairPermissions.IsDisableAnimations()
 | |
|             || prevPermissions.IsDisableSounds() != dto.GroupPairPermissions.IsDisableSounds()
 | |
|             || prevPermissions.IsDisableVFX() != dto.GroupPairPermissions.IsDisableVFX())
 | |
|         {
 | |
|             RecreateLazy();
 | |
|             var group = _allGroups[dto.Group];
 | |
|             GroupPairs[group].ForEach(p => p.ApplyLastReceivedData());
 | |
|         }
 | |
|         RecreateLazy();
 | |
|     }
 | |
| 
 | |
|     protected override void Dispose(bool disposing)
 | |
|     {
 | |
|         base.Dispose(disposing);
 | |
| 
 | |
|         _dalamudContextMenu.OnOpenGameObjectContextMenu -= DalamudContextMenuOnOnOpenGameObjectContextMenu;
 | |
| 
 | |
|         DisposePairs();
 | |
|     }
 | |
| 
 | |
|     private void DalamudContextMenuOnOnOpenGameObjectContextMenu(GameObjectContextMenuOpenArgs args)
 | |
|     {
 | |
|         if (args.ObjectId == 0xE000000) return;
 | |
|         if (!_configurationService.Current.EnableRightClickMenus) return;
 | |
| 
 | |
|         foreach (var pair in _allClientPairs.Where((p => p.Value.IsVisible)))
 | |
|         {
 | |
|             pair.Value.AddContextMenu(args);
 | |
|         }
 | |
|     }
 | |
| 
 | |
|     private Lazy<List<Pair>> DirectPairsLazy() => new(() => _allClientPairs.Select(k => k.Value).Where(k => k.UserPair != null).ToList());
 | |
| 
 | |
|     private void DisposePairs()
 | |
|     {
 | |
|         Logger.LogDebug("Disposing all Pairs");
 | |
|         Parallel.ForEach(_allClientPairs, item =>
 | |
|         {
 | |
|             item.Value.MarkOffline();
 | |
|         });
 | |
| 
 | |
|         RecreateLazy();
 | |
|     }
 | |
| 
 | |
|     private Lazy<Dictionary<GroupFullInfoDto, List<Pair>>> GroupPairsLazy()
 | |
|     {
 | |
|         return new Lazy<Dictionary<GroupFullInfoDto, List<Pair>>>(() =>
 | |
|         {
 | |
|             Dictionary<GroupFullInfoDto, List<Pair>> outDict = new();
 | |
|             foreach (var group in _allGroups)
 | |
|             {
 | |
|                 outDict[group.Value] = _allClientPairs.Select(p => p.Value).Where(p => p.GroupPair.Any(g => GroupDataComparer.Instance.Equals(group.Key, g.Key.Group))).ToList();
 | |
|             }
 | |
|             return outDict;
 | |
|         });
 | |
|     }
 | |
| 
 | |
|     private void ReapplyPairData()
 | |
|     {
 | |
|         foreach (var pair in _allClientPairs.Select(k => k.Value))
 | |
|         {
 | |
|             pair.ApplyLastReceivedData(true);
 | |
|         }
 | |
|     }
 | |
| 
 | |
|     private void RecreateLazy()
 | |
|     {
 | |
|         _directPairsInternal = DirectPairsLazy();
 | |
|         _groupPairsInternal = GroupPairsLazy();
 | |
|     }
 | |
| } |