hunter-open-sdk 2.0.0-beta.2 → 2.0.0-beta.20

package/lib/abilities/activate/index.js

@@ -0,0 +1,48 @@
+"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.getCurrentEnvPath)(__dirname, './worker.js'));
+      this.worker = new _webWorker.default(workerPath, {
+        type: 'module'
+      });
+    }
+  }
+}
+const entity = new Activate();
+function activate(params, callback) {
+  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,48 @@
+"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.getCurrentEnvPath)(__dirname, './worker.js'));
+      this.worker = new _webWorker.default(workerPath, {
+        type: 'module'
+      });
+    }
+  }
+}
+const entity = new Erase();
+function erase(params, callback) {
+  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,338 @@
+"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 _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: '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: 'type_update_result',
+          eventCode: -1 // eventCode 0 更新成功 -1 更新失败
+        }
+      });
+      return;
+    }
+    console.log('========写入version成功');
+    // 写入成功后 重新初始化dll
+    (0, _utils4.initHunterInspectionKit)();
+    saasCallBack({
+      data: {
+        eventName: '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: '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: 'type_need_update',
+          eventCode: isHot == 1 ? 0 : -1 // eventCode 0 需要更新 -1 不需要更新
+        }
+      });
+    } else {
+      // 错误的时候不需要更新
+      saasCallBack({
+        data: {
+          eventName: '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: 'type_need_install_driver',
+            eventCode: 0,
+            eventContent: JSON.stringify({
+              percent // 下载进度
+            })
+          }
+        });
+      }).then(() => {
+        // 成功
+        console.log('驱动下载并安装成功');
+        saasCallBack({
+          data: {
+            eventName: 'type_install_driver_result',
+            eventCode: 0 // 0 成功 -1 失败
+          }
+        });
+        resolve();
+      }).catch(err => {
+        // 失败
+        console.error('驱动下载或安装失败', err);
+        Sentry.captureException('驱动下载或安装失败:' + JSON.stringify(err));
+        saasCallBack({
+          data: {
+            eventName: '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: 'type_usb_device_result',
+        udid,
+        eventCode: 0
+      }
+    });
+    _deviceHelper.deviceHelper.addUdid(udid);
+    // 有驱动，直接运行脚本
+    DevicePair(udid, params);
+  }).on('detached', function (udid) {
+    // 拔出设备
+    saasCallBack({
+      data: {
+        eventName: 'type_usb_device_result',
+        udid,
+        eventCode: -1
+      }
+    });
+    _deviceHelper.deviceHelper.removeUdid(udid);
+  }).on('error', function (err) {
+    console.log('usb连接错误：', err);
+    Sentry.captureException('usb连接错误:' + JSON.stringify(err));
+    const errMsg = `usb连接错误：${err}`;
+    throw new Error(errMsg);
+  });
+};
+exports.registerEvent = registerEvent;
+const unregisterEvent = () => {
+  usbListener && usbListener.destroy();
+  _deviceHelper.deviceHelper.clearUdidList();
+};
+exports.unregisterEvent = unregisterEvent;

package/lib/abilities/getAndroidSn/index.js

