using BotSharp.Abstraction.Functions.Models;
using BotSharp.Abstraction.Routing;
using BotSharp.Abstraction.Routing.Enums;
using BotSharp.Abstraction.Routing.Settings;
using System.Diagnostics.Metrics;

namespace BotSharp.Core.Routing.Hooks;

public class RoutingAgentHook : AgentHookBase
{
    private readonly RoutingSettings _routingSetting;
    public override string SelfId => string.Empty;

    public RoutingAgentHook(IServiceProvider services, AgentSettings settings, RoutingSettings routingSetting) 
        : base(services, settings)
    {
        _routingSetting = routingSetting;
    }

    public override bool OnInstructionLoaded(string template, Dictionary<string, object> dict)
    {
        if (_agent.Type != AgentType.Routing)
        {
            return base.OnInstructionLoaded(template, dict);
        }
        dict["router"] = _agent;

        var routing = _services.GetRequiredService<IRoutingService>();
        var agents = routing.GetRoutableAgents(_agent.Profiles);
        dict["routing_agents"] = agents;
        dict["routing_handlers"] = routing.GetHandlers(_agent);

        return base.OnInstructionLoaded(template, dict);
    }

    public override bool OnFunctionsLoaded(List<FunctionDef> functions)
    {
        if (_agent.Type == AgentType.Task)
        {
            // check if enabled the routing rule
            var routing = _services.GetRequiredService<IRoutingService>();
            var rule = routing.GetRulesByAgentId(_agent.Id)
                .FirstOrDefault(x => x.Type == RuleType.Fallback);
            if (rule != null)
            {
                var agentService = _services.GetRequiredService<IAgentService>();
                var redirectAgent = agentService.GetAgent(rule.RedirectTo).Result;

                var json = JsonSerializer.Serialize(new
                {
                    user_goal_agent = new
                    {
                        type = "string",
                        description = $"{_agent.Name}"
                    },
                    next_action_agent = new
                    {
                        type = "string",
                        description = $"{redirectAgent.Name}"
                    },
                    reason = new
                    {
                        type = "string",
                        description = $"the reason why you need to fallback to [{redirectAgent.Name}] from [{_agent.Name}]"
                    },
                });
                functions.Add(new FunctionDef
                {
                    Name = "fallback_to_router",
                    Description = $"If the user's request is beyond your capabilities, you can call this function to handle by other agent ({redirectAgent.Name}).",
                    Parameters =
                    {
                        Properties = JsonSerializer.Deserialize<JsonDocument>(json),
                        Required = new List<string>
                        {
                            "user_goal_agent",
                            "next_action_agent",
                            "reason"
                        }
                    }
                });
            }
        }

        return base.OnFunctionsLoaded(functions);
    }
}