mirrors/nginx-ui
1
0
Fork 0
mirror of https://github.com/0xJacky/nginx-ui.git synced 2025-05-12 02:45:49 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 6

chore: dry run core upgrade in demo

Browse source
This commit is contained in:
0xJacky 2023-05-21 09:46:59 +08:00
parent 0265e44c17
commit b1b6e8fec8
No known key found for this signature in database
GPG key ID: B6E4A6E4A561BAF0
2 changed files with 122 additions and 120 deletions
Download patch file Download diff file Expand all files Collapse all files

1
resources/demo/app.ini
View file

@ -6,6 +6,7 @@ Email = test@jackyu.cn
HTTPChallengePort = 9180 HTTPChallengePort = 9180
StartCmd = bash StartCmd = bash
NodeSecret = fdc7764f-92d2-454c-9640-6a09be121139 NodeSecret = fdc7764f-92d2-454c-9640-6a09be121139
Demo = true
[nginx_log] [nginx_log]
AccessLogPath = /var/log/nginx/access.local.log AccessLogPath = /var/log/nginx/access.local.log

241
server/api/upgrade.go
View file

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