修改初始化

zhaochongyi

2024-08-26 d1849709bd583ca114e632ec5197991c5febae7d

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
const { app, session } = require('electron');
const _ = require('lodash');
const fs = require('fs');
const path = require('path');
const Log = require('ee-core/log');
 
/**
 * 扩展插件 （electron自身对该功能并不完全支持，官方也不建议使用）
 * @class
 */
class ChromeExtensionAddon {
 
  constructor() {
  }
 
  /**
   * 创建
   */
  async create () {
    Log.info('[addon:chromeExtension] load');
 
    const extensionIds = this.getAllIds();
    for (let i = 0; i < extensionIds.length; i++) {
      await this.load(extensionIds[i]);
    }
  }
 
  /**
   * 获取扩展id列表（crx解压后的目录名，即是该扩展的id）
   */
  getAllIds () {
    const extendsionDir = this.getDirectory();
    const ids = this.getDirs(extendsionDir);
 
    return ids;
  }
 
  /**
   * 扩展所在目录
   */
  getDirectory () {
    let extensionDirPath = '';
    let variablePath = 'build'; // 打包前路径
    if (app.isPackaged) {
      variablePath = '..'; // 打包后路径
    }
    extensionDirPath = path.join(app.getAppPath(), variablePath, "extraResources", "chromeExtension");
  
    return extensionDirPath;
  }
 
  /**
   * 加载扩展
   */
  async load (extensionId = '') {
    if (_.isEmpty(extensionId)) {
      return false
    }
    
    try {
      const extensionPath = path.join(this.getDirectory(), extensionId);
      Log.info('[addon:chromeExtension] extensionPath:', extensionPath);
      await session.defaultSession.loadExtension(extensionPath, { allowFileAccess: true });
    } catch (e) {
      Log.info('[addon:chromeExtension] load extension error extensionId:%s, errorInfo:%s', extensionId, e.toString());
      return false
    }
  
    return true
  }
  
  /**
   * 获取目录下所有文件夹
   */
  getDirs(dir) {
    if (!dir) {
      return [];
    }
 
    const components = [];
    const files = fs.readdirSync(dir);
    files.forEach(function(item, index) {
      const stat = fs.lstatSync(dir + '/' + item);
      if (stat.isDirectory() === true) {
        components.push(item);
      }
    });
 
    return components;
  };
}
 
ChromeExtensionAddon.toString = () => '[class ChromeExtensionAddon]';
module.exports = ChromeExtensionAddon;