@@ -21,16 +21,14 @@ import {
2121} from '@/components/v1/ui/tooltip' ;
2222import { useCurrentTenantId , useTenantDetails } from '@/hooks/use-tenant' ;
2323import { queries } from '@/lib/api' ;
24- import { controlPlaneApi } from '@/lib/api/api' ;
24+ import { cloudApi } from '@/lib/api/api' ;
2525import {
2626 TenantSubscription ,
2727 SubscriptionPlan ,
2828 SubscriptionPlanCode ,
2929 SubscriptionPeriod ,
3030 Coupon ,
31- UpdateTenantSubscriptionResponse ,
32- } from '@/lib/api/generated/control-plane/data-contracts' ;
33- import { ContentType } from '@/lib/api/generated/control-plane/http-client' ;
31+ } from '@/lib/api/generated/cloud/data-contracts' ;
3432import { useApiError } from '@/lib/hooks' ;
3533import queryClient from '@/query-client' ;
3634import { useMutation , useQuery } from '@tanstack/react-query' ;
@@ -68,7 +66,7 @@ export const Subscription: React.FC<SubscriptionProps> = ({
6866 const { handleApiError } = useApiError ( { } ) ;
6967 const [ portalLoading , setPortalLoading ] = useState ( false ) ;
7068 const creditBalanceQuery = useQuery ( {
71- ...queries . controlPlane . creditBalance ( tenantId ) ,
69+ ...queries . cloud . creditBalance ( tenantId ) ,
7270 } ) ;
7371
7472 const creditBalance = useMemo ( ( ) => {
@@ -112,12 +110,7 @@ export const Subscription: React.FC<SubscriptionProps> = ({
112110 return ;
113111 }
114112 setPortalLoading ( true ) ;
115- const link = await controlPlaneApi . request < { url ?: string } > ( {
116- path : `/api/v1/control-plane/billing/tenants/${ tenantId } /billing-portal-link` ,
117- method : 'GET' ,
118- secure : true ,
119- format : 'json' ,
120- } ) ;
113+ const link = await cloudApi . billingPortalLinkGet ( tenantId ) ;
121114 window . open ( link . data . url , '_blank' ) ;
122115 } catch ( e ) {
123116 handleApiError ( e as any ) ;
@@ -131,22 +124,14 @@ export const Subscription: React.FC<SubscriptionProps> = ({
131124 mutationFn : async ( { plan_code } : { plan_code : string } ) => {
132125 const [ plan , period ] = plan_code . split ( '_' ) ;
133126 setLoading ( plan_code ) ;
134- const response =
135- await controlPlaneApi . request < UpdateTenantSubscriptionResponse > ( {
136- path : `/api/v1/control-plane/billing/tenants/${ tenantId } /subscription` ,
137- method : 'PATCH' ,
138- body : {
139- plan : plan as SubscriptionPlanCode ,
140- period : period as SubscriptionPeriod ,
141- } ,
142- secure : true ,
143- type : ContentType . Json ,
144- format : 'json' ,
145- } ) ;
127+ const response = await cloudApi . tenantSubscriptionUpdate ( tenantId , {
128+ plan : plan as SubscriptionPlanCode ,
129+ period : period as SubscriptionPeriod ,
130+ } ) ;
146131 return response . data ;
147132 } ,
148133 onSuccess : async ( data ) => {
149- if ( data ?. checkoutUrl ) {
134+ if ( data && ' checkoutUrl' in data ) {
150135 window . location . href = data . checkoutUrl ;
151136 return ;
152137 }
@@ -156,7 +141,7 @@ export const Subscription: React.FC<SubscriptionProps> = ({
156141 queryKey : queries . tenantResourcePolicy . get ( tenantId ) . queryKey ,
157142 } ) ,
158143 queryClient . invalidateQueries ( {
159- queryKey : queries . controlPlane . billing ( tenantId ) . queryKey ,
144+ queryKey : queries . cloud . billing ( tenantId ) . queryKey ,
160145 } ) ,
161146 ] ) ;
162147
0 commit comments