录制和网络路径

src/main/recorder.js

@@ -0,0 +1,117 @@
+const { desktopCapturer, ipcMain, app } = require('electron');
+const fs = require('fs');
+const path = require('path');
+
+class ScreenRecorder {
+  constructor() {
+    this.mediaRecorder = null;
+    this.recordedChunks = [];
+    this.isRecording = false;
+    this.mainWindow = null;
+  }
+
+  setMainWindow(window) {
+    this.mainWindow = window;
+  }
+
+  async startRecording() {
+    if (this.isRecording) {
+      return { success: false, message: '正在录制中' };
+    }
+
+    try {
+      // 获取屏幕源
+      const sources = await desktopCapturer.getSources({
+        types: ['screen'],
+        thumbnailSize: { width: 1920, height: 1080 }
+      });
+
+      if (sources.length === 0) {
+        return { success: false, message: '无法获取屏幕源' };
+      }
+
+      // 发送源ID到渲染进程
+      const sourceId = sources[0].id;
+      this.isRecording = true;
+      
+      return { 
+        success: true, 
+        sourceId: sourceId,
+        message: '开始录制' 
+      };
+    } catch (error) {
+      console.error('启动录制失败:', error);
+      return { success: false, message: error.message };
+    }
+  }
+
+  async stopRecording(videoData) {
+    if (!this.isRecording) {
+      return { success: false, message: '当前没有进行录制' };
+    }
+
+    try {
+      this.isRecording = false;
+      
+      // 确保videos目录存在 - 在应用根目录下
+      const appPath = process.env.NODE_ENV === 'development' 
+        ? path.join(__dirname, '../../') 
+        : path.dirname(app.getPath('exe'));
+      const videosDir = path.join(appPath, 'videos');
+      if (!fs.existsSync(videosDir)) {
+        fs.mkdirSync(videosDir, { recursive: true });
+      }
+
+      // 生成文件名：soonworker+时间戳
+      const timestamp = new Date().toISOString().replace(/[:.]/g, '-').slice(0, -5);
+      const fileName = `soonworker_${timestamp}.webm`;
+      const filePath = path.join(videosDir, fileName);
+
+      // 将base64数据转换为Buffer并保存
+      const buffer = Buffer.from(videoData.replace('data:video/webm;base64,', ''), 'base64');
+      fs.writeFileSync(filePath, buffer);
+
+      console.log('视频已保存到:', filePath);
+      
+      return { 
+        success: true, 
+        filePath: filePath,
+        fileName: fileName,
+        message: '录制已保存' 
+      };
+    } catch (error) {
+      console.error('停止录制失败:', error);
+      return { success: false, message: error.message };
+    }
+  }
+
+  // 获取videos目录路径
+  getVideosPath() {
+    const appPath = process.env.NODE_ENV === 'development' 
+      ? path.join(__dirname, '../../') 
+      : path.dirname(app.getPath('exe'));
+    return path.join(appPath, 'videos');
+  }
+}
+
+// 创建录制器实例
+const recorder = new ScreenRecorder();
+
+// IPC事件处理
+ipcMain.handle('start-recording', async () => {
+  return await recorder.startRecording();
+});
+
+ipcMain.handle('stop-recording', async (event, videoData) => {
+  return await recorder.stopRecording(videoData);
+});
+
+ipcMain.handle('get-videos-path', () => {
+  return recorder.getVideosPath();
+});
+
+// 使用全局变量方式，避免webpack模块转换问题
+global.ScreenRecorderInstance = recorder;
+global.setMainWindow = (window) => recorder.setMainWindow(window);
+
+// 不导出任何内容，完全依赖全局变量