import {type QueryClient} from '@tanstack/vue-query';
import {type Plugin, type Shipment} from '@myparcel-pdk/common';
import {type OneOrMore, toArray} from '@myparcel/ts-utils';
import {setQueryOrder, setQueryShipment} from '../utils';
import {QUERY_KEY_ORDER} from '../actions';

export const fillShipmentsQueryData = (
  queryClient: QueryClient,
  shipments: OneOrMore<Shipment.ModelShipment>,
  order?: Plugin.ModelPdkOrder,
): void => {
  const shipmentsArray = toArray(shipments);
  const orderIds = new Set(shipmentsArray.map((shipment) => shipment.orderId));

  if (order) {
    orderIds.add(order.externalIdentifier);
  }

  orderIds.forEach((orderId) => {
    let newOrder = order;

    if (!newOrder) {
      const queryOrder = queryClient.getQueryData([QUERY_KEY_ORDER, {id: orderId}]) as Plugin.ModelPdkOrder;

      if (!queryOrder) {
        return;
      }

      newOrder = {
        ...queryOrder,
        shipments: queryOrder.shipments.map((shipment) => ({
          ...shipment,
          updated: shipmentsArray.find((item) => item.id === shipment.id)?.updated ?? shipment.updated,
        })),
      };
    }

    setQueryOrder(queryClient, newOrder);
  });

  shipmentsArray.forEach((shipment) => {
    setQueryShipment(queryClient, shipment);
  });
};