From 35968da051f978bf0dd5c5a41d409993f3ceeeee Mon Sep 17 00:00:00 2001
From: Jiachi Liu <inbox@huozhi.im>
Date: Tue, 9 Nov 2021 00:38:24 +0100
Subject: [PATCH] resolve piper on complete shell

---
 packages/next/build/entries.ts                |  1 +
 .../next-middleware-ssr-loader/index.ts       | 43 +++++++++++--------
 packages/next/server/dev/hot-reloader.ts      |  1 +
 3 files changed, 28 insertions(+), 17 deletions(-)

diff --git a/packages/next/build/entries.ts b/packages/next/build/entries.ts
index 3c4feca3e299a..42492d511d853 100644
--- a/packages/next/build/entries.ts
+++ b/packages/next/build/entries.ts
@@ -162,6 +162,7 @@ export function createEntrypoints(
           page,
           absoluteAppPath: pages['/_app'],
           absoluteDocumentPath: pages['/_document'],
+          absoluteErrorPath: pages['/500'] || pages['/_error'],
           absolutePagePath,
           isServerComponent: isFlight,
           buildId,
diff --git a/packages/next/build/webpack/loaders/next-middleware-ssr-loader/index.ts b/packages/next/build/webpack/loaders/next-middleware-ssr-loader/index.ts
index 25f664f31b6fc..810569b6ecd63 100644
--- a/packages/next/build/webpack/loaders/next-middleware-ssr-loader/index.ts
+++ b/packages/next/build/webpack/loaders/next-middleware-ssr-loader/index.ts
@@ -5,6 +5,7 @@ export default async function middlewareRSCLoader(this: any) {
     absolutePagePath,
     absoluteAppPath,
     absoluteDocumentPath,
+    absoluteErrorPath,
     basePath,
     isServerComponent: isServerComponentQuery,
     assetPrefix,
@@ -14,13 +15,15 @@ export default async function middlewareRSCLoader(this: any) {
   const isServerComponent = isServerComponentQuery === 'true'
   const stringifiedAbsolutePagePath = stringifyRequest(this, absolutePagePath)
   const stringifiedAbsoluteAppPath = stringifyRequest(this, absoluteAppPath)
+  const stringifiedAbsoluteErrorPath = stringifyRequest(this, absoluteErrorPath)
   const stringifiedAbsoluteDocumentPath = stringifyRequest(
     this,
     absoluteDocumentPath
   )
 
-  let appDefinition = `const App = require(${stringifiedAbsoluteAppPath}).default`
-  let documentDefinition = `const Document = require(${stringifiedAbsoluteDocumentPath}).default`
+  const appDefinition = `const App = require(${stringifiedAbsoluteAppPath}).default`
+  const documentDefinition = `const Document = require(${stringifiedAbsoluteDocumentPath}).default`
+  const errorDefinition = `const ErrorPage = require(${stringifiedAbsoluteErrorPath}).default`
 
   const transformed = `
         import { adapter } from 'next/dist/server/web/adapter'
@@ -40,6 +43,7 @@ export default async function middlewareRSCLoader(this: any) {
 
         ${appDefinition}
         ${documentDefinition}
+        ${errorDefinition}
         
         const {
           default: Page,
@@ -134,6 +138,7 @@ export default async function middlewareRSCLoader(this: any) {
               : ''
           }
 
+          const req = { url: pathname }
           const renderOpts = {
             Component,
             pageConfig: config || {},
@@ -162,32 +167,36 @@ export default async function middlewareRSCLoader(this: any) {
           const transformStream = new TransformStream()
           const writer = transformStream.writable.getWriter()
           const encoder = new TextEncoder()
-          
+          let result
+          let statusCode = 200
           try {
-            const result = await renderToHTML(
-              { url: pathname },
+            result = await renderToHTML(
+              req,
               {},
               pathname,
               query,
               renderOpts
             )
-            result.pipe({
-              write: str => writer.write(encoder.encode(str)),
-              end: () => writer.close()
-            })
-            .catch(() => writer.close())
           } catch (err) {
-            return new Response(
-              (err || 'An error occurred while rendering ' + pathname + '.').toString(),
-              {
-                status: 500,
-                headers: { 'x-middleware-ssr': '1' }
-              }
+            statusCode = 500
+            const errorRes = { statusCode, err }
+            result = await renderToHTML(
+              req,
+              errorRes,
+              pathname,
+              query,
+              { ...renderOpts, Component: ErrorPage }
             )
           }
 
+          result.pipe({
+            write: str => writer.write(encoder.encode(str)),
+            end: () => writer.close()
+          })
+
           return new Response(transformStream.readable, {
-            headers: { 'x-middleware-ssr': '1' }
+            headers: { 'x-middleware-ssr': '1' },
+            status: statusCode
           })
         }
 
diff --git a/packages/next/server/dev/hot-reloader.ts b/packages/next/server/dev/hot-reloader.ts
index 7a0164864c2c5..3887dace73eda 100644
--- a/packages/next/server/dev/hot-reloader.ts
+++ b/packages/next/server/dev/hot-reloader.ts
@@ -528,6 +528,7 @@ export default class HotReloader {
                     page,
                     absoluteAppPath: this.pagesMapping['/_app'],
                     absoluteDocumentPath: this.pagesMapping['/_document'],
+                    absoluteErrorPath: this.pagesMapping['/_error'],
                     absolutePagePath,
                     isServerComponent,
                     buildId: this.buildId,