From e971f2533d6c9de1a03118f11751bf9d281c372a Mon Sep 17 00:00:00 2001
From: Mike Cao <moocao@gmail.com>
Date: Sat, 3 Feb 2024 23:19:29 -0800
Subject: [PATCH] Updated search queries.

---
 src/app/(main)/dashboard/Dashboard.tsx        |  8 +--
 src/app/(main)/settings/websites/Websites.tsx |  2 +-
 src/app/(main)/websites/page.tsx              |  2 +-
 src/components/input/ProfileButton.tsx        |  2 +-
 src/index.ts                                  |  4 ++
 src/lib/db.ts                                 | 18 ++---
 src/lib/prisma.ts                             | 44 +++++++++---
 src/pages/api/reports/index.ts                | 16 +++--
 src/queries/admin/report.ts                   | 72 +++++--------------
 src/queries/admin/team.ts                     | 35 +--------
 src/queries/admin/teamUser.ts                 | 10 +--
 src/queries/admin/user.ts                     | 24 +------
 12 files changed, 90 insertions(+), 147 deletions(-)

diff --git a/src/app/(main)/dashboard/Dashboard.tsx b/src/app/(main)/dashboard/Dashboard.tsx
index ed976f43..cf6cc210 100644
--- a/src/app/(main)/dashboard/Dashboard.tsx
+++ b/src/app/(main)/dashboard/Dashboard.tsx
@@ -12,14 +12,14 @@ import useDashboard from 'store/dashboard';
 
 export function Dashboard() {
   const { formatMessage, labels, messages } = useMessages();
-  const { teamId } = useTeamContext();
+  const { teamId, renderTeamUrl } = useTeamContext();
   const { showCharts, editing } = useDashboard();
   const { dir } = useLocale();
   const pageSize = 10;
 
-  const { result, query, params, setParams } = useWebsites({}, { pageSize });
+  const { result, query, params, setParams } = useWebsites({ teamId }, { pageSize });
   const { page } = params;
-  const hasData = !!result?.data;
+  const hasData = !!result?.data?.length;
 
   const handlePageChange = (page: number) => {
     setParams({ ...params, page });
@@ -36,7 +36,7 @@ export function Dashboard() {
       </PageHeader>
       {!hasData && (
         <EmptyPlaceholder message={formatMessage(messages.noWebsitesConfigured)}>
-          <Link href="/settings/websites">
+          <Link href={renderTeamUrl('/settings/websites')}>
             <Button>
               <Icon rotate={dir === 'rtl' ? 180 : 0}>
                 <Icons.ArrowRight />
diff --git a/src/app/(main)/settings/websites/Websites.tsx b/src/app/(main)/settings/websites/Websites.tsx
index 09750656..ffa197ad 100644
--- a/src/app/(main)/settings/websites/Websites.tsx
+++ b/src/app/(main)/settings/websites/Websites.tsx
@@ -8,7 +8,7 @@ export default function Websites({ teamId }: { teamId: string }) {
 
   return (
     <>
-      <WebsitesHeader showActions={user.role !== 'view-only'} />
+      <WebsitesHeader teamId={teamId} showActions={user.role !== 'view-only'} />
       <WebsitesDataTable teamId={teamId} userId={user.id} allowEdit={true} />
     </>
   );
diff --git a/src/app/(main)/websites/page.tsx b/src/app/(main)/websites/page.tsx
index a1cc5216..3062bbba 100644
--- a/src/app/(main)/websites/page.tsx
+++ b/src/app/(main)/websites/page.tsx
@@ -5,7 +5,7 @@ import { Metadata } from 'next';
 export default function WebsitesPage({ params: { teamId, userId } }) {
   return (
     <>
-      <WebsitesHeader showActions={false} />
+      <WebsitesHeader teamId={teamId} showActions={false} />
       <WebsitesBrowse teamId={teamId} userId={userId} />
     </>
   );
diff --git a/src/components/input/ProfileButton.tsx b/src/components/input/ProfileButton.tsx
index ec0deb3b..c689cb2c 100644
--- a/src/components/input/ProfileButton.tsx
+++ b/src/components/input/ProfileButton.tsx
@@ -6,8 +6,8 @@ import { useMessages } from 'components/hooks';
 import { useLogin } from 'components/hooks';
 import { useLocale } from 'components/hooks';
 import { CURRENT_VERSION } from 'lib/constants';
-import styles from './ProfileButton.module.css';
 import Avatar from 'components/common/Avatar';
+import styles from './ProfileButton.module.css';
 
 export function ProfileButton() {
   const { formatMessage, labels } = useMessages();
diff --git a/src/index.ts b/src/index.ts
index abd00fbf..1c00d3f0 100644
--- a/src/index.ts
+++ b/src/index.ts
@@ -17,6 +17,7 @@ export * from 'components/hooks/useMessages';
 export * from 'components/hooks/useNavigation';
 export * from 'components/hooks/useSticky';
 export * from 'components/hooks/useTheme';
+export * from 'components/hooks/useTeamContext';
 export * from 'components/hooks/useTimezone';
 
 export * from './app/(main)/settings/teams/[teamId]/TeamEditForm';
@@ -60,3 +61,6 @@ export * from 'components/common/HoverTooltip';
 export * from 'components/common/LinkButton';
 export * from 'components/common/MobileMenu';
 export * from 'components/common/Pager';
+
+export * from 'components/input/TeamsButton';
+export * from 'components/input/ThemeButton';
diff --git a/src/lib/db.ts b/src/lib/db.ts
index 46f73aa2..0ffedd0d 100644
--- a/src/lib/db.ts
+++ b/src/lib/db.ts
@@ -17,27 +17,23 @@ export function getDatabaseType(url = process.env.DATABASE_URL) {
     return POSTGRESQL;
   }
 
-  if (process.env.CLICKHOUSE_URL) {
-    return CLICKHOUSE;
-  }
-
   return type;
 }
 
 export async function runQuery(queries: any) {
-  const db = getDatabaseType(process.env.CLICKHOUSE_URL || process.env.DATABASE_URL);
-
-  if (db === POSTGRESQL || db === MYSQL) {
-    return queries[PRISMA]();
-  }
-
-  if (db === CLICKHOUSE) {
+  if (process.env.CLICKHOUSE_URL) {
     if (queries[KAFKA]) {
       return queries[KAFKA]();
     }
 
     return queries[CLICKHOUSE]();
   }
+
+  const db = getDatabaseType();
+
+  if (db === POSTGRESQL || db === MYSQL) {
+    return queries[PRISMA]();
+  }
 }
 
 export function notImplemented() {
diff --git a/src/lib/prisma.ts b/src/lib/prisma.ts
index ad8e45a9..a210b72b 100644
--- a/src/lib/prisma.ts
+++ b/src/lib/prisma.ts
@@ -24,7 +24,7 @@ const POSTGRESQL_DATE_FORMATS = {
 };
 
 function getAddIntervalQuery(field: string, interval: string): string {
-  const db = getDatabaseType(process.env.DATABASE_URL);
+  const db = getDatabaseType();
 
   if (db === POSTGRESQL) {
     return `${field} + interval '${interval}'`;
@@ -36,7 +36,7 @@ function getAddIntervalQuery(field: string, interval: string): string {
 }
 
 function getDayDiffQuery(field1: string, field2: string): string {
-  const db = getDatabaseType(process.env.DATABASE_URL);
+  const db = getDatabaseType();
 
   if (db === POSTGRESQL) {
     return `${field1}::date - ${field2}::date`;
@@ -48,7 +48,7 @@ function getDayDiffQuery(field1: string, field2: string): string {
 }
 
 function getCastColumnQuery(field: string, type: string): string {
-  const db = getDatabaseType(process.env.DATABASE_URL);
+  const db = getDatabaseType();
 
   if (db === POSTGRESQL) {
     return `${field}::${type}`;
@@ -92,7 +92,7 @@ function getTimestampDiffQuery(field1: string, field2: string): string {
   }
 }
 
-function mapFilter(column, operator, name, type = 'varchar') {
+function mapFilter(column: string, operator: string, name: string, type = 'varchar') {
   switch (operator) {
     case OPERATORS.equals:
       return `${column} = {{${name}::${type}}}`;
@@ -208,16 +208,44 @@ function getQueryMode(): Prisma.QueryMode {
   return 'default';
 }
 
+function getSearchParameters(query: string, filters: { [key: string]: any }[]) {
+  if (!query) return;
+
+  const mode = getQueryMode();
+  const parseFilter = (filter: { [key: string]: any }) => {
+    const [[key, value]] = Object.entries(filter);
+
+    return {
+      [key]:
+        typeof value === 'string'
+          ? {
+              [value]: query,
+              mode,
+            }
+          : parseFilter(value),
+    };
+  };
+
+  const params = filters.map(filter => parseFilter(filter));
+
+  return {
+    AND: {
+      OR: params,
+    },
+  };
+}
+
 export default {
   ...prisma,
   getAddIntervalQuery,
-  getDayDiffQuery,
   getCastColumnQuery,
+  getDayDiffQuery,
   getDateQuery,
-  getTimestampDiffQuery,
   getFilterQuery,
-  parseFilters,
+  getSearchParameters,
+  getTimestampDiffQuery,
   getQueryMode,
-  rawQuery,
   pagedQuery,
+  parseFilters,
+  rawQuery,
 };
diff --git a/src/pages/api/reports/index.ts b/src/pages/api/reports/index.ts
index 86032ef5..b5478018 100644
--- a/src/pages/api/reports/index.ts
+++ b/src/pages/api/reports/index.ts
@@ -3,9 +3,10 @@ import { useAuth, useCors, useValidate } from 'lib/middleware';
 import { NextApiRequestQueryBody } from 'lib/types';
 import { pageInfo } from 'lib/schema';
 import { NextApiResponse } from 'next';
-import { methodNotAllowed, ok } from 'next-basics';
+import { methodNotAllowed, ok, unauthorized } from 'next-basics';
 import { createReport, getReports } from 'queries';
 import * as yup from 'yup';
+import { canViewTeam, canViewWebsite } from 'lib/auth';
 
 export interface ReportRequestBody {
   websiteId: string;
@@ -51,16 +52,23 @@ export default async (
     const { page, query, pageSize, websiteId, teamId } = req.query;
     const filters = {
       page,
-      pageSize: +pageSize || undefined,
+      pageSize,
       query,
     };
 
+    if (
+      (websiteId && !(await canViewWebsite(req.auth, websiteId))) ||
+      (teamId && !(await canViewTeam(req.auth, teamId)))
+    ) {
+      return unauthorized(res);
+    }
+
     const data = await getReports(
       {
         where: {
+          userId: !teamId && !websiteId ? userId : undefined,
+          websiteId,
           website: {
-            id: websiteId,
-            userId: !websiteId && !teamId ? userId : undefined,
             teamId,
           },
         },
diff --git a/src/queries/admin/report.ts b/src/queries/admin/report.ts
index 92b04ff8..fee0b442 100644
--- a/src/queries/admin/report.ts
+++ b/src/queries/admin/report.ts
@@ -20,68 +20,30 @@ export async function getReports(
   criteria: ReportFindManyArgs,
   filters: ReportSearchFilter = {},
 ): Promise<FilterResult<Report[]>> {
-  const mode = prisma.getQueryMode();
-  const { query, userId, websiteId } = filters;
+  const { query } = filters;
 
   const where: Prisma.ReportWhereInput = {
-    userId,
-    websiteId,
     ...criteria.where,
-    AND: [
+    ...prisma.getSearchParameters(query, [
+      { name: 'contains' },
+      { description: 'contains' },
+      { type: 'contains' },
       {
-        OR: [
-          {
-            userId,
-          },
-        ],
+        user: {
+          username: 'contains',
+        },
       },
       {
-        OR: [
-          {
-            name: {
-              contains: query,
-              mode,
-            },
-          },
-          {
-            description: {
-              contains: query,
-              mode,
-            },
-          },
-          {
-            type: {
-              contains: query,
-              mode,
-            },
-          },
-          {
-            user: {
-              username: {
-                contains: query,
-                mode,
-              },
-            },
-          },
-          {
-            website: {
-              name: {
-                contains: query,
-                mode,
-              },
-            },
-          },
-          {
-            website: {
-              domain: {
-                contains: query,
-                mode,
-              },
-            },
-          },
-        ],
+        website: {
+          name: 'contains',
+        },
       },
-    ],
+      {
+        website: {
+          domain: 'contains',
+        },
+      },
+    ]),
   };
 
   return prisma.pagedQuery('report', { ...criteria, where }, filters);
diff --git a/src/queries/admin/team.ts b/src/queries/admin/team.ts
index 0686fc26..e381d5ce 100644
--- a/src/queries/admin/team.ts
+++ b/src/queries/admin/team.ts
@@ -24,41 +24,12 @@ export async function getTeams(
   criteria: TeamFindManyArgs,
   filters: TeamSearchFilter = {},
 ): Promise<FilterResult<Team[]>> {
-  const mode = prisma.getQueryMode();
-  const { userId, query } = filters;
+  const { getSearchParameters } = prisma;
+  const { query } = filters;
 
   const where: Prisma.TeamWhereInput = {
     ...criteria.where,
-    ...(userId && {
-      teamUser: {
-        some: { userId },
-      },
-    }),
-    ...(query && {
-      AND: {
-        OR: [
-          {
-            name: {
-              startsWith: query,
-              mode,
-            },
-          },
-          {
-            teamUser: {
-              some: {
-                role: ROLES.teamOwner,
-                user: {
-                  username: {
-                    startsWith: query,
-                    mode,
-                  },
-                },
-              },
-            },
-          },
-        ],
-      },
-    }),
+    ...getSearchParameters(query, [{ name: 'contains' }]),
   };
 
   return prisma.pagedQuery<TeamFindManyArgs>(
diff --git a/src/queries/admin/teamUser.ts b/src/queries/admin/teamUser.ts
index 34e4a1a5..83f788de 100644
--- a/src/queries/admin/teamUser.ts
+++ b/src/queries/admin/teamUser.ts
@@ -18,18 +18,10 @@ export async function getTeamUsers(
   filters?: TeamUserSearchFilter,
 ): Promise<FilterResult<TeamUser[]>> {
   const { query } = filters;
-  const mode = prisma.getQueryMode();
 
   const where: Prisma.TeamUserWhereInput = {
     ...criteria.where,
-    user: {
-      username: query
-        ? {
-            contains: query,
-            mode,
-          }
-        : undefined,
-    },
+    ...prisma.getSearchParameters(query, [{ user: { username: 'contains' } }]),
   };
 
   return prisma.pagedQuery(
diff --git a/src/queries/admin/user.ts b/src/queries/admin/user.ts
index f9eb23a4..a9a90c5a 100644
--- a/src/queries/admin/user.ts
+++ b/src/queries/admin/user.ts
@@ -52,29 +52,11 @@ export async function getUsers(
   criteria: UserFindManyArgs,
   filters?: UserSearchFilter,
 ): Promise<FilterResult<User[]>> {
-  const { teamId, query } = filters;
-  const mode = prisma.getQueryMode();
+  const { query } = filters;
 
   const where: Prisma.UserWhereInput = {
-    ...(teamId && {
-      teamUser: {
-        some: {
-          teamId,
-        },
-      },
-    }),
-    ...(query && {
-      AND: {
-        OR: [
-          {
-            username: {
-              contains: query,
-              mode,
-            },
-          },
-        ],
-      },
-    }),
+    ...criteria.where,
+    ...prisma.getSearchParameters(query, [{ username: 'contains' }]),
     deletedAt: null,
   };