using System.Collections.Generic;
using UnityEngine;

namespace LLM.Commands
{
    public class LLMCommandRouter : MonoBehaviour
    {
        [Header("Registry")]
        [SerializeField] private bool autoDiscoverReceiversOnAwake = true;
        [SerializeField] private List<LLMCommandReceiverBase> receivers = new List<LLMCommandReceiverBase>();

        [Header("Debug")]
        [SerializeField] private bool debugLogs = true;

        private readonly Dictionary<string, LLMCommandReceiverBase> map = new Dictionary<string, LLMCommandReceiverBase>();

        private void Awake()
        {
            if (autoDiscoverReceiversOnAwake)
            {
                AutoDiscoverReceivers();
            }

            RebuildMap();
        }

        public void AutoDiscoverReceivers()
        {
            receivers.Clear();
            receivers.AddRange(FindObjectsOfType<LLMCommandReceiverBase>(true));
        }

        public void RebuildMap()
        {
            map.Clear();

            for (int i = 0; i < receivers.Count; i++)
            {
                var r = receivers[i];
                if (r == null) continue;
                if (string.IsNullOrWhiteSpace(r.TargetId)) continue;

                if (map.ContainsKey(r.TargetId))
                {
                    if (debugLogs) Debug.LogWarning($"[LLMCommandRouter] Duplicate targetId='{r.TargetId}'.", r);
                    continue;
                }

                map.Add(r.TargetId, r);
            }

            if (debugLogs) Debug.Log($"[LLMCommandRouter] Registry built. targets={map.Count}", this);
        }

        public int ExecuteAll(LLMCommand[] commands)
        {
            if (commands == null || commands.Length == 0) return 0;

            int executed = 0;
            for (int i = 0; i < commands.Length; i++)
            {
                var cmd = commands[i];
                if (cmd == null) continue;
                if (string.IsNullOrWhiteSpace(cmd.targetId)) continue;

                if (!map.TryGetValue(cmd.targetId, out var receiver) || receiver == null) continue;

                if (receiver.TryExecute(cmd)) executed++;
            }

            if (debugLogs) Debug.Log($"[LLMCommandRouter] Executed commands={executed}/{commands.Length}", this);
            return executed;
        }
    }
}