inabox/InABox.Poster.Custom/CustomPosterEngine.cs

using InABox.Core;
using InABox.Scripting;
using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Threading.Tasks;

namespace InABox.Poster.Custom;

public class CustomPosterEngine<TPostable> : PosterEngine<TPostable, ICustomPoster<TPostable>, CustomPosterSettings>
    where TPostable : Entity, IPostable, IRemotable, IPersistent, new()
{
    private ScriptDocument? _script;
    private bool _hasCheckedScript;

    private ScriptDocument? GetScriptDocument()
    {
        if (_hasCheckedScript)
        {
            return _script;
        }

        var settings = GetSettings();
        if (settings.ScriptEnabled && !string.IsNullOrWhiteSpace(settings.Script))
        {
            var document = new ScriptDocument(settings.Script);
            if (!document.Compile())
            {
                throw new Exception("Script failed to compile!");
            }
            _script = document;
        }
        else
        {
            _script = null;
        }

        _hasCheckedScript = true;
        return _script;
    }

    public override bool BeforePost(IDataModel<TPostable> model)
    {
        if (GetScriptDocument() is ScriptDocument script)
        {
            return script.Execute(methodname: "BeforePost", parameters: new object[] { model });
        }
        return false;
    }

    protected override IPostResult<TPostable> DoProcess(IDataModel<TPostable> model)
    {
        if (GetScriptDocument() is ScriptDocument script)
        {
            if(!script.Execute(methodname: "Process", parameters: new object[] { model }))
            {
                throw new Exception("Post Failed.");
            }

            var resultsObject = script.GetValue("Results");
            return (resultsObject as IPostResult<TPostable>)
                ?? throw new Exception($"Script 'Results' property expected to be IPostResult<{typeof(TPostable)}>, got {resultsObject}");
        }
        else
        {
            throw new Exception("Post Failed.");
        }
    }
    public override void AfterPost(IDataModel<TPostable> model, IPostResult<TPostable> result)
    {
        if (GetScriptDocument() is ScriptDocument script)
        {
            script.Execute(methodname: "AfterPost", parameters: new object[] { model });
        }
    }
}