BadHtmlTemplate.cs

Go to the documentation of this file.

using System.IO;
 
using BadScript2;
using BadScript2.IO;
using BadScript2.Runtime;
using BadScript2.Runtime.Objects;
using BadScript2.Runtime.Objects.Types;
 
using HtmlAgilityPack;
 
namespace BadHtml;
 
public class BadHtmlTemplate
{
    private readonly IFileSystem m_FileSystem;
 
    private string? m_Source;
 
    private BadHtmlTemplate(string filePath, IFileSystem fileSystem)
    {
        FilePath = filePath;
        m_FileSystem = fileSystem;
    }
 
 
    public string FilePath { get; }
 
    private string GetSource()
    {
        if (m_Source == null)
        {
            Reload();
        }
 
        return m_Source!;
    }
 
    public void Reload()
    {
        m_Source = m_FileSystem.ReadAllText(FilePath);
    }
 
    public HtmlDocument RunTemplate(object? model = null,
                                    BadHtmlTemplateOptions? options = null,
                                    BadScope? caller = null)
    {
        options ??= new BadHtmlTemplateOptions();
        string src = GetSource();
 
        // ReSharper disable once UseObjectOrCollectionInitializer
        HtmlDocument input = new HtmlDocument();
        input.OptionUseIdAttribute = true;
        input.LoadHtml(src);
 
        // ReSharper disable once UseObjectOrCollectionInitializer
        HtmlDocument output = new HtmlDocument();
        output.OptionUseIdAttribute = true;
        output.LoadHtml("");
 
        BadExecutionContext executionContext =
            (options.Runtime ?? new BadRuntime()).CreateContext(Path.GetDirectoryName(m_FileSystem.GetFullPath(FilePath)
                                                                    ) ??
                                                                "/"
                                                               );
        executionContext.Scope.SetCaller(caller);
 
        if (model != null)
        {
            BadObject mod = model as BadObject ?? BadObject.Wrap(model);
 
            executionContext.Scope.DefineVariable("Model",
                                                  mod,
                                                  executionContext.Scope,
                                                  new BadPropertyInfo(BadAnyPrototype.Instance, true)
                                                 );
        }
 
        foreach (HtmlNode node in input.DocumentNode.ChildNodes)
        {
            BadHtmlContext ctx =
                new BadHtmlContext(node, output.DocumentNode, executionContext, FilePath, src, options, m_FileSystem);
            BadHtmlNodeTransformer.Transform(ctx);
        }
 
        return output;
    }
 
    public string Run(object? model = null, BadHtmlTemplateOptions? options = null, BadScope? caller = null)
    {
        return RunTemplate(model, options, caller)
               .DocumentNode.OuterHtml;
    }
 
    public static BadHtmlTemplate Create(string file, IFileSystem? fileSystem = null)
    {
        return new BadHtmlTemplate(file, fileSystem ?? BadFileSystem.Instance);
    }
}