Mango.Nop.Plugins/Nop.Plugin.Misc.AIPlugin/Factories/CustomOrderModelFactory.cs

using AyCode.Core.Extensions;
using FruitBank.Common.Interfaces;
using Microsoft.AspNetCore.Mvc.Infrastructure;
using Microsoft.AspNetCore.Mvc.Routing;
using Microsoft.AspNetCore.Mvc.TagHelpers;
using Nop.Core;
using Nop.Core.Domain.Catalog;
using Nop.Core.Domain.Common;
using Nop.Core.Domain.Directory;
using Nop.Core.Domain.Orders;
using Nop.Core.Domain.Shipping;
using Nop.Core.Domain.Tax;
using Nop.Plugin.Misc.FruitBankPlugin.Helpers;
using Nop.Plugin.Misc.FruitBankPlugin.Models;
using Nop.Plugin.Misc.FruitBankPlugin.Services;
using Nop.Services.Affiliates;
using Nop.Services.Catalog;
using Nop.Services.Common;
using Nop.Services.Configuration;
using Nop.Services.Customers;
using Nop.Services.Directory;
using Nop.Services.Discounts;
using Nop.Services.Helpers;
using Nop.Services.Localization;
using Nop.Services.Media;
using Nop.Services.Orders;
using Nop.Services.Payments;
using Nop.Services.Security;
using Nop.Services.Seo;
using Nop.Services.Shipping;
using Nop.Services.Stores;
using Nop.Services.Tax;
using Nop.Services.Vendors;
using Nop.Web.Areas.Admin.Factories;
using Nop.Web.Areas.Admin.Models.Orders;
using System.Collections;
using System.Reflection;
using System.Runtime.Intrinsics.Arm;

namespace Nop.Plugin.Misc.FruitBankPlugin.Factories
{
    public class CustomOrderModelFactory : OrderModelFactory
    {
        private readonly IOrderMeasurementService _orderMeasurementService;
        private readonly IGenericAttributeService _genericAttributeService;

        public CustomOrderModelFactory(
            IOrderMeasurementService orderMeasurementService,
            AddressSettings addressSettings,
        CatalogSettings catalogSettings,
        CurrencySettings currencySettings,
        IActionContextAccessor actionContextAccessor,
        IAddressModelFactory addressModelFactory,
        IAddressService addressService,
        IAffiliateService affiliateService,
        IBaseAdminModelFactory baseAdminModelFactory,
        ICountryService countryService,
        ICurrencyService currencyService,
        ICustomerService customerService,
        IDateTimeHelper dateTimeHelper,
        IDiscountService discountService,
        IDownloadService downloadService,
        IEncryptionService encryptionService,
        IGiftCardService giftCardService,
        ILocalizationService localizationService,
        IMeasureService measureService,
        IOrderProcessingService orderProcessingService,
        IOrderReportService orderReportService,
        IOrderService orderService,
        IPaymentPluginManager paymentPluginManager,
        IPaymentService paymentService,
        IPictureService pictureService,
        IPriceCalculationService priceCalculationService,
        IPriceFormatter priceFormatter,
        IProductAttributeService productAttributeService,
        IProductService productService,
        IReturnRequestService returnRequestService,
        IRewardPointService rewardPointService,
        ISettingService settingService,
        IShipmentService shipmentService,
        IShippingService shippingService,
        IStateProvinceService stateProvinceService,
        IStoreService storeService,
        ITaxService taxService,
        IUrlHelperFactory urlHelperFactory,
        IVendorService vendorService,
        IWorkContext workContext,
        MeasureSettings measureSettings,
        NopHttpClient nopHttpClient,
        OrderSettings orderSettings,
        ShippingSettings shippingSettings,
        IUrlRecordService urlRecordService,
        TaxSettings taxSettings,
        IGenericAttributeService genericAttributeService
        ) : base(addressSettings,
            catalogSettings,
            currencySettings,
            actionContextAccessor,
            addressModelFactory,
            addressService,
            affiliateService,
            baseAdminModelFactory,
            countryService,
            currencyService,
            customerService,
            dateTimeHelper,
            discountService,
            downloadService,
            encryptionService,
            giftCardService,
            localizationService,
            measureService,
            orderProcessingService,
            orderReportService,
            orderService,
            paymentPluginManager,
            paymentService,
            pictureService,
            priceCalculationService,
            priceFormatter,
            productAttributeService,
            productService,
            returnRequestService,
            rewardPointService,
            settingService,
            shipmentService,
            shippingService,
            stateProvinceService,
            storeService,
            taxService,
            urlHelperFactory,
            vendorService,
            workContext,
            measureSettings,
            nopHttpClient,
            orderSettings,
            shippingSettings,
            urlRecordService,
            taxSettings
            )
        {
            _orderMeasurementService = orderMeasurementService;
            _genericAttributeService = genericAttributeService;
        }

