52 lines
1.9 KiB
C#
52 lines
1.9 KiB
C#
using BotSharp.Abstraction.Agents.Models;
|
|
using BotSharp.Abstraction.Functions.Models;
|
|
using BotSharp.Abstraction.Models;
|
|
using BotSharp.Abstraction.Routing.Models;
|
|
using BotSharp.Abstraction.Templating;
|
|
using BotSharp.Abstraction.Translation.Models;
|
|
using Fluid;
|
|
|
|
namespace BotSharp.Core.Templating;
|
|
|
|
public class TemplateRender : ITemplateRender
|
|
{
|
|
private readonly IServiceProvider _services;
|
|
private readonly ILogger _logger;
|
|
private static readonly FluidParser _parser = new FluidParser();
|
|
private TemplateOptions _options;
|
|
|
|
public TemplateRender(IServiceProvider services, ILogger<TemplateRender> logger)
|
|
{
|
|
_services = services;
|
|
_logger = logger;
|
|
_options = new TemplateOptions();
|
|
_options.MemberAccessStrategy.MemberNameStrategy = MemberNameStrategies.SnakeCase;
|
|
|
|
_options.MemberAccessStrategy.Register<NameDesc>();
|
|
_options.MemberAccessStrategy.Register<ParameterPropertyDef>();
|
|
_options.MemberAccessStrategy.Register<RoleDialogModel>();
|
|
_options.MemberAccessStrategy.Register<Agent>();
|
|
_options.MemberAccessStrategy.Register<RoutableAgent>();
|
|
_options.MemberAccessStrategy.Register<RoutingHandlerDef>();
|
|
_options.MemberAccessStrategy.Register<FunctionDef>();
|
|
_options.MemberAccessStrategy.Register<FunctionParametersDef>();
|
|
_options.MemberAccessStrategy.Register<UserIdentity>();
|
|
_options.MemberAccessStrategy.Register<TranslationInput>();
|
|
}
|
|
|
|
public string Render(string template, Dictionary<string, object> dict)
|
|
{
|
|
if (_parser.TryParse(template, out var t, out var error))
|
|
{
|
|
var context = new TemplateContext(dict, _options);
|
|
template = t.Render(context);
|
|
return template;
|
|
}
|
|
else
|
|
{
|
|
_logger.LogWarning(error);
|
|
return template;
|
|
}
|
|
}
|
|
}
|