DanhengServer-OpenSource/GameServer/Game/Challenge/ChallengeInstance.cs

using System.Text.Json.Serialization;
using EggLink.DanhengServer.Data;
using EggLink.DanhengServer.Data.Excel;
using EggLink.DanhengServer.Database.Challenge;
using EggLink.DanhengServer.Enums.Item;
using EggLink.DanhengServer.Enums.Mission;
using EggLink.DanhengServer.GameServer.Game.Battle;
using EggLink.DanhengServer.GameServer.Game.Player;
using EggLink.DanhengServer.GameServer.Game.Scene;
using EggLink.DanhengServer.GameServer.Game.Scene.Entity;
using EggLink.DanhengServer.GameServer.Server.Packet.Send.Challenge;
using EggLink.DanhengServer.GameServer.Server.Packet.Send.Lineup;
using EggLink.DanhengServer.Proto;
using EggLink.DanhengServer.Util;

namespace EggLink.DanhengServer.GameServer.Game.Challenge;

public class ChallengeInstance
{
    public ChallengeInstance(PlayerInstance player, ChallengeConfigExcel excel)
    {
        Player = player;
        Excel = excel;
        ChallengeId = excel.GetId();
        StartPos = new Position();
        StartRot = new Position();
        CurrentStage = 1;
        RoundsLeft = Excel.IsStory() ? 5 : Excel.ChallengeCountDown;
        SetStatus(ChallengeStatus.ChallengeDoing);
        SetCurrentExtraLineup(ExtraLineupType.LineupChallenge);
    }

    public ChallengeInstance(PlayerInstance player, ChallengeConfigExcel excel, ChallengeInstanceData data)
    {
        Player = player;
        Excel = excel;

        StartPos = data.StartPos;
        StartRot = data.StartRot;
        ChallengeId = data.ChallengeId;
        CurrentStage = data.CurrentStage;
        CurrentExtraLineup = data.CurrentExtraLineup;
        Status = data.Status;
        HasAvatarDied = data.HasAvatarDied;
        SavedMp = data.SavedMp;
        RoundsLeft = data.RoundsLeft;
        Stars = data.Stars;
        ScoreStage1 = data.ScoreStage1;
        ScoreStage2 = data.ScoreStage2;
        StoryBuffs = data.StoryBuffs;
        BossBuffs = data.BossBuffs;
    }

    public Position StartPos { get; set; }
    public Position StartRot { get; set; }
    public int ChallengeId { get; set; }
    public int CurrentStage { get; set; }
    public int CurrentExtraLineup { get; set; }
    public int Status { get; set; }
    public bool HasAvatarDied { get; set; }

    public int SavedMp { get; set; }
    public int RoundsLeft { get; set; }
    public int Stars { get; set; }
    public int ScoreStage1 { get; set; }
    public int ScoreStage2 { get; set; }

    [JsonIgnore] public List<BattleTarget>? BossTarget1 { get; set; }

    [JsonIgnore] public List<BattleTarget>? BossTarget2 { get; set; }

    [JsonIgnore] private PlayerInstance Player { get; }

    [JsonIgnore] public ChallengeConfigExcel Excel { get; set; }

    public List<int> StoryBuffs { get; set; } = [];
    public List<int> BossBuffs { get; set; } = [];

    public SceneInstance GetScene()
    {
        return Player.SceneInstance!;
    }

    public int GetChallengeId()
    {
        return Excel.GetId();
    }

    public bool IsStory()
    {
        return Excel.IsStory();
    }

    public bool IsBoss()
    {
        return Excel.IsBoss();
    }

    public void SetStatus(ChallengeStatus status)
    {
        Status = (int)status;
    }

    public void SetCurrentExtraLineup(ExtraLineupType type)
    {
        CurrentExtraLineup = (int)type;
    }

    public int GetRoundsElapsed()
    {
        return Excel.ChallengeCountDown - RoundsLeft;
    }

    public int GetTotalScore()
    {
        return ScoreStage1 + ScoreStage2;
    }

