TorrServerJellyfin/server/cmd/main.go

package main

import (
	"context"
	"fmt"
	"net"
	"os"
	"time"

	"github.com/alexflint/go-arg"
	"github.com/pkg/browser"

	"server"
	"server/log"
	"server/settings"
	"server/version"
)

type args struct {
	Port       string `arg:"-p" help:"web server port"`
	Path       string `arg:"-d" help:"database path"`
	LogPath    string `arg:"-l" help:"log path"`
	WebLogPath string `arg:"-w" help:"web log path"`
	RDB        bool   `arg:"-r" help:"start in read-only DB mode"`
	HttpAuth   bool   `arg:"-a" help:"http auth on all requests"`
	DontKill   bool   `arg:"-k" help:"dont kill server on signal"`
	UI         bool   `arg:"-u" help:"run page torrserver in browser"`
}

func (args) Version() string {
	return "TorrServer " + version.Version
}

var params args

func main() {
	arg.MustParse(&params)

	if params.Path == "" {
		params.Path, _ = os.Getwd()
	}

	if params.Port == "" {
		params.Port = "8090"
	}

	settings.Path = params.Path
	settings.HttpAuth = params.HttpAuth
	log.Init(params.LogPath, params.WebLogPath)

	dnsResolve()
	Preconfig(params.DontKill)

	if params.UI {
		go func() {
			time.Sleep(time.Second)
			browser.OpenURL("http://127.0.0.1:" + params.Port)
		}()
	}

	server.Start(params.Port, params.RDB)
	log.TLogln(server.WaitServer())
	log.Close()
	time.Sleep(time.Second * 3)
	os.Exit(0)
}

func dnsResolve() {
	addrs, err := net.LookupHost("www.google.com")
	if len(addrs) == 0 {
		fmt.Println("Check dns", addrs, err)

		fn := func(ctx context.Context, network, address string) (net.Conn, error) {
			d := net.Dialer{}
			return d.DialContext(ctx, "udp", "1.1.1.1:53")
		}

		net.DefaultResolver = &net.Resolver{
			Dial: fn,
		}

		addrs, err = net.LookupHost("www.themoviedb.org")
		fmt.Println("Check new dns", addrs, err)
	}
}