        public override async Task<OrderSearchModel> PrepareOrderSearchModelAsync(OrderSearchModel searchModel)
        {
            // let base prepare default model first
            var baseModel = await base.PrepareOrderSearchModelAsync(searchModel);

            //foreach (var order in baseModel.order)


            // create derived/extended instance
            //var extended = new OrderSearchModelExtended();

            // copy all public instance properties from baseModel to extended
            //CopyModelHelper.CopyPublicProperties(baseModel, extended);

            // try to obtain NeedsMeasurement from the incoming searchModel (if it's extended)
            //bool? needsMeasurement = null;
            //var prop = searchModel?.GetType().GetProperty("NeedsMeasurement", BindingFlags.Public | BindingFlags.Instance);
            //if (prop != null && prop.PropertyType == typeof(bool?))
            //{
            //    needsMeasurement = (bool?)prop.GetValue(searchModel);
            //}

            //extended.NeedsMeasurement = needsMeasurement;

            // return the extended object (it's assignable to OrderSearchModel)
            //return extended;
            return baseModel;
        }

        public override async Task<OrderListModel> PrepareOrderListModelAsync(OrderSearchModel searchModel)
            => await base.PrepareOrderListModelAsync(searchModel);

        public async Task<OrderListModelExtended> PrepareOrderListModelExtendedAsync(OrderSearchModel searchModel)
        {
            var orderListModel = await PrepareOrderListModelAsync(searchModel);
            var extendedRows = new List<OrderModelExtended>(orderListModel.RecordsFiltered);

            foreach (var orderModel in orderListModel.Data)
            {
                var orderModelExtended = new OrderModelExtended();
                PropertyHelper.CopyPublicValueTypeProperties(orderModel, orderModelExtended);

                orderModelExtended.IsMeasurable = await ShouldMarkAsNeedsMeasurementAsync(orderModel);
                orderModelExtended.DateOfReceipt = await GetPickupDateTimeAsync(orderModel);

                Console.WriteLine(orderModelExtended.Id);
                extendedRows.Add(orderModelExtended);
            }

            orderListModel.Data = null;

            var orderListModelExtended = orderListModel.ToJson().JsonTo<OrderListModelExtended>();
            orderListModelExtended.Data = extendedRows;

            return orderListModelExtended;
        }

        // example async custom logic
        private async Task<bool> ShouldMarkAsNeedsMeasurementAsync(OrderModel order)
        {
            // TODO: your logic (e.g. check if order has products that need measuring)
            if (order == null)
                return false;

            var fullOrder = await _orderService.GetOrderByIdAsync(order.Id);
            if (fullOrder != null)
            {
                return await _orderMeasurementService.IsPendingMeasurementAsync(fullOrder);
            }
            return await Task.FromResult(false);
        }

        private async Task<DateTime?> GetPickupDateTimeAsync(OrderModel order)
        {           
            DateTime? dateTime = DateTime.MinValue;
            var fullOrder = await _orderService.GetOrderByIdAsync(order.Id);
            if (fullOrder != null)
            {
                dateTime = await _genericAttributeService.GetAttributeAsync<DateTime>(fullOrder, nameof(IOrderDto.DateOfReceipt));
                if(dateTime == DateTime.MinValue || !dateTime.HasValue)
                {
                    dateTime = null;
                }
            }
            return dateTime;
        }
    }
}