    public bool IsWin()
    {
        return Status == (int)ChallengeStatus.ChallengeFinish;
    }

    #region Serialization

    public CurChallenge ToProto()
    {
        var proto = new CurChallenge
        {
            ChallengeId = (uint)Excel.GetId(),
            Status = (ChallengeStatus)Status,
            ScoreId = (uint)ScoreStage1,
            ScoreTwo = (uint)ScoreStage2,
            RoundCount = (uint)GetRoundsElapsed(),
            ExtraLineupType = (ExtraLineupType)CurrentExtraLineup,
            StageInfo = new ChallengeStoryInfo()
        };

        if (Excel.IsBoss())
            proto.StageInfo.CurBossBuffs = new ChallengeBossBuffList
            {
                ChallengeBossConst = 1
            };

        if (Excel.IsStory()) proto.StageInfo.CurStoryBuffs = new ChallengeStoryBuffList();

        if (StoryBuffs.Count >= CurrentStage)
            proto.StageInfo.CurStoryBuffs.BuffList.Add(StoryBuffs.Select(x => (uint)x));

        if (BossBuffs.Count >= CurrentStage)
            proto.StageInfo.CurBossBuffs.BuffList.Add(BossBuffs.Select(x => (uint)x));

        return proto;
    }

    public ChallengeStageInfo ToStageInfo()
    {
        var proto = new ChallengeStageInfo();

        if (Excel.IsBoss())
        {
            proto.BossInfo = new ChallengeBossInfo
            {
                FirstNode = new ChallengeBossSingleNodeInfo
                {
                    BuffId = (uint)BossBuffs[0]
                },
                SecondNode = new ChallengeBossSingleNodeInfo
                {
                    BuffId = (uint)BossBuffs[1]
                },
                BCMDAAKCNBA = true
            };

            foreach (var lineupAvatar in Player.LineupManager?.GetExtraLineup(ExtraLineupType.LineupChallenge)
                         ?.BaseAvatars ?? [])
            {
                var avatar = Player.AvatarManager?.GetAvatar(lineupAvatar.BaseAvatarId);
                if (avatar == null) continue;
                proto.BossInfo.FirstLineup.Add((uint)avatar.GetAvatarId());
                var equip = Player.InventoryManager?.GetItem(0, avatar.GetCurPathInfo().EquipId,
                    ItemMainTypeEnum.Equipment);
                if (equip != null)
                    proto.BossInfo.ChallengeAvatarEquipmentMap.Add((uint)avatar.GetAvatarId(),
                        equip.ToChallengeEquipmentProto());

                var relicProto = new ChallengeBossAvatarRelicInfo();

                foreach (var relicUniqueId in avatar.GetCurPathInfo().Relic)
                {
                    var relic = Player.InventoryManager?.GetItem(0, relicUniqueId.Value, ItemMainTypeEnum.Relic);
                    if (relic == null) continue;
                    relicProto.AvatarRelicSlotMap.Add((uint)relicUniqueId.Key, relic.ToChallengeRelicProto());
                }

                proto.BossInfo.ChallengeAvatarRelicMap.Add((uint)avatar.GetAvatarId(), relicProto);
            }

            foreach (var lineupAvatar in Player.LineupManager?.GetExtraLineup(ExtraLineupType.LineupChallenge2)
                         ?.BaseAvatars ?? [])
            {
                var avatar = Player.AvatarManager?.GetAvatar(lineupAvatar.BaseAvatarId);
                if (avatar == null) continue;
                proto.BossInfo.SecondLineup.Add((uint)avatar.GetAvatarId());
                var equip = Player.InventoryManager?.GetItem(0, avatar.GetCurPathInfo().EquipId,
                    ItemMainTypeEnum.Equipment);
                if (equip != null)
                    proto.BossInfo.ChallengeAvatarEquipmentMap.Add((uint)avatar.GetAvatarId(),
                        equip.ToChallengeEquipmentProto());

                var relicProto = new ChallengeBossAvatarRelicInfo();

                foreach (var relicUniqueId in avatar.GetCurPathInfo().Relic)
                {
                    var relic = Player.InventoryManager?.GetItem(0, relicUniqueId.Value, ItemMainTypeEnum.Relic);
                    if (relic == null) continue;
                    relicProto.AvatarRelicSlotMap.Add((uint)relicUniqueId.Key, relic.ToChallengeRelicProto());
                }

                proto.BossInfo.ChallengeAvatarRelicMap.Add((uint)avatar.GetAvatarId(), relicProto);
            }
        }

        return proto;
    }

