refactor: structure of api-router directory

api/system/install.go

@@ -0,0 +1,78 @@
+package system
+
+import (
+	"github.com/0xJacky/Nginx-UI/api"
+	"github.com/0xJacky/Nginx-UI/internal/boot"
+	"github.com/0xJacky/Nginx-UI/model"
+	"github.com/0xJacky/Nginx-UI/query"
+	"github.com/0xJacky/Nginx-UI/settings"
+	"github.com/gin-gonic/gin"
+	"github.com/google/uuid"
+	"golang.org/x/crypto/bcrypt"
+	"net/http"
+)
+
+func installLockStatus() bool {
+	return "" != settings.ServerSettings.JwtSecret
+}
+
+func InstallLockCheck(c *gin.Context) {
+	c.JSON(http.StatusOK, gin.H{
+		"lock": installLockStatus(),
+	})
+}
+
+type InstallJson struct {
+	Email    string `json:"email" binding:"required,email"`
+	Username string `json:"username" binding:"required,max=255"`
+	Password string `json:"password" binding:"required,max=255"`
+	Database string `json:"database"`
+}
+
+func InstallNginxUI(c *gin.Context) {
+	// Visit this api after installed is forbidden
+	if installLockStatus() {
+		c.JSON(http.StatusForbidden, gin.H{
+			"error": "installed",
+		})
+		return
+	}
+	var json InstallJson
+	ok := api.BindAndValid(c, &json)
+	if !ok {
+		return
+	}
+
+	settings.ServerSettings.JwtSecret = uuid.New().String()
+	settings.ServerSettings.NodeSecret = uuid.New().String()
+	settings.ServerSettings.Email = json.Email
+	if "" != json.Database {
+		settings.ServerSettings.Database = json.Database
+	}
+	settings.ReflectFrom()
+
+	err := settings.Save()
+	if err != nil {
+		api.ErrHandler(c, err)
+		return
+	}
+
+	// Init model
+	boot.InitDatabase()
+
+	pwd, _ := bcrypt.GenerateFromPassword([]byte(json.Password), bcrypt.DefaultCost)
+
+	u := query.Auth
+	err = u.Create(&model.Auth{
+		Name:     json.Username,
+		Password: string(pwd),
+	})
+
+	if err != nil {
+		api.ErrHandler(c, err)
+		return
+	}
+	c.JSON(http.StatusOK, gin.H{
+		"message": "ok",
+	})
+}

api/system/router.go

@@ -0,0 +1,20 @@
+package system
+
+import (
+    "github.com/gin-gonic/gin"
+)
+
+func InitPublicRouter(r *gin.RouterGroup) {
+    r.GET("install", InstallLockCheck)
+    r.POST("install", InstallNginxUI)
+    r.GET("translation/:code", GetTranslation)
+}
+
+func InitPrivateRouter(r *gin.RouterGroup) {
+    r.GET("settings", GetSettings)
+    r.POST("settings", SaveSettings)
+
+    r.GET("upgrade/release", GetRelease)
+    r.GET("upgrade/current", GetCurrentVersion)
+    r.GET("upgrade/perform", PerformCoreUpgrade)
+}

api/system/settings.go

@@ -0,0 +1,42 @@
+package system
+
+import (
+    "github.com/0xJacky/Nginx-UI/api"
+    "github.com/0xJacky/Nginx-UI/settings"
+    "github.com/gin-gonic/gin"
+    "net/http"
+)
+
+func GetSettings(c *gin.Context) {
+    c.JSON(http.StatusOK, gin.H{
+        "server": settings.ServerSettings,
+        "nginx":  settings.NginxSettings,
+        "openai": settings.OpenAISettings,
+    })
+}
+
+func SaveSettings(c *gin.Context) {
+    var json struct {
+        Server settings.Server `json:"server"`
+        Nginx  settings.Nginx  `json:"nginx"`
+        Openai settings.OpenAI `json:"openai"`
+    }
+
+    if !api.BindAndValid(c, &json) {
+        return
+    }
+
+    settings.ServerSettings = json.Server
+    settings.NginxSettings = json.Nginx
+    settings.OpenAISettings = json.Openai
+
+    settings.ReflectFrom()
+
+    err := settings.Save()
+    if err != nil {
+        api.ErrHandler(c, err)
+        return
+    }
+
+    GetSettings(c)
+}

api/system/translation.go

