SPT-Server-Build/Libraries/SPTarkov.Server.Core/Migration/Migrations/ThreeTenToThreeEleven.cs

using System.Text.Json.Nodes;
using SPTarkov.DI.Annotations;
using SPTarkov.Server.Core.Extensions;
using SPTarkov.Server.Core.Helpers;
using SPTarkov.Server.Core.Models.Eft.Common.Tables;
using SPTarkov.Server.Core.Models.Eft.Profile;
using SPTarkov.Server.Core.Models.Enums;
using SPTarkov.Server.Core.Services;
using Range = SemanticVersioning.Range;

namespace SPTarkov.Server.Core.Migration.Migrations
{
    [Injectable]
    public class ThreeTenToThreeEleven(
        DatabaseService databaseService,
        // Yes, referencing the helpers directly causes a circular dependency. Too bad!
        IServiceProvider serviceProvider
    ) : AbstractProfileMigration
    {
        private List<string> _oldSuiteData = [];

        public override string FromVersion
        {
            get { return "~3.10"; }
        }

        public override string ToVersion
        {
            get { return "3.11"; }
        }

        public override string MigrationName
        {
            get { return "310x-SPTSharp"; }
        }

        public override IEnumerable<Type> PrerequisiteMigrations
        {
            get { return [typeof(HideoutSeed)]; }
        }

        public override bool CanMigrate(
            JsonObject profile,
            IEnumerable<IProfileMigration> previouslyRanMigrations
        )
        {
            var profileVersion = GetProfileVersion(profile);

            var fromRange = Range.Parse(FromVersion);

            var versionMatches = fromRange.IsSatisfied(profileVersion);

            return versionMatches;
        }

        public override JsonObject? Migrate(JsonObject profile)
        {
            if (profile["suits"] is JsonArray suitsArray)
            {
                _oldSuiteData = suitsArray
                    .Select(node => node?.GetValue<string>())
                    .Where(suit => suit != null)
                    .ToList()!;
            }

            profile.Remove("suits");

            return profile;
        }

        public override bool PostMigrate(SptProfile profile)
        {
            if (profile.CustomisationUnlocks is null)
            {
                profile.CustomisationUnlocks = [];
                profile.AddCustomisationUnlocksToProfile();
            }

            profile.CharacterData.PmcData.Prestige ??= [];

            if (profile.CharacterData.PmcData.Inventory.HideoutCustomizationStashId is null)
            {
                profile.CharacterData.PmcData.Inventory.HideoutCustomizationStashId = new(
                    "676db384777490e23c45b657"
                );

                //Directly injecting CreateProfileService causes a circular dependency which I can't be bothered to fix just for this
                serviceProvider
                    .GetService<CreateProfileService>()!
                    .AddMissingInternalContainersToProfile(profile.CharacterData.PmcData);
            }

            if (profile.CharacterData.PmcData.Hideout.Customization is null)
            {
                profile.CharacterData.PmcData.Hideout.Customization = new Dictionary<
                    string,
                    Models.Common.MongoId
                >
                {
                    { "Wall", new("675844bdf94a97cbbe096f1a") },
                    { "Floor", new("6758443ff94a97cbbe096f18") },
                    { "Light", new("675fe8abbc3deae49a0b947f") },
                    { "Ceiling", new("673b3f977038192ee006aa09") },
                    { "ShootingRangeMark", new("67585d416c72998cf60ed85a") },
                };
            }

            if (profile.CharacterData.PmcData.Info.Side == "Bear")
            {
                ProcessBearProfile(profile);
            }

            if (profile.CharacterData.PmcData.Info.Side == "Usec")
            {
                ProcessUsecProfile(profile);
            }

            if (profile.CharacterData.PmcData.Achievements.Count > 0)
            {
                var achievementsDb = databaseService.GetTemplates().Achievements;

                foreach (var achievementId in profile.CharacterData.PmcData.Achievements.Keys)
                {
                    var achievementDb = achievementsDb.FirstOrDefault(a => a.Id == achievementId);
                    var rewards = achievementDb?.Rewards;

                    if (rewards == null)
                    {
                        continue;
                    }

                    // Only hand out the new hideout customization rewards.
                    var filteredRewards = rewards
                        .Where(r => r.Type == RewardType.CustomizationDirect)
                        .ToList();

                    //Directly injecting RewardHelper causes a circular dependency which I can't be bothered to fix just for this
                    serviceProvider
                        .GetService<RewardHelper>()!
                        .ApplyRewards(
                            filteredRewards,
                            CustomisationSource.ACHIEVEMENT,
                            profile,
                            profile.CharacterData.PmcData,
                            achievementId
                        );
                }
            }

            return true;
        }

        private void ProcessBearProfile(SptProfile profile)
        {
            // Reset clothing customization back to default as customization changed in 3.11
            profile.CharacterData.PmcData.Customization.Body = new("5cc0858d14c02e000c6bea66");
            profile.CharacterData.PmcData.Customization.Feet = new("5cc085bb14c02e000e67a5c5");
            profile.CharacterData.PmcData.Customization.Hands = new("5cc0876314c02e000c6bea6b");
            profile.CharacterData.PmcData.Customization.DogTag = new("674731c8bafff850080488bb");

            if (profile.CharacterData.PmcData.Info.GameVersion == "edge_of_darkness")
            {
                profile.CharacterData.PmcData.Customization.DogTag = new(
                    "6746fd09bafff85008048838"
                );
            }

            if (profile.CharacterData.PmcData.Info.GameVersion == "unheard_edition")
            {
                profile.CharacterData.PmcData.Customization.DogTag = new(
                    "67471928d17d6431550563b5"
                );
            }

            foreach (var oldSuite in _oldSuiteData)
            {
                // Default Bear clothing, dont need to add this
                if (
                    oldSuite == "5cd946231388ce000d572fe3"
                    || oldSuite == "5cd945d71388ce000a659dfb"
                    || oldSuite == "666841a02537107dc508b704"
                )
                {
                    continue;
                }

                var trader = databaseService.GetTrader("5ac3b934156ae10c4430e83c");
                var traderClothing = trader?.Suits?.FirstOrDefault(s => s.SuiteId == oldSuite);

                if (traderClothing != null)
                {
                    var clothingToAdd = new CustomisationStorage
                    {
                        Id = traderClothing.SuiteId,
                        Source = CustomisationSource.UNLOCKED_IN_GAME,
                        Type = CustomisationType.SUITE,
                    };

                    profile.CustomisationUnlocks.Add(clothingToAdd);
                }
            }
        }

        private void ProcessUsecProfile(SptProfile profile)
        {
            // Reset clothing customization back to default as customization changed in 3.11
            profile.CharacterData.PmcData.Customization.Body = new("5cde95d97d6c8b647a3769b0"); //Usec default clothing
            profile.CharacterData.PmcData.Customization.Feet = new("5cde95ef7d6c8b04713c4f2d");
            profile.CharacterData.PmcData.Customization.Hands = new("5cde95fa7d6c8b04737c2d13");
            profile.CharacterData.PmcData.Customization.DogTag = new("674731d1170146228c0d222a"); //Usec base dogtag

            if (profile.CharacterData.PmcData.Info.GameVersion == "edge_of_darkness")
            {
                profile.CharacterData.PmcData.Customization.DogTag = new(
                    "67471938bafff850080488b7"
                );
            }

            if (profile.CharacterData.PmcData.Info.GameVersion == "unheard_edition")
            {
                profile.CharacterData.PmcData.Customization.DogTag = new(
                    "6747193f170146228c0d2226"
                );
            }

            foreach (var oldSuite in _oldSuiteData)
            {
                // Default Usec clothing, dont need to add this
                if (
                    oldSuite == "5cde9ec17d6c8b04723cf479"
                    || oldSuite == "5cde9e957d6c8b0474535da7"
                    || oldSuite == "666841a02537107dc508b704"
                )
                {
                    continue;
                }

                var trader = databaseService.GetTrader("5ac3b934156ae10c4430e83c");
                var traderClothing = trader?.Suits?.FirstOrDefault(s => s.SuiteId == oldSuite);

                if (traderClothing != null)
                {
                    var clothingToAdd = new CustomisationStorage
                    {
                        Id = traderClothing.SuiteId,
                        Source = CustomisationSource.UNLOCKED_IN_GAME,
                        Type = CustomisationType.SUITE,
                    };

                    profile.CustomisationUnlocks.Add(clothingToAdd);
                }
            }
        }
    }
}