|
@@ -1,147 +0,0 @@
|
|
|
-package controller
|
|
|
-
|
|
|
-import (
|
|
|
- "encoding/json"
|
|
|
- "fmt"
|
|
|
- "net/http"
|
|
|
-
|
|
|
- "github.com/google/uuid"
|
|
|
-
|
|
|
- "github.com/gorilla/mux"
|
|
|
- "github.com/gravitl/netmaker/logger"
|
|
|
- "github.com/gravitl/netmaker/logic"
|
|
|
- "github.com/gravitl/netmaker/models"
|
|
|
- "github.com/gravitl/netmaker/mq"
|
|
|
-)
|
|
|
-
|
|
|
-func relayHandlers(r *mux.Router) {
|
|
|
- r.HandleFunc("/api/nodes/{network}/{nodeid}/createrelay", logic.SecurityCheck(true, http.HandlerFunc(createGateway))).Methods(http.MethodPost)
|
|
|
- r.HandleFunc("/api/nodes/{network}/{nodeid}/deleterelay", logic.SecurityCheck(true, http.HandlerFunc(deleteGateway))).Methods(http.MethodDelete)
|
|
|
-}
|
|
|
-
|
|
|
-// @Summary Create a relay
|
|
|
-// @Router /api/nodes/{network}/{nodeid}/createrelay [post]
|
|
|
-// @Tags PRO
|
|
|
-// @Accept json
|
|
|
-// @Produce json
|
|
|
-// @Param network path string true "Network ID"
|
|
|
-// @Param nodeid path string true "Node ID"
|
|
|
-// @Param body body models.RelayRequest true "Relay request parameters"
|
|
|
-// @Success 200 {object} models.ApiNode
|
|
|
-// @Failure 400 {object} models.ErrorResponse
|
|
|
-// @Failure 500 {object} models.ErrorResponse
|
|
|
-func createRelay(w http.ResponseWriter, r *http.Request) {
|
|
|
- var relayRequest models.RelayRequest
|
|
|
- var params = mux.Vars(r)
|
|
|
- w.Header().Set("Content-Type", "application/json")
|
|
|
- err := json.NewDecoder(r.Body).Decode(&relayRequest)
|
|
|
- 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
|
|
|
- }
|
|
|
- relayRequest.NetID = params["network"]
|
|
|
- relayRequest.NodeID = params["nodeid"]
|
|
|
- _, relayNode, err := logic.CreateRelay(relayRequest)
|
|
|
- if err != nil {
|
|
|
- logger.Log(
|
|
|
- 0,
|
|
|
- r.Header.Get("user"),
|
|
|
- fmt.Sprintf(
|
|
|
- "failed to create relay on node [%s] on network [%s]: %v",
|
|
|
- relayRequest.NodeID,
|
|
|
- relayRequest.NetID,
|
|
|
- err,
|
|
|
- ),
|
|
|
- )
|
|
|
- logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
|
|
|
- return
|
|
|
- }
|
|
|
- for _, relayedNodeID := range relayNode.RelayedNodes {
|
|
|
- relayedNode, err := logic.GetNodeByID(relayedNodeID)
|
|
|
- if err == nil {
|
|
|
- if relayedNode.FailedOverBy != uuid.Nil {
|
|
|
- go logic.ResetFailedOverPeer(&relayedNode)
|
|
|
- }
|
|
|
-
|
|
|
- }
|
|
|
- }
|
|
|
- go mq.PublishPeerUpdate(false)
|
|
|
- logger.Log(
|
|
|
- 1,
|
|
|
- r.Header.Get("user"),
|
|
|
- "created relay on node",
|
|
|
- relayRequest.NodeID,
|
|
|
- "on network",
|
|
|
- relayRequest.NetID,
|
|
|
- )
|
|
|
- apiNode := relayNode.ConvertToAPINode()
|
|
|
- w.WriteHeader(http.StatusOK)
|
|
|
- json.NewEncoder(w).Encode(apiNode)
|
|
|
-}
|
|
|
-
|
|
|
-// @Summary Remove a relay
|
|
|
-// @Router /api/nodes/{network}/{nodeid}/deleterelay [delete]
|
|
|
-// @Tags PRO
|
|
|
-// @Accept json
|
|
|
-// @Produce json
|
|
|
-// @Param network path string true "Network ID"
|
|
|
-// @Param nodeid path string true "Node ID"
|
|
|
-// @Success 200 {object} models.ApiNode
|
|
|
-// @Failure 400 {object} models.ErrorResponse
|
|
|
-// @Failure 500 {object} models.ErrorResponse
|
|
|
-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)
|
|
|
- go func() {
|
|
|
- for _, relayedNode := range updateNodes {
|
|
|
- err = mq.NodeUpdate(&relayedNode)
|
|
|
- if err != nil {
|
|
|
- logger.Log(
|
|
|
- 1,
|
|
|
- "relayed node update ",
|
|
|
- relayedNode.ID.String(),
|
|
|
- "on network",
|
|
|
- relayedNode.Network,
|
|
|
- ": ",
|
|
|
- err.Error(),
|
|
|
- )
|
|
|
-
|
|
|
- }
|
|
|
- h, err := logic.GetHost(relayedNode.HostID.String())
|
|
|
- if err == nil {
|
|
|
- if h.OS == models.OS_Types.IoT {
|
|
|
- nodes, err := logic.GetAllNodes()
|
|
|
- if err != nil {
|
|
|
- return
|
|
|
- }
|
|
|
- node.IsRelay = true // for iot update to recognise that it has to delete relay peer
|
|
|
- if err = mq.PublishSingleHostPeerUpdate(h, nodes, &node, nil, false, nil); err != nil {
|
|
|
- logger.Log(1, "failed to publish peer update to host", h.ID.String(), ": ", err.Error())
|
|
|
- }
|
|
|
- }
|
|
|
- }
|
|
|
- }
|
|
|
- mq.PublishPeerUpdate(false)
|
|
|
- }()
|
|
|
- logger.Log(
|
|
|
- 1,
|
|
|
- r.Header.Get("user"),
|
|
|
- "deleted relay on node",
|
|
|
- node.ID.String(),
|
|
|
- "on network",
|
|
|
- node.Network,
|
|
|
- )
|
|
|
- apiNode := node.ConvertToAPINode()
|
|
|
- w.WriteHeader(http.StatusOK)
|
|
|
- json.NewEncoder(w).Encode(apiNode)
|
|
|
-}
|