132 lines
4.2 KiB
C#
132 lines
4.2 KiB
C#
using BotSharp.Abstraction.Models;
|
|
using BotSharp.Abstraction.Routing.Models;
|
|
using BotSharp.Abstraction.Templating;
|
|
using BotSharp.Abstraction.Translation.Models;
|
|
using Fluid;
|
|
using Fluid.Ast;
|
|
using System.Collections;
|
|
using System.IO;
|
|
using System.Reflection;
|
|
using System.Text.Encodings.Web;
|
|
|
|
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>();
|
|
|
|
_parser.RegisterIdentifierTag("link", RenderIdentifierTag);
|
|
}
|
|
|
|
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);
|
|
}
|
|
else
|
|
{
|
|
_logger.LogWarning(error);
|
|
}
|
|
|
|
return template;
|
|
}
|
|
|
|
public void RegisterType(Type type)
|
|
{
|
|
if (type == null || IsStringType(type)) return;
|
|
|
|
if (IsListType(type))
|
|
{
|
|
if (type.IsGenericType)
|
|
{
|
|
var genericType = type.GetGenericArguments()[0];
|
|
RegisterType(genericType);
|
|
}
|
|
}
|
|
else if (IsTrackToNextLevel(type))
|
|
{
|
|
_options.MemberAccessStrategy.Register(type);
|
|
var props = type.GetProperties();
|
|
foreach (var prop in props)
|
|
{
|
|
RegisterType(prop.PropertyType);
|
|
}
|
|
}
|
|
}
|
|
|
|
|
|
#region Private methods
|
|
private static async ValueTask<Completion> RenderIdentifierTag(string identifier, TextWriter writer, TextEncoder encoder, TemplateContext context)
|
|
{
|
|
try
|
|
{
|
|
var value = await context.Model.GetValueAsync(TemplateRenderConstant.RENDER_AGENT, context);
|
|
var agent = value?.ToObjectValue() as Agent;
|
|
var found = agent?.Templates?.FirstOrDefault(x => x.Name.IsEqualTo(identifier));
|
|
var key = $"{agent?.Id} | {identifier}";
|
|
|
|
if (found == null || (context.AmbientValues.TryGetValue(key, out var visited) && (bool)visited))
|
|
{
|
|
writer.Write(string.Empty);
|
|
}
|
|
else if (_parser.TryParse(found.Content, out var t, out _))
|
|
{
|
|
context.AmbientValues[key] = true;
|
|
var rendered = t.Render(context);
|
|
writer.Write(rendered);
|
|
context.AmbientValues.Remove(key);
|
|
}
|
|
else
|
|
{
|
|
writer.Write(string.Empty);
|
|
}
|
|
}
|
|
catch
|
|
{
|
|
writer.Write(string.Empty);
|
|
}
|
|
|
|
return Completion.Normal;
|
|
}
|
|
|
|
private static bool IsStringType(Type type)
|
|
{
|
|
return type == typeof(string);
|
|
}
|
|
|
|
private static bool IsListType(Type type)
|
|
{
|
|
var interfaces = type.GetTypeInfo().ImplementedInterfaces;
|
|
return type.IsArray || interfaces.Any(x => x.Name == typeof(IEnumerable).Name);
|
|
}
|
|
|
|
private static bool IsTrackToNextLevel(Type type)
|
|
{
|
|
return type.IsClass || type.IsInterface || type.IsAbstract;
|
|
}
|
|
#endregion
|
|
}
|