hunter-open-sdk 2.0.0-beta.3 → 2.0.0-beta.30

package/lib/abilities/activate/index.js

@@ -0,0 +1,51 @@
+"use strict";
+
+Object.defineProperty(exports, "__esModule", {
+  value: true
+});
+exports.activate = activate;
+var _webWorker = _interopRequireDefault(require("web-worker"));
+var _url = _interopRequireDefault(require("url"));
+var _utils = require("../../utils/utils.common");
+var _utils2 = require("../../utils/utils.path");
+var _utils3 = require("../../utils/utils.ability");
+function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }
+class Activate {
+  constructor() {
+    this.initWorker();
+    this.initWorkerListener();
+  }
+  worker = null;
+  callback = null;
+  exec(payload, callback) {
+    this.callback = typeof callback === 'function' ? callback : _utils3.defaultCallback;
+    this.worker && this.worker.postMessage({
+      payload
+    });
+  }
+  initWorkerListener() {
+    this.worker && this.worker.addEventListener('message', event => {
+      const {
+        payload
+      } = event.data;
+      this.callback(payload);
+      this.callback = null;
+    });
+  }
+  initWorker() {
+    if (!this.worker) {
+      const workerPath = new _url.default.URL('file://' + (0, _utils2.convertPathWithCurrentEnv)(__dirname, './worker.js'));
+      this.worker = new _webWorker.default(workerPath, {
+        type: 'module'
+      });
+    }
+  }
+}
+let entity = null;
+function activate(params, callback) {
+  if (!entity) {
+    entity = new Activate();
+  }
+  const payload = (0, _utils.parsePayload)(params);
+  entity.exec(payload, callback);
+}

package/lib/abilities/activate/worker.js

@@ -0,0 +1,19 @@
+"use strict";
+
+const {
+  initWorkerNodePath
+} = require("../../utils/utils.worker");
+
+// 先注册NodePath后才能正常使用node包，请勿挪动位置
+initWorkerNodePath();
+const {
+  abilityCallKit
+} = require("../../utils/utils.global");
+function onMainMessage(event) {
+  const {
+    payload
+  } = event.data;
+  const res = abilityCallKit('activate', payload);
+  postMessage(res);
+}
+addEventListener('message', onMainMessage);

package/lib/abilities/erase/index.js

@@ -0,0 +1,51 @@
+"use strict";
+
+Object.defineProperty(exports, "__esModule", {
+  value: true
+});
+exports.erase = erase;
+var _webWorker = _interopRequireDefault(require("web-worker"));
+var _url = _interopRequireDefault(require("url"));
+var _utils = require("../../utils/utils.common");
+var _utils2 = require("../../utils/utils.path");
+var _utils3 = require("../../utils/utils.ability");
+function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }
+class Erase {
+  constructor() {
+    this.initWorker();
+    this.initWorkerListener();
+  }
+  worker = null;
+  callback = null;
+  exec(payload, callback) {
+    this.callback = typeof callback === 'function' ? callback : _utils3.defaultCallback;
+    this.worker && this.worker.postMessage({
+      payload
+    });
+  }
+  initWorkerListener() {
+    this.worker && this.worker.addEventListener('message', event => {
+      const {
+        payload
+      } = event.data;
+      this.callback(payload);
+      this.callback = null;
+    });
+  }
+  initWorker() {
+    if (!this.worker) {
+      const workerPath = new _url.default.URL('file://' + (0, _utils2.convertPathWithCurrentEnv)(__dirname, './worker.js'));
+      this.worker = new _webWorker.default(workerPath, {
+        type: 'module'
+      });
+    }
+  }
+}
+let entity = null;
+function erase(params, callback) {
+  if (!entity) {
+    entity = new Erase();
+  }
+  const payload = (0, _utils.parsePayload)(params);
+  entity.exec(payload, callback);
+}

package/lib/abilities/erase/worker.js

@@ -0,0 +1,19 @@
+"use strict";
+
+const {
+  initWorkerNodePath
+} = require("../../utils/utils.worker");
+
+// 先注册NodePath后才能正常使用node包，请勿挪动位置
+initWorkerNodePath();
+const {
+  abilityCallKit
+} = require("../../utils/utils.global");
+function onMainMessage(event) {
+  const {
+    payload
+  } = event.data;
+  const res = abilityCallKit('erase', payload);
+  postMessage(res);
+}
+addEventListener('message', onMainMessage);

package/lib/abilities/eventManager/index.js

