import {type QueryClient} from '@tanstack/vue-query';
import {type Plugin, type Shipment} from '@myparcel-pdk/common';
import {isOfType, type Replace} from '@myparcel/ts-utils';
import {QUERY_KEY_ORDER} from '../../actions';
import {setQueryData} from './setQueryData';

type FilteredShipment = {id: number; updated: unknown};

type FilteredOrder = Replace<Plugin.ModelPdkOrder, 'shipments', FilteredShipment[]>;

export const setQueryOrder = (queryClient: QueryClient, order: Plugin.ModelPdkOrder | FilteredOrder): void => {
  let filteredOrder: FilteredOrder;

  if (order.shipments?.length && isOfType<Shipment.ModelShipment>(order.shipments[0], 'orderId')) {
    filteredOrder = {
      ...order,
      shipments: (order as Plugin.ModelPdkOrder).shipments
        .filter((shipment) => !shipment.deleted)
        .map(({id, updated}) => ({id, updated})),
    };
  } else {
    filteredOrder = order;
  }

  setQueryData(queryClient, [QUERY_KEY_ORDER, {id: order.externalIdentifier}], filteredOrder);
};