diff --git a/src/ageAssurance/data.tsx b/src/ageAssurance/data.tsx index cfea297f67..2839d2b488 100644 --- a/src/ageAssurance/data.tsx +++ b/src/ageAssurance/data.tsx @@ -183,7 +183,7 @@ export async function getServerState({agent}: {agent: AtpAgent}) { const geolocation = device.get(['mergedGeolocation']) if (!geolocation || !geolocation.countryCode) { logger.error(`getServerState: missing geolocation countryCode`) - return + return null } const {data} = await agent.app.bsky.ageassurance.getState({ countryCode: geolocation.countryCode, @@ -225,7 +225,9 @@ export async function prefetchServerState({agent}: {agent: AtpAgent}) { try { logger.debug(`prefetchServerState: resolving...`) const res = await networkRetry(3, () => getServerState({agent})) - qc.setQueryData(qk, res) + if (res) { + qc.setQueryData(qk, res) + } } catch (err) { const e = err as Error logger.warn(`prefetchServerState: failed`, { @@ -238,10 +240,12 @@ export async function refetchServerState({agent}: {agent: AtpAgent}) { if (!did) return logger.debug(`refetchServerState: fetching...`) const res = await networkRetry(3, () => getServerState({agent})) - qc.setQueryData( - createServerStateQueryKey({did}), - res, - ) + if (res) { + qc.setQueryData( + createServerStateQueryKey({did}), + res, + ) + } return res } export function usePatchServerState() {