umami/lib/auth.js

63 lines
1.5 KiB
JavaScript

import { validate } from 'uuid';
import { parseSecureToken, parseToken } from 'next-basics';
import { getWebsite } from 'queries';
import { SHARE_TOKEN_HEADER } from 'lib/constants';
import { secret } from 'lib/crypto';
export function getAuthToken(req) {
try {
const token = req.headers.authorization;
return parseSecureToken(token.split(' ')[1], secret());
} catch {
return null;
}
}
export function getShareToken(req) {
try {
return parseSecureToken(req.headers[SHARE_TOKEN_HEADER], secret());
} catch {
return null;
}
}
export function isValidToken(token, validation) {
try {
const result = parseToken(token, secret());
if (typeof validation === 'object') {
return !Object.keys(validation).find(key => result[key] !== validation[key]);
} else if (typeof validation === 'function') {
return validation(result);
}
} catch (e) {
return false;
}
return false;
}
export async function allowQuery(req, skipToken) {
const { id } = req.query;
const token = req.headers[SHARE_TOKEN_HEADER];
const website = await getWebsite(validate(id) ? { websiteUuid: id } : { id: +id });
if (website) {
if (token && token !== 'undefined' && !skipToken) {
return isValidToken(token, { websiteId: website.id });
}
const authToken = await getAuthToken(req);
if (authToken) {
const { userId, isAdmin } = authToken;
return isAdmin || website.userId === userId;
}
}
return false;
}