PalacePal/Pal.Client/Floors/Tasks/SaveNewLocations.cs

using System;
using System.Collections.Generic;
using System.Linq;
using Microsoft.Extensions.DependencyInjection;
using Microsoft.Extensions.Logging;
using Pal.Client.Database;
using Pal.Common;

namespace Pal.Client.Floors.Tasks;

internal sealed class SaveNewLocations : DbTask<SaveNewLocations>
{
    private readonly MemoryTerritory _territory;
    private readonly List<PersistentLocation> _newLocations;

    public SaveNewLocations(IServiceScopeFactory serviceScopeFactory, MemoryTerritory territory,
        List<PersistentLocation> newLocations)
        : base(serviceScopeFactory)
    {
        _territory = territory;
        _newLocations = newLocations;
    }

    protected override void Run(PalClientContext dbContext, ILogger<SaveNewLocations> logger)
    {
        Run(_territory, dbContext, logger, _newLocations);
    }

    public static void Run<T>(
        MemoryTerritory territory,
        PalClientContext dbContext,
        ILogger<T> logger,
        List<PersistentLocation> locations)
    {
        lock (territory.LockObj)
        {
            logger.LogInformation("Saving {Count} new locations for territory {Territory}", locations.Count,
                territory.TerritoryType);

            Dictionary<PersistentLocation, ClientLocation> mapping =
                locations.ToDictionary(x => x, x => ToDatabaseLocation(x, territory.TerritoryType));
            dbContext.Locations.AddRange(mapping.Values);
            dbContext.SaveChanges();

            foreach ((PersistentLocation persistentLocation, ClientLocation clientLocation) in mapping)
            {
                persistentLocation.LocalId = clientLocation.LocalId;
            }
        }
    }

    private static ClientLocation ToDatabaseLocation(PersistentLocation location, ETerritoryType territoryType)
    {
        return new ClientLocation
        {
            TerritoryType = (ushort)territoryType,
            Type = ToDatabaseType(location.Type),
            X = location.Position.X,
            Y = location.Position.Y,
            Z = location.Position.Z,
            Seen = location.Seen,
            Source = location.Source,
            SinceVersion = typeof(Plugin).Assembly.GetName().Version!.ToString(2),
        };
    }

    private static ClientLocation.EType ToDatabaseType(MemoryLocation.EType type)
    {
        return type switch
        {
            MemoryLocation.EType.Trap => ClientLocation.EType.Trap,
            MemoryLocation.EType.Hoard => ClientLocation.EType.Hoard,
            _ => throw new ArgumentOutOfRangeException(nameof(type), type, null)
        };
    }
}