DanhengServer-OpenSource/GameServer/Game/GridFight/GridFightInstance.cs

using EggLink.DanhengServer.Data;
using EggLink.DanhengServer.GameServer.Game.Battle;
using EggLink.DanhengServer.GameServer.Game.GridFight.Component;
using EggLink.DanhengServer.GameServer.Game.GridFight.PendingAction;
using EggLink.DanhengServer.GameServer.Game.GridFight.Sync;
using EggLink.DanhengServer.GameServer.Game.Player;
using EggLink.DanhengServer.GameServer.Server.Packet.Send.GridFight;
using EggLink.DanhengServer.Proto;

namespace EggLink.DanhengServer.GameServer.Game.GridFight;

public class GridFightInstance(PlayerInstance player, uint season, uint divisionId, bool isOverLock, uint uniqueId)
{
    public uint Season { get; } = season;
    public uint DivisionId { get; } = divisionId;
    public bool IsOverLock { get; } = isOverLock;
    public uint UniqueId { get; } = uniqueId;
    public List<BaseGridFightComponent> Components { get; } = [];
    public PlayerInstance Player { get; } = player;

    public BattleInstance? StartBattle()
    {
        var battle = Player.BattleManager!.StartGridFightBattle(this);

        return battle;
    }

    public async ValueTask EndBattle(BattleInstance battle, PVEBattleResultCsReq req)
    {
        if (battle.BattleEndStatus == BattleEndStatus.BattleEndQuit) return;

        List<BaseGridFightSyncData> syncs = [];

        var basicComp = GetComponent<GridFightBasicComponent>();
        var levelComp = GetComponent<GridFightLevelComponent>();
        var prevData = basicComp.Data.Clone();

        var expNum = 5u;
        var baseCoin = levelComp.CurrentSection.Excel.BasicGoldRewardNum;
        var interestCoin = basicComp.Data.CurGold / 10;

        if (battle.BattleEndStatus == BattleEndStatus.BattleEndWin)
        {
            basicComp.Data.ComboNum++;
        }
        else
        {
            basicComp.Data.ComboNum = 0;

            // cost hp
            await basicComp.UpdateLineupHp(-5, false);
        }

        var comboCoin = basicComp.Data.ComboNum switch
        {
            >= 5 => 3u,
            2 or 3 or 4 => 2u,
            0 => 0u,
            _ => 1u
        };

        await basicComp.UpdateGoldNum((int)(baseCoin + interestCoin + comboCoin), false);
        await basicComp.AddLevelExp(expNum, false);

        List<GridFightRoleDamageSttInfo> sttList = [];
        foreach (var roleBattleStt in req.Stt.GridFightBattleStt.RoleBattleStt)
        {
            var res = await levelComp.AddRoleDamageStt(roleBattleStt.RoleBasicId, roleBattleStt.Damage, false);
            if (res.Item2 != null)
                sttList.Add(res.Item2);
        }

        var curData = basicComp.Data.Clone();
        await Player.SendPacket(new PacketGridFightEndBattleStageNotify(this, expNum, prevData, curData,
            sttList, battle.BattleEndStatus == BattleEndStatus.BattleEndWin, baseCoin, interestCoin,
            comboCoin));


        syncs.Add(new GridFightGoldSyncData(GridFightSrc.KGridFightSrcBattleEnd, basicComp.Data, 0, levelComp.CurrentSection.ChapterId, levelComp.CurrentSection.SectionId));
        syncs.Add(new GridFightPlayerLevelSyncData(GridFightSrc.KGridFightSrcBattleEnd, basicComp.Data));
        syncs.Add(new GridFightLineupHpSyncData(GridFightSrc.KGridFightSrcBattleEnd, basicComp.Data));
        syncs.Add(new GridFightComboNumSyncData(GridFightSrc.KGridFightSrcBattleEnd, basicComp.Data));
        syncs.Add(new GridFightRoleDamageSttSyncData(GridFightSrc.KGridFightSrcBattleEnd, levelComp));
        syncs.AddRange(await levelComp.EnterNextSection(false));

        await Player.SendPacket(new PacketGridFightSyncUpdateResultScNotify(syncs));
    }

