diff --git a/src/components/ageAssurance/useInitAgeAssurance.ts b/src/components/ageAssurance/useInitAgeAssurance.ts index 8571a67844..73102477fc 100644 --- a/src/components/ageAssurance/useInitAgeAssurance.ts +++ b/src/components/ageAssurance/useInitAgeAssurance.ts @@ -15,6 +15,7 @@ import { import {isNetworkError} from '#/lib/hooks/useCleanError' import {logger} from '#/logger' import {createAgeAssuranceQueryKey} from '#/state/age-assurance' +import {useGeolocation} from '#/state/geolocation' import {useAgent} from '#/state/session' const APPVIEW = __DEV__ ? DEV_ENV_APPVIEW : PUBLIC_APPVIEW @@ -23,8 +24,15 @@ const APPVIEW_DID = __DEV__ ? DEV_ENV_APPVIEW_DID : PUBLIC_APPVIEW_DID export function useInitAgeAssurance() { const qc = useQueryClient() const agent = useAgent() + const {geolocation} = useGeolocation() return useMutation({ - async mutationFn(props: AppBskyUnspeccedInitAgeAssurance.InputSchema) { + async mutationFn( + props: Omit, + ) { + if (!geolocation?.countryCode) { + throw new Error(`Geolocation not available, cannot init age assurance.`) + } + const { data: {token}, } = await agent.com.atproto.server.getServiceAuth({ @@ -43,7 +51,11 @@ export function useInitAgeAssurance() { */ const {data} = await wait( 2e3, - appView.app.bsky.unspecced.initAgeAssurance(props), + appView.app.bsky.unspecced.initAgeAssurance({ + ...props, + // @ts-expect-error + countryCode: geolocation?.countryCode, + }), ) qc.setQueryData(