From b919948967d11eaa0954fffd88688223e29e837a Mon Sep 17 00:00:00 2001
From: wangsijie <wangsijie@silverhand.io>
Date: Thu, 21 Jul 2022 21:23:23 +0800
Subject: [PATCH] feat(next): handleAuthRoutes

---
 .../next-sample/pages/api/logto/[action].ts   |  3 +
 .../next-sample/pages/api/sign-in-callback.ts |  3 -
 packages/next-sample/pages/api/sign-in.ts     |  3 -
 packages/next-sample/pages/api/sign-out.ts    |  3 -
 packages/next-sample/pages/api/user.ts        |  3 -
 packages/next-sample/pages/index.tsx          |  6 +-
 packages/next-sample/pages/profile-ssr.tsx    |  2 +-
 packages/next-sample/pages/protected.tsx      |  2 +-
 packages/next/src/index.test.ts               | 77 ++++++++++++++++++-
 packages/next/src/index.ts                    | 28 ++++++-
 10 files changed, 109 insertions(+), 21 deletions(-)
 create mode 100644 packages/next-sample/pages/api/logto/[action].ts
 delete mode 100644 packages/next-sample/pages/api/sign-in-callback.ts
 delete mode 100644 packages/next-sample/pages/api/sign-in.ts
 delete mode 100644 packages/next-sample/pages/api/sign-out.ts
 delete mode 100644 packages/next-sample/pages/api/user.ts

