1
0
mirror of https://github.com/kremalicious/umami.git synced 2024-12-22 17:23:54 +01:00
umami/lib/request.js

89 lines
2.0 KiB
JavaScript
Raw Normal View History

import path from 'path';
2020-07-17 10:03:38 +02:00
import requestIp from 'request-ip';
import { browserName, detectOS } from 'detect-browser';
2020-07-23 05:45:09 +02:00
import isLocalhost from 'is-localhost-ip';
2020-07-18 09:25:29 +02:00
import maxmind from 'maxmind';
2020-09-27 05:46:20 +02:00
2020-08-07 04:37:19 +02:00
import {
DESKTOP_OS,
MOBILE_OS,
DESKTOP_SCREEN_WIDTH,
LAPTOP_SCREEN_WIDTH,
MOBILE_SCREEN_WIDTH,
} from './constants';
2020-07-18 06:01:49 +02:00
2021-04-26 03:32:06 +02:00
let lookup;
2020-07-17 10:03:38 +02:00
export function getIpAddress(req) {
// Custom header
if (req.headers[process.env.CLIENT_IP_HEADER]) {
return req.headers[process.env.CLIENT_IP_HEADER];
}
2020-07-18 09:25:29 +02:00
// Cloudflare
else if (req.headers['cf-connecting-ip']) {
2020-07-17 10:03:38 +02:00
return req.headers['cf-connecting-ip'];
}
2020-07-18 09:25:29 +02:00
2020-07-17 10:03:38 +02:00
return requestIp.getClientIp(req);
}
2020-08-07 04:14:44 +02:00
export function getDevice(screen, browser, os) {
2020-08-07 04:37:19 +02:00
if (!screen) return;
2020-08-07 04:14:44 +02:00
const [width] = screen.split('x');
2020-07-17 10:03:38 +02:00
2020-08-07 04:14:44 +02:00
if (DESKTOP_OS.includes(os)) {
if (os === 'Chrome OS' || width < DESKTOP_SCREEN_WIDTH) {
return 'laptop';
}
return 'desktop';
} else if (MOBILE_OS.includes(os)) {
2020-08-08 05:36:57 +02:00
if (os === 'Amazon OS' || width > MOBILE_SCREEN_WIDTH) {
2020-08-07 04:14:44 +02:00
return 'tablet';
}
return 'mobile';
}
2020-08-07 04:37:19 +02:00
if (width >= DESKTOP_SCREEN_WIDTH) {
return 'desktop';
} else if (width >= LAPTOP_SCREEN_WIDTH) {
return 'laptop';
} else if (width >= MOBILE_SCREEN_WIDTH) {
return 'tablet';
} else {
return 'mobile';
}
2020-07-17 10:03:38 +02:00
}
2020-07-18 09:25:29 +02:00
export async function getCountry(req, ip) {
// Cloudflare
if (req.headers['cf-ipcountry']) {
return req.headers['cf-ipcountry'];
}
// Ignore local ips
if (await isLocalhost(ip)) {
return;
}
// Database lookup
2021-04-26 03:32:06 +02:00
if (!lookup) {
2021-07-19 08:48:30 +02:00
lookup = await maxmind.open(path.resolve('./public/geo/GeoLite2-Country.mmdb'));
2021-04-26 03:32:06 +02:00
}
2020-07-18 09:25:29 +02:00
const result = lookup.get(ip);
return result?.country?.iso_code;
2020-07-17 10:03:38 +02:00
}
2020-08-07 04:14:44 +02:00
export async function getClientInfo(req, { screen }) {
2020-08-21 04:17:27 +02:00
const userAgent = req.headers['user-agent'];
2020-08-07 04:14:44 +02:00
const ip = getIpAddress(req);
const country = await getCountry(req, ip);
const browser = browserName(userAgent);
const os = detectOS(userAgent);
const device = getDevice(screen, browser, os);
return { userAgent, browser, os, ip, country, device };
}