m/umami
1
0
Fork 0
mirror of https://github.com/kremalicious/umami.git synced 2025-02-14 21:10:34 +01:00
Code Issues Packages Projects Releases Wiki Activity

Add incr on block.

Browse Source
This commit is contained in:
Brian Cao 2023-06-01 12:49:28 -07:00
parent 968574dda1
commit 7e587198dd
2 changed files with 8 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
lib/cache.ts
View File

@ -54,6 +54,11 @@ async function fetchUserBlock(userId: string) {
return redis.get(key); return redis.get(key);
} }
async function incrementUserBlock(userId: string) {
const key = `user:block:${userId}`;
return redis.incr(key);
}
export default { export default {
fetchWebsite, fetchWebsite,
storeWebsite, storeWebsite,
@ -65,5 +70,6 @@ export default {
storeSession, storeSession,
deleteSession, deleteSession,
fetchUserBlock, fetchUserBlock,
incrementUserBlock,
enabled: redis.enabled, enabled: redis.enabled,
}; };

2
lib/session.ts
View File

@ -99,6 +99,8 @@ export async function findSession(req: NextApiRequestCollect) {
async function checkUserBlock(userId: string) { async function checkUserBlock(userId: string) {
if (process.env.ENABLE_BLOCKER && (await cache.fetchUserBlock(userId))) { if (process.env.ENABLE_BLOCKER && (await cache.fetchUserBlock(userId))) {
await cache.incrementUserBlock(userId);
throw new Error('Usage Limit.'); throw new Error('Usage Limit.');
} }
} }