ee-core 1.2.9 → 1.2.10-beta.1

package/bin/tools.js

@@ -2,6 +2,7 @@
 
 const codeCompress = require('../tools/codeCompress');
 const replaceDist = require('../tools/replaceDist');
+const encrypt = require('../tools/encrypt');
 
 // argv
 const args = process.argv;
@@ -20,3 +21,7 @@ if (cmd == 'compress') {
 if (cmd == 'restore') {
   codeCompress.restore();
 }
+
+if (cmd == 'encrypt') {
+  encrypt.run();
+}

package/lib/application.js

@@ -2,6 +2,7 @@ const {app} = require('electron');
 const path = require('path');
 const EeApp = require('./eeApp');
 const debug = require('debug')('ee-core:Appliaction');
+const fs = require('fs');
 
 class Appliaction extends EeApp {
   constructor() {
@@ -45,6 +46,13 @@ class Appliaction extends EeApp {
       options.execDir = path.dirname(app.getPath('exe'));
     }
 
+    // Use encryption, base directory is public/electron
+    const encryptDir = path.join(app.getAppPath(), 'public', 'electron');
+    let isEncrypted = fs.existsSync(encryptDir);
+    if (options.env == 'prod' && isEncrypted) {
+      options.baseDir = encryptDir;
+    }
+
     // normalize env
     env.NODE_ENV = 'production';
     env.EE_HOME = options.homeDir;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "ee-core",
-  "version": "1.2.9",
+  "version": "1.2.10-beta.1",
   "description": "ee core",
   "main": "index.js",
   "scripts": {

package/tools/encrypt.js

@@ -0,0 +1,182 @@
+'use strict';
+
+const path = require('path');
+const fs = require('fs');
+const fsPro = require('fs-extra');
+const UglifyJS = require('uglify-js');
+
+class Encrypt {
+  constructor() {
+
+    // argv
+    this.type = '';
+    for (let i = 0; i < process.argv.length; i++) {
+      let tmpArgv = process.argv[i];
+      if (tmpArgv.indexOf('--type=') !== -1) {
+        this.type = tmpArgv.substring(7);
+      }
+    }
+
+    const directory = [
+      'electron',
+    ];
+    this.dirs = [];
+
+    this.basePath = process.cwd();
+    // if (this.type == 'uglify') {
+    //   this.encryptCodeDir = path.join(this.basePath, 'public');
+    // }
+    this.encryptCodeDir = path.join(this.basePath, 'public');
+
+    // 检查存在的目录
+    for (let i = 0; i < directory.length; i++) {
+      let codeDirPath = path.join(this.basePath, directory[i]);
+      if (fs.existsSync(codeDirPath)) {
+        this.dirs.push(directory[i]);
+      }
+    }
+    console.log('dirs:', this.dirs);
+  }
+
+  /**
+   * 备份 electron目录代码
+   */
+  backup () {
+    console.log('[ee-core] [encrypt] backup start');
+
+    for (let i = 0; i < this.dirs.length; i++) {
+      // check code dir
+      let codeDirPath = path.join(this.basePath, this.dirs[i]);
+      if (!fs.existsSync(codeDirPath)) {
+        console.log('[ee-core] [encrypt] backup ERROR: %s is not exist', codeDirPath);
+        return
+      }
+
+      let targetDir = path.join(this.encryptCodeDir, this.dirs[i]);
+
+      // remove old
+      this.rmBackup(targetDir);
+
+      // copy
+      console.log('[ee-core] [encrypt] backup targetDir:', targetDir);
+      if (!fs.existsSync(targetDir)) {
+        this.mkdir(targetDir);
+        this.chmodPath(targetDir, '777');
+      }
+
+      fsPro.copySync(codeDirPath, targetDir);
+    }
+    console.log('[ee-core] [encrypt] backup success');
+  }
+
+  /**
+   * 压缩代码
+   */
+  compress () {
+    console.log('[ee-core] [encrypt] compress start');
+    for (let i = 0; i < this.dirs.length; i++) {
+      let codeDirPath = path.join(this.encryptCodeDir, this.dirs[i]);
+      this.compressLoop(codeDirPath);
+    }
+    console.log('[ee-core] [encrypt] compress success');
+  };
+
+  compressLoop (dirPath) {
+    let files = [];
+    if (fs.existsSync(dirPath)) {
+      files = fs.readdirSync(dirPath);
+      files.forEach((file, index) => {
+        let curPath = dirPath + '/' + file;        
+        if (fs.statSync(curPath).isDirectory()) {
+          this.compressLoop(curPath);
+        } else {
+          if (path.extname(curPath) === '.js') {
+            this.miniFile(curPath);
+          }
+        }
+      });
+    }
+  }
+
+  miniFile (file) {
+    let code = fs.readFileSync(file, "utf8");
+    const options = {
+      mangle: {
+        toplevel: false,
+      },
+    };
+    
+    let result = UglifyJS.minify(code, options);
+    fs.writeFileSync(file, result.code, "utf8"); 
+  }
+
+  /**
+   * 移除备份
+   */
+  rmBackup (dir) {
+    if (fs.existsSync(dir)) {
+      console.log('[ee-core] [encrypt] clean old directory:', dir);
+      fs.rmSync(dir, {recursive: true, force: true});
+    }
+    return;
+  }
+
+  /**
+   * 检查文件是否存在
+   */
+  fileExist (filePath) {
+    try {
+      return fs.statSync(filePath).isFile();
+    } catch (err) {
+      return false;
+    }
+  };
+
+  mkdir (dirpath, dirname) {
+    // 判断是否是第一次调用
+    if (typeof dirname === 'undefined') {
+      if (fs.existsSync(dirpath)) {
+        return;
+      }
+      this.mkdir(dirpath, path.dirname(dirpath));
+    } else {
+      // 判断第二个参数是否正常，避免调用时传入错误参数
+      if (dirname !== path.dirname(dirpath)) {
+        this.mkdir(dirpath);
+        return;
+      }
+      if (fs.existsSync(dirname)) {
+        fs.mkdirSync(dirpath);
+      } else {
+        this.mkdir(dirname, path.dirname(dirname));
+        fs.mkdirSync(dirpath);
+      }
+    }
+  };
+
+  chmodPath (path, mode) {
+    let files = [];
+    if (fs.existsSync(path)) {
+      files = fs.readdirSync(path);
+      files.forEach((file, index) => {
+        const curPath = path + '/' + file;
+        if (fs.statSync(curPath).isDirectory()) {
+          this.chmodPath(curPath, mode); // 递归删除文件夹
+        } else {
+          fs.chmodSync(curPath, mode);
+        }
+      });
+      fs.chmodSync(path, mode);
+    }
+  };
+}
+
+const run = () => {
+  const e = new Encrypt();
+  e.backup();
+  e.compress();
+}
+
+module.exports = {
+  run
+};