mirrors/nginx-ui
1
0
Fork 0
mirror of https://github.com/0xJacky/nginx-ui.git synced 2025-05-11 10:25:52 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 4

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
StartCmd = bash
NodeSecret = fdc7764f-92d2-454c-9640-6a09be121139
Demo = true
[nginx_log]
AccessLogPath = /var/log/nginx/access.local.log

241
server/api/upgrade.go
View file

@ -1,144 +1,145 @@
package api
import (
"github.com/0xJacky/Nginx-UI/server/internal/logger"
"github.com/0xJacky/Nginx-UI/server/internal/upgrader"
"github.com/gin-gonic/gin"
"github.com/gorilla/websocket"
"net/http"
"os"
"github.com/0xJacky/Nginx-UI/server/internal/logger"
"github.com/0xJacky/Nginx-UI/server/internal/upgrader"
"github.com/0xJacky/Nginx-UI/server/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 {
ErrHandler(c, err)
return
}
runtimeInfo, err := upgrader.GetRuntimeInfo()
if err != nil {
ErrHandler(c, err)
return
}
type resp struct {
upgrader.TRelease
upgrader.RuntimeInfo
}
c.JSON(http.StatusOK, resp{
data, runtimeInfo,
})
data, err := upgrader.GetRelease(c.Query("channel"))
if err != nil {
ErrHandler(c, err)
return
}
runtimeInfo, err := upgrader.GetRuntimeInfo()
if err != nil {
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 {
ErrHandler(c, err)
return
}
curVer, err := upgrader.GetCurrentVersion()
if err != nil {
ErrHandler(c, err)
return
}
c.JSON(http.StatusOK, curVer)
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 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"`
}
var control struct {
DryRun bool `json:"dry_run"`
Channel string `json:"channel"`
}
err = ws.ReadJSON(&control)
err = ws.ReadJSON(&control)
if err != nil {
logger.Error(err)
return
}
if err != nil {
logger.Error(err)
return
}
_ = ws.WriteJSON(gin.H{
"status": "info",
"message": "Initialing core upgrader",
})
_ = ws.WriteJSON(gin.H{
"status": "info",
"message": "Initialing core upgrader",
})
u, err := upgrader.NewUpgrader(control.Channel)
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,
})
}
}()
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)
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
}
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 {
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
}
_ = 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
}
}