UPS live rates

import { enrichItemDetails } from "./modules.js"; // Enriches item data with metafields like dimensions

// 🔐 Generate a UPS OAuth token using Basic Authorization headers
async function getToken(env) {
    const formData = {
        grant_type: "client_credentials"
    };

    const url = "https://onlinetools.ups.com/security/v1/oauth/token";

    const resp = await fetch(url, {
        method: "POST",
        headers: {
            "Content-Type": "application/x-www-form-urlencoded",
            "x-merchant-id": env.UPS_SHIPPER_NUMBER, // Required UPS field
            "Authorization": "Basic " + btoa(`${env.UPS_CLIENT_ID}:${env.UPS_CLIENT_SECRET}`) // Base64(client_id:client_secret)
        },
        body: new URLSearchParams(formData).toString()
    });

    const data = await resp.json();
    return data.access_token;
}

// 🕒 Add business days (skipping weekends)
function addBusinessDays(numDays) {
    const currentDate = new Date();
    let daysAdded = 0;

    while (daysAdded < numDays) {
        currentDate.setDate(currentDate.getDate() + 1);
        const day = currentDate.getDay();
        if (day !== 0 && day !== 6) daysAdded++; // Skip Sunday (0) and Saturday (6)
    }

    const year = currentDate.getFullYear();
    const month = String(currentDate.getMonth() + 1).padStart(2, "0");
    const day = String(currentDate.getDate()).padStart(2, "0");

    return `${year}${month}${day}`; // Format: YYYYMMDD
}

// 📆 Format UPS date/time into Shopify ISO format
function formatToShopifyMaxDeliveryDate(date, time) {
    const year = date.substring(0, 4);
    const month = date.substring(4, 6);
    const day = date.substring(6, 8);
    return `${year}-${month}-${day}`; // YYYY-MM-DD (Shopify-friendly)
}

