inabox/InABox.Poster.CSV/CSVPosterEngine.cs

using CsvHelper;
using InABox.Core;
using InABox.Core.Postable;
using InABox.Scripting;
using Microsoft.Win32;
using System;
using System.Collections.Generic;
using System.Globalization;
using System.IO;
using System.Linq;
using System.Text;
using System.Threading.Tasks;

namespace InABox.Poster.CSV
{
    public class CSVPosterEngine<TPostable> : PosterEngine<TPostable, ICSVPoster<TPostable>, CSVPosterSettings>
        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);
                document.Properties.Add(new ScriptProperty("Results", null));
                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 });
            }
            else
            {
                return Poster.BeforePost(model);
            }
        }

        protected override IPostResult<TPostable> DoProcess(IDataModel<TPostable> model)
        {
            var settings = GetSettings();

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

                var resultsObject = script.GetValue("Results");
                results = (resultsObject as ICSVExport<TPostable>)
                    ?? throw new Exception($"Script 'Results' property expected to be ICSVExport<{typeof(TPostable)}>, got {resultsObject}");
            }
            else
            {
                results = Poster.Process(model);
            }

            var dlg = new SaveFileDialog()
            {
                FileName = settings.DefaultOutputFile,
                Filter = "CSV Files (*.csv)|*.csv"
            };

            if(dlg.ShowDialog() == true)
            {
                using var writer = new StreamWriter(dlg.FileName);
                using var csv = new CsvWriter(writer, CultureInfo.InvariantCulture);
                csv.Context.RegisterClassMap(results.ClassMap.ClassMap);

                var method = typeof(CsvWriter).GetMethods()
                    .Where(x => x.Name == nameof(CsvWriter.WriteRecords) && x.GetGenericArguments().Length == 1).First()
                    .MakeGenericMethod(results.Type);
                method.Invoke(csv, new object?[] { results.Records });
                return results;
            }
            else
            {
                throw new PostCancelledException();
            }
        }

        public override void AfterPost(IDataModel<TPostable> model, IPostResult<TPostable> result)
        {
            if (GetScriptDocument() is ScriptDocument script)
            {
                // Ignoring 'result', because this is actually the 'Results' field of the script class.
                script.Execute(methodname: "AfterPost", parameters: new object[] { model });
            }
            else
            {
                Poster.AfterPost(model, result);
            }
        }
    }
}