migrate

Wavelet/internal/apps/openflare/node/routers.go

@@ -0,0 +1,192 @@
+// Copyright 2026 Arctel.net
+// SPDX-License-Identifier: Apache-2.0
+
+package node
+
+import (
+	"encoding/json"
+	"errors"
+	"io"
+
+	"github.com/Rain-kl/Wavelet/internal/apps/openflare/compat"
+	"github.com/gin-gonic/gin"
+	"gorm.io/gorm"
+)
+
+func handleLogicError(c *gin.Context, err error) bool {
+	if err == nil {
+		return false
+	}
+	if errors.Is(err, gorm.ErrRecordNotFound) {
+		compat.Fail(c, errNodeNotFound)
+		return true
+	}
+	compat.Fail(c, err.Error())
+	return true
+}
+
+// ListNodesHandler lists all nodes.
+func ListNodesHandler(c *gin.Context) {
+	nodes, err := ListNodes(c.Request.Context())
+	if handleLogicError(c, err) {
+		return
+	}
+	compat.OK(c, nodes)
+}
+
+// CreateNodeHandler creates a node.
+func CreateNodeHandler(c *gin.Context) {
+	var input Input
+	if !compat.BindJSON(c, &input) {
+		return
+	}
+	view, err := CreateNode(c.Request.Context(), input)
+	if handleLogicError(c, err) {
+		return
+	}
+	compat.OK(c, view)
+}
+
+// UpdateNodeHandler updates a node.
+func UpdateNodeHandler(c *gin.Context) {
+	id, ok := compat.IDParam(c)
+	if !ok {
+		return
+	}
+	var input Input
+	if !compat.BindJSON(c, &input) {
+		return
+	}
+	view, err := UpdateNode(c.Request.Context(), id, input)
+	if handleLogicError(c, err) {
+		return
+	}
+	compat.OK(c, view)
+}
+
+// DeleteNodeHandler deletes a node.
+func DeleteNodeHandler(c *gin.Context) {
+	id, ok := compat.IDParam(c)
+	if !ok {
+		return
+	}
+	if err := DeleteNode(c.Request.Context(), id); handleLogicError(c, err) {
+		return
+	}
+	compat.OKMessage(c, "")
+}
+
+// GetBootstrapTokenHandler returns the global discovery token.
+func GetBootstrapTokenHandler(c *gin.Context) {
+	view, err := GetBootstrapToken(c.Request.Context())
+	if handleLogicError(c, err) {
+		return
+	}
+	compat.OK(c, view)
+}
+
+// RotateBootstrapTokenHandler rotates the global discovery token.
+func RotateBootstrapTokenHandler(c *gin.Context) {
+	view, err := RotateBootstrapToken(c.Request.Context())
+	if handleLogicError(c, err) {
+		return
+	}
+	compat.OK(c, view)
+}
+
+// GetAgentReleaseHandler returns the latest agent release for a node.
+func GetAgentReleaseHandler(c *gin.Context) {
+	id, ok := compat.IDParam(c)
+	if !ok {
+		return
+	}
+	release, err := GetAgentRelease(c.Request.Context(), id, c.Query("channel"))
+	if handleLogicError(c, err) {
+		return
+	}
+	compat.OK(c, release)
+}
+
+// RequestAgentUpdateHandler requests agent self-update on a node.
+func RequestAgentUpdateHandler(c *gin.Context) {
+	id, ok := compat.IDParam(c)
+	if !ok {
+		return
+	}
+	var request AgentUpdateInput
+	if c.Request.ContentLength > 0 {
+		if err := bindOptionalJSON(c.Request.Body, &request); err != nil {
+			compat.Fail(c, "参数错误")
+			return
+		}
+	}
+	view, err := RequestAgentUpdate(c.Request.Context(), id, request)
+	if handleLogicError(c, err) {
+		return
+	}
+	compat.OK(c, view)
+}
+
+// RequestOpenrestyRestartHandler requests openresty restart on a node.
+func RequestOpenrestyRestartHandler(c *gin.Context) {
+	id, ok := compat.IDParam(c)
+	if !ok {
+		return
+	}
+	view, err := RequestOpenrestyRestart(c.Request.Context(), id)
+	if handleLogicError(c, err) {
+		return
+	}
+	compat.OK(c, view)
+}
+
+// RequestForceSyncHandler requests force sync on a node.
+func RequestForceSyncHandler(c *gin.Context) {
+	id, ok := compat.IDParam(c)
+	if !ok {
+		return
+	}
+	view, err := RequestForceSync(c.Request.Context(), id)
+	if handleLogicError(c, err) {
+		return
+	}
+	compat.OK(c, view)
+}
+
+// GetObservabilityHandler returns node observability details.
+func GetObservabilityHandler(c *gin.Context) {
+	id, ok := compat.IDParam(c)
+	if !ok {
+		return
+	}
+	var query ObservabilityQuery
+	if err := c.ShouldBindQuery(&query); err != nil {
+		compat.Fail(c, "参数错误")
+		return
+	}
+	view, err := GetObservability(c.Request.Context(), id, query)
+	if handleLogicError(c, err) {
+		return
+	}
+	compat.OK(c, view)
+}
+
+// CleanupHealthEventsHandler cleans up node health events.
+func CleanupHealthEventsHandler(c *gin.Context) {
+	id, ok := compat.IDParam(c)
+	if !ok {
+		return
+	}
+	result, err := CleanupHealthEvents(c.Request.Context(), id)
+	if handleLogicError(c, err) {
+		return
+	}
+	compat.OK(c, result)
+}
+
+func bindOptionalJSON(body io.Reader, target any) error {
+	if err := json.NewDecoder(body).Decode(target); err != nil && !errors.Is(err, io.EOF) {
+		return err
+	}
+	return nil
+}