crabatool 1.0.856 → 1.0.857

package/index.js

@@ -338,6 +338,7 @@ async function checkFast(args) {
   if (args.includes('-runEv')) {
     start.bindConfigByArgv('-evSavePath');
     start.bindConfigByArgv('-port');
+    start.bindConfigByArgv('-local', 'boolean');
     start.runEvServer();
     return false;
   }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "crabatool",
-  "version": "1.0.856",
+  "version": "1.0.857",
   "description": "crabatool",
   "main": "index.js",
   "bin": {

package/tool/evsockettool.js

@@ -19,6 +19,9 @@ var evConfig = {
 // 监控的文件
 var monitoringFiles = {};
 
+// 已完成的文件（防止重复处理）
+var completedFiles = {};
+
 // 确保输出目录存在
 if (!fs.existsSync(evConfig.outputDir)) {
   fs.mkdirSync(evConfig.outputDir, { recursive: true });
@@ -62,6 +65,9 @@ function init() {
   // 直接启动runEv服务时，什么都不用做
   // 等待前端连接和start消息，如果等到的是连接消息，记录连接时间。
   log('evsockettool 初始化完成，等待前端连接');
+  if (config.local) {
+    startEvRecording();
+  }
 }
 
 // 启动ev录屏
@@ -132,10 +138,21 @@ function stopEvRecording(conn) {
   // 清理监控文件
   for (var filePath in monitoringFiles) {
     if (monitoringFiles.hasOwnProperty(filePath)) {
-      clearInterval(monitoringFiles[filePath]);
+      if (monitoringFiles[filePath].timer) {
+        clearInterval(monitoringFiles[filePath].timer);
+      }
       delete monitoringFiles[filePath];
     }
   }
+  log('清理监控文件');
+
+  // 清理已完成文件标记
+  for (var completedFile in completedFiles) {
+    if (completedFiles.hasOwnProperty(completedFile)) {
+      delete completedFiles[completedFile];
+    }
+  }
+  log('清理已完成文件标记');
 
   // 清理watcher
   if (evConfig.watcher) {
@@ -169,15 +186,21 @@ function watchEvOutput(conn) {
   evConfig.watcher.on('add', function(filePath) {
     var ext = path.extname(filePath).toLowerCase();
     if (ext === '.mp3' || ext === '.wav' || ext === '.mp4') {
-      log('发现新的媒体文件:', filePath);
+      // 检查文件是否已完成
+      if (completedFiles[filePath]) {
+        log('文件已处理完成，跳过:', filePath);
+        return;
+      }
+      console.log('\r\n\r\n');
+      log('---------发现新的媒体文件---------', filePath);
 
       // 检查文件创建时间是否大于启动时间
       try {
         var stats = fs.statSync(filePath);
         var fileCreateTime = stats.ctime.getTime();
         if (fileCreateTime > evConfig.startTime) {
-          log('录屏开始，文件已生成:', filePath);
           sendEvMessage(conn, 'msg', '录屏已开始，正在录制中');
+          startFileMonitoring(conn, filePath);
         }
       } catch (ex) {
         log('检查文件信息失败:', ex.message);
@@ -188,25 +211,29 @@ function watchEvOutput(conn) {
   evConfig.watcher.on('change', function(filePath) {
     var ext = path.extname(filePath).toLowerCase();
     if (ext === '.mp3' || ext === '.wav' || ext === '.mp4') {
-      log('文件变化:', filePath);
+      //log('文件变化:', filePath);
+
+      // 检查文件是否已完成
+      if (completedFiles[filePath]) {
+        log('文件已处理完成，跳过:', filePath);
+        return;
+      }
 
       // 检查文件创建时间和大小
       try {
         var stats = fs.statSync(filePath);
         var fileCreateTime = stats.ctime.getTime();
-
         // 文件创建时间必须大于启动时间
         if (fileCreateTime > evConfig.startTime) {
-          log('文件创建时间符合要求:', new Date(fileCreateTime).toISOString());
-
-          // 如果文件还没有被监控，开始监控
-          if (!monitoringFiles[filePath]) {
-            log('开始监控文件:', filePath, '当前大小:', stats.size, 'bytes');
-            // 先标记为监控中，防止短时间内重复触发
-            monitoringFiles[filePath] = true;
-            startMonitoringFile(conn, filePath);
+          log('文件发生改变:', filePath, '当前大小:', stats.size, 'bytes');
+
+          // 如果文件已经在监控中，重置成功计数
+          if (monitoringFiles[filePath]) {
+            log('文件在监控中发生变化，重置成功计数:', filePath);
+            monitoringFiles[filePath].successCount = 0;
           } else {
-            log('文件已在监控中，跳过:', filePath);
+            // 开始监控文件
+            startFileMonitoring(conn, filePath);
           }
         } else {
           log('文件创建时间不符合要求，跳过:', filePath);
@@ -222,74 +249,60 @@ function watchEvOutput(conn) {
   });
 }
 
-// 监控文件大小变化，判断录制是否真正结束
-function startMonitoringFile(conn, filePath) {
-  // 如果已经是一个有效的interval，说明已经在监控中
-  if (monitoringFiles[filePath] && monitoringFiles[filePath] !== true) {
-    log('文件已在监控中，跳过:', filePath);
+// 开始监控文件，每隔1秒尝试重命名检测，连续3次成功才算完成
+function startFileMonitoring(conn, filePath) {
+  // 如果已经在监控中，跳过
+  if (monitoringFiles[filePath]) {
+    //log('文件已在监控中，跳过:', filePath);
     return;
   }
+  var stats = fs.statSync(filePath);
+  log('开始监控文件是否录制完成:', filePath, '当前大小:', stats.size, 'bytes');
+
+  // 标记为监控中，存储状态
+  monitoringFiles[filePath] = {
+    timer: null,
+    successCount: 0,
+    requiredSuccessCount: 3 // 需要连续3次成功
+  };
 
-  var lastSize = 0;
-  var stableCount = 0;
-  var hasGrowth = false;
-
-  // 根据文件类型设置不同的阈值
-  var ext = path.extname(filePath).toLowerCase();
-  var isVideo = ext === '.mp4';
-  var isAudio = ext === '.mp3' || ext === '.wav';
-
-  // MP4视频：帧率20fps，文件增长快，检查间隔1秒，稳定3次(3秒)即可判定完成
-  // MP3音频：码率128kbps，文件增长慢，检查间隔3秒，稳定2次(6秒)即可判定完成
-  var checkInterval = isVideo ? 1000 : (isAudio ? 3000 : 2000); // MP4: 1秒, MP3: 3秒, 其他: 2秒
-  var stableThreshold = isVideo ? 3 : (isAudio ? 2 : 3); // MP4: 3次(3秒), MP3: 2次(6秒), 其他: 3次
-  var minFileSize = isVideo ? 5000 : (isAudio ? 3000 : 1000); // MP4: 5KB, MP3: 3KB, 其他: 1KB
-
-  log('开始监控文件:', path.basename(filePath), '类型:', isVideo ? '视频' : (isAudio ? '音频' : '其他'),
-    '检查间隔:', checkInterval + 'ms', '稳定阈值:', stableThreshold + '次');
-
-  // 清除之前的标记，设置为真正的interval
-  if (monitoringFiles[filePath] === true) {
-    delete monitoringFiles[filePath];
-  }
-
-  monitoringFiles[filePath] = setInterval(function() {
-    try {
-      var stats = fs.statSync(filePath);
-      var currentSize = stats.size;
-
-      if (currentSize > lastSize) {
-        hasGrowth = true;
-        stableCount = 0; // 重置计数
-        log('文件大小增长:', filePath, '当前大小:', currentSize, 'bytes');
-      } else if (currentSize === lastSize) {
-        stableCount++;
-        log('文件大小稳定:', filePath, '连续稳定次数:', stableCount, '稳定时间:', (stableCount * checkInterval / 1000).toFixed(1), '秒');
-      }
-
-      lastSize = currentSize;
-
-      // 当文件大小连续稳定5次（15秒）且至少有过一次增长，并且文件大小大于最小阈值时，认为录制真正结束
-      if (stableCount >= stableThreshold && hasGrowth && currentSize >= minFileSize) {
-        clearInterval(monitoringFiles[filePath]);
-        delete monitoringFiles[filePath];
-        log('文件录制真正完成:', filePath, '最终大小:', currentSize, 'bytes');
-        processEvFile(conn, filePath);
+  const state = monitoringFiles[filePath];
+
+  state.timer = setInterval(function() {
+    // 尝试重命名检测文件是否被占用
+    checkFileLock(filePath, function(isLocked) {
+      if (!isLocked) {
+        // 重命名成功，增加成功计数
+        state.successCount++;
+        log('文件重命名成功，连续成功次数:', state.successCount + '/' + state.requiredSuccessCount, filePath);
+
+        // 连续3次成功，说明文件未被占用，录制完成
+        if (state.successCount >= state.requiredSuccessCount) {
+          clearInterval(state.timer);
+          delete monitoringFiles[filePath];
+          // 标记文件为已完成
+          completedFiles[filePath] = true;
+          log('【文件录制完成】:', filePath);
+          processEvFile(conn, filePath);
+        }
+      } else {
+        // 重命名失败，重置成功计数，文件仍被占用，继续监控
+        state.successCount = 0;
+        //log('文件仍被占用，重置成功计数:', filePath);
       }
-    } catch (ex) {
-      log('监控文件失败:', ex.message);
-      clearInterval(monitoringFiles[filePath]);
-      delete monitoringFiles[filePath];
-    }
-  }, checkInterval);
+    });
+  }, 1000); // 每隔1秒检查一次
 }
 
 // 处理ev生成的文件
 function processEvFile(conn, filePath) {
+  var stats = fs.statSync(filePath);
+  log('【开始上传文件到七牛】:', filePath, '当前大小:', stats.size, 'bytes');
+
   // 检查是否有七牛token
   if (!evConfig.qiniuToken) {
-    log('没有七牛token，无法上传文件');
-    sendEvMessage(conn, 'error', '录制完成，但没有七牛token，无法上传文件');
+    log('但没有七牛token，无法上传文件');
+    sendEvMessage(conn, 'error', '【录制完成】，但没有七牛token，无法上传文件');
     return;
   }
 
@@ -307,8 +320,6 @@ function processEvFile(conn, filePath) {
     var formUploader = new qiniu.form_up.FormUploader();
     var putExtra = new qiniu.form_up.PutExtra();
 
-    log('开始上传文件到七牛服务器:', filePath);
-
     // 上传文件
     formUploader.putFile(evConfig.qiniuToken, qiniuFileName, filePath, putExtra, function(respErr, respBody, respInfo) {
       if (respErr) {
@@ -343,6 +354,9 @@ function processEvFile(conn, filePath) {
 
 // 发送ev消息给前端
 function sendEvMessage(conn, status, message, data) {
+  if (!conn) {
+    return;
+  }
   var response = {
     mode: 'evRecord',
     status: status,
@@ -359,6 +373,50 @@ function sendEvMessage(conn, status, message, data) {
   }
 }
 
+// 检查文件是否被占用（通过重命名检测）
+function checkFileLock(filePath, callback) {
+  // 生成临时文件名（在原文件名后添加递增数字）
+  const dir = path.dirname(filePath);
+  const ext = path.extname(filePath);
+  const baseName = path.basename(filePath, ext);
+
+  // 生成带递增数字的临时文件名
+  let tempName = baseName + '_temp' + ext;
+  let tempPath = path.join(dir, tempName);
+  let counter = 1;
+
+  // 确保临时文件名不冲突
+  while (fs.existsSync(tempPath)) {
+    tempName = baseName + '_temp' + counter + ext;
+    tempPath = path.join(dir, tempName);
+    counter++;
+  }
+
+  //log('尝试重命名文件:', filePath, '→', tempPath);
+
+  // 尝试重命名文件
+  fs.rename(filePath, tempPath, function(err) {
+    if (err) {
+      // 重命名失败，说明文件被占用
+      //log('文件被占用，重命名失败:', err.message);
+      callback(true);
+    } else {
+      // 重命名成功，说明文件未被占用
+      //log('文件未被占用，重命名成功');
+
+      // 恢复原文件名
+      fs.rename(tempPath, filePath, function(restoreErr) {
+        if (restoreErr) {
+          log('恢复文件名失败:', restoreErr.message);
+        } else {
+          log('文件未被占用，重命名成功:', filePath);
+        }
+        callback(false);
+      });
+    }
+  });
+}
+
 // 自动检测ev录屏工具路径
 function autoDetectEvPath() {
   var possiblePaths = [

package/tool/sockettool.js

@@ -53,7 +53,6 @@ function createSocket() {
         }
       } catch (ex) { }
     });
-
     conn.on('connect', function(code, reason) {
       utils.log('connect连接成功');
       broadcast('ws connect success');
@@ -61,9 +60,7 @@ function createSocket() {
     conn.on("close", function(code, reason) {
       utils.debug("connect关闭");
     });
-    conn.on("error", function(code, reason) {
-
-    });
+    conn.on("error", function(code, reason) {});
   }).listen(config.socketPort);
 
   socketServer.on('connection', function(code, reason) {