dify/web/service/use-billing.ts

import { useMutation, useQuery } from '@tanstack/react-query'
import { consoleClient, consoleQuery } from '@/service/client'
import { fetchCurrentPlanVectorSpace } from './billing'

const currentPlanVectorSpaceQueryKey = ['billing', 'current-plan-vector-space'] as const

export const useBindPartnerStackInfo = () => {
  return useMutation({
    mutationKey: consoleQuery.billing.partners.byPartnerKey.tenants.put.mutationKey(),
    mutationFn: (data: { partnerKey: string; clickId: string }) =>
      consoleClient.billing.partners.byPartnerKey.tenants.put({
        params: { partner_key: data.partnerKey },
        body: { click_id: data.clickId },
      }),
  })
}

export const useBillingUrl = (enabled: boolean) => {
  return useQuery({
    queryKey: consoleQuery.billing.invoices.get.queryKey(),
    enabled,
    queryFn: async () => {
      const res = await consoleClient.billing.invoices.get()
      return res.url
    },
  })
}

export const useCurrentPlanVectorSpace = (enabled = true) => {
  return useQuery({
    queryKey: currentPlanVectorSpaceQueryKey,
    queryFn: () => fetchCurrentPlanVectorSpace(),
    enabled,
  })
}