From 67fa74a0e3c7f71d9d6f9e769659c3959513941a Mon Sep 17 00:00:00 2001
From: Blake Wilson <blake.wilson@wpengine.com>
Date: Thu, 1 Jul 2021 12:49:41 -0500
Subject: [PATCH] feat: (#321) support `redirect` and `notFound`

---
 packages/next/src/getProps.ts | 35 ++++++++++++++++++++++++++++++++++-
 1 file changed, 34 insertions(+), 1 deletion(-)

diff --git a/packages/next/src/getProps.ts b/packages/next/src/getProps.ts
index 91a7e24a7..6d8743b7a 100644
--- a/packages/next/src/getProps.ts
+++ b/packages/next/src/getProps.ts
@@ -1,12 +1,13 @@
 /* eslint-disable react/no-children-prop */
 import { CategoryIdType, PageIdType, PostIdType } from '@faustjs/core';
-import { isNumber, isObject } from 'lodash';
+import { isBoolean, isNumber, isObject } from 'lodash';
 import isNil from 'lodash/isNil';
 import {
   GetServerSidePropsContext,
   GetStaticPropsContext,
   GetStaticPropsResult,
   GetServerSidePropsResult,
+  Redirect,
 } from 'next';
 import { RouterContext } from 'next/dist/next-server/lib/router-context';
 
@@ -29,6 +30,8 @@ export interface GetNextServerSidePropsConfig<Props = Record<string, unknown>> {
   client: ReturnType<typeof getClient>;
   Page?: FunctionComponent | ComponentClass;
   props?: Props;
+  notFound?: boolean;
+  redirect?: Redirect;
 }
 
 export interface GetNextStaticPropsConfig<Props = Record<string, unknown>> {
@@ -36,6 +39,8 @@ export interface GetNextStaticPropsConfig<Props = Record<string, unknown>> {
   Page?: FunctionComponent | ComponentClass;
   props?: Props;
   revalidate?: number | boolean;
+  notFound?: boolean;
+  redirect?: Redirect;
 }
 
 export interface PageProps<Props> {
@@ -194,6 +199,20 @@ export async function getNextServerSideProps<Props>(
   context: GetServerSidePropsContext,
   config: GetNextServerSidePropsConfig,
 ): Promise<GetServerSidePropsResult<Props>> {
+  const { notFound, redirect } = config;
+
+  if (isBoolean(notFound) && notFound === true) {
+    return {
+      notFound,
+    };
+  }
+
+  if (isObject(redirect)) {
+    return {
+      redirect,
+    };
+  }
+
   return getProps(context, config);
 }
 
@@ -201,6 +220,20 @@ export async function getNextStaticProps<Props>(
   context: GetStaticPropsContext,
   config: GetNextStaticPropsConfig,
 ): Promise<GetStaticPropsResult<Props>> {
+  const { notFound, redirect } = config;
+
+  if (isBoolean(notFound) && notFound === true) {
+    return {
+      notFound,
+    };
+  }
+
+  if (isObject(redirect)) {
+    return {
+      redirect,
+    };
+  }
+
   const pageProps: GetStaticPropsResult<Props> = await getProps(
     context,
     config,