ee-core 1.2.10 → 1.3.0-beta.1

package/config/config.default.js

@@ -247,6 +247,17 @@ module.exports = appInfo => {
     enable: false
   }; 
 
+  /**
+   * 加密配置
+   */
+  config.encrypt = {
+    type: 'bytecode', // bytecode | confusion
+    directory: [
+      'electron'
+    ],
+    fileExt: ['.js'],
+  };
+
   /**
    * TODO storage
    */

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "ee-core",
-  "version": "1.2.10",
+  "version": "1.3.0-beta.1",
   "description": "ee core",
   "main": "index.js",
   "scripts": {
@@ -13,7 +13,6 @@
   },
   "dependencies": {
     "agentkeepalive": "^4.2.0",
-    "better-sqlite3": "^7.6.0",
     "bytenode": "^1.3.6",
     "co": "^4.6.0",
     "debug": "^4.3.3",

package/tools/encrypt.js

@@ -6,19 +6,20 @@ const fsPro = require('fs-extra');
 const is = require('is-type-of');
 const UglifyJS = require('uglify-js');
 const bytenode = require('bytenode');
+const utility = require('utility');
+const crypto = require('crypto');
 
 class Encrypt {
   constructor() {
     this.basePath = process.cwd();
-    const directory = [
-      'electron',
-    ];
     this.dirs = [];
-    this.type = '';
-    this.configPath = '';
-    this.config = null;
-    this.filesExt = ['.js', '.json', '.node'];
     this.encryptCodeDir = path.join(this.basePath, 'public');
+    this.config = this.loadConfig('encrypt');
+    this.filesExt = this.config.fileExt || ['.js'];
+    this.type = this.config.type || 'bytecode';
+    const directory = this.config.directory || ['electron'];
+    this.tmpFile = '';
+    this.mapFile = '';
 
     // argv
     for (let i = 0; i < process.argv.length; i++) {
@@ -26,11 +27,6 @@ class Encrypt {
       if (tmpArgv.indexOf('--type=') !== -1) {
         this.type = tmpArgv.substring(7);
       }
-      if (tmpArgv.indexOf('--config=') !== -1) {
-        let configPathStr = tmpArgv.substring(9);
-        this.configPath = path.join(this.basePath, configPathStr);
-        this.config = fs.existsSync(this.configPath) ? require(this.configPath) : null;
-      }
     }
 
     // 检查存在的目录
@@ -43,17 +39,6 @@ class Encrypt {
     console.log('[ee-core] [encrypt] dirs:', this.dirs);
   }
 
-  /**
-   * 检查
-   */
-  check () {
-    if (this.configPath.length > 0 && !is.object(this.config)) {
-      console.log('[ee-core] [encrypt] ERROR: config file is invalid');
-      return false;
-    }
-    return true;
-  }
-
   /**
    * 备份 electron 目录代码
    */
@@ -85,6 +70,23 @@ class Encrypt {
     console.log('[ee-core] [encrypt] backup end');
     return true;
   }
+  
+  prepare () {
+    if (this.type == 'bytecode') {
+      let filename = this.config.mangle || this.config.mangle.file || null;
+      if (filename) {
+        this.tmpFile = path.join(this.encryptCodeDir, 'electron', 'tmp.json');
+        this.mapFile = path.join(this.encryptCodeDir, 'electron', filename);
+        fs.writeFileSync(this.mapFile, '');
+        const content = {
+          nameMap: {}
+        };
+        utility.writeJSONSync(this.tmpFile, content);
+      }
+    }
+
+    return true;
+  }
 
   /**
    * 加密代码
@@ -171,7 +173,7 @@ class Encrypt {
       output: jscFile,
       electron: true
     });
-    fs.rmSync(curPath, {force: true});
+	  fsPro.removeSync(curPath);
   }
 
   /**
@@ -180,7 +182,7 @@ class Encrypt {
   rmBackup (dir) {
     if (fs.existsSync(dir)) {
       console.log('[ee-core] [encrypt] clean old directory:', dir);
-      fs.rmSync(dir, {recursive: true, force: true});
+      fsPro.removeSync(dir);
     }
     return;
   }
@@ -233,12 +235,32 @@ class Encrypt {
       fs.chmodSync(path, mode);
     }
   };
+
+  loadConfig (prop) {
+    const filepath = path.join(this.basePath, 'electron', 'config', 'config.default.js');
+    const obj = require(filepath);
+    if (!obj) return obj;
+
+    let ret = obj;
+    if (is.function(obj) && !is.class(obj)) {
+      ret = obj();
+    }
+    return ret[prop] || {};
+  };
+
+  md5 (file) {
+    const buffer = fs.readFileSync(file);
+    const hash = crypto.createHash('md5');
+    hash.update(buffer, 'utf8');
+    const str = hash.digest('hex');
+    return str;
+  }
 }
 
 const run = () => {
   const e = new Encrypt();
-  if (!e.check()) return;
   if (!e.backup()) return;
+  //if (!e.prepare()) return;
   e.encrypt();
 }