From 442ad61779c80224e124a580b68df12715e32100 Mon Sep 17 00:00:00 2001
From: Mike Cao <mike@mikecao.com>
Date: Tue, 12 Dec 2023 21:23:12 -0800
Subject: [PATCH] Created admin API endpoints.

---
 .../(main)/settings/users/UsersDataTable.tsx  |  2 +-
 .../settings/websites/WebsiteSettings.tsx     |  6 +-
 .../settings/websites/WebsitesTable.tsx       |  6 +-
 src/app/Providers.tsx                         |  5 +-
 src/pages/api/admin/users.ts                  | 53 +++++++++++++++
 src/pages/api/admin/websites.ts               | 66 +++++++++++++++++++
 src/pages/api/users/index.ts                  | 16 +----
 src/pages/api/websites/index.ts               | 28 +-------
 8 files changed, 133 insertions(+), 49 deletions(-)
 create mode 100644 src/pages/api/admin/users.ts
 create mode 100644 src/pages/api/admin/websites.ts

diff --git a/src/app/(main)/settings/users/UsersDataTable.tsx b/src/app/(main)/settings/users/UsersDataTable.tsx
index b7716451..2495d023 100644
--- a/src/app/(main)/settings/users/UsersDataTable.tsx
+++ b/src/app/(main)/settings/users/UsersDataTable.tsx
@@ -11,7 +11,7 @@ export function UsersDataTable() {
   const modified = useCache((state: any) => state?.users);
   const queryResult = useFilterQuery({
     queryKey: ['users', { modified }],
-    queryFn: (params: { [key: string]: any }) => get(`/users`, params),
+    queryFn: (params: { [key: string]: any }) => get(`/admin/users`, params),
   });
 
   return (
diff --git a/src/app/(main)/settings/websites/WebsiteSettings.tsx b/src/app/(main)/settings/websites/WebsiteSettings.tsx
index 4607b423..0c5ce614 100644
--- a/src/app/(main)/settings/websites/WebsiteSettings.tsx
+++ b/src/app/(main)/settings/websites/WebsiteSettings.tsx
@@ -17,7 +17,7 @@ export function WebsiteSettings({ websiteId, openExternal = false }) {
   const { formatMessage, labels, messages } = useMessages();
   const { get, useQuery } = useApi();
   const { showToast } = useToasts();
-  const { websitesUrl, settingsUrl } = useContext(SettingsContext);
+  const { websitesUrl, websitesPath, settingsPath } = useContext(SettingsContext);
   const { data, isLoading } = useQuery({
     queryKey: ['website', websiteId],
     queryFn: () => get(`${websitesUrl}/${websiteId}`),
@@ -38,7 +38,7 @@ export function WebsiteSettings({ websiteId, openExternal = false }) {
 
   const handleReset = async (value: string) => {
     if (value === 'delete') {
-      router.push(settingsUrl);
+      router.push(settingsPath);
     } else if (value === 'reset') {
       showSuccess();
     }
@@ -57,7 +57,7 @@ export function WebsiteSettings({ websiteId, openExternal = false }) {
   return (
     <>
       <PageHeader title={values?.name}>
-        <Link href={`${websitesUrl}/${websiteId}`} target={openExternal ? '_blank' : null}>
+        <Link href={`${websitesPath}/${websiteId}`} target={openExternal ? '_blank' : null}>
           <Button variant="primary">
             <Icon>
               <Icons.External />
diff --git a/src/app/(main)/settings/websites/WebsitesTable.tsx b/src/app/(main)/settings/websites/WebsitesTable.tsx
index 5f99a86d..069d0c2f 100644
--- a/src/app/(main)/settings/websites/WebsitesTable.tsx
+++ b/src/app/(main)/settings/websites/WebsitesTable.tsx
@@ -25,7 +25,7 @@ export function WebsitesTable({
   const { formatMessage, labels } = useMessages();
   const { user } = useUser();
   const breakpoint = useBreakpoint();
-  const { settingsUrl, websitesUrl } = useContext(SettingsContext);
+  const { settingsPath, websitesPath } = useContext(SettingsContext);
 
   return (
     <GridTable data={data} cardMode={['xs', 'sm', 'md'].includes(breakpoint)}>
@@ -52,7 +52,7 @@ export function WebsitesTable({
             return (
               <>
                 {allowEdit && (!showTeam || ownerId === user.id) && (
-                  <Link href={`${settingsUrl}/${id}`}>
+                  <Link href={`${settingsPath}/${id}`}>
                     <Button>
                       <Icon>
                         <Icons.Edit />
@@ -62,7 +62,7 @@ export function WebsitesTable({
                   </Link>
                 )}
                 {allowView && (
-                  <Link href={`${websitesUrl}/${id}`}>
+                  <Link href={`${websitesPath}/${id}`}>
                     <Button>
                       <Icon>
                         <Icons.External />
diff --git a/src/app/Providers.tsx b/src/app/Providers.tsx
index c2ddd2ff..f1460b05 100644
--- a/src/app/Providers.tsx
+++ b/src/app/Providers.tsx
@@ -33,10 +33,11 @@ function SettingsProvider({ children }) {
     const hostUrl = process.env.hostUrl || window?.location.origin;
 
     setConfig({
-      settingsUrl: '/settings/websites',
       shareUrl: hostUrl,
       trackingCodeUrl: hostUrl,
-      websitesUrl: `/websites`,
+      websitesUrl: '/websites',
+      settingsPath: '/settings/websites',
+      websitesPath: `/websites`,
     });
   }, []);
 
diff --git a/src/pages/api/admin/users.ts b/src/pages/api/admin/users.ts
new file mode 100644
index 00000000..d16895b7
--- /dev/null
+++ b/src/pages/api/admin/users.ts
@@ -0,0 +1,53 @@
+import { canViewUsers } from 'lib/auth';
+import { useAuth, useValidate } from 'lib/middleware';
+import { NextApiRequestQueryBody, Role, SearchFilter, User } from 'lib/types';
+import { pageInfo } from 'lib/schema';
+import { NextApiResponse } from 'next';
+import { methodNotAllowed, ok, unauthorized } from 'next-basics';
+import { getUsers } from 'queries';
+import * as yup from 'yup';
+
+export interface UsersRequestQuery extends SearchFilter {}
+export interface UsersRequestBody {
+  username: string;
+  password: string;
+  id: string;
+  role: Role;
+}
+
+const schema = {
+  GET: yup.object().shape({
+    ...pageInfo,
+  }),
+  POST: yup.object().shape({
+    username: yup.string().max(255).required(),
+    password: yup.string().required(),
+    id: yup.string().uuid(),
+    role: yup
+      .string()
+      .matches(/admin|user|view-only/i)
+      .required(),
+  }),
+};
+
+export default async (
+  req: NextApiRequestQueryBody<UsersRequestQuery, UsersRequestBody>,
+  res: NextApiResponse<User[] | User>,
+) => {
+  await useAuth(req, res);
+  await useValidate(schema, req, res);
+
+  if (req.method === 'GET') {
+    if (!(await canViewUsers(req.auth))) {
+      return unauthorized(res);
+    }
+
+    const { page, query, pageSize } = req.query;
+
+    const users = await getUsers({ page, query, pageSize: +pageSize || undefined });
+
+    return ok(res, users);
+  }
+
+  return methodNotAllowed(res);
+};
diff --git a/src/pages/api/admin/websites.ts b/src/pages/api/admin/websites.ts
new file mode 100644
index 00000000..13c19072
--- /dev/null
+++ b/src/pages/api/admin/websites.ts
@@ -0,0 +1,66 @@
+import { canViewAllWebsites } from 'lib/auth';
+import { useAuth, useCors, useValidate } from 'lib/middleware';
+import { NextApiRequestQueryBody, SearchFilter } from 'lib/types';
+import { NextApiResponse } from 'next';
+import { methodNotAllowed, ok, unauthorized } from 'next-basics';
+import { getWebsites } from 'queries';
+import * as yup from 'yup';
+import { pageInfo } from 'lib/schema';
+
+export interface WebsitesRequestQuery extends SearchFilter {}
+
+export interface WebsitesRequestBody {
+  name: string;
+  domain: string;
+  shareId: string;
+}
+
+const schema = {
+  GET: yup.object().shape({
+    ...pageInfo,
+  }),
+  POST: yup.object().shape({
+    name: yup.string().max(100).required(),
+    domain: yup.string().max(500).required(),
+    shareId: yup.string().max(50).nullable(),
+  }),
+};
+
+export default async (
+  req: NextApiRequestQueryBody<WebsitesRequestQuery, WebsitesRequestBody>,
+  res: NextApiResponse,
+) => {
+  await useCors(req, res);
+  await useAuth(req, res);
+  await useValidate(schema, req, res);
+
+  if (req.method === 'GET') {
+    if (!(await canViewAllWebsites(req.auth))) {
+      return unauthorized(res);
+    }
+
+    const websites = await getWebsites(req.query, {
+      include: {
+        teamWebsite: {
+          include: {
+            team: {
+              select: {
+                name: true,
+              },
+            },
+          },
+        },
+        user: {
+          select: {
+            username: true,
+            id: true,
+          },
+        },
+      },
+    });
+
+    return ok(res, websites);
+  }
+
+  return methodNotAllowed(res);
+};
diff --git a/src/pages/api/users/index.ts b/src/pages/api/users/index.ts
index de49a37b..7f8c9174 100644
--- a/src/pages/api/users/index.ts
+++ b/src/pages/api/users/index.ts
@@ -1,4 +1,4 @@
-import { canCreateUser, canViewUsers } from 'lib/auth';
+import { canCreateUser } from 'lib/auth';
 import { ROLES } from 'lib/constants';
 import { uuid } from 'lib/crypto';
 import { useAuth, useValidate } from 'lib/middleware';
@@ -6,7 +6,7 @@ import { NextApiRequestQueryBody, Role, SearchFilter, User } from 'lib/types';
 import { pageInfo } from 'lib/schema';
 import { NextApiResponse } from 'next';
 import { badRequest, hashPassword, methodNotAllowed, ok, unauthorized } from 'next-basics';
-import { createUser, getUserByUsername, getUsers } from 'queries';
+import { createUser, getUserByUsername } from 'queries';
 import * as yup from 'yup';
 
 export interface UsersRequestQuery extends SearchFilter {}
@@ -39,18 +39,6 @@ export default async (
   await useAuth(req, res);
   await useValidate(schema, req, res);
 
-  if (req.method === 'GET') {
-    if (!(await canViewUsers(req.auth))) {
-      return unauthorized(res);
-    }
-
-    const { page, query, pageSize } = req.query;
-
-    const users = await getUsers({ page, query, pageSize: +pageSize || undefined });
-
-    return ok(res, users);
-  }
-
   if (req.method === 'POST') {
     if (!(await canCreateUser(req.auth))) {
       return unauthorized(res);
diff --git a/src/pages/api/websites/index.ts b/src/pages/api/websites/index.ts
index 89162067..b30681cf 100644
--- a/src/pages/api/websites/index.ts
+++ b/src/pages/api/websites/index.ts
@@ -1,10 +1,10 @@
-import { canCreateWebsite, canViewAllWebsites } from 'lib/auth';
+import { canCreateWebsite } from 'lib/auth';
 import { uuid } from 'lib/crypto';
 import { useAuth, useCors, useValidate } from 'lib/middleware';
 import { NextApiRequestQueryBody, SearchFilter } from 'lib/types';
 import { NextApiResponse } from 'next';
 import { methodNotAllowed, ok, unauthorized } from 'next-basics';
-import { createWebsite, getWebsites } from 'queries';
+import { createWebsite } from 'queries';
 import userWebsites from 'pages/api/users/[id]/websites';
 import * as yup from 'yup';
 import { pageInfo } from 'lib/schema';
@@ -41,30 +41,6 @@ export default async (
   } = req.auth;
 
   if (req.method === 'GET') {
-    if (await canViewAllWebsites(req.auth)) {
-      const websites = await getWebsites(req.query, {
-        include: {
-          teamWebsite: {
-            include: {
-              team: {
-                select: {
-                  name: true,
-                },
-              },
-            },
-          },
-          user: {
-            select: {
-              username: true,
-              id: true,
-            },
-          },
-        },
-      });
-
-      return ok(res, websites);
-    }
-
     if (!req.query.id) {
       req.query.id = userId;
     }