main.go

package main

import (
	"flag"
	"fmt"
	"os"
	"path"
	"runtime/pprof"
	"strings"

	"searchproxy/server"
	"searchproxy/util/miscellaneous"

	log "github.com/sirupsen/logrus"
)

// Global vars for versioning
var (
	Build     = "unset" // nolint
	BuildDate = "unset" // nolint
	GoVersion = "unset" // nolint
	Version   = "unset" // nolint
)

func main() { // nolint funlen
	var (
		geoIPDBFile    = flag.String("geoipdb", "GeoLite2-City.mmdb", "Path to Maxmind's DB file")
		bind           = flag.String("bind", "0.0.0.0:8000", "Address to bind to, host:port")
		readTimeout    = flag.Int("readt", 30, "Read timeout, seconds")
		requestTimeout = flag.Int("reqt", 30, "HTTP Request timeout (for mirror checks), seconds")
		writeTimeout   = flag.Int("writet", 30, "Write timeout, seconds")
		mirrorsPath    = flag.String("mirrors", "./mirrors.yml", "Path to mirrors.yml file")
		version        = flag.Bool("version", false, "Print version and exit")
		// Debugging
		debug      = flag.Bool("debug", false, "enable debug")
		cpuprofile = flag.String("cpuprofile", "", "write cpu profile to file. Enables debug")
	)

	flag.Parse()

	if *version {
		fmt.Printf("%s version %s\n", server.ProductName, Version)
		fmt.Printf("Build: %s\n", Build)
		fmt.Printf("BuildDate: %s\n", BuildDate)
		fmt.Printf("Go: %s\n\n", GoVersion)
		os.Exit(0)
	}

	buildInfo := &miscellaneous.BuildInfo{
		Build:     Build,
		BuildDate: BuildDate,
		GoVersion: GoVersion,
		Version:   Version,
	}
	searchProxyServer := server.New(*bind, *readTimeout, *writeTimeout, *requestTimeout, buildInfo)

	if *cpuprofile != "" {
		*debug = true
		f, err := os.Create(*cpuprofile)

		if err != nil {
			log.Fatal(err)
		}

		_ = pprof.StartCPUProfile(f)

		defer pprof.StopCPUProfile()
	}

	searchProxyServer.SetDebug(*debug)
	searchProxyServer.SetGeoIPDBFile(*geoIPDBFile)

	directory, file := path.Split(*mirrorsPath)
	filePattern := strings.TrimRight(file, ".yml")

	searchProxyServer.ConfigFromFile(filePattern, directory)
	searchProxyServer.Run()
}