MikuSB/GameServer/Server/CallGS/Handlers/Rogue3D/Rogue3DStateHelper.cs

using MikuSB.Data;
using MikuSB.Database.Player;
using MikuSB.GameServer.Game.Player;
using MikuSB.Proto;

namespace MikuSB.GameServer.Server.CallGS.Handlers.Rogue3D;

internal static class Rogue3DStateHelper
{
    private const uint GroupId = 124;
    private const uint LevelPassStart = 20;
    private const uint DailyBuffStart = 51;
    private const uint DailyBuffEnd = 65;
    private const int DailyBuffBitCount = 10;
    private const int DailyBuffBitsPerValue = DailyBuffBitCount + 1;
    private const uint DailyBuffMask = (1u << DailyBuffBitCount) - 1;
    private const uint UnlockDiff1Sid = LevelPassStart + 1;
    private const uint UnlockDiff2Sid = LevelPassStart + 2;
    private const uint UnlockDiff3Sid = LevelPassStart + 3;
    private const uint UnlockDiff4Sid = LevelPassStart + 4;
    private static uint[]? ShuffledDailyBuffIds;

    public static NtfSyncPlayer EnsureUnlockState(PlayerInstance player)
    {
        var sync = new NtfSyncPlayer();

        EnsureMinAttr(player, UnlockDiff1Sid, 1, sync);
        EnsureMinAttr(player, UnlockDiff2Sid, 1, sync);
        EnsureMinAttr(player, UnlockDiff3Sid, 1, sync);
        EnsureMinAttr(player, UnlockDiff4Sid, 1, sync);

        foreach (var scienceSid in GetUnlockTalentScienceSids())
        {
            EnsureMinAttr(player, scienceSid, 1, sync);
        }

        EnsureDailyBuffAttrs(player, sync);

        return sync;
    }

    private static IEnumerable<uint> GetUnlockTalentScienceSids()
    {
        return GameData.Rogue3DTalentData.Values
            .Select(x => x.UnlockCondition)
            .Where(x => x > 0)
            .Distinct()
            .OrderBy(x => x);
    }

    private static void EnsureDailyBuffAttrs(PlayerInstance player, NtfSyncPlayer sync)
    {
        var buffIds = GetOrCreateDailyBuffIds()
            .Take((int)(DailyBuffEnd - DailyBuffStart + 1) * 3)
            .ToArray();

        var index = 0;
        for (var sid = DailyBuffStart; sid <= DailyBuffEnd; sid++)
        {
            uint packed = 0;
            for (var slot = 0; slot < 3 && index < buffIds.Length; slot++, index++)
            {
                packed |= (buffIds[index] & DailyBuffMask) << (slot * DailyBuffBitsPerValue);
            }

            SetAttr(player, sid, packed, sync);
        }
    }

    private static uint[] GetOrCreateDailyBuffIds()
    {
        if (ShuffledDailyBuffIds != null)
        {
            return ShuffledDailyBuffIds;
        }

        var groupedBuffIds = GameData.Rogue3DDailyBuffData.Values
            .Where(x => x.ScoreBuffId > 0 && x.ScoreBuffId <= DailyBuffMask)
            .GroupBy(x => x.GroupId)
            .OrderBy(x => x.Key)
            .Select(x => x
                .OrderBy(y => y.Id)
                .Select(y => y.ScoreBuffId)
                .Distinct()
                .ToList())
            .ToList();

        var random = new Random();
        foreach (var group in groupedBuffIds)
        {
            Shuffle(group, random);
        }
        Shuffle(groupedBuffIds, random);

        var buffIds = new List<uint>();
        var indexByGroup = new int[groupedBuffIds.Count];
        var hasRemaining = true;

        while (hasRemaining)
        {
            hasRemaining = false;
            for (var i = 0; i < groupedBuffIds.Count; i++)
            {
                var group = groupedBuffIds[i];
                var index = indexByGroup[i];
                if (index >= group.Count)
                {
                    continue;
                }

                buffIds.Add(group[index]);
                indexByGroup[i] = index + 1;
                hasRemaining = true;
            }
        }

        ShuffledDailyBuffIds = buffIds.ToArray();
        return ShuffledDailyBuffIds;
    }

    private static IEnumerable<uint> GetDailyBuffIds()
    {
        return GetOrCreateDailyBuffIds();
    }

    private static void Shuffle<T>(IList<T> list, Random random)
    {
        for (var i = list.Count - 1; i > 0; i--)
        {
            var swapIndex = random.Next(i + 1);
            (list[i], list[swapIndex]) = (list[swapIndex], list[i]);
        }
    }

    private static void EnsureMinAttr(PlayerInstance player, uint sid, uint value, NtfSyncPlayer sync, bool overwrite = false)
    {
        var attr = player.Data.Attrs.FirstOrDefault(x => x.Gid == GroupId && x.Sid == sid);
        if (attr == null)
        {
            attr = new PlayerAttr { Gid = GroupId, Sid = sid, Val = value };
            player.Data.Attrs.Add(attr);
            AddSync(player, sync, sid, value);
            return;
        }

        if ((!overwrite && attr.Val >= value) || (overwrite && attr.Val == value))
        {
            return;
        }

        attr.Val = value;
        AddSync(player, sync, sid, value);
    }

    private static void SetAttr(PlayerInstance player, uint sid, uint value, NtfSyncPlayer sync)
    {
        var attr = player.Data.Attrs.FirstOrDefault(x => x.Gid == GroupId && x.Sid == sid);
        if (attr == null)
        {
            attr = new PlayerAttr { Gid = GroupId, Sid = sid };
            player.Data.Attrs.Add(attr);
        }

        if (attr.Val == value)
        {
            return;
        }

        attr.Val = value;
        AddSync(player, sync, sid, value);
    }

    private static void AddSync(PlayerInstance player, NtfSyncPlayer sync, uint sid, uint value)
    {
        sync.Custom[player.ToPackedAttrKey(GroupId, sid)] = value;
        sync.Custom[player.ToShiftedAttrKey(GroupId, sid)] = value;
    }
}