netmaker/pro/controllers/relay.go

package controllers

import (
	"encoding/json"
	"fmt"
	"net/http"

	"github.com/google/uuid"
	proLogic "github.com/gravitl/netmaker/pro/logic"

	"github.com/gorilla/mux"
	controller "github.com/gravitl/netmaker/controllers"
	"github.com/gravitl/netmaker/logger"
	"github.com/gravitl/netmaker/logic"
	"github.com/gravitl/netmaker/models"
	"github.com/gravitl/netmaker/mq"
)

// RelayHandlers - handle Pro Relays
func RelayHandlers(r *mux.Router) {

	r.HandleFunc("/api/nodes/{network}/{nodeid}/createrelay", logic.SecurityCheck(true, http.HandlerFunc(createRelay))).Methods(http.MethodPost)
	r.HandleFunc("/api/nodes/{network}/{nodeid}/deleterelay", logic.SecurityCheck(true, http.HandlerFunc(deleteRelay))).Methods(http.MethodDelete)
	r.HandleFunc("/api/v1/host/{hostid}/failoverme", controller.Authorize(true, false, "host", http.HandlerFunc(failOverME))).Methods(http.MethodPost)
}

// @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 := proLogic.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 := proLogic.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); 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)
}