DanhengServer-OpenSource/Common/Database/Lineup/LineupData.cs

using System.ComponentModel;
using EggLink.DanhengServer.Database.Avatar;
using EggLink.DanhengServer.Database.Quests;
using EggLink.DanhengServer.Proto;
using Newtonsoft.Json;
using SqlSugar;

namespace EggLink.DanhengServer.Database.Lineup;

[SugarTable("Lineup")]
public class LineupData : BaseDatabaseDataHelper
{
    public int CurLineup { get; set; } // index of current lineup
    public int CurExtraLineup { get; set; } = -1; // index of current extra lineup
    [SugarColumn(IsIgnore = true)] public int ExtraMpCount { get; set; }
    [SugarColumn(IsJson = true)] public Dictionary<int, LineupInfo> Lineups { get; set; } = []; // 9 * 4

    public int GetCurLineupIndex()
    {
        return CurExtraLineup == -1 ? CurLineup : CurExtraLineup;
    }
}

public class LineupInfo
{
    public string? Name { get; set; }
    public int LineupType { get; set; }
    public int LeaderAvatarId { get; set; }
    public List<LineupAvatarInfo>? BaseAvatars { get; set; }
    [DefaultValue(5)] public int Mp { get; set; } = 5;

    [JsonIgnore] public LineupData? LineupData { get; set; }

    [JsonIgnore] public AvatarData? AvatarData { get; set; }

    public int GetSlot(int avatarId)
    {
        return BaseAvatars?.FindIndex(item => item.BaseAvatarId == avatarId) ?? -1;
    }

    public bool Heal(int count, bool allowRevive)
    {
        var result = false;
        if (BaseAvatars != null && AvatarData != null)
            foreach (var avatar in BaseAvatars)
            {
                BaseAvatarInfo? avatarInfo;
                if (avatar.SpecialAvatarId > 0)
                    avatarInfo = AvatarData?.TrialAvatars?.Find(item => item.SpecialAvatarId == avatar.SpecialAvatarId);
                else
                    avatarInfo = AvatarData?.FormalAvatars?.Find(item => item.BaseAvatarId == avatar.BaseAvatarId);

                if (avatarInfo != null)
                {
                    if (avatarInfo.GetCurHp(IsExtraLineup()) <= 0 && !allowRevive) continue;
                    if (avatarInfo.GetCurHp(IsExtraLineup()) >= 10000 && count > 0) continue; // full hp
                    if (avatarInfo.GetCurHp(IsExtraLineup()) <= 0 && count < 0) continue; // dead
                    avatarInfo.SetCurHp(Math.Max(Math.Min(avatarInfo.GetCurHp(IsExtraLineup()) + count, 10000), 0),
                        IsExtraLineup());
                    result = true;
                }
            }

        return result;
    }

    public bool CostNowPercentHp(double count)
    {
        var result = false;
        if (BaseAvatars != null && AvatarData != null)
            foreach (var avatar in BaseAvatars)
            {
                BaseAvatarInfo? avatarInfo;
                if (avatar.SpecialAvatarId > 0)
                    avatarInfo = AvatarData?.TrialAvatars?.Find(item => item.SpecialAvatarId == avatar.SpecialAvatarId);
                else
                    avatarInfo = AvatarData?.FormalAvatars?.Find(item => item.BaseAvatarId == avatar.BaseAvatarId);

                if (avatarInfo != null)
                {
                    if (avatarInfo.CurrentHp <= 0) continue;
                    avatarInfo.SetCurHp((int)Math.Max(avatarInfo.GetCurHp(IsExtraLineup()) * (1 - count), 100),
                        IsExtraLineup());
                    result = true;
                }
            }

        return result;
    }

    public bool AddPercentSp(int count)
    {
        var result = false;
        if (BaseAvatars != null && AvatarData != null)
            foreach (var avatar in BaseAvatars)
            {
                BaseAvatarInfo? avatarInfo;
                if (avatar.SpecialAvatarId > 0)
                    avatarInfo = AvatarData?.TrialAvatars?.Find(item => item.SpecialAvatarId == avatar.SpecialAvatarId);
                else
                    avatarInfo = AvatarData?.FormalAvatars?.Find(item => item.BaseAvatarId == avatar.BaseAvatarId);

                if (avatarInfo != null)
                {
                    if (avatarInfo.CurrentHp <= 0) continue;
                    avatarInfo.SetCurSp(Math.Min(avatarInfo.GetCurSp(IsExtraLineup()) + count, 10000), IsExtraLineup());
                    result = true;
                }
            }

        return result;
    }

    public bool IsExtraLineup()
    {
        return LineupType != 0;
    }

    public Proto.LineupInfo ToProto()
    {
        Mp = Math.Max(5 + (LineupData?.ExtraMpCount ?? 0), Mp);
        Proto.LineupInfo info = new()
        {
            Name = Name,
            MaxMp = (uint)(5 + (LineupData?.ExtraMpCount ?? 0)),
            Mp = (uint)Mp,
            ExtraLineupType = (ExtraLineupType)(LineupType == (int)ExtraLineupType.LineupHeliobus
                ? (int)ExtraLineupType.LineupNone
                : LineupType),
            Index = (uint)(LineupData?.Lineups?.Values.ToList().IndexOf(this) ?? 0)
        };

        if (LineupType != (int)ExtraLineupType.LineupNone) info.Index = 0;

        if (BaseAvatars?.Find(item => item.BaseAvatarId == LeaderAvatarId) != null) // find leader,if not exist,set to 0
            info.LeaderSlot = (uint)BaseAvatars.IndexOf(BaseAvatars.Find(item => item.BaseAvatarId == LeaderAvatarId)!);
        else
            info.LeaderSlot = 0;

        if (BaseAvatars != null)
            foreach (var avatar in BaseAvatars)
                if (avatar.AssistUid != 0) // assist avatar
                {
                    var assistPlayer = DatabaseHelper.Instance?.GetInstance<AvatarData>(avatar.AssistUid);
                    if (assistPlayer != null)
                        info.AvatarList.Add(assistPlayer?.FormalAvatars
                            ?.Find(item => item.BaseAvatarId == avatar.BaseAvatarId)
                            ?.ToLineupInfo(BaseAvatars.IndexOf(avatar), this,
                                AvatarType.AvatarAssistType)); // assist avatar may not work
                }
                else if (avatar.SpecialAvatarId != 0) // special avatar
                {
                    info.AvatarList.Add(AvatarData?.TrialAvatars
                        ?.Find(item => item.SpecialAvatarId == avatar.SpecialAvatarId)
                        ?.ToLineupInfo(BaseAvatars.IndexOf(avatar), this, AvatarType.AvatarTrialType));
                }
                else // normal avatar
                {
                    info.AvatarList.Add(AvatarData?.FormalAvatars
                        ?.Find(item => item.BaseAvatarId == avatar.BaseAvatarId)
                        ?.ToLineupInfo(BaseAvatars.IndexOf(avatar), this));
                }

        var storyId = DatabaseHelper.Instance!.GetInstance<StoryLineData>(AvatarData!.Uid)?.CurStoryLineId;
        if (storyId != null && storyId != 0)
        {
            info.GameStoryLineId = (uint)storyId;
            BaseAvatars?.ForEach(item =>
            {
                if (item.SpecialAvatarId != 0) info.StoryLineAvatarIdList.Add((uint)item.SpecialAvatarId);
            });
        }

        return info;
    }
}

public class LineupAvatarInfo
{
    public int BaseAvatarId { get; set; }
    public int AssistUid { get; set; }
    public int SpecialAvatarId { get; set; }
}