RoadSign/pkg/cmd/server/main.go

package main

import (
	"os"
	"os/signal"
	"strings"
	"syscall"

	roadsign "code.smartsheep.studio/goatworks/roadsign/pkg"
	"code.smartsheep.studio/goatworks/roadsign/pkg/hypertext"
	"code.smartsheep.studio/goatworks/roadsign/pkg/sideload"
	"code.smartsheep.studio/goatworks/roadsign/pkg/sign"
	"github.com/google/uuid"
	"github.com/rs/zerolog"
	"github.com/rs/zerolog/log"
	"github.com/spf13/viper"
)

func init() {
	zerolog.TimeFieldFormat = zerolog.TimeFormatUnix
	log.Logger = log.Output(zerolog.ConsoleWriter{Out: os.Stdout})
}

func main() {
	// Configure settings
	viper.AddConfigPath(".")
	viper.AddConfigPath("..")
	viper.SetConfigName("settings")
	viper.SetConfigType("yaml")

	// Load settings
	if err := viper.ReadInConfig(); err != nil {
		log.Panic().Err(err).Msg("An error occurred when loading settings.")
	}

	// Present settings
	if len(viper.GetString("security.credential")) <= 0 {
		credential := strings.ReplaceAll(uuid.NewString(), "-", "")
		viper.Set("security.credential", credential)
		_ = viper.WriteConfig()

		log.Warn().Msg("There isn't any api credential configured in settings.yml, auto generated a credential for api accessing.")
		log.Warn().Msgf("RoadSign auto generated api credential is %s", credential)
	}

	// Load & init sign
	if err := sign.ReadInConfig(viper.GetString("paths.configs")); err != nil {
		log.Panic().Err(err).Msg("An error occurred when loading configurations.")
	} else {
		log.Info().Int("count", len(sign.App.Sites)).Msg("All configuration has been loaded.")
	}

	// Preheat processes
	log.Info().Msg("Preheating processes...")
	sign.App.PreheatProcesses(func(total int, success int) {
		log.Info().Int("requested", total).Int("succeed", success).Msgf("Preheat processes completed!")
	})

	// Init hypertext server
	hypertext.RunServer(
		hypertext.InitServer(),
		viper.GetStringSlice("hypertext.ports"),
		viper.GetStringSlice("hypertext.secured_ports"),
		viper.GetString("hypertext.certificate.pem"),
		viper.GetString("hypertext.certificate.key"),
	)

	// Init sideload server
	hypertext.RunServer(
		sideload.InitSideload(),
		viper.GetStringSlice("hypertext.sideload_ports"),
		viper.GetStringSlice("hypertext.sideload_secured_ports"),
		viper.GetString("hypertext.certificate.sideload_pem"),
		viper.GetString("hypertext.certificate.sideload_key"),
	)

	log.Info().Msgf("RoadSign v%s is started...", roadsign.AppVersion)

	quit := make(chan os.Signal, 1)
	signal.Notify(quit, syscall.SIGINT, syscall.SIGTERM)
	<-quit

	log.Info().Msgf("RoadSign v%s is quitting...", roadsign.AppVersion)
}