dt-admin-pc-v2/electron/electron.js

// electron/electron.js
const path = require('path');
const { app, BrowserWindow, ipcMain } = require('electron');

const isDev = process.env.IS_DEV == "true" ? true : false;

// rtsp相关,ffmpeg-static会自动下载对应平台windos，linux，mac(intle)，mac(M1版本)的ffmpeg二进制文件
const ffmpegPath = require('ffmpeg-static')
/**
  stream相关配置及方法
  name = options.name
  streamUrl = options.streamUrl
  width = options.width
  height = options.height
  wsPort = options.wsPort
  nputStreamStarted = false
  stream = undefined
  ffmpegPath = options?.ffmpegPath ?? ffmpeg
  stop()
 */
  const Stream = require('node-rtsp-stream')
  /**
   * rtsp列表
   * interface {
   *   rtspUrl: {
   *     ws: websocket地址
   *     stream: stream实例
   *   }
   * }
   */
  const rtspOpenders = {}
  let addPort = 9000

/** 
 * 开启rtsp
 * @param rtsp {String} rtsp流地址
 */


ipcMain.on('openRtsp', (event, rtsp) => {
  /** 判断是否已开启,若已开启,直接返回ws地址, 未开启则先开启再返回 */
  if (rtspOpenders[rtsp]) {
    event.returnValue = {
      code: 200,
      msg: '开启成功',
      ws: rtspOpenders[rtsp].ws
    }
  } else {
    addPort++
    const stream = new Stream({
      name: `socket-${addPort}`,
      streamUrl: rtsp,
      wsPort: addPort,
      ffmpegPath: app.isPackaged ? ffmpegPath.replace('app.asar', 'app.asar.unpacked') : ffmpegPath,
      ffmpegOptions: {
        '-stats': '',
        '-r': 30
      }
    }).on('exitWithError', () => {
      stream.stop()
      delete rtspOpenders[rtsp]
      event.returnValue = {
        code: 400,
        msg: '开启失败'
      }
    })
    rtspOpenders[rtsp] = {
      ws: `ws://localhost:${addPort}`,
      stream: stream
    }
    event.returnValue = {
      code: 200,
      msg: '开启成功',
      ws: rtspOpenders[rtsp].ws
    }
  }
})

/**
 * 关闭rtsp
 */
ipcMain.on('closeRtsp', (event, rtsp) => {
  if (rtspOpenders[rtsp]) {
    // 停止解析
    rtspOpenders[rtsp].stream.stop()
    // 删除该项
    delete rtspOpenders[rtsp]
    // 返回结果
    event.returnValue = {
      code: 200,
      msg: '关闭成功'
    }
  } else {
    event.returnValue = {
      code: 400,
      msg: '未找到该rtsp'
    }
  }
})


///////////////////////////////////////////

function createWindow() {
  // Create the browser window.
  const mainWindow = new BrowserWindow({
    width: 800,
    height: 600,
    webPreferences: {
      //preload: path.join(__dirname, 'preload.js'),
      javascript: true,
      plugins: true,
      allowRunningInsecureContent: true,
      // 允许web断使用node
      nodeIntegration: true,
      contextIsolation: false,
      // 同源策略关闭
      webSecurity: false,
    },
  });

  // and load the index.html of the app.
  // win.loadFile("index.html");

  mainWindow.loadURL(
    isDev
      ? 'http://localhost:3200'
      : `file://${path.join(__dirname, '../dist/index.html')}`
  );

  // mainWindow.loadURL('http://127.0.0.1:5500');
  // Open the DevTools.
  // if (isDev) {
    mainWindow.webContents.openDevTools();
  // }
}

// This method will be called when Electron has finished
// initialization and is ready to create browser windows.
// Some APIs can only be used after this event occurs.
app.whenReady().then(() => {
  createWindow()
  app.on('activate', function () {
    // On macOS it's common to re-create a window in the app when the
    // dock icon is clicked and there are no other windows open.
    if (BrowserWindow.getAllWindows().length === 0) createWindow()
  })
});

// Quit when all windows are closed, except on macOS. There, it's common
// for applications and their menu bar to stay active until the user quits
// explicitly with Cmd + Q.
app.on('window-all-closed', () => {
  if (process.platform !== 'darwin') {
    app.quit();
  }
});