using System;
using System.Collections.Generic;
using UnityEngine;

namespace CleverCrow.Fluid.FSMs {
    public class State : IState {
        private readonly Dictionary<string, ITransition> _transitions = new Dictionary<string, ITransition>();
        
        public Enum Id { get; }
        public List<IAction> Actions { get; } = new List<IAction>();
        public IFsm ParentFsm { get; }

        public State (IFsm fsm, Enum id) {
            ParentFsm = fsm;
            Id = id;
        }

        public void AddTransition (ITransition transition) {
            _transitions[transition.Name] = transition;
        }
        
        public ITransition GetTransition (string name) {
            ITransition result;
            _transitions.TryGetValue(name, out result);
            
            return result;
        }
        
        public void Update () {
            foreach (var action in Actions) {
                action.Update();
            }
        }

        public void Enter () {
            foreach (var action in Actions) {
                action.Enter();
            }
        }

        public void Exit () {
            foreach (var action in Actions) {
                action.Exit();
            }
        }

        public void Transition (string id) {
            var transition = GetTransition(id);
            if (transition == null) return;
            
            ParentFsm.SetState(transition.Target);
        }
    }
}