using System;
using System.Runtime.InteropServices;
using gaemstone.ECS.Internal;
using gaemstone.ECS.Utility;
using static flecs_hub.flecs;

namespace gaemstone.ECS;

public static class ObserverExtensions
{
	public static unsafe Entity<TContext> InitObserver<TContext>(this Entity<TContext> entity,
		FilterDesc filter, Action<Iterator<TContext>> callback, params Entity[] events)
	{
		if (events.Length == 0) throw new ArgumentException("Must specify at least 1 event", nameof(events));
		if (events.Length >  8) throw new ArgumentException("Must specify at most 8 events", nameof(events));

		var world = entity.World;
		using var alloc = TempAllocator.Use();
		var desc = new ecs_observer_desc_t {
			entity = entity,
			filter = filter.ToFlecs(alloc),
			binding_ctx      = (void*)CallbackContextHelper.Create((world, callback)),
			binding_ctx_free = new() { Data = new() { Pointer = &FreeContext } },
			callback         = new() { Data = new() { Pointer = &Callback } },
		};
		var span = desc.events;
		for (var i = 0; i < events.Length; i++)
			span[i] = events[i];
		if (ecs_observer_init(world, &desc).Data == default)
			throw new InvalidOperationException();
		return entity;
	}

	[UnmanagedCallersOnly]
	private static unsafe void Callback(ecs_iter_t* iter)
	{
		var (world, callback) = CallbackContextHelper
			.Get<(World, Action<Iterator>)>((nint)iter->binding_ctx);
		callback(new Iterator(iter));
	}

	[UnmanagedCallersOnly]
	private static unsafe void FreeContext(void* context)
		=> CallbackContextHelper.Free((nint)context);
}