main.go

package main

import (
	"net/http"
	"os"
	"path"
	"path/filepath"

	cli "github.com/alecthomas/kingpin/v2"
	"github.com/packaged/logger/v2"
	"github.com/packaged/logger/v2/ld"
	"go.uber.org/zap"
)

var (
	configPath     = cli.Flag("config", "Path to the config yaml").Short('c').String()
	devEnvironment = cli.Flag("development", "Development Environment").Short('d').Bool()
	verboseLog     = cli.Flag("verbose", "Verbose logging").Short('v').Bool()
)

var logs *logger.Logger

func main() {
	cli.Parse()

	var opts []logger.Option
	if *verboseLog {
		opts = append(opts, logger.Debug)
	} else {
		opts = append(opts, func(config *zap.Config) { logger.DisableStacktrace(config) })
	}
	if *devEnvironment {
		opts = append(opts, logger.WithConsoleEncoding)
	}
	logs = logger.Instance(opts...)

	var configPaths []string
	if *configPath != "" {
		// config file specified
		if filepath.IsAbs(*configPath) {
			// absolute path
			configPaths = append(configPaths, *configPath)
		} else if cwd, err := os.Getwd(); err == nil {
			// relative path
			configPaths = append(configPaths, path.Join(cwd, *configPath))
		}
	} else {
		if cwd, err := os.Getwd(); err == nil {
			// no config specified, search in current directory
			configPaths = append(configPaths, path.Join(cwd, "config.yaml"))
		}

		if binPath, err := filepath.Abs(filepath.Dir(os.Args[0])); err == nil {
			// search in binary directory
			configPaths = append(configPaths, path.Join(binPath, "config.yaml"))
		}
	}

	var cfg *Config
	for _, configFile := range configPaths {
		info, err := os.Stat(configFile)
		if !os.IsNotExist(err) && !info.IsDir() {
			cfg, err = LoadConfig(configFile)
			logs.FatalIf(err, "loading config")
			break
		}
	}

	if cfg == nil {
		logs.Fatal("Config file not found")
	}

	go startSshTunnel(cfg)

	p := NewProxy(cfg)
	httpServer := http.Server{Addr: cfg.ListenAddress, Handler: p}

	logs.Debug("Listening", ld.TrustedString("host", cfg.ListenAddress))
	if cfg.Tls {
		logs.Debug("Serving with TLS")
	}
	if cfg.Tls {
		logs.FatalIf(httpServer.ListenAndServeTLS(cfg.TlsCertFile, cfg.TlsKeyFile), "serve")
	}
	logs.FatalIf(httpServer.ListenAndServe(), "serve")
}