using System.Collections.Generic;

namespace SWB.Base;

/// <summary>
/// Attach this component somewhere in the root of your scene.
/// Register all weapons that you want to use in here.
/// A custom WeaponRegistry can be used by basing it off this one. 
/// </summary>
[Group( "SWB" )]
[Title( "Weapon Registry" )]
public class WeaponRegistry : Component
{
	public static WeaponRegistry Instance { get; private set; }

	[Property] public List<PrefabScene> WeaponPrefabs { get; set; } = new();
	public Dictionary<string, Weapon> Weapons { get; set; } = new();

	protected override void OnDestroy()
	{
		if ( Instance == this )
			Instance = null;
	}

	protected override void OnAwake()
	{
		Instance = this;

		WeaponPrefabs.ForEach( weaponPrefab =>
		{
			CloneConfig config = new()
			{
				StartEnabled = false
			};

			var weaponGO = weaponPrefab.Clone( config );
			weaponGO.SetParent( this.GameObject );

			// Makes it so we can clone this with additional components added at runtime
			weaponGO.BreakFromPrefab();

			var weapon = weaponGO.Components.Get<Weapon>( true );
			Weapons.TryAdd( weapon.ClassName, weapon );

			weaponGO.Name = weapon.ClassName;
		} );
	}

	public Weapon Get( string className )
	{
		if ( string.IsNullOrWhiteSpace( className ) ) return null;

		Weapons.TryGetValue( className, out var weapon );
		return weapon;
	}
}