@@ -0,0 +1,165 @@
+"use strict";
+
+Object.defineProperty(exports, "__esModule", {
+  value: true
+});
+exports.unregisterGetAndroidSn = exports.registerGetAndroidSn = void 0;
+var _usb = require("usb");
+var _child_process = _interopRequireDefault(require("child_process"));
+var _utils = require("../../utils/utils.env");
+var _utils2 = require("../../utils/utils.sn");
+function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }
+const exec = _child_process.default.exec;
+let hadRegisterUsb = false; // 是否已经注册过usb事件
+let isPause = false; // 是否暂停获取sn
+
+// mac获取sn
+const getMacSn = async (device, callBack) => {
+  // 拿到当前连接的设备deviceAddress 作为标识去usb列表筛选
+  const attachDeviceId = device.deviceAddress;
+  const customWebUSB = new _usb.WebUSB({
+    allowAllDevices: true
+  });
+  const deviceList = await customWebUSB.getDevices();
+
+  // 如果设备没有attachDeviceId 则直接返回
+  if (!attachDeviceId) return;
+
+  // 删选出当前的设备信息 为获取到serialNumber
+  const currentDevice = deviceList.filter(item => item.device.deviceAddress == attachDeviceId);
+  if (!currentDevice.length) {
+    callBack({
+      code: -1,
+      data: {},
+      message: '获取SN失败'
+    });
+    return;
+  }
+
+  // 检查设备是否为支持的三星、华为、荣耀手机品牌
+  const deviceInfo = currentDevice[0];
+  const productName = deviceInfo.productName?.toLowerCase() || '';
+  const manufacturerName = deviceInfo.manufacturerName?.toLowerCase() || '';
+
+  // 支持的三星、华为、荣耀品牌关键词
+  const supportedBrands = ['samsung', 'galaxy', 'huawei', 'honor', 'honor magic'];
+  const isSupportedBrand = supportedBrands.some(brand => productName.includes(brand) || manufacturerName.includes(brand));
+  if (!isSupportedBrand) {
+    callBack({
+      code: -1,
+      data: {},
+      message: '暂不支持该品牌手机查询SN'
+    });
+    return;
+  }
+  callBack({
+    code: 0,
+    data: {
+      SN: currentDevice[0].serialNumber
+    },
+    message: '获取SN成功'
+  });
+};
+
+// win获取sn
+const getWinSn = async (device, callBack) => {
+  // windows 需要运行命令获取sn
+  const cmdStr = `wmic path Win32_PnPEntity where "DeviceID like \'USB%\'" get Name, PNPDeviceID /value`;
+  exec(cmdStr, function (err, stdout, stderr) {
+    if (err) {
+      // 运行失败 则直接返回获取失败
+      callBack({
+        code: -1,
+        data: {},
+        message: `windows运行程序获取SN失败err: ${err}`
+      });
+      return;
+    }
+    if (stderr) {
+      // 运行失败 则直接返回获取失败
+      callBack({
+        code: -1,
+        data: {},
+        message: `windows运行程序获取SN失败stderr: ${stderr}`
+      });
+      return;
+    }
+
+    // 处理WMIC输出并提取手机序列号
+    const phoneDevices = (0, _utils2.parseWmicOutput)(stdout);
+    if (phoneDevices.length === 0) {
+      callBack({
+        code: -1,
+        data: {},
+        message: `暂不支持该品牌手机查询SN`
+      });
+    } else {
+      console.log('检测到的手机设备:');
+      const sn = phoneDevices[0].serial;
+      callBack({
+        code: 0,
+        data: {
+          SN: sn
+        },
+        message: `获取SN成功`
+      });
+      // phoneDevices.forEach(device => {
+      //   console.log(`设备名称: ${device.name}`);
+      //   console.log(`序列号: ${device.serial}`);
+      //   console.log(`PNPDeviceID: ${device.pnpDeviceId}`);
+      //   console.log('---');
+      // });
+    }
+  });
+};
+
+/**
+ * 获取Android SN
+ * @param {Object} params - 参数
+ * @param {Function} callBack - 回调函数
+ */
+const registerGetAndroidSn = (params, callBack) => {
+  if (hadRegisterUsb) {
+    isPause = false;
+    return;
+  }
+  hadRegisterUsb = true;
+  _usb.usb.on('attach', async device => {
+    if (isPause) {
+      return;
+    }
+    const {
+      idVendor
+    } = device.deviceDescriptor;
+    if (idVendor == 1452) {
+      // 表示该机器为苹果产品，则不进行获取sn
+      return;
+    }
+    callBack({
+      data: {
+        eventName: 'type_usb_android_device_result',
+        eventCode: 0
+      }
+    });
+    if ((0, _utils.isMac)()) {
+      // mac获取sn
+      getMacSn(device, callBack);
+      return;
+    }
+    if ((0, _utils.isWin)()) {
+      // win获取sn
+      getWinSn(device, callBack);
+      return;
+    }
+  });
+  _usb.usb.on('detach', () => {
+    if (isPause) {
+      return;
+    }
+  });
+};
+exports.registerGetAndroidSn = registerGetAndroidSn;
+const unregisterGetAndroidSn = () => {
+  isPause = true;
+};
+exports.unregisterGetAndroidSn = unregisterGetAndroidSn;

package/lib/abilities/getGeneralReport/index.js

@@ -0,0 +1,64 @@
+"use strict";
+
+Object.defineProperty(exports, "__esModule", {
+  value: true
+});
+exports.getGeneralReport = getGeneralReport;
+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 }; }
+const DEFAULT_POOL_SIZE = 8;
+class WorkerPool {
+  constructor(size) {
+    this.poolSize = size;
+    this.idleWorkers = [];
+    this.taskQueue = [];
+    for (let i = 0; i < this.poolSize; i++) {
+      this.idleWorkers.push(this._createWorker());
+    }
+  }
+  _createWorker() {
+    const workerPath = new _url.default.URL('file://' + (0, _utils2.getCurrentEnvPath)(__dirname, './worker.js'));
+    return new _webWorker.default(workerPath, {
+      type: 'module'
+    });
+  }
+  exec(payload, callback) {
+    if (this.idleWorkers.length > 0) {
+      this._run(this.idleWorkers.pop(), payload, callback);
+    } else {
+      this.taskQueue.push({
+        payload,
+        callback
+      });
+    }
+  }
+  _run(worker, payload, callback) {
+    const onMessage = event => {
+      worker.removeEventListener('message', onMessage);
+      callback(event.data.payload);
+      if (this.taskQueue.length > 0) {
+        const next = this.taskQueue.shift();
+        this._run(worker, next.payload, next.callback);
+      } else {
+        this.idleWorkers.push(worker);
+      }
+    };
+    worker.addEventListener('message', onMessage);
+    worker.postMessage({
+      payload
+    });
+  }
+}
+let pool = new WorkerPool(DEFAULT_POOL_SIZE);
+function getGeneralReport(params, callback, options = {}) {
+  const cb = typeof callback === 'function' ? callback : _utils3.defaultCallback;
+  const payload = (0, _utils.parsePayload)(params);
+  if (options?.poolSize != null && options.poolSize !== pool.poolSize) {
+    pool = new WorkerPool(options.poolSize);
+  }
+  pool.exec(payload, cb);
+}