mirror of
https://github.com/gravitl/netmaker.git
synced 2025-10-06 11:56:39 +08:00
128 lines
3.8 KiB
Go
128 lines
3.8 KiB
Go
package controller
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"net/http"
|
|
|
|
"github.com/gorilla/mux"
|
|
"github.com/gravitl/netmaker/logger"
|
|
"github.com/gravitl/netmaker/logic"
|
|
"github.com/gravitl/netmaker/models"
|
|
"github.com/gravitl/netmaker/mq"
|
|
"github.com/gravitl/netmaker/nm-proxy/manager"
|
|
"golang.zx2c4.com/wireguard/wgctrl/wgtypes"
|
|
)
|
|
|
|
// swagger:route POST /api/nodes/{network}/{nodeid}/createrelay nodes createRelay
|
|
//
|
|
// Create a relay.
|
|
//
|
|
// Schemes: https
|
|
//
|
|
// Security:
|
|
// oauth
|
|
//
|
|
// Responses:
|
|
// 200: nodeResponse
|
|
func createRelay(w http.ResponseWriter, r *http.Request) {
|
|
var relay models.RelayRequest
|
|
var params = mux.Vars(r)
|
|
w.Header().Set("Content-Type", "application/json")
|
|
err := json.NewDecoder(r.Body).Decode(&relay)
|
|
if err != nil {
|
|
logger.Log(0, r.Header.Get("user"), "error decoding request body: ", err.Error())
|
|
logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
|
|
return
|
|
}
|
|
relay.NetID = params["network"]
|
|
relay.NodeID = params["nodeid"]
|
|
updatenodes, node, err := logic.CreateRelay(relay)
|
|
if err != nil {
|
|
logger.Log(0, r.Header.Get("user"),
|
|
fmt.Sprintf("failed to create relay on node [%s] on network [%s]: %v", relay.NodeID, relay.NetID, err))
|
|
logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
|
|
return
|
|
}
|
|
|
|
relayPeersMap := make(map[string][]wgtypes.PeerConfig)
|
|
logger.Log(1, r.Header.Get("user"), "created relay on node", relay.NodeID, "on network", relay.NetID)
|
|
for _, relayedNode := range updatenodes {
|
|
peers, err := logic.GetPeersForProxy(&relayedNode)
|
|
if err == nil {
|
|
relayPeersMap[relayedNode.PublicKey] = peers
|
|
}
|
|
|
|
// relayEndpoint, err := net.ResolveUDPAddr("udp", fmt.Sprintf("%s:%d", node.Endpoint, node.LocalListenPort))
|
|
// if err != nil {
|
|
// logger.Log(1, "failed to resolve relay node endpoint: ", err.Error())
|
|
// }
|
|
|
|
// err = mq.ProxyUpdate(&manager.ManagerAction{
|
|
// Action: manager.AddInterface,
|
|
// Payload: manager.ManagerPayload{
|
|
// InterfaceName: relayedNode.Interface,
|
|
// IsRelayed: true,
|
|
// Peers: peers,
|
|
// RelayedTo: relayEndpoint,
|
|
// },
|
|
// }, &node)
|
|
// if err != nil {
|
|
// logger.Log(1, "failed to send proxy update for relayed node: ", err.Error())
|
|
// }
|
|
|
|
err = mq.NodeUpdate(&relayedNode)
|
|
if err != nil {
|
|
logger.Log(1, "error sending update to relayed node ", relayedNode.Name, "on network", relay.NetID, ": ", err.Error())
|
|
}
|
|
}
|
|
// send proxy update for node that is relaying traffic
|
|
logger.Log(0, "--------> sending relay update to proxy")
|
|
err = mq.ProxyUpdate(&manager.ManagerAction{
|
|
Action: manager.RelayPeers,
|
|
Payload: manager.ManagerPayload{
|
|
IsRelay: true,
|
|
RelayedPeers: relayPeersMap,
|
|
},
|
|
}, &node)
|
|
if err != nil {
|
|
logger.Log(1, "failed to send proxy update: ", err.Error())
|
|
}
|
|
w.WriteHeader(http.StatusOK)
|
|
json.NewEncoder(w).Encode(node)
|
|
runUpdates(&node, true)
|
|
}
|
|
|
|
// swagger:route DELETE /api/nodes/{network}/{nodeid}/deleterelay nodes deleteRelay
|
|
//
|
|
// Remove a relay.
|
|
//
|
|
// Schemes: https
|
|
//
|
|
// Security:
|
|
// oauth
|
|
//
|
|
// Responses:
|
|
// 200: nodeResponse
|
|
func deleteRelay(w http.ResponseWriter, r *http.Request) {
|
|
w.Header().Set("Content-Type", "application/json")
|
|
var params = mux.Vars(r)
|
|
nodeid := params["nodeid"]
|
|
netid := params["network"]
|
|
updatenodes, node, err := logic.DeleteRelay(netid, nodeid)
|
|
if err != nil {
|
|
logger.Log(0, r.Header.Get("user"), "error decoding request body: ", err.Error())
|
|
logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
|
|
return
|
|
}
|
|
logger.Log(1, r.Header.Get("user"), "deleted relay server", nodeid, "on network", netid)
|
|
for _, relayedNode := range updatenodes {
|
|
err = mq.NodeUpdate(&relayedNode)
|
|
if err != nil {
|
|
logger.Log(1, "error sending update to relayed node ", relayedNode.Name, "on network", netid, ": ", err.Error())
|
|
}
|
|
}
|
|
w.WriteHeader(http.StatusOK)
|
|
json.NewEncoder(w).Encode(node)
|
|
runUpdates(&node, true)
|
|
}
|