Fix usage.

src/queries/admin/website.ts

@@ -3,6 +3,7 @@ import redis from '@umami/redis-client';
 import prisma from 'lib/prisma';
 import { PageResult, PageParams } from 'lib/types';
 import WebsiteFindManyArgs = Prisma.WebsiteFindManyArgs;
+import { ROLES } from 'lib/constants';
 
 async function findWebsite(criteria: Prisma.WebsiteFindUniqueArgs): Promise<Website> {
   return prisma.client.website.findUnique(criteria);
@@ -52,6 +53,27 @@ export async function getAllWebsites(userId: string) {
   });
 }
 
+export async function getAllUserWebsitesIncludingTeamOwner(userId: string) {
+  return prisma.client.website.findMany({
+    where: {
+      OR: [
+        { userId },
+        {
+          team: {
+            deletedAt: null,
+            teamUser: {
+              some: {
+                role: ROLES.teamOwner,
+                userId,
+              },
+            },
+          },
+        },
+      ],
+    },
+  });
+}
+
 export async function getUserWebsites(
   userId: string,
   filters?: PageParams,