3 posts tagged with "Electron"

function getVideoDetail(fileInfo: fileInfoDTO): Promise<VideoDetailDTO | null> {
    return new Promise(async (resolve) => {
        const url = await tempVideo(fileInfo)
        fs.stat(url, (err, stats) => {
            if (err) {
                resolve(null)
                return
            }
            const fileSizeInBytes = stats.size
            const fileSizeInMB = (fileSizeInBytes / (1024 * 1024)).toFixed(2)

            ffmpeg.ffprobe(url, (err, metadata) => {
                if (err) {
                    console.error('获取视频元数据时出错:', err)
                    resolve(null)
                    return
                }

                // 获取视频流
                const videoStream = metadata.streams.find((stream) => stream.codec_type === 'video')
                // const audioStream = metadata.streams.find((stream) => stream.codec_type === 'audio')

                if (videoStream) {
                    const frameRateArray = videoStream.r_frame_rate.split('/');
                    const frameRate = parseInt(frameRateArray[0], 10);
                    console.log(frameRate); 

                    const obj: VideoDetailDTO = {
                        url: url,
                        name: fileInfo.name,
                        size: fileSizeInMB,
                        width: videoStream.width,
                        height: videoStream.height,
                        duration: metadata.format.duration,
                        codec_name: videoStream.codec_name,
                        r_frame_rate: frameRate,
                        bit_rate: Number((videoStream.bit_rate / 1000).toFixed(0)),
                        color_space: videoStream.color_space,
                        format_name: metadata.format.format_name
                    }
                    resolve(obj)
                } else {
                    resolve(null)
                }
            })
        })
    })
}

/**
   * 选择 目录
   * @param map 
   * @param event 
   * @param obj 
   */
  SELECT_FOLDER:  (map: getMapFuns, event: Electron.IpcMainEvent): void => {
    console.log('---')
    const win = map?.getMainWin?.()
    if (win) {
      dialog.showOpenDialog(win, {
        properties: ['openDirectory'] // 设置属性为 'openDirectory' 以仅打开目录选择器。
      }).then(result => {
        console.log(result)
        if (!result.canceled) { // 检查用户是否点击了取消按钮。
          event.reply('SELECT_FOLDER_RESP', result.filePaths[0])
        } else {
          event.reply('SELECT_FOLDER_RESP',  null)
        }
      }).catch(err => {
        console.error(err); // 处理可能的错误。
        event.reply('SELECT_FOLDER_RESP',  null)
      });
    }
  }

    const ffmpeg = require("fluent-ffmpeg");
    const ffmpegPath = require('ffmpeg-static');
    ffmpeg.setFfmpegPath(ffmpegPath);

import ffmpeg from 'fluent-ffmpeg'
import ffmpegPath from 'ffmpeg-static'
ffmpeg.setFfmpegPath(ffmpegPath)
import fs from 'fs'
import path from 'path'
import { tmpdir } from 'os'

interface fileInfoDto {
  name: string
  content: any
}

// 将 timemark 字符串转换为秒
function timemarkToSeconds(timemark): number {
  const parts = timemark.split(':')
  const hours = parseInt(parts[0], 10)
  const minutes = parseInt(parts[1], 10)
  const seconds = parseFloat(parts[2])
  return hours * 3600 + minutes * 60 + seconds
}

// 生成文件返回地址
const tempVideo = (fileInfo: fileInfoDto): Promise<string> => {
  return new Promise((resolve) => {
    const { name, content } = fileInfo
    // 生成临时文件路径
    const tempFilePath = path.join(tmpdir(), name)
    // 将文件内容保存为临时文件
    fs.writeFile(tempFilePath, Buffer.from(content), (err) => {
      if (err) {
        console.error('保存临时文件出错:', err)
        resolve('')
        return
      }
      resolve(tempFilePath)
    })
  })
}

const VideoConversion = async (options: { fileInfo: fileInfoDto; outputFilePath: string }) => {
  console.log(options)
  const inputFilePath = await tempVideo(options.fileInfo)
  let totalFrames = 0
  if (inputFilePath) {
    // ffmpeg -i /var/folders/h7/zj0ytbjd62x1b7kp7f7t9_9c0000gn/T/录屏2025-03-09 10.53.04.mov -y /Users/mao/code/electron/ElectronConvert/o.mp4
    ffmpeg(inputFilePath)
      .output(options.outputFilePath)
      .on('start', (commandLine) => {
        console.log('开始转码: ' + commandLine)
      })
      .on('codecData', (data) => {
        totalFrames = timemarkToSeconds(data.duration)
        console.log('视频长度', totalFrames, '秒')
      })
      .on('progress', (progress) => {
        const timemark = progress.timemark
        // 将 timemark 转换为秒
        const currentTime = timemarkToSeconds(timemark)
        const progressPercentage = (currentTime / totalFrames) * 100
        console.log(`当前转码进度: ${progressPercentage.toFixed(2)}%`)
      })
      .on('end', () => {
        console.log('转码完成')
      })
      .on('error', (err) => {
        console.error('转码出错: ' + err.message)
      })
      .run()
  }
}

export default {
  VideoConversion,
  tempVideo
}