    public void InitializeComponents()
    {
        Components.Add(new GridFightBasicComponent(this));
        Components.Add(new GridFightShopComponent(this));
        Components.Add(new GridFightLevelComponent(this));
        Components.Add(new GridFightRoleComponent(this));
        Components.Add(new GridFightAugmentComponent(this));
        Components.Add(new GridFightTraitComponent(this));

        _ = GetComponent<GridFightRoleComponent>().AddAvatar(1414, 3, false);
        _ = GetComponent<GridFightShopComponent>().RefreshShop(true, false);

        _ = CreatePendingAction<GridFightPortalBuffPendingAction>(sendPacket:false);
        _ = CreatePendingAction<GridFightElitePendingAction>(sendPacket: false);
        _ = CreatePendingAction<GridFightEnterNodePendingAction>(sendPacket: false);
    }

    public T GetComponent<T>() where T : BaseGridFightComponent
    {
        return (T)Components.First(c => c is T);
    }

    public uint GetDivisionDifficulty()
    {
        return GameData.GridFightDivisionStageData.TryGetValue(DivisionId, out var excel)
            ? excel.EnemyDifficultyLevel
            : 0;
    }

    public GridFightCurrentInfo ToProto()
    {
        return new GridFightCurrentInfo
        {
            DivisionId = DivisionId,
            Season = Season,
            IsOverlock = IsOverLock,
            UniqueId = UniqueId,
            PendingAction = GetCurAction().ToProto(),
            GridFightGameData = ToGameDataInfo(),
            RogueCurrentGameInfo = { ToGameInfos() }
        };
    }

    public List<GridFightGameInfo> ToGameInfos()
    {
        return (from c in Components select c.ToProto()).ToList();
    }

    public GridFightGameData ToGameDataInfo()
    {
        return new GridFightGameData
        {
            GameItemInfoList = { new GridFightGameItemInfo
            {
                UniqueId = 18,
                JCDHFKOCDOL = new()
            } }
        };
    }

    #region Pending Action

    public SortedDictionary<uint, BaseGridFightPendingAction> PendingActions { get; set; } = new();
    private uint _curQueuePos = 1;

    public BaseGridFightPendingAction GetCurAction()
    {
        if (PendingActions.Count > 0)
        {
            return PendingActions.First().Value;
        }

        return new GridFightEmptyPendingAction(this);
    }

    public uint AddPendingAction(BaseGridFightPendingAction action)
    {
        var pos = _curQueuePos++;

        action.QueuePosition = pos;
        PendingActions[pos] = action;

        return pos;
    }

    public async ValueTask<List<BaseGridFightSyncData>> CreatePendingAction<T>(GridFightSrc src = GridFightSrc.KGridFightSrcEnterNode, bool sendPacket = true) where T: BaseGridFightPendingAction
    {
        var action = (T)Activator.CreateInstance(typeof(T), this)!;
        var basicComp = GetComponent<GridFightBasicComponent>();

        AddPendingAction(action);

        basicComp.Data.LockReason = (uint)GridFightLockReason.KGridFightLockReasonPendingAction;
        basicComp.Data.LockType = (uint)GridFightLockType.KGridFightLockTypeAll;

        var res = new GridFightPendingActionSyncData(src, action);
        if (sendPacket)
           await Player.SendPacket(new PacketGridFightSyncUpdateResultScNotify(res));

        return [res, new GridFightLockInfoSyncData(src, basicComp.Data.Clone())];
    }