diff --git a/packages/next-sample/pages/api/logto/[action].ts b/packages/next-sample/pages/api/logto/[action].ts
new file mode 100644
index 00000000..1ce3265a
--- /dev/null
+++ b/packages/next-sample/pages/api/logto/[action].ts
@@ -0,0 +1,3 @@
+import { logtoClient } from '../../../libraries/logto';
+
+export default logtoClient.handleAuthRoutes();
diff --git a/packages/next-sample/pages/api/sign-in-callback.ts b/packages/next-sample/pages/api/sign-in-callback.ts
deleted file mode 100644
index 53018887..00000000
--- a/packages/next-sample/pages/api/sign-in-callback.ts
+++ /dev/null
@@ -1,3 +0,0 @@
-import { logtoClient } from '../../libraries/logto';
-
-export default logtoClient.handleSignInCallback();
diff --git a/packages/next-sample/pages/api/sign-in.ts b/packages/next-sample/pages/api/sign-in.ts
deleted file mode 100644
index ee1e3928..00000000
--- a/packages/next-sample/pages/api/sign-in.ts
+++ /dev/null
@@ -1,3 +0,0 @@
-import { logtoClient } from '../../libraries/logto';
-
-export default logtoClient.handleSignIn();
diff --git a/packages/next-sample/pages/api/sign-out.ts b/packages/next-sample/pages/api/sign-out.ts
deleted file mode 100644
index f9bd1291..00000000
--- a/packages/next-sample/pages/api/sign-out.ts
+++ /dev/null
@@ -1,3 +0,0 @@
-import { logtoClient } from '../../libraries/logto';
-
-export default logtoClient.handleSignOut();
diff --git a/packages/next-sample/pages/api/user.ts b/packages/next-sample/pages/api/user.ts
deleted file mode 100644
index c7983994..00000000
--- a/packages/next-sample/pages/api/user.ts
+++ /dev/null
@@ -1,3 +0,0 @@
-import { logtoClient } from '../../libraries/logto';
-
-export default logtoClient.handleUser();
diff --git a/packages/next-sample/pages/index.tsx b/packages/next-sample/pages/index.tsx
index aa61fe0b..a2e415a3 100644
--- a/packages/next-sample/pages/index.tsx
+++ b/packages/next-sample/pages/index.tsx
@@ -4,7 +4,7 @@ import { useMemo } from 'react';
 import useSWR from 'swr';
 
 const Home = () => {
-  const { data } = useSWR<LogtoUser>('/api/user');
+  const { data } = useSWR<LogtoUser>('/api/logto/user');
   const { data: protectedResource } = useSWR<{ data: string }>('/api/protected-resource');
 
   const userInfo = useMemo(() => {
@@ -42,11 +42,11 @@ const Home = () => {
       </header>
       <nav>
         {data?.isAuthenticated ? (
-          <Link href="/api/sign-out">
+          <Link href="/api/logto/sign-out">
             <a>Sign Out</a>
           </Link>
         ) : (
-          <Link href="/api/sign-in">
+          <Link href="/api/logto/sign-in">
             <a>Sign In</a>
           </Link>
         )}
diff --git a/packages/next-sample/pages/profile-ssr.tsx b/packages/next-sample/pages/profile-ssr.tsx
index 359cd56f..aa8a1eba 100644
--- a/packages/next-sample/pages/profile-ssr.tsx
+++ b/packages/next-sample/pages/profile-ssr.tsx
@@ -52,7 +52,7 @@ export const getServerSideProps = logtoClient.withLogtoSsr(async function ({ req
   const { user } = req;
 
   if (!user.isAuthenticated) {
-    res.setHeader('location', '/api/login');
+    res.setHeader('location', '/api/logto/login');
     // eslint-disable-next-line @silverhand/fp/no-mutation
     res.statusCode = 302;
     res.end();
diff --git a/packages/next-sample/pages/protected.tsx b/packages/next-sample/pages/protected.tsx
index 5a2dbdec..49bdeb00 100644
--- a/packages/next-sample/pages/protected.tsx
+++ b/packages/next-sample/pages/protected.tsx
@@ -10,7 +10,7 @@ const Protected = () => {
   useEffect(() => {
     if (error?.message === 'Unauthorized') {
       // eslint-disable-next-line @silverhand/fp/no-mutating-methods
-      void Router.push('/api/sign-in');
+      void Router.push('/api/logto/sign-in');
     }
   }, [error]);
 
diff --git a/packages/next/src/index.test.ts b/packages/next/src/index.test.ts
index 1628a1d2..ead8cb10 100644
--- a/packages/next/src/index.test.ts
+++ b/packages/next/src/index.test.ts
@@ -1,3 +1,4 @@
+import { NextApiResponse } from 'next';
 import { testApiHandler } from 'next-test-api-route-handler';
 
 import LogtoClient from '.';
@@ -26,6 +27,10 @@ const getIdTokenClaims = jest.fn(() => ({
 const signOut = jest.fn();
 const getAccessToken = jest.fn(async () => true);
 
+const mockResponse = (_: unknown, response: NextApiResponse) => {
+  response.status(200).end();
+};
+
 jest.mock('./storage', () =>
   jest.fn(() => ({
     setItem,
@@ -72,7 +77,7 @@ describe('Next', () => {
       const client = new LogtoClient(configs);
       await testApiHandler({
         handler: client.handleSignIn(),
-        url: '/api/sign-in',
+        url: '/api/logto/sign-in',
         test: async ({ fetch }) => {
           const response = await fetch({ method: 'GET', redirect: 'manual' });
           const headers = response.headers as Map<string, string>;
@@ -89,7 +94,7 @@ describe('Next', () => {
       const client = new LogtoClient(configs);
       await testApiHandler({
         handler: client.handleSignInCallback(),
-        url: '/api/sign-in-callback',
+        url: '/api/logto/sign-in-callback',
         test: async ({ fetch }) => {
           const response = await fetch({ method: 'GET', redirect: 'manual' });
           const headers = response.headers as Map<string, string>;
@@ -142,7 +147,7 @@ describe('Next', () => {
       const client = new LogtoClient(configs);
       await testApiHandler({
         handler: client.handleSignOut(),
-        url: '/api/sign-out',
+        url: '/api/logto/sign-out',
         test: async ({ fetch }) => {
           const response = await fetch({ method: 'GET', redirect: 'manual' });
           const headers = response.headers as Map<string, string>;
@@ -153,4 +158,70 @@ describe('Next', () => {
       expect(signOut).toHaveBeenCalled();
     });
   });
+
+  describe('handleAuthRoutes', () => {
+    it('should call handleSignIn for "sign-in"', async () => {
+      const client = new LogtoClient(configs);
+      jest.spyOn(client, 'handleSignIn').mockImplementation(() => mockResponse);
+      await testApiHandler({
+        handler: client.handleAuthRoutes(),
+        paramsPatcher: (parameters) => {
+          // eslint-disable-next-line @silverhand/fp/no-mutation
+          parameters.action = 'sign-in';
+        },
+        test: async ({ fetch }) => {
+          await fetch({ method: 'GET', redirect: 'manual' });
+          expect(client.handleSignIn).toHaveBeenCalled();
+        },
+      });
+    });
+
+    it('should call handleSignInCallback for "sign-in-callback"', async () => {
+      const client = new LogtoClient(configs);
+      jest.spyOn(client, 'handleSignInCallback').mockImplementation(() => mockResponse);
+      await testApiHandler({
+        handler: client.handleAuthRoutes(),
+        paramsPatcher: (parameters) => {
+          // eslint-disable-next-line @silverhand/fp/no-mutation
+          parameters.action = 'sign-in-callback';
+        },
+        test: async ({ fetch }) => {
+          await fetch({ method: 'GET', redirect: 'manual' });
+          expect(client.handleSignInCallback).toHaveBeenCalled();
+        },
+      });
+    });
+
+    it('should call handleSignOut for "sign-out"', async () => {
+      const client = new LogtoClient(configs);
+      jest.spyOn(client, 'handleSignOut').mockImplementation(() => mockResponse);
+      await testApiHandler({
+        handler: client.handleAuthRoutes(),
+        paramsPatcher: (parameters) => {
+          // eslint-disable-next-line @silverhand/fp/no-mutation
+          parameters.action = 'sign-out';
+        },
+        test: async ({ fetch }) => {
+          await fetch({ method: 'GET', redirect: 'manual' });
+          expect(client.handleSignOut).toHaveBeenCalled();
+        },
+      });
+    });
+
+    it('should call handleUser for "user"', async () => {
+      const client = new LogtoClient(configs);
+      jest.spyOn(client, 'handleUser').mockImplementation(() => mockResponse);
+      await testApiHandler({
+        handler: client.handleAuthRoutes(),
+        paramsPatcher: (parameters) => {
+          // eslint-disable-next-line @silverhand/fp/no-mutation
+          parameters.action = 'user';
+        },
+        test: async ({ fetch }) => {
+          await fetch({ method: 'GET', redirect: 'manual' });
+          expect(client.handleUser).toHaveBeenCalled();
+        },
+      });
+    });
+  });
 });
diff --git a/packages/next/src/index.ts b/packages/next/src/index.ts
index 72ba586e..d1c52f3b 100644
--- a/packages/next/src/index.ts
+++ b/packages/next/src/index.ts
@@ -14,7 +14,9 @@ export default class LogtoClient {
   private storage?: NextStorage;
   constructor(private readonly config: LogtoNextConfig) {}
 
-  handleSignIn = (redirectUri = `${this.config.baseUrl}/api/sign-in-callback`): NextApiHandler =>
+  handleSignIn = (
+    redirectUri = `${this.config.baseUrl}/api/logto/sign-in-callback`
+  ): NextApiHandler =>
     withIronSessionApiRoute(async (request, response) => {
       const nodeClient = this.createNodeClient(request);
       await nodeClient.signIn(redirectUri);
@@ -54,6 +56,30 @@ export default class LogtoClient {
       response.json(request.user);
     }, config);
 
+  handleAuthRoutes =
+    (configs?: WithLogtoConfig): NextApiHandler =>
+    (request, response) => {
+      const { action } = request.query;
+
+      if (action === 'sign-in') {
+        return this.handleSignIn()(request, response);
+      }
+
+      if (action === 'sign-in-callback') {
+        return this.handleSignInCallback()(request, response);
+      }
+
+      if (action === 'sign-out') {
+        return this.handleSignOut()(request, response);
+      }
+
+      if (action === 'user') {
+        return this.handleUser(configs)(request, response);
+      }
+
+      response.status(404).end();
+    };
+
   withLogtoApiRoute = (handler: NextApiHandler, config: WithLogtoConfig = {}): NextApiHandler =>
     withIronSessionApiRoute(async (request, response) => {
       const user = await this.getLogtoUserFromRequest(request, config.getAccessToken);