main.go

/*
AVI Video Compression: ffmpeg -i input.avi -vcodec msmpeg4v2 output.avi
MP4 Video Compression: ffmpeg -i input.mp4 -acodec mp2 output.mp4
*/

package main

import (
	"flag"
	"fmt"
	"os"
	"path/filepath"

	"github.com/g-kutty/v-comp/compressor"
	"github.com/g-kutty/v-comp/graceful"
	"github.com/g-kutty/v-comp/logger"
	"github.com/g-kutty/v-comp/watcher"
)

const appVersion = "1.0"

var (
	path    string
	help    bool
	version bool
	threads int
)

// Read command line arguments before start.
func init() {
	flag.StringVar(&path, "p", "", "")
	flag.StringVar(&path, "path", "", "")
	flag.IntVar(&threads, "t", 4, "")
	flag.IntVar(&threads, "thread", 4, "")
	flag.BoolVar(&version, "v", false, "")
	flag.BoolVar(&version, "version", false, "")
	flag.BoolVar(&help, "h", false, "")
	flag.BoolVar(&help, "help", false, "")
	flag.Usage = func() {
		fmt.Fprintf(os.Stderr, "Usage: v-comp [options]\n")
		fmt.Fprintf(os.Stderr, "options:\n")
		fmt.Fprintf(os.Stderr, "\t-p, -path      Directory                The directory to watch.\n")
		fmt.Fprintf(os.Stderr, "\t-t, -thread    Threads                  Number of threads run at a time.\n")
		fmt.Fprintf(os.Stderr, "\t-v, -version   Version                  Prints the version.\n")
		fmt.Fprintf(os.Stderr, "\t-h, -help      Help                     Show this help.\n")
	}
}

func main() {

	// gracefull shutdown.
	go graceful.ActivateGracefulShutdown()
	parseFlags()

	// video files.
	var files []string

	// log
	logger.Info().Command("watching", "w").Message(logger.FormattedMessage(path)).Log()

	// walk through files.
	err := filepath.Walk(path, watcher.Visit(&files))
	if err != nil {
		logger.Error().Message(err.Error()).Log()
		return
	}

	// compress all video files.
	err = compressor.Compress(files, threads)
	if err != nil {
		logger.Error().Message(err.Error()).Log()
		return
	}
}

// parseFlags read command line arguments.
func parseFlags() {
	flag.Parse()

	// display version.
	if version {
		fmt.Printf("v-comp v%s\n", appVersion)
		os.Exit(0)
	}

	// display help guides for command line arguments.
	if help {
		flag.Usage()
		os.Exit(0)
	}
	validateFlags()
}

// validateFlags validate the flag values.
func validateFlags() {
	if path == "" {
		// Get the current working directory if no path is specified.
		dir, _ := os.Getwd()
		path, _ = filepath.Abs(dir)
	} else {
		dir, err := os.Stat(path)
		if err != nil {
			logger.Error().Message("Cannot find path,", logger.FormattedMessage(path)).Log()
			os.Exit(1)
		}
		if !dir.IsDir() {
			logger.Error().Message(fmt.Sprintf("Invalid path, %s. Path must be directory.", logger.FormattedMessage(path))).Log()
			os.Exit(1)
		}
		path, _ = filepath.Abs(path)
	}
}