Fix missing _app in AppTree rendering

vercel · Apr 15, 2022 · 95d7b5a · 95d7b5a
1 parent 9c7311b
commit 95d7b5a
Showing 1 changed file with 9 additions and 5 deletions.
diff --git a/packages/next/server/render.tsx b/packages/next/server/render.tsx
@@ -197,17 +197,21 @@ function enhanceComponents(
 
 function renderFlight(AppMod: any, ComponentMod: any, props: any) {
   const isServerComponent = !!ComponentMod.__next_rsc__
+  const isServerApp = !!AppMod.__next_rsc__
   const App = interopDefault(AppMod)
   const Component = interopDefault(ComponentMod)
   const AppServer = isServerComponent
     ? (App as React.ComponentType)
     : React.Fragment
 
-  return (
-    <AppServer>
-      <Component {...props} />
-    </AppServer>
-  )
+  if (isServerApp) {
+    return (
+      <AppServer>
+        <Component {...props} />
+      </AppServer>
+    )
+  }
+  return <App Component={Component} {...props} />
 }
 
 export type RenderOptsPartial = {