    #endregion

    #region Management

    public void OnBattleStart(BattleInstance battle)
    {
        battle.RoundLimit = RoundsLeft;

        battle.Buffs.Add(new MazeBuff(Excel.MazeBuffID, 1, -1));

        if (StoryBuffs.Count >= CurrentStage)
        {
            var buffId = StoryBuffs[CurrentStage - 1];
            battle.Buffs.Add(new MazeBuff(buffId, 1, -1));
        }

        if (Excel.StoryExcel != null)
        {
            battle.AddBattleTarget(1, 10002, GetTotalScore());

            foreach (var id in Excel.StoryExcel.BattleTargetID!) battle.AddBattleTarget(5, id, GetTotalScore());
        }

        if (Excel.BossExcel != null)
        {
            battle.AddBattleTarget(1, 90004, 0);
            battle.AddBattleTarget(1, 90005, 0);
        }
    }

    public virtual async ValueTask OnBattleEnd(BattleInstance battle, PVEBattleResultCsReq req)
    {
        if (IsStory())
        {
            // Calculate score for current stage
            var stageScore = (int)req.Stt.ChallengeScore - GetTotalScore();

            // Set score
            if (CurrentStage == 1)
                ScoreStage1 = stageScore;
            else
                ScoreStage2 = stageScore;
        }

        if (IsBoss())
        {
            // Calculate score for current stage
            var stageScore = 0;
            foreach (var battleTarget in req.Stt.BattleTargetInfo[1].BattleTargetList_)
                stageScore += (int)battleTarget.Progress;

            // Set score
            if (CurrentStage == 1)
                ScoreStage1 = stageScore;
            else
                ScoreStage2 = stageScore;
        }

        switch (req.EndStatus)
        {
            case BattleEndStatus.BattleEndWin:
                // Check if any avatar in the lineup has died
                foreach (var avatar in battle.Lineup.AvatarData!.Avatars)
                    if (avatar.CurrentHp <= 0)
                        HasAvatarDied = true;

                // Get monster count in stage
                long monsters = Player.SceneInstance!.Entities.Values.OfType<EntityMonster>().Count();

                if (monsters == 0) await AdvanceStage(req);

                // Calculate rounds left
                if (IsStory()) RoundsLeft = (int)Math.Min(Math.Max(RoundsLeft - req.Stt.RoundCnt, 1), RoundsLeft);

                // Set saved technique points (This will be restored if the player resets the challenge)
                SavedMp = Player.LineupManager!.GetCurLineup()!.Mp;
                break;
            case BattleEndStatus.BattleEndQuit:
                // Reset technique points and move back to start position
                var lineup = Player.LineupManager!.GetCurLineup()!;
                lineup.Mp = SavedMp;
                await Player.MoveTo(StartPos, StartRot);
                await Player.SendPacket(new PacketSyncLineupNotify(lineup));
                break;
            default:
                // Determine challenge result
                if ((IsStory() || IsBoss()) && req.Stt.EndReason == BattleEndReason.TurnLimit)
                {
                    await AdvanceStage(req);
                }
                else
                {
                    // Fail challenge
                    Status = (int)ChallengeStatus.ChallengeFailed;

                    // Send challenge result data
                    await Player.SendPacket(new PacketChallengeSettleNotify(this));
                }

                break;
        }
    }

