using Sandbox.UI;
namespace Sandbox;

public partial class BaseSpawnMenu : Panel
{
	SpawnMenuOption activeOption;
	PanelSwitcher Switcher = default;

	protected Panel MenuFooter;
	bool _addedFooter;

	protected override void OnParametersSet()
	{
		base.OnParametersSet();

		options.Clear();
		Rebuild();
		StateHasChanged();
	}

	protected override void OnAfterTreeRender( bool firstTime )
	{
		base.OnAfterTreeRender( firstTime );

		if ( !_addedFooter && MenuFooter.IsValid() )
		{
			_addedFooter = true;
			OnMenuFooter( MenuFooter );
		}
	}

	protected bool _firstViewed;

	public override void Tick()
	{
		base.Tick();

		if ( !IsVisible ) return;
		if ( _firstViewed ) return;
		if ( options.Count == 0 ) return;
		if ( !Switcher.IsValid() ) return;

		_firstViewed = true;
		SwitchOption( options.Where( x => x.PanelCreator != null || x.Panel != null ).FirstOrDefault() );
	}


	protected virtual void Rebuild()
	{

	}

	protected virtual void OnMenuFooter( Panel footer )
	{

	}

	public void AddHeader( string name )
	{
		var o = new SpawnMenuOption
		{
			Type = "header",
			Name = name
		};

		options.Add( o );
	}

	public void AddGrow()
	{
		var o = new SpawnMenuOption
		{
			Type = "grow"
		};

		options.Add( o );
	}

	public void AddOption( string name, Func<Panel> createPanelFunction )
	{
		var o = new SpawnMenuOption
		{
			Name = name,
			PanelCreator = createPanelFunction
		};

		options.Add( o );
	}

	public void AddOption( string icon, string name, Func<Panel> createPanelFunction )
	{
		var o = new SpawnMenuOption
		{
			Icon = icon,
			Name = name,
			PanelCreator = createPanelFunction
		};

		options.Add( o );
	}

	public void AddOption( string icon, string name, Func<Panel> createPanelFunction, Action onRightClick )
	{
		var o = new SpawnMenuOption
		{
			Icon = icon,
			Name = name,
			PanelCreator = createPanelFunction,
			OnRightClick = onRightClick
		};

		options.Add( o );
	}

	void OnOptionClick( SpawnMenuOption o )
	{
		if ( o.OnClick != null )
		{
			o.OnClick.Invoke();
			return;
		}

		SwitchOption( o );
	}

	void OnOptionRightClick( SpawnMenuOption o )
	{
		o.OnRightClick?.Invoke();
	}

	void OnOptionMouseDown( SpawnMenuOption o, PanelEvent e )
	{
		if ( e is MousePanelEvent me && me.MouseButton == MouseButtons.Right && o.OnRightClick != null )
		{
			o.OnRightClick.Invoke();
			e.StopPropagation();
		}
	}

	void SwitchOption( SpawnMenuOption o )
	{
		if ( o == activeOption ) return;

		activeOption?.Panel?.SetClass( "hidden", true );

		activeOption = o;

		if ( activeOption.Panel == null && activeOption.PanelCreator != null )
		{
			activeOption.Panel = activeOption.PanelCreator.Invoke();
			Switcher.AddChild( activeOption.Panel );
		}

		activeOption?.Panel?.SetClass( "hidden", false );
		StateHasChanged();
	}

	public void AddAction( string icon, string name, Action action )
	{
		var o = new SpawnMenuOption
		{
			Icon = icon,
			Name = name,
			OnClick = action
		};

		options.Add( o );
	}

	public void SelectOption( string name )
	{
		var option = options.FirstOrDefault( o => o.Name == name && (o.PanelCreator != null || o.Panel != null) );
		if ( option != null ) SwitchOption( option );
	}

	public void DeselectOption()
	{
		activeOption?.Panel?.SetClass( "hidden", true );
		activeOption = null;
		StateHasChanged();
	}

	public void AddSkeletons( int count )
	{
		for ( int i = 0; i < count; i++ )
		{
			options.Add( new SpawnMenuOption { Type = "skeleton" } );
		}
		StateHasChanged();
	}

	class SpawnMenuOption
	{
		public string Type { get; set; } = "option";
		public string Name { get; set; }
		public string Icon { get; set; }
		public Func<Panel> PanelCreator { get; set; }
		public Panel Panel { get; set; }
		public Action OnClick { get; set; }
		public Action OnRightClick { get; set; }
	}

	List<SpawnMenuOption> options = new();
}