    public async ValueTask HandleResultRequest(GridFightHandlePendingActionCsReq req)
    {
        var basicComp = GetComponent<GridFightBasicComponent>();
        var levelComp = GetComponent<GridFightLevelComponent>();

        var curAction = GetCurAction();

        // end
        var isFinish = true;
        GridFightSrc src;

        List<BaseGridFightSyncData> syncs = [];

        switch (req.GridFightActionTypeCase)
        {
            case GridFightHandlePendingActionCsReq.GridFightActionTypeOneofCase.PortalBuffAction:
                src = GridFightSrc.KGridFightSrcSelectPortalBuff;

                syncs.AddRange(await GetComponent<GridFightLevelComponent>().AddPortalBuff(req.PortalBuffAction.SelectPortalBuffId, false, src));

                // initial supply
                await basicComp.UpdateGoldNum(5, false, GridFightSrc.KGridFightSrcInitialSupplySelect);
                syncs.Add(new GridFightGoldSyncData(GridFightSrc.KGridFightSrcInitialSupplySelect, basicComp.Data));
                break;
            case GridFightHandlePendingActionCsReq.GridFightActionTypeOneofCase.PortalBuffRerollAction:
                if (curAction is GridFightPortalBuffPendingAction portalBuffAction)
                {
                    isFinish = false;
                    await portalBuffAction.RerollBuff();
                }

                break;
            case GridFightHandlePendingActionCsReq.GridFightActionTypeOneofCase.AugmentAction:
                src = GridFightSrc.KGridFightSrcSelectAugment;

                syncs.AddRange(await GetComponent<GridFightAugmentComponent>().AddAugment(req.AugmentAction.AugmentId, false, src));
                break;

            case GridFightHandlePendingActionCsReq.GridFightActionTypeOneofCase.RerollAugmentAction:
                if (curAction is GridFightAugmentPendingAction augmentAction)
                {
                    isFinish = false;
                    await augmentAction.RerollAugment(req.RerollAugmentAction.AugmentId);
                }

                break;
            case GridFightHandlePendingActionCsReq.GridFightActionTypeOneofCase.EliteAction:
                break;
            case GridFightHandlePendingActionCsReq.GridFightActionTypeOneofCase.EliteBranchAction:
                var target = req.EliteBranchAction.EliteBranchId;
                levelComp.CurrentSection.BranchId = target;
                // sync
                syncs.Add(new GridFightLevelSyncData(GridFightSrc.KGridFightSrcNone, levelComp));

                break;
            case GridFightHandlePendingActionCsReq.GridFightActionTypeOneofCase.SupplyAction:
                src = GridFightSrc.KGridFightSrcSelectSupply;

                PendingActions.Remove(curAction.QueuePosition);

                if (curAction is GridFightSupplyPendingAction supplyAction)
                {
                    foreach (var supply in req.SupplyAction.SelectSupplyIndexes)
                    {
                        var role = supplyAction.RoleList[(int)supply];

                        syncs.AddRange(await GetComponent<GridFightRoleComponent>().AddAvatar(role.RoleId, 1, false, true,
                            GridFightSrc.KGridFightSrcSelectSupply, 0, 0, req.SupplyAction.SelectSupplyIndexes.ToArray()));
                    }
                }

                syncs.AddRange(await CheckCurNodeFinish(src));

                break;
        }

        if (isFinish)
        {
            PendingActions.Remove(curAction.QueuePosition);
            syncs.Add(new GridFightFinishPendingActionSyncData(GridFightSrc.KGridFightSrcNone, curAction.QueuePosition));

            // unlock
            basicComp.Data.LockReason = (uint)GridFightLockReason.KGridFightLockReasonUnknown;
            basicComp.Data.LockType = (uint)GridFightLockType.KGridFightLockTypeNone;

            syncs.Add(new GridFightLockInfoSyncData(GridFightSrc.KGridFightSrcNone, basicComp.Data.Clone()));
        }

        if (PendingActions.Count > 0)
        {
            basicComp.Data.LockReason = (uint)GridFightLockReason.KGridFightLockReasonPendingAction;
            basicComp.Data.LockType = (uint)GridFightLockType.KGridFightLockTypeAll;

            syncs.Add(new GridFightPendingActionSyncData(GridFightSrc.KGridFightSrcNone, GetCurAction(), 1));
            syncs.Add(new GridFightLockInfoSyncData(GridFightSrc.KGridFightSrcNone, basicComp.Data.Clone(), 1));
        }

        await Player.SendPacket(new PacketGridFightSyncUpdateResultScNotify(syncs));
    }

    public async ValueTask<List<BaseGridFightSyncData>> CheckCurNodeFinish(GridFightSrc src)
    {
        var levelComp = GetComponent<GridFightLevelComponent>();
        var curSection = levelComp.CurrentSection;

        if (curSection.Encounters.Count != 0) return [];

        if (PendingActions.Count != 0) return [];

        // next
        return await levelComp.EnterNextSection(src:GridFightSrc.KGridFightSrcNone);
    }

    #endregion
}