// 🚚 Main shipping rate calculator
export async function calculateShippingRates(DATA, env) {
    try {
        // 1. Enrich item dimensions via metafields
        DATA = await enrichItemDetails(DATA, [{
            namespace: "custom",
            size: 10
        }]);
        
        const token = await getToken(env); // 2. Get UPS OAuth token
        console.log("token",token);

        const url = `https://onlinetools.ups.com/api/rating/v2409/shop?additionalinfo=timeintransit`;

        const destination = DATA.destination;
        const origin = DATA.origin;

        // 3. Prepare box/package data for each item
        const boxes = DATA.items.map(item => {
            const dims = item.metafields?.custom || {};
            const weightLbs = (item?.grams || 100) / 453.592; // Convert grams to pounds
            return {
                totalWeightInPounds: weightLbs,
                dimensions: {
                    length: Number(dims.length || 10),
                    width: Number(dims.width || 10),
                    height: Number(dims.height || 10)
                }
            };
        });

        // 4. Calculate total shipment weight
        const totalWeight = boxes.reduce((sum, b) => sum + b.totalWeightInPounds, 0);
        const ShipmentTotalWeight = {
            UnitOfMeasurement: {
                Code: "LBS",
                Description: "Pounds"
            },
            Weight: totalWeight.toFixed(2)
        };

        // 5. Construct package list for UPS API
        const packages = boxes.map(box => ({
            PackagingType: {
                Code: "02",
                Description: "Package"
            },
            PackageWeight: {
                UnitOfMeasurement: {
                    Code: "LBS",
                    Description: "Pounds"
                },
                Weight: box.totalWeightInPounds.toFixed(2)
            },
            Dimensions: {
                UnitOfMeasurement: {
                    Code: "IN",
                    Description: "Inches"
                },
                Length: box.dimensions.length.toString(),
                Width: box.dimensions.width.toString(),
                Height: box.dimensions.height.toString()
            }
        }));

        const pickupDate = addBusinessDays(2); // Assume 2 business days prep time

        // 6. Construct UPS API request payload
        const requestBody = {
            RateRequest: {
                Request: {
                    TransactionReference: {
                        CustomerContext: "CustomerContext"
                    }
                },
                PickupType: {
                    Code: "01", // Daily pickup
                    Description: "Daily Pickup"
                },
                CustomerClassification: {
                    Code: "00", // Rates Associated with Shipper Number
                    Description: "Rates Associated with Shipper Number"
                },
                Shipment: {
                    Shipper: {
                        Name: "Store",
                        ShipperNumber: env.UPS_SHIPPER_NUMBER,
                        Address: {
                            AddressLine: [origin.address1],
                            City: origin.city,
                            StateProvinceCode: origin.province,
                            PostalCode: origin.postal_code.replace(/\s/g, ""),
                            CountryCode: origin.country
                        }
                    },
                    ShipTo: {
                        Name: "ShipToName",
                        Address: {
                            AddressLine: [destination.address1],
                            City: destination.city,
                            StateProvinceCode: destination.province,
                            PostalCode: destination.postal_code.replace(/\s/g, ""),
                            CountryCode: destination.country
                        }
                    },
                    ShipFrom: {
                        Name: "Store",
                         Address: {
                            AddressLine: [origin.address1],
                            City: origin.city,
                            StateProvinceCode: origin.province,
                            PostalCode: origin.postal_code.replace(/\s/g, ""),
                            CountryCode: origin.country
                        }
                    },
                    PaymentDetails: {
                        ShipmentCharge: [{
                            Type: "01", // Bill to Shipper
                            BillShipper: {
                                AccountNumber: env.UPS_SHIPPER_NUMBER
                            }
                        }]
                    },
                    ShipmentRatingOptions: {
                        TPFCNegotiatedRatesIndicator: "Y",
                        NegotiatedRatesIndicator: "Y"
                    },
                    ShipmentTotalWeight,
                    NumOfPieces: boxes.length.toString(),
                    Package: packages,
                    RatingMethodRequestedIndicator: "Y",
                    DeliveryTimeInformation: {
                        PackageBillType: "03", //  Non-Document
                        Pickup: {
                            Date: pickupDate
                        }
                    }
                }
            }
        };
        
        console.log("requestBody",requestBody);

        // 7. Fetch UPS rates with transit info
        const response = await fetch(url, {
            method: "POST",
            headers: {
                "Content-Type": "application/json",
                transId: "transactionId123", // Arbitrary transaction ID
                transactionSrc: "Shopify", // Source ID for tracking
                Authorization: `Bearer ${token}`
            },
            body: JSON.stringify(requestBody)
        });

        const data = await response.json();
        
         console.log("response data",data);
         
        const ups_service_codes = ["03","01","02","07","54","65","11","08"]; // Added international codes
        const service_names = ["UPS® Ground","UPS Next Day Air®","UPS 2nd Day Air®","UPS Worldwide Express®","UPS Worldwide Express Plus®","UPS Worldwide Saver®","UPS Standard®","UPS Worldwide Expedited®"];
        const shopify_service_codes = ["UPS_Ground","UPS_Next_Day_Air","UPS_2nd_Day_Air","UPS_Worldwide_Express","UPS_Worldwide_Express_Plus","UPS_Worldwide_Saver","UPS_Standard","UPS_Worldwide_Expedited"];

        const shipments = data?.RateResponse?.RatedShipment || [];
        const filtered= shipments.filter(shipment => ups_service_codes.includes(shipment.Service.Code) );


        // 8. Transform into Shopify rate objects
        const rates = filtered.map(svc => {
            const ups_code = svc?.Service?.Code;
            const code_index = ups_service_codes.indexOf(ups_code);
            
            const arrival = svc?.TimeInTransit?.ServiceSummary?.EstimatedArrival?.Arrival || {};
            const deliveryDate = arrival.Date && arrival.Time ? formatToShopifyMaxDeliveryDate(arrival.Date, arrival.Time) : "";
            
            let description = "";

            const arrivalDays = svc?.GuaranteedDelivery || svc?.EstimatedArrival;
    
            if (arrivalDays?.BusinessDaysInTransit) {
              description += `Estimated to arrive by ${arrivalDays.BusinessDaysInTransit} business days`;
            }

            return {
                service_name: service_names[code_index],
                service_code: shopify_service_codes[code_index],
                total_price: ( parseFloat(svc?.NegotiatedRateCharges?.TotalCharge?.MonetaryValue) * 100 ).toFixed(0),
                currency: DATA.currency,
                description,
                min_delivery_date: deliveryDate,
                max_delivery_date: deliveryDate
            };
        });

        return {
            rates
        };
    } catch (error) {
        console.error("UPS Rate Error:", error.message);
        return {
            rates: []
        };
    }
}