@@ -0,0 +1,347 @@
+"use strict";
+
+Object.defineProperty(exports, "__esModule", {
+  value: true
+});
+exports.unregisterEvent = exports.registerEvent = void 0;
+var _usbmux = _interopRequireDefault(require("usbmux"));
+var _path = _interopRequireDefault(require("path"));
+var _request = _interopRequireDefault(require("request"));
+var _fs = _interopRequireDefault(require("fs"));
+var Sentry = _interopRequireWildcard(require("@sentry/browser"));
+var _utils = require("../../utils/utils.env");
+var _constants = require("../../constants");
+var _utils2 = require("../../utils/utils.common");
+var _winDriver = require("../../events/driver/winDriver");
+var _utils3 = require("../../utils/utils.global");
+var _deviceHelper = require("../../helper/deviceHelper");
+var _utils4 = require("../../utils/utils.initDll");
+function _getRequireWildcardCache(e) { if ("function" != typeof WeakMap) return null; var r = new WeakMap(), t = new WeakMap(); return (_getRequireWildcardCache = function (e) { return e ? t : r; })(e); }
+function _interopRequireWildcard(e, r) { if (!r && e && e.__esModule) return e; if (null === e || "object" != typeof e && "function" != typeof e) return { default: e }; var t = _getRequireWildcardCache(r); if (t && t.has(e)) return t.get(e); var n = { __proto__: null }, a = Object.defineProperty && Object.getOwnPropertyDescriptor; for (var u in e) if ("default" !== u && Object.prototype.hasOwnProperty.call(e, u)) { var i = a ? Object.getOwnPropertyDescriptor(e, u) : null; i && (i.get || i.set) ? Object.defineProperty(n, u, i) : n[u] = e[u]; } return n.default = e, t && t.set(e, n), n; }
+function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }
+// saas回调
+let saasCallBack;
+
+// usb监听
+let usbListener = null;
+/**
+ * 获取资源文件的基础路径
+ */
+const getResourcePath = () => {
+  if ((0, _utils.isDev)()) {
+    // 开发环境：使用源码目录
+    return _path.default.resolve(__dirname, '../resources');
+  } else {
+    // 生产环境：使用 Electron 的资源路径
+    return _path.default.resolve(process.resourcesPath, 'resources');
+  }
+};
+
+/**
+ * 发布配对
+ * @param {String} udid 设备id
+ * @param {String} params 参数
+ * @returns {String} 发布配对结果
+ */
+const publishPair = (udid = '', params) => {
+  // 通过调用 trust_status 来判断信任状态
+  const kitParamsObj = JSON.parse(params);
+
+  // 合并参数
+  Object.assign(kitParamsObj, {
+    cmd: 'publish_pair',
+    udid
+  });
+  return (0, _utils3.callHunterInspectionKit)('TheForceKit', JSON.stringify(kitParamsObj));
+};
+
+/**
+ * 获取信任状态
+ * @param {String} udid 设备id
+ * @returns
+ */
+const getTrustStatus = async (udid = '', params, once = false) => {
+  if (!_deviceHelper.deviceHelper.hasUdid(udid)) {
+    // 防止一直循环占用cpu资源
+    return false;
+  }
+
+  // 合并参数
+  const paramsObj = Object.assign(JSON.parse(params), {
+    udid
+  });
+  let label = 'Windows';
+  if ((0, _utils.isMac)()) {
+    label = 'Mac';
+  }
+
+  // 通过调用 trust_status 来判断信任状态
+  const kitParamsObj = {
+    cmd: 'trust_status',
+    label,
+    params: JSON.stringify(paramsObj)
+  };
+
+  // 获取信任状态
+  const trustStatusRes = (0, _utils3.callHunterInspectionKit)('TheForceKit', JSON.stringify(kitParamsObj));
+  if (trustStatusRes) {
+    const trustStatusResObj = JSON.parse(trustStatusRes);
+    const {
+      Code,
+      Data
+    } = trustStatusResObj;
+    if (Code == 0 && (Data.imei || Data.serialNumber)) {
+      return Data;
+    } else if (!once) {
+      await (0, _utils2.sleep)();
+      return await getTrustStatus(udid, params);
+    }
+  }
+  return false;
+};
+const getTrustStatusOnce = async (udid, params) => {
+  return await getTrustStatus(udid, params, true);
+};
+
+/**
+ * 配对并获取信任状态
+ * @param {String} udid 设备id
+ * @returns
+ */
+const DevicePair = async (udid = '', params) => {
+  const callBack = () => {
+    saasCallBack({
+      data: {
+        eventName: _constants.EVENT_TYPE.TRUST_RESULT,
+        udid,
+        eventCode: 0
+      }
+    });
+  };
+
+  // 单次获取信任状态
+  const trustStatusOnceRes = await getTrustStatusOnce(udid, params);
+
+  // 未信任弹起信任窗口
+  if (!trustStatusOnceRes) {
+    // 发布配对
+    publishPair(udid, params);
+
+    // 通过 trust_status 来判断信任状态
+    const trustStatusRes = await getTrustStatus(udid, params);
+    if (trustStatusRes) {
+      callBack();
+    }
+    return;
+  }
+
+  // 已信任 直接返回
+  callBack();
+};
+
+// 写入dll文件version
+const writeDllVersion = version => {
+  const resourcePath = getResourcePath();
+  const versionPath = _path.default.resolve(resourcePath, 'version.online');
+  _fs.default.writeFile(versionPath, version, err => {
+    if (err) {
+      saasCallBack({
+        data: {
+          eventName: _constants.EVENT_TYPE.UPDATE_RESULT,
+          eventCode: -1 // eventCode 0 更新成功 -1 更新失败
+        }
+      });
+      return;
+    }
+    console.log('========写入version成功');
+    // 写入成功后 重新初始化dll
+    (0, _utils4.initHunterInspectionKit)();
+    saasCallBack({
+      data: {
+        eventName: _constants.EVENT_TYPE.UPDATE_RESULT,
+        eventCode: 0 // eventCode 0 更新成功 -1 更新失败
+      }
+    });
+  });
+};
+
+// 下载更新dll
+const updateDll = (packageUrl, targetVersion) => {
+  // 若需要更新 则下载完成后 覆盖本地dll文件
+  let dllOutPath;
+  const resourcePath = getResourcePath();
+  if ((0, _utils.isMac)()) {
+    if ((0, _utils.isArm64)()) {
+      dllOutPath = _path.default.resolve(resourcePath, `mac/libinspectionkit_arm64_${targetVersion}.dylib`);
+    }
+  } else {
+    dllOutPath = _path.default.resolve(resourcePath, `dll/libinspectionkit_x64_${targetVersion}.dll`);
+  }
+  const dllStream = _fs.default.createWriteStream(dllOutPath);
+  (0, _request.default)(packageUrl).pipe(dllStream).on('close', function (err) {
+    if (err) {
+      saasCallBack({
+        data: {
+          eventName: _constants.EVENT_TYPE.UPDATE_RESULT,
+          eventCode: -1 // eventCode 0 更新成功 -1 更新失败
+        }
+      });
+    } else {
+      // 写version文件内容
+      writeDllVersion(targetVersion);
+    }
+  });
+};
+
+/**
+ * 更新下载dll文件 并重写本地版本号
+ * @param {String} udid 设备id
+ * @returns
+ */
+const updateDllFileCheck = () => {
+  // mac的amd不支持更新
+  if ((0, _utils.isMac)() && (0, _utils.isAmd64)()) return;
+  const updateParams = {
+    cmd: 'hot_update',
+    label: (0, _utils.isWin)() ? 'Windows' : 'Mac',
+    params: {}
+  };
+  const updateRes = (0, _utils3.callHunterInspectionKit)('TheForceKit', JSON.stringify(updateParams));
+  if (updateRes) {
+    const updateResObj = JSON.parse(updateRes);
+    const {
+      Code,
+      Data
+    } = updateResObj;
+    if (Code == 0) {
+      const {
+        isHot,
+        url,
+        targetVersion
+      } = Data;
+
+      // 需要更新 则直接下载
+      isHot == 1 && updateDll(url, targetVersion);
+      saasCallBack({
+        data: {
+          eventName: _constants.EVENT_TYPE.NEED_UPDATE,
+          eventCode: isHot == 1 ? 0 : -1 // eventCode 0 需要更新 -1 不需要更新
+        }
+      });
+    } else {
+      // 错误的时候不需要更新
+      saasCallBack({
+        data: {
+          eventName: _constants.EVENT_TYPE.NEED_UPDATE,
+          eventCode: -1 // eventCode 0 需要更新 -1 不需要更新
+        }
+      });
+    }
+  }
+};
+
+/**
+ * 检查并安装驱动
+ */
+const checkAndInstallDriver = () => {
+  return new Promise((resolve, reject) => {
+    if (!(0, _utils.isWin)()) {
+      resolve();
+      return;
+    }
+    (0, _winDriver.checkHasDriver)().then(res => {
+      if (res) {
+        resolve();
+        return;
+      }
+      (0, _winDriver.downloadDriver)(percent => {
+        saasCallBack({
+          data: {
+            eventName: _constants.EVENT_TYPE.NEED_INSTALL_DRIVER,
+            eventCode: 0,
+            eventContent: JSON.stringify({
+              percent // 下载进度
+            })
+          }
+        });
+      }).then(() => {
+        // 成功
+        console.log('驱动下载并安装成功');
+        saasCallBack({
+          data: {
+            eventName: _constants.EVENT_TYPE.INSTALL_DRIVER_RESULT,
+            eventCode: 0 // 0 成功 -1 失败
+          }
+        });
+        resolve();
+      }).catch(err => {
+        // 失败
+        console.error('驱动下载或安装失败', err);
+        Sentry.captureException('驱动下载或安装失败:' + JSON.stringify(err));
+        saasCallBack({
+          data: {
+            eventName: _constants.EVENT_TYPE.INSTALL_DRIVER_RESULT,
+            eventCode: -1 // 0 成功 -1 失败
+          }
+        });
+        reject();
+      });
+    }).catch(err => {
+      console.error('驱动检测失败', err);
+      Sentry.captureException('驱动检测失败:' + JSON.stringify(err));
+      reject();
+    });
+  });
+};
+const registerEvent = async (params, callBack) => {
+  saasCallBack = callBack;
+
+  // 检查是否需要更新dll
+  updateDllFileCheck();
+
+  // 在插线之前就检测驱动状态
+  await checkAndInstallDriver();
+  usbListener = new _usbmux.default.createListener().on('attached', async udid => {
+    if (!udid) return;
+    // 插入设备通知
+    saasCallBack({
+      data: {
+        eventName: _constants.EVENT_TYPE.USB_DEVICE_RESULT,
+        udid,
+        eventCode: 0
+      }
+    });
+    _deviceHelper.deviceHelper.addUdid(udid);
+    // 有驱动，直接运行脚本
+    DevicePair(udid, params);
+  }).on('detached', function (udid) {
+    // 拔出设备
+    saasCallBack({
+      data: {
+        eventName: _constants.EVENT_TYPE.USB_DISCONNECT,
+        udid,
+        eventCode: 0
+      }
+    });
+    _deviceHelper.deviceHelper.removeUdid(udid);
+  }).on('error', function (err) {
+    const errStr = 'usb连接错误:' + JSON.stringify(err);
+    console.log(errStr);
+    Sentry.captureException(errStr);
+    saasCallBack({
+      data: {
+        eventName: _constants.EVENT_TYPE.USB_ERROR,
+        eventCode: -1,
+        eventContent: JSON.stringify({
+          message: errStr
+        })
+      }
+    });
+  });
+};
+exports.registerEvent = registerEvent;
+const unregisterEvent = () => {
+  usbListener && usbListener.destroy();
+  _deviceHelper.deviceHelper.clearUdidList();
+};
+exports.unregisterEvent = unregisterEvent;

