ffui/main.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)})
	}
}