chore: prepare v2.0.0-beta.37

internal/kernel/boot.go

@@ -0,0 +1,128 @@
+package kernel
+
+import (
+	"crypto/rand"
+	"encoding/hex"
+	"github.com/0xJacky/Nginx-UI/internal/analytic"
+	"github.com/0xJacky/Nginx-UI/internal/cache"
+	"github.com/0xJacky/Nginx-UI/internal/cert"
+	"github.com/0xJacky/Nginx-UI/internal/cluster"
+	"github.com/0xJacky/Nginx-UI/internal/cron"
+	"github.com/0xJacky/Nginx-UI/internal/passkey"
+	"github.com/0xJacky/Nginx-UI/internal/validation"
+	"github.com/0xJacky/Nginx-UI/model"
+	"github.com/0xJacky/Nginx-UI/query"
+	"github.com/0xJacky/Nginx-UI/settings"
+	"github.com/google/uuid"
+	"github.com/uozi-tech/cosy"
+	sqlite "github.com/uozi-tech/cosy-driver-sqlite"
+	"github.com/uozi-tech/cosy/logger"
+	cSettings "github.com/uozi-tech/cosy/settings"
+	"mime"
+	"path"
+	"runtime"
+)
+
+func Boot() {
+	defer recovery()
+
+	async := []func(){
+		InitJsExtensionType,
+		InitDatabase,
+		InitNodeSecret,
+		InitCryptoSecret,
+		validation.Init,
+		cache.Init,
+	}
+
+	syncs := []func(){
+		analytic.RecordServerAnalytic,
+	}
+
+	for _, v := range async {
+		v()
+	}
+
+	for _, v := range syncs {
+		go v()
+	}
+}
+
+func InitAfterDatabase() {
+	syncs := []func(){
+		registerPredefinedUser,
+		cert.InitRegister,
+		cron.InitCronJobs,
+		cluster.RegisterPredefinedNodes,
+		analytic.RetrieveNodesStatus,
+		passkey.Init,
+		RegisterAcmeUser,
+	}
+
+	for _, v := range syncs {
+		go v()
+	}
+}
+
+func recovery() {
+	if err := recover(); err != nil {
+		buf := make([]byte, 1024)
+		runtime.Stack(buf, false)
+		logger.Errorf("%s\n%s", err, buf)
+	}
+}
+
+func InitDatabase() {
+	// Skip install
+	if settings.NodeSettings.SkipInstallation {
+		skipInstall()
+	}
+
+	if "" != cSettings.AppSettings.JwtSecret {
+		db := cosy.InitDB(sqlite.Open(path.Dir(cSettings.ConfPath), settings.DatabaseSettings))
+		model.Use(db)
+		query.Init(db)
+
+		InitAfterDatabase()
+	}
+}
+
+func InitNodeSecret() {
+	if "" == settings.NodeSettings.Secret {
+		logger.Info("Secret is empty, generating...")
+		uuidStr := uuid.New().String()
+		settings.NodeSettings.Secret = uuidStr
+
+		err := settings.Save()
+		if err != nil {
+			logger.Error("Error save settings", err)
+		}
+		logger.Info("Generated Secret: ", uuidStr)
+	}
+}
+
+func InitCryptoSecret() {
+	if "" == settings.CryptoSettings.Secret {
+		logger.Info("Secret is empty, generating...")
+
+		key := make([]byte, 32)
+		if _, err := rand.Read(key); err != nil {
+			logger.Error("Generate Secret failed: ", err)
+			return
+		}
+
+		settings.CryptoSettings.Secret = hex.EncodeToString(key)
+
+		err := settings.Save()
+		if err != nil {
+			logger.Error("Error save settings", err)
+		}
+		logger.Info("Secret Generated")
+	}
+}
+
+func InitJsExtensionType() {
+	// 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")
+}

internal/kernel/register_acme_user.go

@@ -0,0 +1,21 @@
+package kernel
+
+import (
+	"github.com/0xJacky/Nginx-UI/query"
+	"github.com/uozi-tech/cosy/logger"
+)
+
+func RegisterAcmeUser() {
+	a := query.AcmeUser
+	users, _ := a.Where(a.RegisterOnStartup.Is(true)).Find()
+	for _, user := range users {
+		err := user.Register()
+		if err != nil {
+			logger.Error(err)
+		}
+		_, err = a.Where(a.ID.Eq(user.ID)).Updates(user)
+		if err != nil {
+			logger.Error(err)
+		}
+	}
+}

internal/kernel/skip_install.go

@@ -0,0 +1,75 @@
+package kernel
+
+import (
+	"github.com/0xJacky/Nginx-UI/model"
+	"github.com/0xJacky/Nginx-UI/query"
+	"github.com/0xJacky/Nginx-UI/settings"
+	"github.com/caarlos0/env/v11"
+	"github.com/google/uuid"
+	"github.com/pkg/errors"
+	"github.com/uozi-tech/cosy/logger"
+	cSettings "github.com/uozi-tech/cosy/settings"
+	"golang.org/x/crypto/bcrypt"
+	"gorm.io/gorm"
+)
+
+type predefinedUser struct {
+	Name     string `json:"name"`
+	Password string `json:"password"`
+}
+
+func skipInstall() {
+	logger.Info("Skip installation mode enabled")
+
+	if cSettings.AppSettings.JwtSecret == "" {
+		cSettings.AppSettings.JwtSecret = uuid.New().String()
+	}
+
+	if settings.NodeSettings.Secret == "" {
+		settings.NodeSettings.Secret = uuid.New().String()
+		logger.Infof("Secret: %s", settings.NodeSettings.Secret)
+	}
+
+	err := settings.Save()
+	if err != nil {
+		logger.Fatal(err)
+	}
+}
+
+func registerPredefinedUser() {
+	// when skip installation mode is enabled, the predefined user will be created
+	if !settings.NodeSettings.SkipInstallation {
+		return
+	}
+	pUser := &predefinedUser{}
+
+	err := env.ParseWithOptions(pUser, env.Options{
+		Prefix:                "NGINX_UI_PREDEFINED_USER_",
+		UseFieldNameByDefault: true,
+	})
+
+	if err != nil {
+		logger.Fatal(err)
+	}
+
+	u := query.User
+
+	_, err = u.First()
+
+	// Only effect when there is no user in the database
+	if !errors.Is(err, gorm.ErrRecordNotFound) || pUser.Name == "" || pUser.Password == "" {
+		return
+	}
+
+	// Create a new user with the predefined name and password
+	pwd, _ := bcrypt.GenerateFromPassword([]byte(pUser.Password), bcrypt.DefaultCost)
+
+	err = u.Create(&model.User{
+		Name:     pUser.Name,
+		Password: string(pwd),
+	})
+
+	if err != nil {
+		logger.Error(err)
+	}
+}