main.go

package main

import (
	"fmt"
	"log"
	"os"
	"strconv"

	"github.com/BurntSushi/toml"
	"github.com/joho/godotenv"

	"github.com/ystv/stv_web/controllers"
	"github.com/ystv/stv_web/store"
	"github.com/ystv/stv_web/structs"
	"github.com/ystv/stv_web/utils"
)

func main() {
	var tomlUsed, local, global bool
	var err error

	config := structs.Config{}

	root := false

	_, err = os.ReadFile("/toml/config.toml")
	if err == nil {
		_, err = toml.DecodeFile("/toml/config.toml", &config)
		root = true
	} else {
		_, err = os.ReadFile("./toml/config.toml")
		if err == nil {
			_, err = toml.DecodeFile("./toml/config.toml", &config)
		}
	}

	tomlUsed = err == nil

	if !tomlUsed {
		err = godotenv.Load(".env") // Load .env
		global = err == nil

		err = godotenv.Overload(".env.local") // Load .env.local
		local = err == nil
	}

	stvAddress := os.Getenv("STV_ADDRESS")
	domainName := os.Getenv("STV_DOMAIN_NAME")

	switch {
	case !global && !local && !tomlUsed && stvAddress == "" && domainName == "":
		log.Fatal("unable to find env files, toml file or env variables")
	case tomlUsed:
		log.Println("using toml file")
	case !global && !local:
		log.Println("using env variables")
	case local && global:
		log.Println("using global and local env files")
	case !local:
		log.Println("using global env file")
	default:
		log.Println("using local env file")
	}

	if !tomlUsed {
		var debug bool
		debug, err = strconv.ParseBool(os.Getenv("STV_DEBUG"))
		if err != nil {
			debug = false
		}

		var adPort int
		adPort, err = strconv.Atoi(os.Getenv("STV_AD_PORT"))
		if err != nil {
			log.Fatalf("failed to get ad port env: %+v", err)
		}

		var adSecurity int
		adSecurity, err = strconv.Atoi(os.Getenv("STV_AD_SECURITY"))
		if err != nil {
			log.Fatalf("failed to get ad security env: %+v", err)
		}

		var mailPort int
		mailPort, err = strconv.Atoi(os.Getenv("STV_MAIL_PORT"))
		if err != nil {
			log.Fatalf("failed to get mail port env: %+v", err)
		}

		if !tomlUsed {
			config = structs.Config{
				Server: structs.Server{
					Debug:      debug,
					Address:    os.Getenv("STV_ADDRESS"),
					DomainName: os.Getenv("STV_DOMAIN_NAME"),
				},
				AD: structs.AD{
					BypassUsername: os.Getenv("STV_AD_BYPASS_USERNAME"),
					BypassPassword: os.Getenv("STV_AD_BYPASS_PASSWORD"),
					Server:         os.Getenv("STV_AD_SERVER"),
					Port:           adPort,
					BaseDN:         os.Getenv("STV_AD_BASE_DN"),
					Security:       adSecurity,
					Bind: structs.ADBind{
						Username: os.Getenv("STV_AD_BIND_USERNAME"),
						Password: os.Getenv("STV_AD_BIND_PASSWORD"),
					},
				},
				Mail: structs.Mail{
					Host:      os.Getenv("STV_MAIL_HOST"),
					User:      os.Getenv("STV_MAIL_USERNAME"),
					Password:  os.Getenv("STV_MAIL_PASSWORD"),
					Port:      mailPort,
					DefaultTo: os.Getenv("STV_MAIL_DEFAULT_TO"),
				},
			}
		}
	}

	if config.Server.Debug {
		log.SetFlags(log.Llongfile)
		fmt.Println()
		log.Println("running in debug mode, do not use in production")
		fmt.Println()
	}

	var mailer *utils.Mailer
	var mailConfig utils.MailConfig
	if config.Mail.Host != "" {
		mailConfig = utils.MailConfig{
			Host:     config.Mail.Host,
			Port:     config.Mail.Port,
			Username: config.Mail.User,
			Password: config.Mail.Password,
		}

		mailer, err = utils.NewMailer(mailConfig)
		if err != nil {
			log.Printf("failed to connect to mail server: %+v", err)
		} else {
			log.Println("Connected to mail server")

			mailer.KeepAlive = true

			mailer.Defaults = utils.Defaults{
				DefaultTo:   config.Mail.DefaultTo,
				DefaultFrom: "YSTV STV <stv@ystv.co.uk>",
			}
		}
	}

	if err != nil {
		log.Fatalf("The web server couldn't be started!\n\n%s\n\nExiting!", err)
	}

	if config.Server.Debug {
		log.Println("Debug Mode - Disabled auth - do not run in production!")
	}

	newStore, err := store.NewStore(root)
	if err != nil {
		log.Fatal("Failed to create store: ", err)
	}

	_, err = newStore.GetAllowRegistration()
	if err != nil {
		_, err = newStore.SetAllowRegistration(false)
		if err != nil {
			log.Fatal("Failed to initialise allow registration in store: ", err)
		}
	}

	controller := controllers.GetController(config.Server.DomainName)

	router1 := New(NewRouter{
		Config: config,
		Repos:  controllers.NewRepos(controller, mailer, newStore, mailConfig),
		Debug:  config.Server.Debug,
		Mailer: mailer,
	})

	err = router1.Start()
	if err != nil {
		log.Fatalf("The web server couldn't be started!\n\n%s\n\nExiting!", err)
	}
}