feat: Add an ability to download only selected databases

Author: Dmitry Shirokov

index.js

@@ -1,15 +1,15 @@
-const path = require("path");
+const path = require('path');
 
-const {selectedDbs} = require('./utils');
-const selected = selectedDbs();
+const { getSelectedDbs } = require('./utils');
+const selected = getSelectedDbs();
 
 const makePath = (type) => path.resolve(__dirname, `dbs/GeoLite2-${type}.mmdb`);
 
-const paths = selected.reduce((a,c) => {
+const paths = selected.reduce((a, c) => {
   a[c.toLowerCase()] = makePath(c);
   return a;
 }, {});
 
 module.exports = {
-  paths
+  paths,
 };

scripts/postinstall.js

@@ -3,7 +3,7 @@ const https = require('https');
 const zlib = require('zlib');
 const tar = require('tar');
 const path = require('path');
-const {getLicense, selectedDbs} = require('../utils');
+const { getLicense, getSelectedDbs } = require('../utils');
 
 let licenseKey;
 try {
@@ -35,14 +35,10 @@ if (!licenseKey) {
 const link = (edition) =>
   `https://download.maxmind.com/app/geoip_download?edition_id=${edition}&license_key=${licenseKey}&suffix=tar.gz`;
 
-
-const selected = selectedDbs();
-const links = [
-  'City',
-  'Country',
-  'ASN',
-].filter(e => selected.includes(e))
- .map(e => link(`GeoLite2-${e}`));
+const selected = getSelectedDbs();
+const links = ['City', 'Country', 'ASN']
+  .filter((e) => selected.includes(e))
+  .map((e) => link(`GeoLite2-${e}`));
 
 const downloadPath = path.join(__dirname, '..', 'dbs');
 

utils.js

@@ -1,7 +1,8 @@
 const path = require('path');
 const fs = require('fs');
 
-const cwdPath = (file) => path.join(process.env['INIT_CWD'] || process.cwd(), file);
+const cwdPath = (file) =>
+  path.join(process.env['INIT_CWD'] || process.cwd(), file);
 
 const getConfig = () => {
   const packageJsonFilename = cwdPath('package.json');
@@ -11,51 +12,55 @@ const getConfig = () => {
 
 const getLicense = () => {
   const envKey = process.env.MAXMIND_LICENSE_KEY;
-  if(envKey) return envKey;
-  
+  if (envKey) return envKey;
+
   const config = getConfig();
   const licenseKey = config['license-key'];
-  if(licenseKey) return licenseKey;
+  if (licenseKey) return licenseKey;
 
   const configFile = config['license-file'];
-  if(!configFile) return;
+  if (!configFile) return;
 
   const configFilePath = cwdPath(configFile);
-  return fs.existsSync(configFilePath) 
-       ? fs.readFileSync(configFilePath, 'utf8').trim()
-       : undefined;
-}
+  return fs.existsSync(configFilePath)
+    ? fs.readFileSync(configFilePath, 'utf8').trim()
+    : undefined;
+};
 
-const selectedDbs = () => {
+const getSelectedDbs = () => {
   const config = getConfig();
   const selected = config['selected-dbs'];
   const valids = ['City', 'Country', 'ASN'];
-  if(!selected) return valids;
-  if(!Array.isArray(selected)) {
+  if (!selected) return valids;
+  if (!Array.isArray(selected)) {
     console.error('selected-dbs property must have be an array.');
     process.exit(1);
   }
 
-  if(selected.length === 0) return valids;
+  if (selected.length === 0) return valids;
 
-  if(selected.length > 3) {
-    console.error('selected-dbs has too many values, there are only three valid values: City, Country, ASN.');
+  if (selected.length > 3) {
+    console.error(
+      `selected-dbs has too many values, there are only three valid values: ${valids.join(', ')}.`
+    );
     process.exit(1);
   }
 
-  for(const value of selected) {
+  for (const value of selected) {
     const index = valids.indexOf(value);
-    if(index === -1) {
-      console.error(`Invalid value in selected-dbs: ${value}. The only valid values are: City, Country, ASN.`);
+    if (index === -1) {
+      console.error(
+        `Invalid value in selected-dbs: ${value}. The only valid values are: ${valids.join(', ')}.`
+      );
       process.exit(1);
     }
   }
 
   return selected;
-}
+};
 
 module.exports = {
   getConfig,
   getLicense,
-  selectedDbs
-}
+  getSelectedDbs,
+};