using System;
using System.Collections.Generic;

namespace MoonTools.ECS
{
	public abstract class EntityComponentReader
	{
		internal EntityStorage EntityStorage;
		internal ComponentDepot ComponentDepot;
		internal RelationDepot RelationDepot;
		protected FilterBuilder FilterBuilder => new FilterBuilder(ComponentDepot);

		internal void RegisterEntityStorage(EntityStorage entityStorage)
		{
			EntityStorage = entityStorage;
		}

		internal void RegisterComponentDepot(ComponentDepot componentDepot)
		{
			ComponentDepot = componentDepot;
		}

		internal void RegisterRelationDepot(RelationDepot relationDepot)
		{
			RelationDepot = relationDepot;
		}

		protected ReadOnlySpan<TComponent> ReadComponents<TComponent>() where TComponent : struct
		{
			return ComponentDepot.ReadComponents<TComponent>();
		}

		protected bool Has<TComponent>(in Entity entity) where TComponent : struct
		{
			return ComponentDepot.Has<TComponent>(entity.ID);
		}

		protected bool Some<TComponent>() where TComponent : struct
		{
			return ComponentDepot.Some<TComponent>();
		}

		protected ref readonly TComponent Get<TComponent>(in Entity entity) where TComponent : struct
		{
			return ref ComponentDepot.Get<TComponent>(entity.ID);
		}

		protected ref readonly TComponent GetSingleton<TComponent>() where TComponent : struct
		{
			return ref ComponentDepot.Get<TComponent>();
		}

		protected Entity GetSingletonEntity<TComponent>() where TComponent : struct
		{
			return ComponentDepot.GetSingletonEntity<TComponent>();
		}

		protected bool Exists(in Entity entity)
		{
			return EntityStorage.Exists(entity);
		}

		protected IEnumerable<(Entity, Entity, TRelationKind)> Relations<TRelationKind>() where TRelationKind : struct
		{
			return RelationDepot.Relations<TRelationKind>();
		}

		protected bool Related<TRelationKind>(in Entity a, in Entity b) where TRelationKind : struct
		{
			return RelationDepot.Related<TRelationKind>(a.ID, b.ID);
		}

		protected IEnumerable<(Entity, TRelationKind)> RelatedToA<TRelationKind>(in Entity entity) where TRelationKind : struct
		{
			return RelationDepot.RelatedToA<TRelationKind>(entity.ID);
		}

		protected IEnumerable<(Entity, TRelationKind)> RelatedToB<TRelationKind>(in Entity entity) where TRelationKind : struct
		{
			return RelationDepot.RelatedToB<TRelationKind>(entity.ID);
		}
	}
}