Location Sharing 2.0 (#125)

Need: Lightless-Sync/LightlessServer#49
Authored-by: Tsubasahane <wozaiha@gmail.com>
Reviewed-on: #125
Reviewed-by: cake <cake@noreply.git.lightless-sync.org>
Co-authored-by: Tsubasa <tsubasa@noreply.git.lightless-sync.org>
Co-committed-by: Tsubasa <tsubasa@noreply.git.lightless-sync.org>

LightlessSync/Services/LocationShareService.cs

@@ -0,0 +1,137 @@
+using LightlessSync.API.Data;
+using LightlessSync.API.Dto.CharaData;
+using LightlessSync.API.Dto.User;
+using LightlessSync.Services.Mediator;
+using LightlessSync.WebAPI;
+using Microsoft.Extensions.Caching.Memory;
+using Microsoft.Extensions.Logging;
+using Microsoft.Extensions.Primitives;
+
+namespace LightlessSync.Services
+{
+    public class LocationShareService : DisposableMediatorSubscriberBase
+    {
+        private readonly DalamudUtilService _dalamudUtilService;
+        private readonly ApiController _apiController;
+        private IMemoryCache _locations = new MemoryCache(new MemoryCacheOptions());
+        private IMemoryCache _sharingStatus = new MemoryCache(new MemoryCacheOptions());
+        private CancellationTokenSource _resetToken = new CancellationTokenSource();
+        
+        public LocationShareService(ILogger<LocationShareService> logger, LightlessMediator mediator, DalamudUtilService dalamudUtilService, ApiController apiController) : base(logger, mediator)
+        {
+            _dalamudUtilService = dalamudUtilService;
+            _apiController = apiController;
+
+
+            Mediator.Subscribe<DisconnectedMessage>(this, (msg) =>
+            {
+                _resetToken.Cancel();
+                _resetToken.Dispose();
+                _resetToken = new CancellationTokenSource();
+            });
+            Mediator.Subscribe<ConnectedMessage>(this, (msg) =>
+            {
+                _ = _apiController.UpdateLocation(new LocationDto(new UserData(_apiController.UID, apiController.DisplayName), _dalamudUtilService.GetMapData()));
+                _ = RequestAllLocation();
+            } );
+            Mediator.Subscribe<LocationSharingMessage>(this, UpdateLocationList);
+            Mediator.Subscribe<MapChangedMessage>(this,
+                msg => _ = _apiController.UpdateLocation(new LocationDto(new UserData(_apiController.UID, _apiController.DisplayName), _dalamudUtilService.GetMapData())));
+        }
+
+        private void UpdateLocationList(LocationSharingMessage msg)
+        {
+            if (_locations.TryGetValue(msg.User.UID, out _) && msg.LocationInfo.ServerId is 0)
+            {
+                _locations.Remove(msg.User.UID);
+                return;
+            }
+
+            if ( msg.LocationInfo.ServerId is not 0 && msg.ExpireAt > DateTime.UtcNow)
+            {
+                AddLocationInfo(msg.User.UID, msg.LocationInfo, msg.ExpireAt);
+            }
+        }
+
+        private void AddLocationInfo(string uid, LocationInfo location, DateTimeOffset expireAt)
+        {
+            var options = new MemoryCacheEntryOptions()
+                .SetAbsoluteExpiration(expireAt)
+                .AddExpirationToken(new CancellationChangeToken(_resetToken.Token));
+            _locations.Set(uid, location, options);
+        }
+
+        private async Task RequestAllLocation()
+        {
+            try
+            {
+                var (data, status) = await _apiController.RequestAllLocationInfo().ConfigureAwait(false);
+                foreach (var dto in data)
+                {
+                    AddLocationInfo(dto.LocationDto.User.UID, dto.LocationDto.Location, dto.ExpireAt);
+                }
+
+                foreach (var dto in status)
+                {
+                    AddStatus(dto.User.UID, dto.ExpireAt);
+                }
+            }
+            catch (Exception e)
+            {
+                Logger.LogError(e,"RequestAllLocation error : ");
+                throw;
+            }
+        }
+
+        private void AddStatus(string uid, DateTimeOffset expireAt)
+        {
+            var options = new MemoryCacheEntryOptions()
+                .SetAbsoluteExpiration(expireAt)
+                .AddExpirationToken(new CancellationChangeToken(_resetToken.Token));
+            _sharingStatus.Set(uid, expireAt, options);
+        }
+
+        public string GetUserLocation(string uid)
+        {
+            try
+            {
+                if (_locations.TryGetValue<LocationInfo>(uid, out var location))
+                {
+                    return _dalamudUtilService.LocationToString(location);
+                }
+                return String.Empty;
+            }
+            catch (Exception e)
+            {
+                Logger.LogError(e,"GetUserLocation error : ");
+                throw;
+            }
+        }
+
+        public DateTimeOffset GetSharingStatus(string uid)
+        {
+            try
+            {
+                if (_sharingStatus.TryGetValue<DateTimeOffset>(uid, out var expireAt))
+                {
+                    return expireAt;
+                }
+                return DateTimeOffset.MinValue;
+            }
+            catch (Exception e)
+            {
+                Logger.LogError(e,"GetSharingStatus error : ");
+                throw;
+            }
+        }
+
+        public void UpdateSharingStatus(List<string> users, DateTimeOffset expireAt)
+        {
+            foreach (var user in users)
+            {
+                AddStatus(user, expireAt);
+            }
+        }
+        
+    }
+}