You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
36 lines
1.2 KiB
TypeScript
36 lines
1.2 KiB
TypeScript
import 'hydrooj';
|
|
|
|
import fs from 'fs';
|
|
import { Reader } from 'maxmind';
|
|
import { findFileSync } from '@hydrooj/utils/lib/utils';
|
|
|
|
const buffer = fs.readFileSync(findFileSync('@hydrooj/geoip/GeoLite2-City.mmdb'));
|
|
const reader = new Reader(buffer);
|
|
|
|
export interface Result {
|
|
location?: string,
|
|
continent?: string,
|
|
country?: string,
|
|
city?: string,
|
|
display: string
|
|
}
|
|
|
|
export function lookup(ip: string, locale: string): Result {
|
|
const res: any = reader.get(ip);
|
|
if (!res) return { display: 'Unknown address'.translate(locale) };
|
|
const ret: Result = { display: '' };
|
|
if (res.location) ret.location = res.location;
|
|
if (res.continent) ret.continent = res.continent.names[locale] || res.continent.names.en;
|
|
if (res.country || res.registered_country) {
|
|
ret.country = (res.country || res.registered_country).names[locale]
|
|
|| (res.country || res.registered_country).names.en;
|
|
}
|
|
if (res.city) ret.city = res.city.names[locale] || res.city.names.en;
|
|
ret.display = `${ret.continent} ${ret.country}${ret.city ? ` ${ret.city}` : ''}`;
|
|
return ret;
|
|
}
|
|
|
|
export const provider = '<a href="http://www.maxmind.com" target="_blank">MaxMind</a>';
|
|
|
|
global.Hydro.lib.geoip = { provider, lookup };
|