    private async ValueTask AdvanceStage(PVEBattleResultCsReq req)
    {
        if (CurrentStage >= Excel.StageNum)
        {
            // Last stage
            Status = (int)ChallengeStatus.ChallengeFinish;
            Stars = CalculateStars();

            // Save history
            Player.ChallengeManager!.AddHistory(ChallengeId, Stars, GetTotalScore());

            // Send challenge result data
            if (IsBoss())
                await Player.SendPacket(new PacketChallengeBossPhaseSettleNotify(this, req.Stt.BattleTargetInfo[1]));
            else
                await Player.SendPacket(new PacketChallengeSettleNotify(this));

            // Call MissionManager
            await Player.MissionManager!.HandleFinishType(MissionFinishTypeEnum.ChallengeFinish, this);
        }
        else
        {
            if (IsBoss())
            {
                await Player.SendPacket(new PacketChallengeBossPhaseSettleNotify(this, req.Stt.BattleTargetInfo[1]));
            }
            else
            {
                // Increment and reset stage
                CurrentStage++;
                // Load scene group for stage 2
                await Player.SceneInstance!.EntityLoader!.LoadGroup(Excel.MazeGroupID2);

                // Change player line up
                SetCurrentExtraLineup(ExtraLineupType.LineupChallenge2);
                await Player.LineupManager!.SetCurLineup(CurrentExtraLineup + 10);
                await Player.SendPacket(new PacketChallengeLineupNotify((ExtraLineupType)CurrentExtraLineup));
                SavedMp = Player.LineupManager.GetCurLineup()!.Mp;

                // Move player
                if (Excel.MapEntranceID2 != 0)
                {
                    await Player.EnterScene(Excel.MapEntranceID2, 0, true);
                    StartPos = Player.Data.Pos!;
                    StartRot = Player.Data.Rot!;
                    await Player.SceneInstance!.EntityLoader!.LoadGroup(Excel.MazeGroupID2);
                }
                else
                {
                    await Player.MoveTo(StartPos, StartRot);
                }
            }
        }
    }

    public async ValueTask NextPhase()
    {
        // Increment and reset stage
        CurrentStage++;
        // Load scene group for stage 2
        await Player.SceneInstance!.EntityLoader!.LoadGroup(Excel.MazeGroupID2);

        // Change player line up
        SetCurrentExtraLineup(ExtraLineupType.LineupChallenge2);
        await Player.LineupManager!.SetCurLineup(CurrentExtraLineup + 10);
        await Player.SendPacket(new PacketChallengeLineupNotify((ExtraLineupType)CurrentExtraLineup));
        SavedMp = Player.LineupManager.GetCurLineup()!.Mp;

        // Move player
        if (Excel.MapEntranceID2 != 0)
        {
            await Player.EnterScene(Excel.MapEntranceID2, 0, false);
            StartPos = Player.Data.Pos!;
            StartRot = Player.Data.Rot!;
            await Player.SceneInstance!.EntityLoader!.LoadGroup(Excel.MazeGroupID2);
        }
        else
        {
            await Player.MoveTo(StartPos, StartRot);
        }
    }

    public void OnUpdate()
    {
        // End challenge if its done
        if (Status != (int)ChallengeStatus.ChallengeDoing) Player.ChallengeManager!.ChallengeInstance = null;
    }

    public int CalculateStars()
    {
        var targets = Excel.ChallengeTargetID!;
        var stars = 0;

        for (var i = 0; i < targets.Count; i++)
        {
            if (!GameData.ChallengeTargetData.ContainsKey(targets[i])) continue;

            var target = GameData.ChallengeTargetData[targets[i]];

            switch (target.ChallengeTargetType)
            {
                case ChallengeTargetExcel.ChallengeType.ROUNDS_LEFT:
                    if (RoundsLeft >= target.ChallengeTargetParam1) stars += 1 << i;
                    break;
                case ChallengeTargetExcel.ChallengeType.DEAD_AVATAR:
                    if (!HasAvatarDied) stars += 1 << i;
                    break;
                case ChallengeTargetExcel.ChallengeType.TOTAL_SCORE:
                    if (GetTotalScore() >= target.ChallengeTargetParam1) stars += 1 << i;
                    break;
            }
        }

        return Math.Min(stars, 7);
    }

    #endregion
}