hunter-open-sdk 2.0.0-beta.5 → 2.0.0-beta.7

package/lib/events/driver/winDriver.js

@@ -6,30 +6,27 @@ Object.defineProperty(exports, "__esModule", {
 });
 exports.checkHasDriver = checkHasDriver;
 exports.downloadDriver = downloadDriver;
-var _regenerator = _interopRequireDefault(require("@babel/runtime/regenerator"));
-var _typeof2 = _interopRequireDefault(require("@babel/runtime/helpers/typeof"));
-var _asyncToGenerator2 = _interopRequireDefault(require("@babel/runtime/helpers/asyncToGenerator"));
 var _utils = require("../../utils/utils");
 var _request = _interopRequireDefault(require("request"));
 var _fsExtra = _interopRequireDefault(require("fs-extra"));
 // import { app } from 'electron'
 // import { command } from '../../utils/utils.command'
 
-var Sentry = require("@sentry/browser");
-var sudo = require('sudo-prompt');
-var path = require('path');
-var fs = require('fs');
-var childProcess = require('child_process');
-var compressing = require('compressing');
-var exec = childProcess.exec;
-var options = {
+const Sentry = require("@sentry/browser");
+const sudo = require('sudo-prompt');
+const path = require('path');
+const fs = require('fs');
+const childProcess = require('child_process');
+const compressing = require('compressing');
+const exec = childProcess.exec;
+const options = {
   name: 'Electron'
 };
-var msiStr = 'AppleMobileDeviceService';
+const msiStr = 'AppleMobileDeviceService';
 // const reddirPath = 'C:/Program Files (x86)/Common Files/Apple/Apple Application Support'
-var cmdStr = process.platform === 'win32' ? 'tasklist' : 'ps aux';
-var CACHE_PATH = (0, _utils.isDev)() ? path.join(process.cwd(), './cache') : path.join(process.resourcesPath, './cache');
-var filePath = path.join(CACHE_PATH, './iTunesOL_Lite_64_12.10.0.7.zip');
+const cmdStr = process.platform === 'win32' ? 'tasklist' : 'ps aux';
+const CACHE_PATH = (0, _utils.isDev)() ? path.join(process.cwd(), './cache') : path.join(process.resourcesPath, './cache');
+const filePath = path.join(CACHE_PATH, './iTunesOL_Lite_64_12.10.0.7.zip');
 
 // 确保缓存目录存在，避免后续写入流抛出 ENOENT
 try {
@@ -38,8 +35,8 @@ try {
   console.error('创建缓存目录失败:', e);
   Sentry.captureException('创建缓存目录失败:' + JSON.stringify(e));
 }
-var checkProcess = function checkProcess() {
-  return new Promise(function (resolve, reject) {
+const checkProcess = () => {
+  return new Promise((resolve, reject) => {
     exec(cmdStr, function (err, stdout, stderr) {
       // console.log('stdout>>>', stdout)
       console.log('stderr>>>', stderr);
@@ -47,10 +44,10 @@ var checkProcess = function checkProcess() {
         return reject(err);
       }
       if (stdout) {
-        var hasRroess = false;
-        stdout.split('\n').filter(function (line) {
-          var processMessage = line.trim().split(/\s+/);
-          var processName = processMessage[0];
+        let hasRroess = false;
+        stdout.split('\n').filter(line => {
+          let processMessage = line.trim().split(/\s+/);
+          let processName = processMessage[0];
           // console.log('names.....', processName)
           if (processName.indexOf(msiStr) > -1) {
             hasRroess = true;
@@ -69,8 +66,8 @@ var checkProcess = function checkProcess() {
 //   })
 // }
 
-var checkCache = function checkCache() {
-  return new Promise(function (resolve) {
+const checkCache = () => {
+  return new Promise(resolve => {
     fs.readFile(filePath, function (err) {
       resolve(err ? false : true);
     });
@@ -78,54 +75,39 @@ var checkCache = function checkCache() {
 };
 
 // 轮询检测是否存在进程
-var processCheckTimer = null;
-var checkProcessTimer = function checkProcessTimer() {
+let processCheckTimer = null;
+const checkProcessTimer = () => {
   // 清除之前的定时器
   if (processCheckTimer) {
     clearTimeout(processCheckTimer);
   }
-  processCheckTimer = setTimeout( /*#__PURE__*/(0, _asyncToGenerator2.default)( /*#__PURE__*/_regenerator.default.mark(function _callee() {
-    var hasDriver;
-    return _regenerator.default.wrap(function _callee$(_context) {
-      while (1) switch (_context.prev = _context.next) {
-        case 0:
-          _context.next = 2;
-          return checkProcess();
-        case 2:
-          hasDriver = _context.sent;
-          if (!hasDriver) {
-            _context.next = 6;
-            break;
-          }
-          // 检测到驱动，清除定时器并返回
-          if (processCheckTimer) {
-            clearTimeout(processCheckTimer);
-            processCheckTimer = null;
-          }
-          return _context.abrupt("return");
-        case 6:
-          // 继续轮询检测
-          checkProcessTimer();
-        case 7:
-        case "end":
-          return _context.stop();
+  processCheckTimer = setTimeout(async () => {
+    let hasDriver = await checkProcess();
+    if (hasDriver) {
+      // 检测到驱动，清除定时器并返回
+      if (processCheckTimer) {
+        clearTimeout(processCheckTimer);
+        processCheckTimer = null;
       }
-    }, _callee);
-  })), 1000);
+      return;
+    }
+    // 继续轮询检测
+    checkProcessTimer();
+  }, 1000);
 };
-var installDriver = function installDriver() {
-  var AppleApplicationSupport = path.join(CACHE_PATH, './AppleApplicationSupport.msi');
-  var AppleApplicationSupport64 = path.join(CACHE_PATH, './AppleApplicationSupport64.msi');
-  var AppleMobileDeviceSupport64 = path.join(CACHE_PATH, './AppleMobileDeviceSupport64.msi');
-  var Bonjour64 = path.join(CACHE_PATH, './Bonjour64.msi');
-  var cmdStr = "start /i /wait ".concat(AppleApplicationSupport, " /qn && start /i /wait ").concat(AppleApplicationSupport64, " /qn && start /i /wait ").concat(AppleMobileDeviceSupport64, " /qn && start /i /wait ").concat(Bonjour64, " /qn");
+const installDriver = () => {
+  const AppleApplicationSupport = path.join(CACHE_PATH, './AppleApplicationSupport.msi');
+  const AppleApplicationSupport64 = path.join(CACHE_PATH, './AppleApplicationSupport64.msi');
+  const AppleMobileDeviceSupport64 = path.join(CACHE_PATH, './AppleMobileDeviceSupport64.msi');
+  const Bonjour64 = path.join(CACHE_PATH, './Bonjour64.msi');
+  const cmdStr = `start /i /wait ${AppleApplicationSupport} /qn && start /i /wait ${AppleApplicationSupport64} /qn && start /i /wait ${AppleMobileDeviceSupport64} /qn && start /i /wait ${Bonjour64} /qn`;
   console.log('开始安装驱动，命令:', cmdStr);
   console.log('检查安装文件是否存在:');
   console.log('AppleApplicationSupport:', fs.existsSync(AppleApplicationSupport));
   console.log('AppleApplicationSupport64:', fs.existsSync(AppleApplicationSupport64));
   console.log('AppleMobileDeviceSupport64:', fs.existsSync(AppleMobileDeviceSupport64));
   console.log('Bonjour64:', fs.existsSync(Bonjour64));
-  return new Promise(function (resolve, reject) {
+  return new Promise((resolve, reject) => {
     sudo.exec(cmdStr, options, function (error, stdout, stderr) {
       console.log('安装 error>>', error);
       console.log('安装 stdout>>>', stdout);
@@ -143,13 +125,13 @@ var installDriver = function installDriver() {
   });
 };
 // 解压文件
-var uncopressingFile = function uncopressingFile(filePath) {
-  return new Promise(function (resolve) {
+const uncopressingFile = filePath => {
+  return new Promise(resolve => {
     console.log('开始解压文件:', filePath);
-    compressing.zip.uncompress(filePath, CACHE_PATH).then(function () {
+    compressing.zip.uncompress(filePath, CACHE_PATH).then(() => {
       console.log('文件解压成功');
       resolve(true);
-    }).catch(function (err) {
+    }).catch(err => {
       console.error('文件解压失败:', err);
       Sentry.captureException('文件解压失败:' + JSON.stringify(err));
       resolve(false);
@@ -160,165 +142,116 @@ var uncopressingFile = function uncopressingFile(filePath) {
 // Promise 版本的 downloadDriver，onProgress 作为参数，成功 resolve，失败 reject
 function downloadDriver(onProgress) {
   console.log('=== downloadDriver 函数被调用 ===');
-  console.log('onProgress 参数:', (0, _typeof2.default)(onProgress));
-  return new Promise(function (resolve, reject) {
+  console.log('onProgress 参数:', typeof onProgress);
+  return new Promise((resolve, reject) => {
     console.log('=== 开始创建Promise ===');
-    var url = 'http://ljtools.zhuanstatic.com/download/iTunesDriver/iTunesOL_Lite_64_12.10.0.7.zip';
-    var dest = filePath;
+    const url = 'http://ljtools.zhuanstatic.com/download/iTunesDriver/iTunesOL_Lite_64_12.10.0.7.zip';
+    const dest = filePath;
     console.log('开始下载驱动文件:', url);
     console.log('目标路径:', dest);
 
     // 检查目标目录是否存在
-    var destDir = path.dirname(dest);
+    const destDir = path.dirname(dest);
     console.log('目标目录:', destDir);
     console.log('目录是否存在:', fs.existsSync(destDir));
-    var fileStream = _fsExtra.default.createWriteStream(dest);
-    var received = 0;
-    var total = 0;
+    const fileStream = _fsExtra.default.createWriteStream(dest);
+    let received = 0;
+    let total = 0;
 
     // 进度回调限流：进度变化超过5%时调用
-    var lastProgressPercent = 0;
-    var PROGRESS_THRESHOLD = 5; // 5%阈值
+    let lastProgressPercent = 0;
+    const PROGRESS_THRESHOLD = 5; // 5%阈值
 
-    var throttledProgressCallback = function throttledProgressCallback(percent) {
-      var shouldCall = Math.abs(percent - lastProgressPercent) >= PROGRESS_THRESHOLD;
+    const throttledProgressCallback = percent => {
+      const shouldCall = Math.abs(percent - lastProgressPercent) >= PROGRESS_THRESHOLD;
       if (shouldCall && onProgress) {
         onProgress(percent);
         lastProgressPercent = percent;
-        console.log("\u4E0B\u8F7D\u8FDB\u5EA6: ".concat(percent, "% (").concat(received, "/").concat(total, ")"));
+        console.log(`下载进度: ${percent}% (${received}/${total})`);
       }
     };
     console.log('=== 开始创建request ===');
-    (0, _request.default)(url).on('response', function (response) {
+    (0, _request.default)(url).on('response', response => {
       console.log('下载响应状态:', response.statusCode);
       if (response.statusCode !== 200) {
-        reject(new Error("\u4E0B\u8F7D\u5931\u8D25: ".concat(response.statusMessage)));
+        reject(new Error(`下载失败: ${response.statusMessage}`));
         return;
       }
       total = parseInt(response.headers['content-length'], 10);
       console.log('文件总大小:', total, 'bytes');
-    }).on('data', function (chunk) {
+    }).on('data', chunk => {
       received += chunk.length;
       if (total) {
-        var percent = Math.round(received / total * 100);
+        const percent = Math.round(received / total * 100);
         throttledProgressCallback(percent);
       }
-    }).on('error', function (err) {
+    }).on('error', err => {
       console.error('驱动下载失败', err);
       Sentry.captureException('驱动下载失败:' + JSON.stringify(err));
       fileStream.end(); // 确保关闭文件流
       reject(err);
-    }).pipe(fileStream).once('close', /*#__PURE__*/(0, _asyncToGenerator2.default)( /*#__PURE__*/_regenerator.default.mark(function _callee2() {
-      var uncopressing;
-      return _regenerator.default.wrap(function _callee2$(_context2) {
-        while (1) switch (_context2.prev = _context2.next) {
-          case 0:
-            console.log('=== CLOSE 事件被触发 ===');
-            console.log('文件下载完成，开始解压...');
+    }).pipe(fileStream).once('close', async () => {
+      console.log('=== CLOSE 事件被触发 ===');
+      console.log('文件下载完成，开始解压...');
 
-            // 确保最后一次进度回调被调用（100%）
-            if (onProgress && total) {
-              onProgress(100);
-              console.log('下载完成: 100%');
-            }
-            _context2.prev = 3;
-            _context2.next = 6;
-            return uncopressingFile(filePath);
-          case 6:
-            uncopressing = _context2.sent;
-            console.log('解压结果:', uncopressing);
-            if (uncopressing) {
-              console.log('开始安装驱动...');
-              installDriver().then(function () {
-                console.log('驱动安装成功');
-                resolve();
-              }).catch(function (err) {
-                console.error('驱动安装失败:', err);
-                Sentry.captureException('驱动安装失败:' + JSON.stringify(err));
-                reject(err);
-              });
-            } else {
-              console.error('解压失败');
-              Sentry.captureException('解压失败');
-              reject(new Error('解压失败'));
-            }
-            _context2.next = 16;
-            break;
-          case 11:
-            _context2.prev = 11;
-            _context2.t0 = _context2["catch"](3);
-            console.error('解压或安装过程中出错:', _context2.t0);
-            Sentry.captureException('解压或安装过程中出错:' + JSON.stringify(_context2.t0));
-            reject(_context2.t0);
-          case 16:
-          case "end":
-            return _context2.stop();
+      // 确保最后一次进度回调被调用（100%）
+      if (onProgress && total) {
+        onProgress(100);
+        console.log('下载完成: 100%');
+      }
+      try {
+        // 下载完成后解压并安装
+        const uncopressing = await uncopressingFile(filePath);
+        console.log('解压结果:', uncopressing);
+        if (uncopressing) {
+          console.log('开始安装驱动...');
+          installDriver().then(() => {
+            console.log('驱动安装成功');
+            resolve();
+          }).catch(err => {
+            console.error('驱动安装失败:', err);
+            Sentry.captureException('驱动安装失败:' + JSON.stringify(err));
+            reject(err);
+          });
+        } else {
+          console.error('解压失败');
+          Sentry.captureException('解压失败');
+          reject(new Error('解压失败'));
         }
-      }, _callee2, null, [[3, 11]]);
-    }))).once('error', function (err) {
+      } catch (err) {
+        console.error('解压或安装过程中出错:', err);
+        Sentry.captureException('解压或安装过程中出错:' + JSON.stringify(err));
+        reject(err);
+      }
+    }).once('error', err => {
       console.error('文件流写入错误:', err);
       Sentry.captureException('文件流写入错误:' + JSON.stringify(err));
       reject(err);
     });
   });
 }
-function checkHasDriver() {
-  return _checkHasDriver.apply(this, arguments);
-}
-function _checkHasDriver() {
-  _checkHasDriver = (0, _asyncToGenerator2.default)( /*#__PURE__*/_regenerator.default.mark(function _callee3() {
-    var hasDriver, hasCache, state, uncopressing;
-    return _regenerator.default.wrap(function _callee3$(_context3) {
-      while (1) switch (_context3.prev = _context3.next) {
-        case 0:
-          _context3.next = 2;
-          return checkProcess();
-        case 2:
-          if (!_context3.sent) {
-            _context3.next = 6;
-            break;
-          }
-          _context3.t0 = 1;
-          _context3.next = 7;
-          break;
-        case 6:
-          _context3.t0 = 0;
-        case 7:
-          hasDriver = _context3.t0;
-          _context3.next = 10;
-          return checkCache();
-        case 10:
-          if (!_context3.sent) {
-            _context3.next = 14;
-            break;
-          }
-          _context3.t1 = 2;
-          _context3.next = 15;
-          break;
-        case 14:
-          _context3.t1 = 0;
-        case 15:
-          hasCache = _context3.t1;
-          state = hasDriver + hasCache;
-          _context3.t2 = state;
-          _context3.next = _context3.t2 === 3 ? 20 : _context3.t2 === 1 ? 20 : _context3.t2 === 2 ? 21 : 26;
-          break;
-        case 20:
-          return _context3.abrupt("return", true);
-        case 21:
-          _context3.next = 23;
-          return uncopressingFile(filePath);
-        case 23:
-          uncopressing = _context3.sent;
-          if (uncopressing) installDriver();
-          return _context3.abrupt("return", true);
-        case 26:
-          return _context3.abrupt("return", false);
-        case 27:
-        case "end":
-          return _context3.stop();
-      }
-    }, _callee3);
-  }));
-  return _checkHasDriver.apply(this, arguments);
+async function checkHasDriver() {
+  // 是否存在驱动进程
+  const hasDriver = (await checkProcess()) ? 1 : 0;
+  // 是否存在安装包
+  const hasCache = (await checkCache()) ? 2 : 0;
+  const state = hasDriver + hasCache;
+  switch (state) {
+    // 存在驱动进程 + 存在安装包
+    case 3:
+
+    // 仅存在驱动进程
+    case 1:
+      return true;
+
+    // 仅存在安装包 进行解压安装
+    case 2:
+      const uncopressing = await uncopressingFile(filePath);
+      if (uncopressing) installDriver();
+      return true;
+
+    // state = 0,
+    default:
+      return false;
+  }
 }