首次上传

main.js

@@ -0,0 +1,143 @@
+const { app, BrowserWindow, ipcMain } = require('electron');
+const path = require('path');
+const express = require('express');
+const http = require('http');
+const socketIo = require('socket.io');
+const cors = require('cors');
+
+let mainWindow;
+let streamingServer;
+let io;
+
+// 创建主窗口
+function createWindow() {
+  mainWindow = new BrowserWindow({
+    width: 1200,
+    height: 800,
+    webPreferences: {
+      nodeIntegration: true,
+      contextIsolation: false,
+      enableRemoteModule: true,
+      webSecurity: false
+    },
+    icon: path.join(__dirname, 'assets', 'icon.svg'),
+    title: 'Trae Video Server'
+  });
+
+  mainWindow.loadFile('index.html');
+  
+  // 开发模式下打开开发者工具
+  if (process.argv.includes('--dev')) {
+    mainWindow.webContents.openDevTools();
+  }
+}
+
+// 创建流媒体服务器
+function createStreamingServer() {
+  const app = express();
+  app.use(cors());
+  app.use(express.static(path.join(__dirname, 'public')));
+  
+  const server = http.createServer(app);
+  io = socketIo(server, {
+    cors: {
+      origin: "*",
+      methods: ["GET", "POST"]
+    }
+  });
+
+  // 提供网页预览接口
+  app.get('/preview', (req, res) => {
+    res.sendFile(path.join(__dirname, 'public', 'preview.html'));
+  });
+
+  // Socket.IO连接处理
+  io.on('connection', (socket) => {
+    console.log('客户端连接到流媒体服务');
+    
+    socket.on('disconnect', () => {
+      console.log('客户端断开连接');
+    });
+  });
+
+  const PORT = 3000;
+  server.listen(PORT, () => {
+    console.log(`流媒体服务器运行在 http://localhost:${PORT}`);
+  });
+
+  return server;
+}
+
+// 获取视频设备列表
+ipcMain.handle('get-video-devices', async () => {
+  try {
+    const devices = await navigator.mediaDevices.enumerateDevices();
+    const videoDevices = devices.filter(device => device.kind === 'videoinput');
+    return videoDevices.map(device => ({
+      deviceId: device.deviceId,
+      label: device.label || `摄像头 ${device.deviceId.slice(0, 8)}...`
+    }));
+  } catch (error) {
+    console.error('获取设备列表失败:', error);
+    return [];
+  }
+});
+
+// 开始预览
+ipcMain.handle('start-preview', async (event, deviceId) => {
+  try {
+    // 如果流媒体服务器还没启动，则启动它
+    if (!streamingServer) {
+      streamingServer = createStreamingServer();
+    }
+    
+    return {
+      success: true,
+      streamUrl: 'http://localhost:3000/preview',
+      message: '预览已启动，流媒体服务已就绪'
+    };
+  } catch (error) {
+    console.error('启动预览失败:', error);
+    return {
+      success: false,
+      message: '启动预览失败: ' + error.message
+    };
+  }
+});
+
+// 停止预览
+ipcMain.handle('stop-preview', async () => {
+  try {
+    if (io) {
+      io.emit('stop-stream');
+    }
+    return { success: true, message: '预览已停止' };
+  } catch (error) {
+    console.error('停止预览失败:', error);
+    return { success: false, message: '停止预览失败: ' + error.message };
+  }
+});
+
+// 广播视频帧到所有连接的客户端
+ipcMain.on('broadcast-frame', (event, frameData) => {
+  if (io) {
+    io.emit('video-frame', frameData);
+  }
+});
+
+app.whenReady().then(createWindow);
+
+app.on('window-all-closed', () => {
+  if (streamingServer) {
+    streamingServer.close();
+  }
+  if (process.platform !== 'darwin') {
+    app.quit();
+  }
+});
+
+app.on('activate', () => {
+  if (BrowserWindow.getAllWindows().length === 0) {
+    createWindow();
+  }
+});