[legacy-framework] allow useQuery to run during SSR or SSG

packages/config/src/index.ts

@@ -74,7 +74,7 @@ export interface BlitzConfig extends Record<string, unknown> {
   target?: string
   experimental?: {
     isomorphicResolverImports?: boolean
-    reactMode?: string
+    reactRoot?: boolean
   }
   cli?: {
     clearConsoleOnBlitzDev?: boolean

packages/core/src/blitz-data.tsx

@@ -12,7 +12,7 @@ export function _getBlitzRuntimeData(): BlitzRuntimeData {
   const config = getConfig()
   return {
     sessionCookiePrefix: (config._meta.packageName || "blitz").replace(/[^a-zA-Z0-9-_]/g, "_"),
-    suspenseEnabled: config.experimental?.reactMode !== "legacy",
+    suspenseEnabled: config.experimental?.reactRoot !== false,
   }
 }
 

packages/core/src/use-query-hooks.ts

@@ -7,9 +7,10 @@ import {
   UseQueryResult,
 } from "react-query"
 import {useSession} from "./auth/auth-client"
+import {getBlitzRuntimeData} from "./blitz-data"
 import {useRouter} from "./router"
 import {FirstParam, PromiseReturnType, QueryFn} from "./types"
-import {isClient} from "./utils"
+import {isServer} from "./utils"
 import {
   emptyQueryFn,
   getQueryCacheFunctions,
@@ -49,14 +50,15 @@ export function useQuery<T extends QueryFn, TResult = PromiseReturnType<T>>(
     throw new Error("useQuery is missing the first argument - it must be a query function")
   }
 
-  const suspense =
-    options?.enabled === false || options?.enabled === null ? false : options?.suspense
+  const suspenseConfig = getBlitzRuntimeData().suspenseEnabled
+  let enabled = isServer && suspenseConfig ? false : options?.enabled ?? options?.enabled !== null
+  const suspense = enabled === false ? false : options?.suspense
   const session = useSession({suspense})
   if (session.isLoading) {
-    options.enabled = false
+    enabled = false
   }
 
-  const routerIsReady = useRouter().isReady && isClient
+  const routerIsReady = useRouter().isReady || (isServer && suspenseConfig)
   const enhancedResolverRpcClient = sanitizeQuery(queryFn)
   const queryKey = getQueryKey(queryFn, params)
 
@@ -66,8 +68,20 @@ export function useQuery<T extends QueryFn, TResult = PromiseReturnType<T>>(
       ? () => enhancedResolverRpcClient(params, {fromQueryHook: true})
       : (emptyQueryFn as any),
     ...options,
+    enabled,
   })
 
+  if (
+    queryRest.isIdle &&
+    isServer &&
+    suspenseConfig !== false &&
+    !data &&
+    (!options || !("suspense" in options) || options.suspense) &&
+    (!options || !("enabled" in options) || options.enabled)
+  ) {
+    throw new Promise(() => {})
+  }
+
   const rest = {
     ...queryRest,
     ...getQueryCacheFunctions<FirstParam<T>, TResult, T>(queryFn, params),

test/integration/no-suspense/blitz.config.js

@@ -1,5 +1,5 @@
 module.exports = {
   experimental: {
-    reactMode: "legacy",
+    reactRoot: false,
   },
 }