CTT/Server/Hotfix/Game/System/Other/MainStoryAIComponentSystem.cs

using System;
using System.Collections.Generic;

namespace ET
{
    public class MainStoryAIComponentAwakeSystem : AwakeSystem<MainStoryAIComponent>
    {
    	public override void Awake(MainStoryAIComponent self)
    	{
            MainStoryAIComponent.instance = self;
    	}
    }
    public static class MainStoryAIComponentSystem
    {
        public static MainStoryAI CreateAI(this MainStoryAIComponent self,Team team,int sceneId)
        {
            MainStoryAI ai = EntityFactory.CreateWithParentAndId<MainStoryAI>(self,team.LeaderId);
            ai.Init(team, sceneId);
            foreach (long item in team.GetUnitIds())
            {
                self.aiIdDic[item] = ai.Id;
            }
            return ai;
        }
        public static MainStoryAI GetAI(this MainStoryAIComponent self, long id)
        {
            MainStoryAI ai = self.GetChild<MainStoryAI>(id);
            return ai;
        }
        public static MainStoryAI GetAI(this MainStoryAIComponent self, Unit unit)
        {
            if(!self.aiIdDic.TryGetValue(unit.Id,out long id))
            {
                return null;
            }
            MainStoryAI ai = self.GetChild<MainStoryAI>(id);
            return ai;
        }
        public static bool DestoryAI(this MainStoryAIComponent self, MainStoryAI ai)
        {
            if (!ai) return false;
            List<long> list = ai.team?.GetUnitIds();
            if (list != null)
            foreach (long item in list)
            {
                self.aiIdDic.Remove(item);
            }
            ai.Dispose();
            return true;
        }
        public static bool DestoryAI(this MainStoryAIComponent self, Unit unit)
        {
           MainStoryAI ai= self.GetAI(unit);
            return DestoryAI(self, ai);
        }
    }
}