odboy
4da4b738e6
- 删除 vars 全局状态，按职责拆分 internal/ 包：scanner/probe/classify/organize/mover/store/digest - 数据驱动归类规则替代图片/视频重复分支，store 改为结构体实例 - build.sh 按系统类型编译：Windows 用 upx 压缩，macOS/Linux 不压缩 - 忽略目录拆分：图片_已忽略 / 视频_已忽略 - 输出构建产物加入 .gitignore
81 lines
2.0 KiB
Go
81 lines
2.0 KiB
Go
// Package probe 负责媒体信息探测：图片尺寸读取、ffprobe 视频/音频解析与内嵌解码器释放。
|
|
package probe
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"os/exec"
|
|
"strconv"
|
|
)
|
|
|
|
// ProbeResult 媒体探测结果（图片/视频/音频通用）
|
|
type ProbeResult struct {
|
|
HasVideo bool
|
|
HasAudio bool
|
|
Width int
|
|
Height int
|
|
Duration float64
|
|
VideoCodec string
|
|
AudioCodec string
|
|
Format string
|
|
RawJSON string
|
|
}
|
|
|
|
type streamInfo struct {
|
|
Index int `json:"index"`
|
|
CodecType string `json:"codec_type"`
|
|
CodecName string `json:"codec_name"`
|
|
Width int `json:"width"`
|
|
Height int `json:"height"`
|
|
}
|
|
|
|
type formatInfo struct {
|
|
Duration string `json:"duration"`
|
|
Format string `json:"format_name"`
|
|
}
|
|
|
|
type probeOutput struct {
|
|
Streams []streamInfo `json:"streams"`
|
|
Format formatInfo `json:"format"`
|
|
}
|
|
|
|
// ProbeMediaFile 用 ffprobe 探测媒体文件，一次调用同时获取视频/音频流与时长
|
|
func ProbeMediaFile(ffprobePath, filePath string) (*ProbeResult, error) {
|
|
cmd := exec.Command(ffprobePath, "-v", "error",
|
|
"-show_entries", "stream=index,codec_type,codec_name,width,height:format=duration,format_name",
|
|
"-of", "json", filePath)
|
|
output, err := cmd.Output()
|
|
if err != nil {
|
|
return nil, fmt.Errorf("ffprobe failed: %w", err)
|
|
}
|
|
var out probeOutput
|
|
if err := json.Unmarshal(output, &out); err != nil {
|
|
return nil, fmt.Errorf("parse ffprobe output: %w", err)
|
|
}
|
|
res := &ProbeResult{
|
|
Format: out.Format.Format,
|
|
RawJSON: string(output),
|
|
}
|
|
if d, err := strconv.ParseFloat(out.Format.Duration, 64); err == nil {
|
|
res.Duration = d
|
|
}
|
|
for _, s := range out.Streams {
|
|
switch s.CodecType {
|
|
case "video":
|
|
res.HasVideo = true
|
|
if res.VideoCodec == "" {
|
|
res.VideoCodec = s.CodecName
|
|
}
|
|
if res.Width == 0 && s.Width > 0 && s.Height > 0 {
|
|
res.Width, res.Height = s.Width, s.Height
|
|
}
|
|
case "audio":
|
|
res.HasAudio = true
|
|
if res.AudioCodec == "" {
|
|
res.AudioCodec = s.CodecName
|
|
}
|
|
}
|
|
}
|
|
return res, nil
|
|
}
|