BotSharp/src/Infrastructure/BotSharp.Core/Templating/ResponseTemplateService.cs

using BotSharp.Abstraction.Repositories;
using BotSharp.Abstraction.Routing.Models;
using BotSharp.Abstraction.Templating;
using System.Reflection;

namespace BotSharp.Core.Templating;

public class ResponseTemplateService : IResponseTemplateService
{
    private readonly IServiceProvider _services;
    private readonly ILogger _logger;

    public ResponseTemplateService(IServiceProvider services, ILogger<ResponseTemplateService> logger)
    {
        _services = services;
        _logger = logger;
    }

    public async Task<string> RenderFunctionResponse(string agentId, RoleDialogModel message)
    {
        // Find response template
        var db = _services.GetRequiredService<IBotSharpRepository>();
        var responses = db.GetAgentResponses(agentId, "func", message.FunctionName);

        if (responses.Count == 0)
        {
            return string.Empty;
        }

        var randomIndex = new Random().Next(0, responses.Count);
        var template = responses[randomIndex];

        var render = _services.GetRequiredService<ITemplateRender>();

        // Convert args and execute data to dictionary
        var dict = new Dictionary<string, object>();

        if (message.FunctionArgs != null)
        {
            ExtractArgs(JsonSerializer.Deserialize<JsonDocument>(message.FunctionArgs), dict);
        }

        if (message.ExecutionData != null)
        {
            ExtractExecuteData(message.ExecutionData, dict);
        }
            
        var text = render.Render(template, dict);

        return text;
    }

    public async Task<string> RenderIntentResponse(string agentId, RoleDialogModel message)
    {
        // Find response template
        //var agentService = _services.GetRequiredService<IAgentService>();
        //var dir = Path.Combine(agentService.GetAgentDataDir(agentId), "responses");
        //if (!Directory.Exists(dir))
        //{
        //    return string.Empty;
        //}
        //var responses = Directory.GetFiles(dir)
        //    .Where(f => f.Split(Path.DirectorySeparatorChar).Last().Split('.')[1] == message.IntentName)
        //    .ToList();

        var db = _services.GetRequiredService<IBotSharpRepository>();
        var context = _services.GetRequiredService<RoutingContext>();
        var responses = db.GetAgentResponses(agentId, "intent", context.IntentName);

        if (responses.Count == 0)
        {
            return string.Empty;
        }

        var randomIndex = new Random().Next(0, responses.Count);
        //var template = File.ReadAllText(responses[randomIndex]);
        var template = responses[randomIndex];

        var render = _services.GetRequiredService<ITemplateRender>();

        // Convert args and execute data to dictionary
        var dict = new Dictionary<string, object>();
        if (!string.IsNullOrEmpty(message.FunctionArgs))
        {
            ExtractArgs(JsonSerializer.Deserialize<JsonDocument>(message.FunctionArgs), dict);
        }

        if (message.ExecutionData != null)
        {
            ExtractExecuteData(message.ExecutionData, dict);
        }

        var text = render.Render(template, dict);

        return text;
    }

    private void ExtractArgs(JsonDocument args, Dictionary<string, object> dict)
    {
        if (args.RootElement is JsonElement root)
        {
            foreach (JsonProperty property in root.EnumerateObject())
            {
                if (!string.IsNullOrEmpty(property.Value.ToString()))
                {
                    dict[property.Name] = property.Value.ToString();
                }
            }
        }
    }

    private void ExtractExecuteData(object data, Dictionary<string, object> dict)
    {
        foreach (PropertyInfo property in data.GetType().GetProperties())
        {
            var value = property.GetValue(data, null);
            if (value != null)
            {
                dict[property.Name] = value;
            }
        }
    }
}