RVC-Models-Downloader/main.go

package main

import (
	"context"
	"flag"
	"fmt"
	"os"
	"runtime"
	"sync"
	"time"

	"github.com/fumiama/terasu/dns"
	"github.com/fumiama/terasu/ip"
	ui "github.com/gizak/termui/v3"
	"github.com/sirupsen/logrus"
	"gopkg.in/yaml.v3"

	_ "rvcmd/console"
)

//go:generate ./pckcfg.sh assets packs tools

const ua = "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/123.0.0.0 Safari/537.36 Edg/123.0.0.0"

var (
	notui = false
	sc    screen
)

func main() {
	ntrs := flag.Bool("notrs", false, "use standard TLS client")
	dnsf := flag.String("dns", "", "custom dns.yaml")
	cust := flag.Bool("c", false, "use custom yaml instruction")
	force := flag.Bool("f", false, "force download even file exists")
	wait := flag.Uint("w", 4, "connection waiting seconds")
	flag.BoolVar(&notui, "notui", false, "use plain text instead of TUI")
	flag.Parse()
	args := flag.Args()
	if len(args) != 1 {
		fmt.Println("Usage:", os.Args[0], "[-notrs] [-dns dns.yaml] 'target/to/download'")
		flag.PrintDefaults()
		fmt.Println("  'target/to/download'\n        like packs/general/latest")
		fmt.Println("All available targets:")
		fmt.Println(cmdlst.String())
		return
	}
	if notui {
		logrus.Infoln("RVC Models Downloader start at", time.Now().Local().Format(time.DateTime+" (MST)"))
		logrus.Infof("operating system: %s, architecture: %s", runtime.GOOS, runtime.GOARCH)
		logrus.Infoln("is ipv6 available:", ip.IsIPv6Available.Get())
	} else {
		if err := ui.Init(); err != nil {
			logrus.Errorln("failed to initialize termui:", err)
			return
		}
		defer ui.Close()
		sc = newscreen()
	}
	ctx, cancel := context.WithCancel(context.Background())
	defer cancel()
	wg := sync.WaitGroup{}
	wg.Add(1)
	go func() {
		defer wg.Done()
		if *dnsf != "" {
			f, err := os.Open(*dnsf)
			if err != nil {
				errorln("open custom dns file", *dnsf, "err:", err)
				return
			}
			m := map[string][]string{}
			err = yaml.NewDecoder(f).Decode(&m)
			if err != nil {
				errorln("decode custom dns file", *dnsf, "err:", err)
				return
			}
			_ = f.Close()
			if ip.IsIPv6Available.Get() {
				dns.IPv6Servers.Add(m)
			} else {
				dns.IPv4Servers.Add(m)
			}
			infoln("custom dns file added")
		}
		usercfg, err := readconfig(args[0], *cust)
		if err != nil {
			errorln(err)
			return
		}
		ch := make(chan struct{})
		go func() {
			err := usercfg.download(args[0], "", time.Second*time.Duration(*wait), *cust, !*ntrs, *force)
			ch <- struct{}{}
			if err != nil {
				errorln(err)
				return
			}
		}()
		select {
		case <-ch:
			infoln("all download tasks finished.")
		case <-ctx.Done():
			logrus.Warnln("download canceled")
		}
	}()
	if notui {
		wg.Wait()
	} else {
		sc.show(time.Second)
	}
}