Add the ability to use local optimization_config

Upd #216

Author: mu-hun

src/main/optimization.js

@@ -1,16 +1,31 @@
 /* eslint-disable global-require */
+import fs from 'fs';
+import path from 'path';
 import { downloadFile } from './utils/webutils';
 
 // Here we can access optimizable filters and its optimization percentages
 // eslint-disable-next-line max-len
-const OPTIMIZATION_PERCENT_URL = 'https://chrome.adtidy.org/optimization_config/percent.json?key=4DDBE80A3DA94D819A00523252FB6380';
-// eslint-disable-next-line max-len
-const OPTIMIZATION_STATS_URL = 'https://chrome.adtidy.org/filters/{0}/stats.json?key=4DDBE80A3DA94D819A00523252FB6380';
+const OPTIMIZATION_KEY = '4DDBE80A3DA94D819A00523252FB6380';
+export const OPTIMIZATION_PERCENT_URL = `https://chrome.adtidy.org/optimization_config/percent.json?key=${OPTIMIZATION_KEY}`;
+
+export const getOptimizationStatsUrl = (filterId) =>
+    // eslint-disable-next-line implicit-arrow-linebreak
+    `https://chrome.adtidy.org/filters/${filterId}/stats.json?key=${OPTIMIZATION_KEY}`;
+
+const downloadOptimizationStats = (filterId) =>
+    // eslint-disable-next-line implicit-arrow-linebreak
+    downloadFile(getOptimizationStatsUrl(filterId));
 
 let optimizationEnabled = true;
 
 let filtersOptimizationPercent = null;
 
+let localOptimizationConfigDir = null;
+
+export const setLocalOptimizationConfigDir = (dir) => {
+    localOptimizationConfigDir = dir;
+};
+
 /**
  * Downloads and caches filters optimization percentages configuration
  */
@@ -20,7 +35,11 @@ export const getFiltersOptimizationPercent = () => {
     }
 
     if (filtersOptimizationPercent === null) {
-        filtersOptimizationPercent = JSON.parse(downloadFile(OPTIMIZATION_PERCENT_URL));
+        const content = localOptimizationConfigDir
+            ? fs.readFileSync(path.join(localOptimizationConfigDir, 'percent.json'), 'utf-8')
+            : downloadFile(OPTIMIZATION_PERCENT_URL);
+
+        filtersOptimizationPercent = JSON.parse(content);
     }
 
     if (filtersOptimizationPercent.config.length === 0) {
@@ -40,12 +59,17 @@ export const getFilterOptimizationConfig = (filterId) => {
     }
 
     // config: [{filterId: 1, percent: 45}, ...]
-    const filterOptimizationPercent = getFiltersOptimizationPercent().config
+    const filterOptimizationPercent = getFiltersOptimizationPercent()
+        .config // eslint-disable-next-line
         .find((config) => config.filterId === filterId);
 
     let optimizationConfig = null;
     if (optimizationEnabled && filterOptimizationPercent) {
-        optimizationConfig = JSON.parse(downloadFile(OPTIMIZATION_STATS_URL.replace('{0}', filterId)));
+        const content = localOptimizationConfigDir
+            ? fs.readFileSync(path.join(localOptimizationConfigDir, `${filterId}_stats.json`), 'utf-8')
+            : downloadOptimizationStats(filterId);
+
+        optimizationConfig = JSON.parse(content);
         if (!optimizationConfig || !optimizationConfig.groups || optimizationConfig.groups.length === 0) {
             throw new Error(`Unable to retrieve optimization stats for ${filterId}`);
         }