mirror of
https://github.com/Rain-kl/OpenFlare.git
synced 2026-10-03 15:06:36 +08:00
ryan
46 lines
1.1 KiB
Go
46 lines
1.1 KiB
Go
// Copyright 2026 Arctel.net
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
package relay
|
|
|
|
import (
|
|
"github.com/Rain-kl/Wavelet/internal/apps/openflare/compat"
|
|
ofws "github.com/Rain-kl/Wavelet/internal/apps/openflare/websocket"
|
|
"github.com/Rain-kl/Wavelet/internal/model"
|
|
"github.com/gin-gonic/gin"
|
|
)
|
|
|
|
// PostHeartbeat handles POST /relay/heartbeat.
|
|
func PostHeartbeat(c *gin.Context) {
|
|
var payload HeartbeatPayload
|
|
if !compat.BindJSON(c, &payload) {
|
|
return
|
|
}
|
|
payload.IP = resolveReportedNodeIP(payload.IP, c.Request.RemoteAddr)
|
|
|
|
authNode, ok := c.Get(ctxRelayNodeKey)
|
|
if !ok {
|
|
compat.Unauthorized(c, errAgentTokenInvalid)
|
|
return
|
|
}
|
|
node := authNode.(*model.OpenFlareNode)
|
|
|
|
result, err := Heartbeat(c.Request.Context(), node, payload)
|
|
if err != nil {
|
|
compat.Fail(c, err.Error())
|
|
return
|
|
}
|
|
compat.OK(c, result)
|
|
}
|
|
|
|
// GetWebSocket handles GET /relay/ws.
|
|
func GetWebSocket(c *gin.Context) {
|
|
authNode, ok := c.Get(ctxRelayNodeKey)
|
|
if !ok {
|
|
compat.Unauthorized(c, errAgentTokenInvalid)
|
|
return
|
|
}
|
|
node := authNode.(*model.OpenFlareNode)
|
|
ofws.ServeRelay(c, node.NodeID)
|
|
}
|