prs/prs.shared/Posters/Timberline/PurchaseOrderTimberlinePoster.cs

using Comal.Classes;
using CsvHelper.Configuration.Attributes;
using CsvHelper;
using InABox.Core.Postable;
using InABox.Core;
using InABox.Poster.Timberline;
using InABox.Scripting;
using System;
using System.Collections.Generic;
using System.Globalization;
using System.IO;
using System.Linq;
using System.Text;
using System.Threading.Tasks;
using Microsoft.Win32;
using CsvHelper.TypeConversion;
using CsvHelper.Configuration;
using System.Reflection;
using System.Windows;

namespace PRS.Shared
{
    public enum PurchaseOrderTimberlineCommitmentType
    {
        Subcontract = 1,
        PO = 2
    }

    public class POTimberlineCommitmentTypeConverter : DefaultTypeConverter
    {
        public override object? ConvertFromString(string? text, IReaderRow row, MemberMapData memberMapData)
        {
            if(Enum.TryParse<PurchaseOrderTimberlineCommitmentType>(text, out var type))
            {
                return type;
            }
            return base.ConvertFromString(text, row, memberMapData);
        }

        public override string? ConvertToString(object? value, IWriterRow row, MemberMapData memberMapData)
        {
            if(value is PurchaseOrderTimberlineCommitmentType type)
            {
                return ((int)type).ToString();
            }
            return "";
        }
    }

    public class PurchaseOrderTimberlineHeader
    {
        [Ignore]
        public List<PurchaseOrderTimberlineLine> Lines { get; set; } = new();

        [Index(0)]
        public string RecordID { get; set; } = "C";

        [Index(1)]
        public string CommitmentID { get; set; }

        [Index(2)]
        [TypeConverter(typeof(POTimberlineCommitmentTypeConverter))]
        public PurchaseOrderTimberlineCommitmentType CommitmentType { get; set; }

        [Index(3)]
        public string Description { get; set; }

        [Index(4)]
        public string VendorID { get; set; }

        [Index(5)]
        public DateTime Date { get; set; }

        [Index(6)]
        public double RetainagePercent { get; set; }

        [Index(7)]
        public bool CommittedToJC { get; set; }

        [Index(8)]
        public bool Closed { get; set; }

        [Index(9)]
        public bool Printed { get; set; }

        /// <summary>
        /// Dictionary of extra fields to write; the key is the 0-based index of the column in which we are writing.
        /// If the index is that of any of the explicit fields of this class, nothing happens.
        /// </summary>
        [Ignore]
        public Dictionary<int, string> AdditionalFields { get; set; } = new();
    }

    public class PurchaseOrderTimberlineLine
    {
        [Index(0)]
        public string RecordID { get; set; } = "CI";

        [Index(1)]
        public string CommitmentID { get; set; }

        [Index(2)]
        public int ItemNumber { get; set; }

        [Index(3)]
        public string Description { get; set; }

        [Index(4)]
        public double RetainagePercent { get; set; }

        [Index(5)]
        public DateTime DeliveryDate { get; set; }

        [Index(6)]
        public string ScopeOfWork { get; set; }

        [Index(7)]
        public string Job { get; set; }

        [Index(8)]
        public string Extra { get; set; }

        [Index(9)]
        public string CostCode { get; set; }

        [Index(10)]
        public string Category { get; set; }

        [Index(11)]
        public string TaxGroup { get; set; }

        [Index(12)]
        public string Tax { get; set; }

        [Index(13)]
        public double Units { get; set; }

        [Index(14)]
        public double UnitCost { get; set; }

        [Index(15)]
        public string UnitDescription { get; set; }

        [Index(16)]
        public double Amount { get; set; }

        [Index(17)]
        public bool BoughtOut { get; set; }

        /// <summary>
        /// Dictionary of extra fields to write; the key is the 0-based index of the column in which we are writing.
        /// If the index is that of any of the explicit fields of this class, nothing happens.
        /// </summary>
        [Ignore]
        public Dictionary<int, string> AdditionalFields { get; set; } = new();
    }

