nginx-ui/server/api/upgrade.go

package api

import (
	"github.com/0xJacky/Nginx-UI/server/service"
	"github.com/gin-gonic/gin"
	"github.com/gorilla/websocket"
	"log"
	"net/http"
	"os"
)

func GetRelease(c *gin.Context) {
	data, err := service.GetRelease()
	if err != nil {
		ErrHandler(c, err)
		return
	}

	c.JSON(http.StatusOK, data)
}

func GetCurrentVersion(c *gin.Context) {
	curVer, err := service.GetCurrentVersion()
	if err != nil {
		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 {
		log.Println("[Error] PerformCoreUpgrade Upgrade", err)
		return
	}
	defer ws.Close()

	_ = ws.WriteJSON(gin.H{
		"status":  "info",
		"message": "Initialing core upgrader",
	})

	u, err := service.NewUpgrader()

	if err != nil {
		_ = ws.WriteJSON(gin.H{
			"status":  "error",
			"message": "Initial core upgrader error",
		})
		_ = ws.WriteJSON(gin.H{
			"status":  "error",
			"message": err.Error(),
		})
		log.Println("[Error] PerformCoreUpgrade service.NewUpgrader()", 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(),
		})
		log.Println("[Error] PerformCoreUpgrade DownloadLatestRelease", err)
		return
	}
	_ = ws.WriteJSON(gin.H{
		"status":  "info",
		"message": "Performing core upgrade",
	})
	_ = os.Remove(u.Release.ExPath)
	// bye, overseer will restart nginx-ui
	err = u.PerformCoreUpgrade(u.Release.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(),
		})
		log.Println("[Error] PerformCoreUpgrade", err)
		return
	}
}