watson/umami
1
0
Fork 0
mirror of https://github.com/umami-software/umami.git synced 2026-02-09 15:17:23 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions 5

Updated session and events queries. Added sessions page.

Browse source
This commit is contained in:
Mike Cao 2024-07-08 01:45:54 -07:00
parent 082a751ffe
commit db36c37d32
39 changed files with 376 additions and 180 deletions
Download patch file Download diff file Expand all files Collapse all files

226
src/queries/admin/website.ts
View file

@ -1,226 +0,0 @@
import { Prisma, Website } from '@prisma/client';
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);
}
export async function getWebsite(websiteId: string) {
return findWebsite({
where: {
id: websiteId,
},
});
}
export async function getSharedWebsite(shareId: string) {
return findWebsite({
where: {
shareId,
},
});
}
export async function getWebsites(
criteria: WebsiteFindManyArgs,
filters: PageParams,
): Promise<PageResult<Website[]>> {
const { query } = filters;
const where: Prisma.WebsiteWhereInput = {
...criteria.where,
...prisma.getSearchParameters(query, [
{
name: 'contains',
},
{ domain: 'contains' },
]),
deletedAt: null,
};
return prisma.pagedQuery('website', { ...criteria, where }, filters);
}
export async function getAllWebsites(userId: string) {
return prisma.client.website.findMany({
where: {
OR: [
{ userId },
{
team: {
deletedAt: null,
teamUser: {
some: {
userId,
},
},
},
},
],
deletedAt: null,
},
});
}
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,
): Promise<PageResult<Website[]>> {
return getWebsites(
{
where: {
userId,
},
include: {
user: {
select: {
username: true,
id: true,
},
},
},
},
{
orderBy: 'name',
...filters,
},
);
}
export async function getTeamWebsites(
teamId: string,
filters?: PageParams,
): Promise<PageResult<Website[]>> {
return getWebsites(
{
where: {
teamId,
},
include: {
createUser: {
select: {
id: true,
username: true,
},
},
},
},
filters,
);
}
export async function createWebsite(
data: Prisma.WebsiteCreateInput | Prisma.WebsiteUncheckedCreateInput,
): Promise<Website> {
return prisma.client.website.create({
data,
});
}
export async function updateWebsite(
websiteId: string,
data: Prisma.WebsiteUpdateInput | Prisma.WebsiteUncheckedUpdateInput,
): Promise<Website> {
return prisma.client.website.update({
where: {
id: websiteId,
},
data,
});
}
export async function resetWebsite(
websiteId: string,
): Promise<[Prisma.BatchPayload, Prisma.BatchPayload, Website]> {
const { client, transaction } = prisma;
const cloudMode = !!process.env.cloudMode;
return transaction([
client.eventData.deleteMany({
where: { websiteId },
}),
client.websiteEvent.deleteMany({
where: { websiteId },
}),
client.session.deleteMany({
where: { websiteId },
}),
client.website.update({
where: { id: websiteId },
data: {
resetAt: new Date(),
},
}),
]).then(async data => {
if (cloudMode) {
await redis.client.set(`website:${websiteId}`, data[3]);
}
return data;
});
}
export async function deleteWebsite(
websiteId: string,
): Promise<[Prisma.BatchPayload, Prisma.BatchPayload, Website]> {
const { client, transaction } = prisma;
const cloudMode = !!process.env.CLOUD_MODE;
return transaction([
client.eventData.deleteMany({
where: { websiteId },
}),
client.websiteEvent.deleteMany({
where: { websiteId },
}),
client.session.deleteMany({
where: { websiteId },
}),
client.report.deleteMany({
where: {
websiteId,
},
}),
cloudMode
? client.website.update({
data: {
deletedAt: new Date(),
},
where: { id: websiteId },
})
: client.website.delete({
where: { id: websiteId },
}),
]).then(async data => {
if (cloudMode) {
await redis.client.del(`website:${websiteId}`);
}
return data;
});
}