    public class PurchaseOrderTimberlineSettings : TimberlinePosterSettings<PurchaseOrder>
    {
        protected override string DefaultScript()
        {
            return @"
using PRS.Shared;
using InABox.Core;
using System.Collections.Generic;

public class Module
{
    public void BeforePost(IDataModel<PurchaseOrder> model)
    {
        // Perform pre-processing
    }

    public void ProcessHeader(IDataModel<PurchaseOrder> model, PurchaseOrder purchaseOrder, PurchaseOrderTimberlineHeader header)
    {
        // Do extra processing for a purchase order
    }

    public void ProcessLine(IDataModel<PurchaseOrder> model, PurchaseOrderItem purchaseOrderItem, PurchaseOrderTimberlineLine line)
    {
        // Do extra processing for a purchase order line
    }

    public void AfterPost(IDataModel<PurchaseOrder> model)
    {
        // Perform post-processing
    }
}";
        }
    }

    public class PurchaseOrderTimberlinePoster : ITimberlinePoster<PurchaseOrder, PurchaseOrderTimberlineSettings>
    {
        public ScriptDocument? Script { get; set; }

        public event ITimberlinePoster<PurchaseOrder, PurchaseOrderTimberlineSettings>.AddFragmentCallback? AddFragment;

        public bool BeforePost(IDataModel<PurchaseOrder> model)
        {
            model.SetIsDefault<Document>(false, alias: "CompanyLogo");
            model.SetIsDefault<CoreTable>(false, alias: "CompanyInformation");
            model.SetIsDefault<Employee>(false);

            model.SetIsDefault<PurchaseOrderItem>(true, alias: "PurchaseOrder_PurchaseOrderItem");

            model.SetColumns(new Columns<PurchaseOrder>(x => x.ID)
                .Add(x => x.PONumber)
                .Add(x => x.Description)
                .Add(x => x.SupplierLink.Code)
                .Add(x => x.IssuedDate)
                .Add(x => x.ClosedDate));
            model.SetColumns(new Columns<PurchaseOrderItem>(x => x.ID)
                .Add(x => x.PurchaseOrderLink.ID)
                .Add(x => x.PostedReference)
                .Add(x => x.Description)
                .Add(x => x.ReceivedDate)
                .Add(x => x.Job.JobNumber)
                .Add(x => x.TaxCode.Code)
                .Add(x => x.Qty)
                .Add(x => x.Cost)
                .Add(x => x.Dimensions.UnitSize)
                .Add(x => x.IncTax),
                alias: "PurchaseOrder_PurchaseOrderItem");

            Script?.Execute(methodname: "BeforePost", parameters: new object[] { model });
            return true;
        }

        private void ProcessHeader(IDataModel<PurchaseOrder> model, PurchaseOrder bill, PurchaseOrderTimberlineHeader header)
        {
            Script?.Execute(methodname: "ProcessHeader", parameters: new object[] { model, bill, header });
        }
        private void ProcessLine(IDataModel<PurchaseOrder> model, PurchaseOrderItem purchaseOrderItem, PurchaseOrderTimberlineLine line)
        {
            Script?.Execute(methodname: "ProcessLine", parameters: new object[] { model, purchaseOrderItem, line });
        }

