125 lines
2.8 KiB
TypeScript
125 lines
2.8 KiB
TypeScript
import { User, Website, Team, TeamUser } from '@prisma/client';
|
|
import redis from '@umami/redis-client';
|
|
import { lightFormat, startOfMonth } from 'date-fns';
|
|
import { getAllWebsitesByUser, getSession, getUser, getViewTotals, getWebsite } from '../queries';
|
|
|
|
const DELETED = 'DELETED';
|
|
|
|
async function fetchObject(key, query) {
|
|
const obj = await redis.get(key);
|
|
|
|
if (obj === DELETED) {
|
|
return null;
|
|
}
|
|
|
|
if (!obj) {
|
|
return query().then(async data => {
|
|
if (data) {
|
|
await redis.set(key, data);
|
|
}
|
|
|
|
return data;
|
|
});
|
|
}
|
|
|
|
return obj;
|
|
}
|
|
|
|
async function storeObject(key, data) {
|
|
return redis.set(key, data);
|
|
}
|
|
|
|
async function deleteObject(key, soft = false) {
|
|
return soft ? redis.set(key, DELETED) : redis.del(key);
|
|
}
|
|
|
|
async function fetchWebsite(id): Promise<
|
|
Website & {
|
|
team?: Team & { teamUsers: TeamUser[] };
|
|
}
|
|
> {
|
|
return fetchObject(`website:${id}`, () => getWebsite({ id }, true));
|
|
}
|
|
|
|
async function storeWebsite(data) {
|
|
const { id } = data;
|
|
const key = `website:${id}`;
|
|
|
|
return storeObject(key, data);
|
|
}
|
|
|
|
async function deleteWebsite(id) {
|
|
return deleteObject(`website:${id}`);
|
|
}
|
|
|
|
async function fetchUser(id): Promise<User> {
|
|
return fetchObject(`user:${id}`, () => getUser({ id }, { includePassword: true }));
|
|
}
|
|
|
|
async function storeUser(data) {
|
|
const { id } = data;
|
|
const key = `user:${id}`;
|
|
|
|
return storeObject(key, data);
|
|
}
|
|
|
|
async function fetchCollectLimit(userId): Promise<Number> {
|
|
const monthDate = startOfMonth(new Date());
|
|
const monthKey = lightFormat(monthDate, 'yyyy-MM');
|
|
|
|
return fetchObject(`collectLimit:${userId}:${monthKey}`, async () => {
|
|
const websiteIds = (await getAllWebsitesByUser(userId)).map(a => a.id);
|
|
|
|
return getViewTotals(websiteIds, monthDate).then(data => data.views);
|
|
});
|
|
}
|
|
|
|
async function deleteCollectLimit(userId): Promise<Number> {
|
|
const monthDate = startOfMonth(new Date());
|
|
const monthKey = lightFormat(monthDate, 'yyyy-MM');
|
|
|
|
return deleteObject(`collectLimit:${userId}:${monthKey}`);
|
|
}
|
|
|
|
async function incrementCollectLimit(userId): Promise<Number> {
|
|
const monthDate = startOfMonth(new Date());
|
|
const monthKey = lightFormat(monthDate, 'yyyy-MM');
|
|
|
|
return redis.incr(`collectLimit:${userId}:${monthKey}`);
|
|
}
|
|
|
|
async function deleteUser(id) {
|
|
return deleteObject(`user:${id}`);
|
|
}
|
|
|
|
async function fetchSession(id) {
|
|
return fetchObject(`session:${id}`, () => getSession({ id }));
|
|
}
|
|
|
|
async function storeSession(data) {
|
|
const { id } = data;
|
|
const key = `session:${id}`;
|
|
|
|
return storeObject(key, data);
|
|
}
|
|
|
|
async function deleteSession(id) {
|
|
return deleteObject(`session:${id}`);
|
|
}
|
|
|
|
export default {
|
|
fetchWebsite,
|
|
storeWebsite,
|
|
deleteWebsite,
|
|
fetchUser,
|
|
storeUser,
|
|
deleteUser,
|
|
fetchSession,
|
|
storeSession,
|
|
deleteSession,
|
|
fetchCollectLimit,
|
|
incrementCollectLimit,
|
|
deleteCollectLimit,
|
|
enabled: redis.enabled,
|
|
};
|