SPT-Server-Build/Libraries/Core/Controllers/PrestigeController.cs

using System.Text.Json;
using SptCommon.Annotations;
using Core.Helpers;
using Core.Models.Eft.Common;
using Core.Models.Eft.Common.Tables;
using Core.Models.Eft.Inventory;
using Core.Models.Eft.Prestige;
using Core.Models.Eft.Profile;
using Core.Models.Enums;
using Core.Models.Utils;
using Core.Routers;
using Core.Servers;
using Core.Services;
using Core.Utils;
using Core.Utils.Cloners;
using SptCommon.Extensions;


namespace Core.Controllers;

[Injectable]
public class PrestigeController(
    ISptLogger<PrestigeController> _logger,
    TimeUtil _timeUtil,
    InventoryHelper _inventoryHelper,
    ProfileHelper _profileHelper,
    EventOutputHolder _eventOutputHolder,
    CreateProfileService _createProfileService,
    DatabaseService _databaseService,
    SaveServer _saveServer,
    ICloner _cloner
)
{
    /// <summary>
    /// Handle /client/prestige/list
    /// </summary>
    /// <param name="sessionId"></param>
    /// <param name="info"></param>
    /// <returns></returns>
    public Prestige GetPrestige(
        string sessionId,
        EmptyRequestData info)
    {
        return _databaseService.GetTemplates().Prestige;
    }

    /// <summary>
    /// Handle /client/prestige/obtain
    /// </summary>
    /// <param name="sessionId"></param>
    /// <param name="request"></param>
    /// <returns></returns>
    public void ObtainPrestige(
        string sessionId,
        ObtainPrestigeRequestList request)
    {
        // Going to prestige 1

        // transfer
        // 5% of skills should be transfered over
        // 5% of mastering should be transfered over
        // earned achievements should be transfered over
        // profile stats should be transfered over
        // prestige progress should be transfered over

        // reset
        // trader standing
        // task progress
        // character level
        // stash
        // hideout progress

        // going to prestige 2
        // most likely the same, but wait for dump of new beginnings quest


        // Clone existing profile, create a new one
        var prePrestigeProfileClone = _cloner.Clone(_profileHelper.GetFullProfile(sessionId));
        var prePrestigePmc = prePrestigeProfileClone.CharacterData.PmcData;
        var createRequest = new ProfileCreateRequestData
        {
            Side = prePrestigePmc.Info.Side,
            Nickname = prePrestigePmc.Info.Nickname,
            HeadId = prePrestigePmc.Customization.Head,
            VoiceId = _databaseService.GetTemplates()
                .Customization.FirstOrDefault(
                    (customisation) => customisation.Value.Name == prePrestigePmc.Info.Voice
                )
                .Value.Id,
            SptForcePrestigeLevel = prePrestigeProfileClone.CharacterData.PmcData.Info.PrestigeLevel.GetValueOrDefault(0) + 1 // Current + 1
        };

        // Reset profile
        _createProfileService.CreateProfile(sessionId, createRequest);

        // Get freshly reset profile ready for editing
        var newProfile = _profileHelper.GetFullProfile(sessionId);

        // Copy skills to new profile
        var commonSkillsToCopy = prePrestigePmc.Skills.Common;
        foreach (var skillToCopy in commonSkillsToCopy)
        {
            // Set progress 5% of what it was
            skillToCopy.Progress = skillToCopy.Progress.Value * 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);
        }

        // Copy mastering to new profile
        var masteringSkillsToCopy = prePrestigePmc.Skills.Mastering;
        foreach (var skillToCopy in masteringSkillsToCopy)
        {
            // Set progress 5% of what it was
            skillToCopy.Progress = skillToCopy.Progress.Value * 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);
        }

        // Add existing completed achievements and new one for prestige
        newProfile.CharacterData.PmcData.Achievements = prePrestigeProfileClone.CharacterData.PmcData.Achievements; // this *should* only contain completed ones

        // Add "Prestigious" achievement
        if (!newProfile.CharacterData.PmcData.Achievements.ContainsKey("676091c0f457869a94017a23"))
            newProfile.CharacterData.PmcData.Achievements.Add("676091c0f457869a94017a23", _timeUtil.GetTimeStamp());
        // TODO: is there one for second prestige

        // Add existing Stats to profile
        newProfile.CharacterData.PmcData.Stats = prePrestigePmc.Stats;

        // Assumes Prestige data is in descending order
        var indexOfPrestigeObtained = (int)Math.Min(createRequest.SptForcePrestigeLevel.Value - 1, 1); // Index starts at 0
        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();
        newProfile.CharacterData.PmcData.Info.PrestigeLevel++;

        if (request is not null)
            // Copy transferred items
            foreach (var transferRequest in request)
            {
                var item = prePrestigePmc.Inventory.Items.FirstOrDefault((item) => item.Id == transferRequest.Id);
                var addItemRequest = new AddItemDirectRequest
                {
                    ItemWithModsToAdd = [item],
                    FoundInRaid = item.Upd?.SpawnedInSession,
                    UseSortingTable = false,
                    Callback = null
                };
                _inventoryHelper.AddItemToStash(
                    sessionId,
                    addItemRequest,
                    newProfile.CharacterData.PmcData,
                    _eventOutputHolder.GetOutput(sessionId)
                );
            }

        // Force save of above changes to disk
        _saveServer.SaveProfile(sessionId);
    }

    private void AddPrestigeRewardsToProfile(string sessionId, SptProfile newProfile, IEnumerable<Reward> rewards)
    {
        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:
                {
                    var addItemRequest = new AddItemDirectRequest
                    {
                        ItemWithModsToAdd = reward.Items,
                        FoundInRaid = reward.Items.FirstOrDefault()?.Upd?.SpawnedInSession,
                        UseSortingTable = false,
                        Callback = null
                    };
                    _inventoryHelper.AddItemToStash(
                        sessionId,
                        addItemRequest,
                        newProfile.CharacterData.PmcData,
                        _eventOutputHolder.GetOutput(sessionId)
                    );
                    break;
                }
                // case "ExtraDailyQuest": {
                //     // todo
                //     break;
                // }
                default:
                    _logger.Error($"Unhandled prestige reward type: {reward.Type}");
                    break;
            }
    }
}