        private List<PurchaseOrderTimberlineHeader> DoProcess(IDataModel<PurchaseOrder> model)
        {
            List<IPostableFragment> Fragments = new List<IPostableFragment>();

            var cs = new List<PurchaseOrderTimberlineHeader>();

            var lines = model.GetTable<PurchaseOrderItem>("PurchaseOrder_PurchaseOrderItem").ToObjects<PurchaseOrderItem>()
                .GroupBy(x => x.PurchaseOrderLink.ID).ToDictionary(x => x.Key, x => x.ToList());
            foreach (var purchaseOrder in model.GetTable<PurchaseOrder>().ToObjects<PurchaseOrder>())
            {
                var c = new PurchaseOrderTimberlineHeader
                {
                    CommitmentID = purchaseOrder.PONumber,
                    CommitmentType = PurchaseOrderTimberlineCommitmentType.PO,
                    Description = purchaseOrder.Description,
                    VendorID = purchaseOrder.SupplierLink.Code,
                    Date = purchaseOrder.IssuedDate,
                    // RetainagePercent
                    // Committed to JC
                    Closed = purchaseOrder.ClosedDate != DateTime.MinValue,
                    // Printed
                };
                ProcessHeader(model, purchaseOrder, c);

                // Dictionary from line number to POItem.
                var items = new Dictionary<int, PurchaseOrderItem>();
                var POItems = lines.GetValueOrDefault(purchaseOrder.ID)?.ToList() ?? new List<PurchaseOrderItem>();
                foreach (var purchaseOrderItem in POItems)
                {
                    if(int.TryParse(purchaseOrderItem.PostedReference, out var itemNumber))
                    {
                        if (items.TryGetValue(itemNumber, out var oldItem))
                        {
                            // Theoretically shouldn't happen, but just in case.
                            MessageBox.Show($"Warning: Multiple PurchaseOrder Items have the same line number for export; the line number for '{purchaseOrderItem.Description}' will be changed in the export.");
                            Logger.Send(LogType.Error, "", $"Purchase Order Post: Multiple POItems with the same Line Number; changing line number of POItem {purchaseOrderItem.ID}");
                            purchaseOrderItem.PostedReference = "";
                        }
                        else
                        {
                            items[itemNumber] = purchaseOrderItem;
                        }
                    }
                }

                foreach (var purchaseOrderItem in POItems)
                {
                    if (!int.TryParse(purchaseOrderItem.PostedReference, out var itemNumber))
                    {
                        itemNumber = 1;
                        while(items.ContainsKey(itemNumber))
                        {
                            ++itemNumber;
                        }

                        items[itemNumber] = purchaseOrderItem;
                        purchaseOrderItem.PostedReference = itemNumber.ToString();
                    }
                    var ci = new PurchaseOrderTimberlineLine
                    {
                        CommitmentID = purchaseOrder.PONumber,
                        ItemNumber = itemNumber,
                        Description = purchaseOrderItem.Description,
                        // RetainagePercent = ,
                        DeliveryDate = purchaseOrderItem.ReceivedDate,
                        //ScopeOfWork
                        Job = purchaseOrderItem.Job.JobNumber,
                        //Extra = purchaseOrderItem.Job
                        //CostCode = purchaseOrderItem.c
                        //Category = purchaseOrderItem.cat
                        TaxGroup = purchaseOrderItem.TaxCode.Code,
                        Units = purchaseOrderItem.Qty,
                        UnitCost = purchaseOrderItem.Cost,
                        UnitDescription = purchaseOrderItem.Dimensions.UnitSize,
                        Amount = purchaseOrderItem.IncTax,
                        // BoughtOut
                    };

                    ProcessLine(model, purchaseOrderItem, ci);
                    c.Lines.Add(ci);

                    AddFragment?.Invoke(purchaseOrderItem);
                }
                cs.Add(c);
            }
            return cs;
        }

        public bool Process(IDataModel<PurchaseOrder> model)
        {
            var POs = DoProcess(model);

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

            if (dlg.ShowDialog() == true)
            {
                using var writer = new StreamWriter(dlg.FileName);
                using var csv = new CsvWriter(writer, CultureInfo.InvariantCulture);
                foreach (var po in POs)
                {
                    // Write the record.
                    csv.WriteRecord(po);
                    
                    // Current 0-based index that the writer is at.
                    int i = csv.Index;
                    foreach(var (index, field) in po.AdditionalFields.OrderBy(x => x.Key))
                    {
                        while(i < index)
                        {
                            csv.WriteField("");
                            ++i;
                        }
                        csv.WriteField(field);
                        ++i;
                    }
                    csv.NextRecord();
                    foreach (var poi in po.Lines)
                    {
                        csv.WriteRecord(poi);

                        // Current 0-based index that the writer is at.
                        i = csv.Index;
                        foreach (var (index, field) in poi.AdditionalFields.OrderBy(x => x.Key))
                        {
                            while (i < index)
                            {
                                csv.WriteField("");
                                ++i;
                            }
                            csv.WriteField(field);
                            ++i;
                        }
                        csv.NextRecord();
                    }
                }
                return true;
            }
            else
            {
                throw new PostCancelledException();
            }
        }
        public void AfterPost(IDataModel<PurchaseOrder> model)
        {
            Script?.Execute(methodname: "AfterPost", parameters: new object[] { model });
        }
    }

    public class PurchaseOrderTimberlinePosterEngine<T> : TimberlinePosterEngine<PurchaseOrder, PurchaseOrderTimberlineSettings>
    {
    }
}