nginx-ui/server/init/init.go

package init

import (
	"encoding/gob"
	"fmt"
	_ "github.com/beego/beego/v2/server/web/session/redis"
	"nginx-ui/server/config"
	"nginx-ui/server/db"
	"nginx-ui/server/models"
	"nginx-ui/server/modules/agent_server"
	"nginx-ui/server/modules/nginx/nginx_service"
	"nginx-ui/server/modules/proxy"
	_ "nginx-ui/server/routers"
	"nginx-ui/server/utils"
	"os"
)

func ensureIndexHtml() {
	if !utils.IsExist("views") {
		os.Mkdir("views", 0777)
	}
	sourceDir := utils.GetStaticDir()
	utils.CopyFile(sourceDir+"/index.html", "views/index.html")
}

func init() {
	fmt.Printf("-------init---")
	gob.Register(models.User{})
	db.Init()
	config.InitAdmin()
	fmt.Println("init success")
	ensureRoutes()
	ensureIndexHtml()
	proxy.Instance.RefreshProxies()

	agent_server.SetMessageHandler(models.RegisterNginxType, nginx_service.AgentRegister)
}