PeachValley_ET6/Unity/Codes/ModelView/Module/UI/UI.cs

using System.Collections.Generic;
using UnityEngine;

namespace ET
{
	[FriendClass(typeof(UI))]
	public static class UISystem
	{
		[ObjectSystem]
		public class UIAwakeSystem : AwakeSystem<UI, string, GameObject>
		{
			public override void Awake(UI self, string name, GameObject gameObject)
			{
				self.nameChildren.Clear();
				gameObject.layer = LayerMask.NameToLayer(LayerNames.UI);
				self.Name = name;
				self.GameObject = gameObject;
			}
		}
		
		[ObjectSystem]
		public class UIDestroySystem : DestroySystem<UI>
		{
			public override void Destroy(UI self)
			{
				foreach (UI ui in self.nameChildren.Values)
				{
					ui.Dispose();
				}
			
				UnityEngine.Object.Destroy(self.GameObject);
				self.nameChildren.Clear();
			}
		}

		public static void SetAsFirstSibling(this UI self)
		{
			self.GameObject.transform.SetAsFirstSibling();
		}

		public static void Add(this UI self, UI ui)
		{
			self.nameChildren.Add(ui.Name, ui);
		}

		public static void Remove(this UI self, string name)
		{
			UI ui;
			if (!self.nameChildren.TryGetValue(name, out ui))
			{
				return;
			}
			self.nameChildren.Remove(name);
			ui.Dispose();
		}

		public static UI Get(this UI self, string name)
		{
			UI child;
			if (self.nameChildren.TryGetValue(name, out child))
			{
				return child;
			}
			GameObject childGameObject = self.GameObject.transform.Find(name)?.gameObject;
			if (childGameObject == null)
			{
				return null;
			}
			child = self.AddChild<UI, string, GameObject>(name, childGameObject);
			self.Add(child);
			return child;
		}
	}
	
	public sealed class UI: Entity, IAwake<string, GameObject>, IDestroy
	{
		public GameObject GameObject { get; set; }
		
		public string Name { get; set; }

		public Dictionary<string, UI> nameChildren = new Dictionary<string, UI>();
	}
}