fetchImages.tsx

/*
 * SPDX-FileCopyrightText: 2025 SAP SE or an SAP affiliate company and Juno contributors
 * SPDX-License-Identifier: Apache-2.0
 */

import { ObservableQuery } from "@apollo/client"
import { GetImagesDocument, GetImagesQuery, ImageFilter, VulnerabilityFilter } from "../generated/graphql"
import { RouteContext } from "../routes/-types"

type FetchImagesParams = Pick<RouteContext, "queryClient" | "apiClient"> & {
  filter: ImageFilter
  after?: string | null
  first?: number
  firstVulnerabilities?: number
  afterVulnerabilities?: string | null
  firstVersions?: number
  afterVersions?: string | null
  vulFilter?: VulnerabilityFilter
}

export const fetchImages = ({
  queryClient,
  apiClient,
  filter,
  after,
  first = 10,
  firstVulnerabilities,
  afterVulnerabilities,
  firstVersions,
  afterVersions,
  vulFilter,
}: FetchImagesParams): Promise<ObservableQuery.Result<GetImagesQuery>> => {
  return queryClient.ensureQueryData({
    queryKey: [
      "images",
      filter,
      after,
      firstVulnerabilities,
      afterVulnerabilities,
      firstVersions,
      afterVersions,
      vulFilter,
    ],
    staleTime: 2.5 * 60 * 1000,
    queryFn: () =>
      apiClient.query<GetImagesQuery>({
        query: GetImagesDocument,
        fetchPolicy: "network-only",
        variables: {
          imgFilter: filter,
          vulFilter,
          first,
          after,
          firstVulnerabilities,
          afterVulnerabilities,
          firstVersions,
          afterVersions,
        },
      }),
  })
}