nginx-ui/api/system/self_check.go

package system

import (
	"io"
	"net/http"

	"github.com/gorilla/websocket"
	"github.com/uozi-tech/cosy/logger"

	"time"

	"github.com/0xJacky/Nginx-UI/internal/self_check"
	"github.com/gin-gonic/gin"
)

func SelfCheck(c *gin.Context) {
	report := self_check.Run()
	c.JSON(http.StatusOK, report)
}

func SelfCheckFix(c *gin.Context) {
	result := self_check.AttemptFix(c.Param("name"))
	c.JSON(http.StatusOK, result)
}

func CheckWebSocket(c *gin.Context) {
	var upgrader = websocket.Upgrader{
		CheckOrigin: func(r *http.Request) bool {
			return true
		},
	}
	ws, err := upgrader.Upgrade(c.Writer, c.Request, nil)
	if err != nil {
		logger.Error(err)
		return
	}
	defer ws.Close()
	err = ws.WriteJSON(gin.H{
		"message": "ok",
	})
	if err != nil {
		logger.Error(err)
		return
	}
}

func CheckSSE(c *gin.Context) {
	notify := c.Writer.CloseNotify()
	for i := 0; i < 10; i++ {
		select {
		case <-notify:
			return
		default:
			c.Stream(func(w io.Writer) bool {
				c.SSEvent("message", time.Now())
				return false
			})
			time.Sleep(time.Second * 2)
		}
	}
}