GhostYgg.go

package main

import (
	"flag"
	"fmt"
	"github.com/GuillaumeMCK/GhostYgg/tui"
	"github.com/GuillaumeMCK/GhostYgg/utils"
	"log"
	"os"
	"path/filepath"
)

var (
	torrentFiles   []string
	output         *string
	helpFlag       *bool
	downloadFolder string
)

func init() {
	torrentFiles = []string{}
	// Filter all args that are not torrent files
	for _, arg := range os.Args[1:] {
		if filepath.Ext(arg) == ".torrent" {
			torrentFiles = append(torrentFiles, arg)
		}
	}
	output = flag.String("o", "", "Download directory")
	helpFlag = flag.Bool("help", false, "Show this message")

	// Define how to use the program
	flag.Usage = func() {
		fmt.Printf("GhostYgg - Torrent client\n\n")
		fmt.Printf("Usage: %s file1.torrent file2.torrent ... [options]\n\n", os.Args[0])
		fmt.Printf("Options:\n")
		flag.PrintDefaults()
	}

	// Parse command-line flags
	flag.Parse()

	// If -help flag is set, show the usage and exit
	if *helpFlag {
		flag.Usage()
		return
	}

	// Determine default download folder
	defaultDownloadFolder, err := utils.GetDefaultDownloadFolder()
	if err != nil {
		log.Fatal(err)
	}

	// Use the value of -download-folder flag, or default download folder
	downloadFolder = *output
	if downloadFolder == "" {
		downloadFolder = defaultDownloadFolder
	}

	// Make sure download folder exists
	err = os.MkdirAll(downloadFolder, os.ModePerm)
	if err != nil {
		log.Fatal(err)
	}
}

func main() {
	err := tui.StartTUI(torrentFiles, downloadFolder)
	if err != nil {
		log.Fatal(err)
	}
}