Add knex AliasesDB

src/db/aliases.js

@@ -0,0 +1,59 @@
+// @flow
+
+import { Database } from '../database';
+import { LoggerFactory } from '@calzoneman/jsli';
+import net from 'net';
+
+const LOGGER = LoggerFactory.getLogger('AliasesDB');
+
+class AliasesDB {
+    db: Database;
+
+    constructor(db: Database) {
+        this.db = db;
+    }
+
+    async addAlias(ip: string, name: string) {
+        return this.db.runTransaction(async tx => {
+            try {
+                await tx.table('aliases')
+                        .where({ ip, name })
+                        .del();
+                await tx.table('aliases')
+                        .insert({ ip, name, time: Date.now() });
+            } catch (error) {
+                LOGGER.error('Failed to save alias: %s (ip=%s, name=%s)',
+                        error.message, ip, name);
+            }
+        });
+    }
+
+    async getAliasesByIP(ip: string): Promise<Array<string>> {
+        return this.db.runTransaction(async tx => {
+            const query = tx.table('aliases');
+            if (net.isIP(ip)) {
+                query.where({ ip: ip })
+            } else {
+                const delimiter = /^[0-9]+\./.test(ip) ? '.' : ':';
+                query.where('ip', 'LIKE', ip + delimiter + '%');
+            }
+
+            const rows = await query.select()
+                    .distinct('name')
+                    .orderBy('time', 'desc')
+                    .limit(5);
+            return rows.map(row => row.name);
+        });
+    }
+
+    async getIPsByName(name: string): Promise<Array<string>> {
+        return this.db.runTransaction(async tx => {
+            const rows = await tx.table('aliases')
+                    .select('ip')
+                    .where({ name });
+            return rows.map(row => row.ip);
+        });
+    }
+}
+
+export { AliasesDB };