nginx-ui/server/server.go

package server

import (
	"github.com/0xJacky/Nginx-UI/server/internal/analytic"
	"github.com/0xJacky/Nginx-UI/server/internal/cert"
	"github.com/0xJacky/Nginx-UI/server/internal/logger"
	"github.com/0xJacky/Nginx-UI/server/internal/nginx"
	"github.com/0xJacky/Nginx-UI/server/internal/upgrader"
	"github.com/0xJacky/Nginx-UI/server/model"
	"github.com/0xJacky/Nginx-UI/server/query"
	"github.com/0xJacky/Nginx-UI/server/router"
	"github.com/0xJacky/Nginx-UI/server/settings"
	"github.com/go-co-op/gocron"
	"github.com/google/uuid"
	"github.com/jpillora/overseer"
	"log"
	"mime"
	"net/http"
	"time"
)

func GetRuntimeInfo() (r upgrader.RuntimeInfo, err error) {
	return upgrader.GetRuntimeInfo()
}

func Program(state overseer.State) {
	defer logger.Sync()
	// Hack: fix wrong Content Type of .js file on some OS platforms
	// See https://github.com/golang/go/issues/32350
	_ = mime.AddExtensionType(".js", "text/javascript; charset=utf-8")

	logger.Info("Nginx config dir path: " + nginx.GetConfPath())

	if "" != settings.ServerSettings.JwtSecret {
		db := model.Init()
		query.Init(db)
	}

	if "" == settings.ServerSettings.NodeSecret {
		logger.Warn("NodeSecret is empty")
		settings.ServerSettings.NodeSecret = uuid.New().String()
		settings.ReflectFrom()

		err := settings.Save()
		if err != nil {
			logger.Error("Error save settings")
		}
		logger.Warn("Generated NodeSecret: ", settings.ServerSettings.NodeSecret)
	}

	s := gocron.NewScheduler(time.UTC)
	job, err := s.Every(30).Minute().SingletonMode().Do(cert.AutoObtain)

	if err != nil {
		logger.Fatalf("AutoCert Job: %v, Err: %v\n", job, err)
	}

	s.StartAsync()

	go analytic.RecordServerAnalytic()
	go analytic.RetrieveNodesStatus()

	err = http.Serve(state.Listener, router.InitRouter())
	if err != nil {
		log.Fatalln(err)
	}
	logger.Info("Server exiting")
}