watson/umami
1
0
Fork 0
mirror of https://github.com/umami-software/umami.git synced 2026-02-20 12:35:38 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

Split logic for findSession.

Browse source
This commit is contained in:
Brian Cao 2023-01-01 23:20:14 -08:00
parent e487da72c3
commit f2fa514a31
5 changed files with 161 additions and 130 deletions
Download patch file Download diff file Expand all files Collapse all files

53
lib/middleware.ts
View file

@ -1,20 +1,57 @@
import { createMiddleware, unauthorized, badRequest, parseSecureToken } from 'next-basics';
import debug from 'debug';
import cors from 'cors';
import { validate } from 'uuid';
import redis from '@umami/redis-client';
import { findSession } from 'lib/session';
import cors from 'cors';
import debug from 'debug';
import { getAuthToken, parseShareToken } from 'lib/auth';
import { secret } from 'lib/crypto';
import { ROLES } from 'lib/constants';
import { secret } from 'lib/crypto';
import { isOverApiLimit, findSession, findWebsite, useSessionCache } from 'lib/session';
import {
badRequest,
createMiddleware,
parseSecureToken,
tooManyRequest,
unauthorized,
} from 'next-basics';
import { NextApiRequestCollect } from 'pages/api/collect';
import { validate } from 'uuid';
import { getUser } from '../queries';
import { getJsonBody } from './detect';
const log = debug('umami:middleware');
export const useCors = createMiddleware(cors());
export const useSession = createMiddleware(async (req: any, res, next) => {
const session = await findSession(req);
export const useSession = createMiddleware(async (req: NextApiRequestCollect, res, next) => {
// Verify payload
const { payload } = getJsonBody(req);
const { website: websiteId } = payload;
if (!payload) {
log('useSession: No payload');
return badRequest(res);
}
// Get session from cache
let session = await useSessionCache(req);
// Get or create session
if (!session) {
const website = await findWebsite(websiteId);
if (!website) {
log('useSession: Website not found');
return badRequest(res);
}
if (process.env.ENABLE_COLLECT_LIMIT) {
if (isOverApiLimit(website)) {
return tooManyRequest(res, 'Collect currently exceeds monthly limit of 10000.');
}
}
session = await findSession(req, payload);
}
if (!session) {
log('useSession: Session not found');