@@ -0,0 +1,13 @@
+package system
+
+import (
+	"github.com/0xJacky/Nginx-UI/internal/translation"
+	"github.com/gin-gonic/gin"
+	"net/http"
+)
+
+func GetTranslation(c *gin.Context) {
+	code := c.Param("code")
+
+	c.JSON(http.StatusOK, translation.GetTranslation(code))
+}

api/system/upgrade.go

@@ -0,0 +1,146 @@
+package system
+
+import (
+	"github.com/0xJacky/Nginx-UI/api"
+	"github.com/0xJacky/Nginx-UI/internal/logger"
+	"github.com/0xJacky/Nginx-UI/internal/upgrader"
+	"github.com/0xJacky/Nginx-UI/settings"
+	"github.com/gin-gonic/gin"
+	"github.com/gorilla/websocket"
+	"net/http"
+	"os"
+)
+
+func GetRelease(c *gin.Context) {
+	data, err := upgrader.GetRelease(c.Query("channel"))
+	if err != nil {
+		api.ErrHandler(c, err)
+		return
+	}
+	runtimeInfo, err := upgrader.GetRuntimeInfo()
+	if err != nil {
+		api.ErrHandler(c, err)
+		return
+	}
+	type resp struct {
+		upgrader.TRelease
+		upgrader.RuntimeInfo
+	}
+	c.JSON(http.StatusOK, resp{
+		data, runtimeInfo,
+	})
+}
+
+func GetCurrentVersion(c *gin.Context) {
+	curVer, err := upgrader.GetCurrentVersion()
+	if err != nil {
+		api.ErrHandler(c, err)
+		return
+	}
+
+	c.JSON(http.StatusOK, curVer)
+}
+
+func PerformCoreUpgrade(c *gin.Context) {
+	var upGrader = websocket.Upgrader{
+		CheckOrigin: func(r *http.Request) bool {
+			return true
+		},
+	}
+	// upgrade http to websocket
+	ws, err := upGrader.Upgrade(c.Writer, c.Request, nil)
+	if err != nil {
+		logger.Error(err)
+		return
+	}
+	defer ws.Close()
+
+	var control struct {
+		DryRun  bool   `json:"dry_run"`
+		Channel string `json:"channel"`
+	}
+
+	err = ws.ReadJSON(&control)
+
+	if err != nil {
+		logger.Error(err)
+		return
+	}
+
+	_ = ws.WriteJSON(gin.H{
+		"status":  "info",
+		"message": "Initialing core upgrader",
+	})
+
+	u, err := upgrader.NewUpgrader(control.Channel)
+
+	if err != nil {
+		_ = ws.WriteJSON(gin.H{
+			"status":  "error",
+			"message": "Initial core upgrader error",
+		})
+		_ = ws.WriteJSON(gin.H{
+			"status":  "error",
+			"message": err.Error(),
+		})
+		logger.Error(err)
+		return
+	}
+	_ = ws.WriteJSON(gin.H{
+		"status":  "info",
+		"message": "Downloading latest release",
+	})
+	progressChan := make(chan float64)
+	go func() {
+		for progress := range progressChan {
+			_ = ws.WriteJSON(gin.H{
+				"status":   "progress",
+				"progress": progress,
+			})
+		}
+	}()
+
+	tarName, err := u.DownloadLatestRelease(progressChan)
+
+	if err != nil {
+		_ = ws.WriteJSON(gin.H{
+			"status":  "error",
+			"message": "Download latest release error",
+		})
+		_ = ws.WriteJSON(gin.H{
+			"status":  "error",
+			"message": err.Error(),
+		})
+		logger.Error(err)
+		return
+	}
+
+	defer func() {
+		_ = os.Remove(tarName)
+		_ = os.Remove(tarName + ".digest")
+	}()
+	_ = ws.WriteJSON(gin.H{
+		"status":  "info",
+		"message": "Performing core upgrade",
+	})
+	// dry run
+	if control.DryRun || settings.ServerSettings.Demo {
+		return
+	}
+
+	_ = os.Remove(u.ExPath)
+	// bye, overseer will restart nginx-ui
+	err = u.PerformCoreUpgrade(u.ExPath, tarName)
+	if err != nil {
+		_ = ws.WriteJSON(gin.H{
+			"status":  "error",
+			"message": "Perform core upgrade error",
+		})
+		_ = ws.WriteJSON(gin.H{
+			"status":  "error",
+			"message": err.Error(),
+		})
+		logger.Error(err)
+		return
+	}
+}