mirror of
https://github.com/kremalicious/umami.git
synced 2024-12-30 16:57:44 +01:00
132 lines
2.9 KiB
JavaScript
132 lines
2.9 KiB
JavaScript
import firstBy from 'thenby';
|
|
import { BROWSERS } from './constants';
|
|
import { removeTrailingSlash, removeWWW, getDomainName } from './url';
|
|
|
|
export const urlFilter = (data, { raw }) => {
|
|
const isValidUrl = url => {
|
|
return url !== '' && !url.startsWith('#');
|
|
};
|
|
|
|
if (raw) {
|
|
return data.filter(({ x }) => isValidUrl(x));
|
|
}
|
|
|
|
const cleanUrl = url => {
|
|
try {
|
|
const { pathname, search } = new URL(url, location.origin);
|
|
|
|
if (search.startsWith('?/')) {
|
|
return `${pathname}${search}`;
|
|
}
|
|
|
|
return removeTrailingSlash(pathname);
|
|
} catch {
|
|
return null;
|
|
}
|
|
};
|
|
|
|
const map = data.reduce((obj, { x, y }) => {
|
|
if (!isValidUrl(x)) {
|
|
return obj;
|
|
}
|
|
|
|
const url = cleanUrl(x);
|
|
|
|
if (url) {
|
|
if (!obj[url]) {
|
|
obj[url] = y;
|
|
} else {
|
|
obj[url] += y;
|
|
}
|
|
}
|
|
|
|
return obj;
|
|
}, {});
|
|
|
|
return Object.keys(map)
|
|
.map(key => ({ x: key, y: map[key] }))
|
|
.sort(firstBy('y', -1).thenBy('x'));
|
|
};
|
|
|
|
export const refFilter = (data, { domain, domainOnly, raw }) => {
|
|
const domainName = getDomainName(domain);
|
|
const regex = new RegExp(`http[s]?://${domainName}`);
|
|
const links = {};
|
|
|
|
const isValidRef = ref => {
|
|
return ref !== '' && !ref.startsWith('/') && !ref.startsWith('#');
|
|
};
|
|
|
|
if (raw) {
|
|
return data.filter(({ x }) => isValidRef(x) && !regex.test(x));
|
|
}
|
|
|
|
const cleanUrl = url => {
|
|
try {
|
|
const { hostname, origin, pathname, searchParams, protocol } = new URL(url);
|
|
|
|
if (hostname === domainName) {
|
|
return null;
|
|
}
|
|
|
|
if (domainOnly && hostname) {
|
|
return hostname;
|
|
}
|
|
|
|
if (!origin || origin === 'null') {
|
|
return `${protocol}${removeTrailingSlash(pathname)}`;
|
|
}
|
|
|
|
if (protocol.startsWith('http')) {
|
|
const path = removeTrailingSlash(pathname);
|
|
const ref = searchParams.get('ref');
|
|
const query = ref ? `?ref=${ref}` : '';
|
|
|
|
return removeTrailingSlash(`${removeWWW(hostname)}${path}`) + query;
|
|
}
|
|
|
|
return null;
|
|
} catch {
|
|
return null;
|
|
}
|
|
};
|
|
|
|
const map = data.reduce((obj, { x, y }) => {
|
|
if (!isValidRef(x)) {
|
|
return obj;
|
|
}
|
|
|
|
const url = cleanUrl(x);
|
|
|
|
if (!domainOnly && !raw) {
|
|
links[url] = x;
|
|
}
|
|
|
|
if (url) {
|
|
if (!obj[url]) {
|
|
obj[url] = y;
|
|
} else {
|
|
obj[url] += y;
|
|
}
|
|
}
|
|
|
|
return obj;
|
|
}, {});
|
|
|
|
return Object.keys(map)
|
|
.map(key => ({ x: key, y: map[key], w: links[key] }))
|
|
.sort(firstBy('y', -1).thenBy('x'));
|
|
};
|
|
|
|
export const browserFilter = data =>
|
|
data.map(({ x, y }) => ({ x: BROWSERS[x] || x, y })).filter(({ x }) => x);
|
|
|
|
export const osFilter = data => data.filter(({ x }) => x);
|
|
|
|
export const deviceFilter = data => data.filter(({ x }) => x);
|
|
|
|
export const percentFilter = data => {
|
|
const total = data.reduce((n, { y }) => n + y, 0);
|
|
return data.map(({ x, y, ...props }) => ({ x, y, z: total ? (y / total) * 100 : 0, ...props }));
|
|
};
|