mirror of
https://github.com/IllusionMan1212/ffui.git
synced 2026-10-24 02:12:35 +00:00
Illusion
24924e0bd1
This allows us to send interrupt signals and kill the process when ctrl+c is pressed. Added `copy` and `None` encoder options to copy the video/audio streams without modifications and to allow the removal of audio from video files. Updated to Go 1.23
141 lines
3.1 KiB
Go
141 lines
3.1 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"log"
|
|
"os"
|
|
"os/exec"
|
|
"path/filepath"
|
|
"strings"
|
|
|
|
tea "github.com/charmbracelet/bubbletea"
|
|
"github.com/gabriel-vasile/mimetype"
|
|
)
|
|
|
|
func main() {
|
|
if len(os.Args) < 2 {
|
|
log.Fatal("No directory or file provided.")
|
|
}
|
|
|
|
path := os.Args[1]
|
|
|
|
if path == "" {
|
|
log.Fatal("No directory or file provided.")
|
|
}
|
|
|
|
absolutePath, err := filepath.Abs(path)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
fileInfo, err := os.Stat(absolutePath)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
ffui := initialModel(fileInfo, absolutePath)
|
|
|
|
p := tea.NewProgram(ffui)
|
|
|
|
ffui.Program = p
|
|
|
|
f, err := tea.LogToFile("ffui.log", "ffui")
|
|
defer f.Close()
|
|
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
err = p.Start()
|
|
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
}
|
|
|
|
func buildFFmpegCmdArgs(fullFilePath string, outFullFilePath string, cfg ParsedConfig, additionalArgs ...string) []string {
|
|
args := make([]string, 0, 10)
|
|
// Input file
|
|
args = append(args, "-i")
|
|
args = append(args, fullFilePath)
|
|
|
|
// Encoding parameters
|
|
args = append(args, "-c:v")
|
|
args = append(args, cfg.VideoEncoder)
|
|
|
|
switch cfg.VideoEncoder {
|
|
case "copy":
|
|
// No options
|
|
case "libx264", "libx265":
|
|
args = append(args, "-crf")
|
|
args = append(args, cfg.CRF)
|
|
|
|
args = append(args, "-preset")
|
|
args = append(args, cfg.Preset)
|
|
case "libvpx-vp9":
|
|
args = append(args, "-crf")
|
|
args = append(args, cfg.CRF)
|
|
case "librav1e":
|
|
// No options
|
|
case "libsvtav1":
|
|
args = append(args, "-crf")
|
|
args = append(args, cfg.CRF)
|
|
|
|
// TODO: preset is a number for this encoder (-2 to 13)
|
|
}
|
|
|
|
switch cfg.AudioEncoder {
|
|
case "None":
|
|
args = append(args, "-an")
|
|
case "copy", "aac", "libopus":
|
|
args = append(args, "-c:a")
|
|
args = append(args, cfg.AudioEncoder)
|
|
}
|
|
|
|
for _, arg := range additionalArgs {
|
|
args = append(args, arg)
|
|
}
|
|
|
|
// Output file
|
|
args = append(args, outFullFilePath)
|
|
|
|
return args
|
|
}
|
|
|
|
func encode(fullFilePath string, fileName string, teaP *tea.Program, cfg ParsedConfig) {
|
|
mType, err := mimetype.DetectFile(fullFilePath)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
if !strings.HasPrefix(mType.String(), "video/") {
|
|
log.Fatalf("%s is not a valid video file\n", fileName)
|
|
}
|
|
|
|
parentDir := filepath.Dir(fullFilePath)
|
|
extensionIndex := strings.LastIndex(fileName, ".")
|
|
newFileName := fileName[:extensionIndex]
|
|
extension := fileName[extensionIndex:]
|
|
newFileFullPath := filepath.Join(parentDir, newFileName+fmt.Sprintf("_[%s]_[%s]", cfg.VideoEncoder, cfg.AudioEncoder)+extension)
|
|
|
|
if _, err := os.Stat(newFileFullPath); err == nil {
|
|
if cfg.SkipEncodedVid {
|
|
log.Printf("%s already exists with the exact same encodings (crf and preset might be different though), skipping.", newFileFullPath)
|
|
teaP.Send(finishedEncodingVideo{})
|
|
return
|
|
} else {
|
|
os.Remove(newFileFullPath)
|
|
}
|
|
}
|
|
|
|
cmdArgs := buildFFmpegCmdArgs(fullFilePath, newFileFullPath, cfg, "-progress", "unix://"+getProgressSocket(fullFilePath, teaP))
|
|
cmd := exec.Command("ffmpeg", cmdArgs...)
|
|
// TODO: send the cmd obj as a tea msg so we can send interrupt signals to the process when ctrl+c is pressed
|
|
//teaP.Send()
|
|
err = cmd.Run()
|
|
|
|
if err != nil {
|
|
teaP.Send(errQuitMsg{msg: fmt.Sprintf("FFmpeg exited with error code: %s\n\nError: %v", err, cmd.Stderr)})
|
|
}
|
|
}
|