mirror of
https://github.com/kremalicious/umami.git
synced 2024-11-16 02:05:04 +01:00
101 lines
2.5 KiB
JavaScript
101 lines
2.5 KiB
JavaScript
import { getPageviewMetrics, getSessionMetrics, getWebsiteById } from 'queries';
|
|
import { ok, methodNotAllowed, unauthorized, badRequest } from 'lib/response';
|
|
import { allowQuery } from 'lib/auth';
|
|
import { useCors } from 'lib/middleware';
|
|
|
|
const sessionColumns = ['browser', 'os', 'device', 'country', 'language'];
|
|
const pageviewColumns = ['url', 'referrer'];
|
|
|
|
function getTable(type) {
|
|
if (type === 'event') {
|
|
return 'event';
|
|
}
|
|
|
|
if (sessionColumns.includes(type)) {
|
|
return 'session';
|
|
}
|
|
|
|
return 'pageview';
|
|
}
|
|
|
|
function getColumn(type) {
|
|
if (type === 'event') {
|
|
return `concat(event_type, '\t', event_value)`;
|
|
}
|
|
return type;
|
|
}
|
|
|
|
export default async (req, res) => {
|
|
if (req.method === 'GET') {
|
|
await useCors(req, res);
|
|
|
|
if (!(await allowQuery(req))) {
|
|
return unauthorized(res);
|
|
}
|
|
|
|
const { id, type, start_at, end_at, url, referrer, os, browser, device, country } = req.query;
|
|
|
|
const websiteId = +id;
|
|
const startDate = new Date(+start_at);
|
|
const endDate = new Date(+end_at);
|
|
|
|
if (sessionColumns.includes(type)) {
|
|
let data = await getSessionMetrics(websiteId, startDate, endDate, type, {
|
|
os,
|
|
browser,
|
|
device,
|
|
country,
|
|
});
|
|
|
|
if (type === 'language') {
|
|
let combined = {};
|
|
|
|
for (let { x, y } of data) {
|
|
x = String(x).toLowerCase().split('-')[0];
|
|
|
|
if (!combined[x]) {
|
|
combined[x] = { x, y };
|
|
} else {
|
|
combined[x].y += y;
|
|
}
|
|
}
|
|
|
|
data = Object.values(combined);
|
|
}
|
|
|
|
return ok(res, data);
|
|
}
|
|
|
|
if (pageviewColumns.includes(type) || type === 'event') {
|
|
let domain;
|
|
if (type === 'referrer') {
|
|
const website = getWebsiteById(websiteId);
|
|
|
|
if (!website) {
|
|
return badRequest(res);
|
|
}
|
|
|
|
domain = website.domain;
|
|
}
|
|
|
|
const column = getColumn(type);
|
|
const table = getTable(type);
|
|
|
|
const data = await getPageviewMetrics(websiteId, startDate, endDate, column, table, {
|
|
domain,
|
|
url: type !== 'url' && table !== 'event' ? url : undefined,
|
|
referrer: type !== 'referrer' ? referrer : undefined,
|
|
os: type !== 'os' ? os : undefined,
|
|
browser: type !== 'browser' ? browser : undefined,
|
|
device: type !== 'device' ? device : undefined,
|
|
country: type !== 'country' ? country : undefined,
|
|
event_url: type !== 'url' && table === 'event' ? url : undefined,
|
|
});
|
|
|
|
return ok(res, data);
|
|
}
|
|
}
|
|
|
|
return methodNotAllowed(res);
|
|
};
|