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

using EggLink.DanhengServer.Data;
using EggLink.DanhengServer.Database.Avatar;
using EggLink.DanhengServer.Database.Player;
using EggLink.DanhengServer.Proto;
using Newtonsoft.Json;
using SqlSugar;
using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Threading.Tasks;

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(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<AvatarInfo>? BaseAvatars { get; set; }
        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)
        {
            bool result = false;
            if (BaseAvatars != null && AvatarData != null)
            {
                foreach (var avatar in BaseAvatars)
                {
                    var avatarInfo = AvatarData?.Avatars?.Find(item => item.GetAvatarId() == avatar.BaseAvatarId);
                    if (avatarInfo != null)
                    {
                        if (avatarInfo.GetCurHp(IsExtraLineup()) <= 0 && !allowRevive)
                        {
                            continue;
                        }
                        if (avatarInfo.GetCurHp(IsExtraLineup()) >= 10000)
                        {
                            continue;
                        }
                        avatarInfo.SetCurHp(Math.Min(avatarInfo.GetCurHp(IsExtraLineup()) + count, 10000), IsExtraLineup());
                        result = true;
                    }
                }
                DatabaseHelper.Instance?.UpdateInstance(AvatarData!);
            }
            return result;
        }

        public bool CostNowPercentHp(double count)
        {
            bool result = false;
            if (BaseAvatars != null && AvatarData != null)
            {
                foreach (var avatar in BaseAvatars)
                {
                    var avatarInfo = AvatarData?.Avatars?.Find(item => item.GetAvatarId() == avatar.BaseAvatarId);
                    if (avatarInfo != null)
                    {
                        if (avatarInfo.CurrentHp <= 0)
                        {
                            continue;
                        }
                        avatarInfo.SetCurHp((int)Math.Max(avatarInfo.GetCurHp(IsExtraLineup()) * (1 - count), 100), IsExtraLineup());
                        result = true;
                    }
                }
                DatabaseHelper.Instance?.UpdateInstance(AvatarData!);
            }
            return result;
        }

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

        public Proto.LineupInfo ToProto()
        {
            Proto.LineupInfo info = new()
            {
                Name = Name,
                MaxMp = 5,
                Mp = (uint)Mp,
                ExtraLineupType = (ExtraLineupType)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?.Avatars?.Find(item => item.GetAvatarId() == avatar.BaseAvatarId)?.ToLineupInfo(BaseAvatars.IndexOf(avatar), this, AvatarType.AvatarAssistType));  // assist avatar may not work
                        }
                    } else if (avatar.SpecialAvatarId != 0)  // special avatar
                    {
                        var specialAvatar = GameData.SpecialAvatarData[avatar.SpecialAvatarId];
                        if (specialAvatar != null)
                        {
                            info.AvatarList.Add(specialAvatar.ToAvatarData(LineupData!.Uid).ToLineupInfo(BaseAvatars.IndexOf(avatar), this, AvatarType.AvatarTrialType));
                        }
                    } else  // normal avatar
                    {
                        info.AvatarList.Add(AvatarData?.Avatars?.Find(item => item.AvatarId == avatar.BaseAvatarId)?.ToLineupInfo(BaseAvatars.IndexOf(avatar), this));
                    }
                }
            }

            return info;
        }
    }

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