#if TOOLS
namespace Pong.addons.CustomTypes
{
	[Tool]
	public partial class Plugin : EditorPlugin
	{
		public override void _EnterTree()
		{
			var baseMenuScript = GD.Load<Script>("res://Scripts/Managers/UI/BaseMenu.cs");
			var basePaddleScript = GD.Load<Script>("res://Scripts/Managers/BasePaddle.cs");
			
			AddCustomType("BaseMenu", "VBoxContainer", baseMenuScript, GetEditorIcon("VBoxContainer"));
			AddCustomType("BasePaddle", "RigidBody2D", basePaddleScript, GetEditorIcon("RigidBody2D"));
		}

		public override void _ExitTree()
		{
			RemoveCustomType("BaseMenu");
			RemoveCustomType("BasePaddle");
		}

		private Texture2D GetEditorIcon(string iconName)
		{
			return GetEditorInterface().GetBaseControl().GetThemeIcon(iconName, "EditorIcons");
		}
	}
}
#endif