mirror of
https://github.com/0xJacky/nginx-ui.git
synced 2025-05-11 02:15:48 +02:00
103 lines
2.3 KiB
Go
103 lines
2.3 KiB
Go
package cluster
|
|
|
|
import (
|
|
"github.com/0xJacky/Nginx-UI/api"
|
|
"github.com/0xJacky/Nginx-UI/internal/analytic"
|
|
"github.com/0xJacky/Nginx-UI/internal/cluster"
|
|
"github.com/0xJacky/Nginx-UI/internal/cosy"
|
|
"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/spf13/cast"
|
|
"gorm.io/gorm"
|
|
"net/http"
|
|
)
|
|
|
|
func GetEnvironment(c *gin.Context) {
|
|
id := cast.ToInt(c.Param("id"))
|
|
|
|
envQuery := query.Environment
|
|
|
|
env, err := envQuery.FirstByID(id)
|
|
if err != nil {
|
|
api.ErrHandler(c, err)
|
|
return
|
|
}
|
|
|
|
c.JSON(http.StatusOK, analytic.GetNode(env))
|
|
}
|
|
|
|
func GetEnvironmentList(c *gin.Context) {
|
|
core := cosy.Core[model.Environment](c).
|
|
SetFussy("name")
|
|
|
|
// fix for sqlite
|
|
if c.Query("enabled") != "" {
|
|
core.GormScope(func(tx *gorm.DB) *gorm.DB {
|
|
return tx.Where("enabled = ?", cast.ToInt(cast.ToBool(c.Query("enabled"))))
|
|
})
|
|
}
|
|
|
|
core.SetTransformer(func(m *model.Environment) any {
|
|
return analytic.GetNode(m)
|
|
}).PagingList()
|
|
}
|
|
|
|
func AddEnvironment(c *gin.Context) {
|
|
cosy.Core[model.Environment](c).SetValidRules(gin.H{
|
|
"name": "required",
|
|
"url": "required,url",
|
|
"token": "required",
|
|
"enabled": "omitempty,boolean",
|
|
}).ExecutedHook(func(c *cosy.Ctx[model.Environment]) {
|
|
go analytic.RestartRetrieveNodesStatus()
|
|
}).Create()
|
|
}
|
|
|
|
func EditEnvironment(c *gin.Context) {
|
|
cosy.Core[model.Environment](c).SetValidRules(gin.H{
|
|
"name": "required",
|
|
"url": "required,url",
|
|
"token": "required",
|
|
"enabled": "omitempty,boolean",
|
|
}).ExecutedHook(func(c *cosy.Ctx[model.Environment]) {
|
|
go analytic.RestartRetrieveNodesStatus()
|
|
}).Modify()
|
|
}
|
|
|
|
func DeleteEnvironment(c *gin.Context) {
|
|
id := cast.ToInt(c.Param("id"))
|
|
envQuery := query.Environment
|
|
|
|
env, err := envQuery.FirstByID(id)
|
|
if err != nil {
|
|
api.ErrHandler(c, err)
|
|
return
|
|
}
|
|
err = envQuery.DeleteByID(env.ID)
|
|
if err != nil {
|
|
api.ErrHandler(c, err)
|
|
return
|
|
}
|
|
|
|
go analytic.RestartRetrieveNodesStatus()
|
|
|
|
c.JSON(http.StatusNoContent, nil)
|
|
}
|
|
|
|
func LoadEnvironmentFromSettings(c *gin.Context) {
|
|
err := settings.ReloadCluster()
|
|
if err != nil {
|
|
api.ErrHandler(c, err)
|
|
return
|
|
}
|
|
|
|
cluster.RegisterPredefinedNodes()
|
|
|
|
go analytic.RestartRetrieveNodesStatus()
|
|
|
|
c.JSON(http.StatusOK, gin.H{
|
|
"message": "ok",
|
|
})
|
|
}
|