geoip: use context.service

packages/geoip/index.ts

@@ -0,0 +1,34 @@
+import fs from 'fs';
+import { Reader } from 'maxmind';
+import { Context, findFileSync, Service } from 'hydrooj';
+
+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 apply() {
+    Context.service('geoip', class GeoIPService extends Service {
+        provider = '<a href="http://www.maxmind.com" target="_blank">MaxMind</a>';
+        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;
+        }
+    });
+}