package/lib/abilities/getAndroidReport/index.js

@@ -0,0 +1,60 @@
+"use strict";
+
+Object.defineProperty(exports, "__esModule", {
+  value: true
+});
+exports.getAndroidReport = getAndroidReport;
+var _webWorker = _interopRequireDefault(require("web-worker"));
+var _url = _interopRequireDefault(require("url"));
+var _utils = require("../../utils/utils.common");
+var _utils2 = require("../../utils/utils.path");
+var _utils3 = require("../../utils/utils.ability");
+function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }
+class GetAndroidReport {
+  constructor() {
+    this.initWorker();
+    this.initWorkerListener();
+  }
+  worker = null;
+  callback = null;
+  exec(payload, callback) {
+    this.callback = typeof callback === 'function' ? callback : _utils3.defaultCallback;
+    this.worker && this.worker.postMessage({
+      payload
+    });
+  }
+  initWorkerListener() {
+    this.worker && this.worker.addEventListener('message', event => {
+      console.log('GetAndroidReport', event);
+      const {
+        payload
+      } = event.data;
+      this.callback(payload);
+      this.callback = null;
+    });
+  }
+  initWorker() {
+    if (!this.worker) {
+      const workerPath = new _url.default.URL('file://' + (0, _utils2.convertPathWithCurrentEnv)(__dirname, './worker.js'));
+      this.worker = new _webWorker.default(workerPath, {
+        type: 'module'
+      });
+    }
+  }
+}
+let entity = null;
+
+/**
+ * 获取Android设备报告
+ * @param {Object} params - 请求参数
+ * @param {string} params.imei - 设备IMEI
+ * @param {string} params.brandCode - 品牌编码
+ * @param {Function} callback - 回调函数，接收报告结果
+ */
+function getAndroidReport(params, callback) {
+  if (!entity) {
+    entity = new GetAndroidReport();
+  }
+  const payload = (0, _utils.parsePayload)(params);
+  entity.exec(payload, callback);
+}

package/lib/abilities/getAndroidReport/worker.js

@@ -0,0 +1,21 @@
+"use strict";
+
+const {
+  initWorkerNodePath
+} = require("../../utils/utils.worker");
+
+// 先注册NodePath后才能正常使用node包，请勿挪动位置
+initWorkerNodePath();
+const {
+  abilityCallKit
+} = require("../../utils/utils.global");
+
+// 处理主进程消息
+function onMainMessage(event) {
+  const {
+    payload
+  } = event.data;
+  const res = abilityCallKit('android_report', payload);
+  postMessage(res);
+}
+addEventListener('message', onMainMessage);