SPT-Server-Build/Libraries/Core/Helpers/PrestigeHelper.cs

using System.Text.Json;
using Core.Models.Eft.Common.Tables;
using Core.Models.Eft.Profile;
using Core.Models.Enums;
using Core.Models.Utils;
using Core.Services;
using Core.Utils;
using SptCommon.Annotations;
using SptCommon.Extensions;

namespace Core.Helpers
{
    [Injectable]
    public class PrestigeHelper
    {
        protected ISptLogger<PrestigeHelper> _logger;
        protected TimeUtil _timeUtil;
        protected DatabaseService _databaseService;
        protected MailSendService _mailSendService;
        protected ProfileHelper _profileHelper;
        protected RewardHelper _rewardHelper;

        public PrestigeHelper(
            ISptLogger<PrestigeHelper> logger,
            TimeUtil timeUtil,
            DatabaseService databaseService,
            MailSendService mailSendService,
            ProfileHelper profileHelper,
            RewardHelper rewardHelper)
        {
            _logger = logger;
            _timeUtil = timeUtil;
            _databaseService = databaseService;
            _mailSendService = mailSendService;
            _profileHelper = profileHelper;
            _rewardHelper = rewardHelper;
        }

        public void ProcessPendingPrestige(SptProfile oldProfile, SptProfile newProfile, PendingPrestige prestige)
        {
            var prePrestigePmc = oldProfile.CharacterData.PmcData;
            var sessionId = newProfile.ProfileInfo.ProfileId;

            // Skill copy

            if (prePrestigePmc.Skills.Common is not null)
            {
                var commonSKillsToCopy = prePrestigePmc.Skills.Common;
                foreach (var skillToCopy in commonSKillsToCopy) {
                    // Set progress 5% of what it was
                    skillToCopy.Progress = skillToCopy.Progress * 0.05;
                    var existingSkill = newProfile.CharacterData.PmcData.Skills.Common.FirstOrDefault((skill) => skill.Id == skillToCopy.Id);
                    if (existingSkill is not null)
                    {
                        existingSkill.Progress = skillToCopy.Progress;
                    }
                    else
                    {
                        newProfile.CharacterData.PmcData.Skills.Common.Add(skillToCopy);
                    }
                }

                var masteringSkillsToCopy = prePrestigePmc.Skills.Mastering;
                foreach (var skillToCopy in masteringSkillsToCopy) {
                    // Set progress 5% of what it was
                    skillToCopy.Progress = skillToCopy.Progress * 0.05;
                    var existingSkill = newProfile.CharacterData.PmcData.Skills.Mastering.FirstOrDefault((skill) => skill.Id == skillToCopy.Id);
                    if (existingSkill is not null)
                    {
                        existingSkill.Progress = skillToCopy.Progress;
                    }
                    else
                    {
                        newProfile.CharacterData.PmcData.Skills.Mastering.Add(skillToCopy);
                    }
                }
            }

            var indexOfPrestigeObtained = Math.Min(prestige.PrestigeLevel.Value - 1, 1); // Index starts at 0

            // Add "Prestigious" achievement
            if (!newProfile.CharacterData.PmcData.Achievements.ContainsKey("676091c0f457869a94017a23"))
            {
                _rewardHelper.AddAchievementToProfile(newProfile, "676091c0f457869a94017a23");
            }

            // Assumes Prestige data is in descending order
            var currentPrestigeData = _databaseService.GetTemplates().Prestige.Elements[indexOfPrestigeObtained];
            var prestigeRewards = _databaseService
                .GetTemplates()
                .Prestige.Elements.Slice(0, indexOfPrestigeObtained + 1)
                .SelectMany(prestige => prestige.Rewards);

            AddPrestigeRewardsToProfile(sessionId, newProfile, prestigeRewards);

            // Flag profile as having achieved this prestige level
            newProfile.CharacterData.PmcData.Prestige[currentPrestigeData.Id] = _timeUtil.GetTimeStamp();

            var itemsToTransfer = new List<Item>();

            // Copy transferred items
            foreach (var transferRequest in prestige.Items ?? []) {
                var item = prePrestigePmc.Inventory.Items.FirstOrDefault((item) => item.Id == transferRequest.Id);
                if (item is null)
                {
                    _logger.Error($"Unable to find item with id: {transferRequest.Id} in profile: {sessionId}, skipping");
                    continue;
                }

                itemsToTransfer.Add(item);
            }

            _mailSendService.SendSystemMessageToPlayer(sessionId, "", itemsToTransfer, 31536000);

            newProfile.CharacterData.PmcData.Info.PrestigeLevel = prestige.PrestigeLevel;
        }

        private void AddPrestigeRewardsToProfile(string sessionId, SptProfile newProfile, IEnumerable<Reward> rewards)
        {
            var itemsToSend = new List<Item>();

            foreach (var reward in rewards)
            {
                switch (reward.Type)
                {
                    case RewardType.CustomizationDirect:
                        {
                            _profileHelper.AddHideoutCustomisationUnlock(newProfile, reward, CustomisationSource.PRESTIGE);
                            break;
                        }
                    case RewardType.Skill:
                        if (Enum.TryParse(reward.Target, out SkillTypes result))
                        {
                            _profileHelper.AddSkillPointsToPlayer(
                                newProfile.CharacterData.PmcData,
                                result,
                                ((JsonElement) reward.Value).ToObject<double>()
                            );
                        }
                        else
                        {
                            _logger.Error($"Unable to parse reward Target to Enum: {reward.Target}");
                        }

                        break;
                    case RewardType.Item:
                        {
                            itemsToSend.AddRange(reward.Items);
                            break;
                        }
                    case RewardType.ExtraDailyQuest:
                        {
                            _logger.Info("additional quests will be added when generating repeatables");
                            break;
                        }
                    default:
                        _logger.Error($"Unhandled prestige reward type: {reward.Type}");
                        break;
                }
            }

            if (itemsToSend.Count > 0)
            {
                _mailSendService.SendSystemMessageToPlayer(sessionId, "", itemsToSend, 31536000);
            }
        }
    }
}