using System.Collections.Generic;
using System.Reflection;
using Godot;
using Godot.Collections;

namespace Cirno.Scripts.Components.FSM;

public abstract partial class BaseState<TKey, TType> : Node2D, IState<TKey, TType>
    where TKey : notnull
    where TType : Node
{
    public virtual TKey StateId { get; }
    public IStateMachine<TKey, TType> StateMachine => _stateMachine;

    private IStateMachine<TKey, TType> _stateMachine;

    public TType MainObject => _stateMachine.MainObject;

    [Export]
    private Array<Node> _moduleNodes = [];

    private readonly List<IModule<TKey, TType>> _modules = [];
    
    public virtual void Init(IStateMachine<TKey, TType> machine)
    {
        _stateMachine = machine;

        foreach (var node in _moduleNodes)
        {
            if (node is IModule<TKey, TType> module)
            {
                _modules.Add(module);
                module.Init(_stateMachine);
            }
        }
    }

    public abstract void EnterState();

    public abstract void ExitState();

    public virtual void ProcessState(double delta)
    {
        _modules.ForEach(module => module.Process(delta));
    }

    public virtual void PhysicsProcessState(double delta)
    {
        _modules.ForEach(module => module.PhysicsProcess(delta));
    }
}