using Comal.Classes;

namespace Comal.Stores
{
    public class PurchaseOrderStore : BaseStore<PurchaseOrder>
    {
        protected override void AfterSave(PurchaseOrder entity)
        {
            base.AfterSave(entity);

            UpdateTrackingKanban<PurchaseOrderKanban, PurchaseOrder, PurchaseOrderLink>(entity, p =>
            {
                return p.Status == PurchaseOrderStatus.Closed
                    ? KanbanCategory.Complete
                    : p.Status.Equals(PurchaseOrderStatus.Issued)
                        ? KanbanCategory.Waiting
                        : KanbanCategory.Open;
            });
        }

        protected override void BeforeDelete(PurchaseOrder entity)
        {
            UnlinkTrackingKanban<PurchaseOrderKanban, PurchaseOrder, PurchaseOrderLink>(entity);
        }
    }
}