netmaker/controllers/hosts.go

package controller

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

	"github.com/google/uuid"
	"github.com/gorilla/mux"
	"github.com/gravitl/netmaker/database"
	"github.com/gravitl/netmaker/logger"
	"github.com/gravitl/netmaker/logic"
	"github.com/gravitl/netmaker/models"
	"github.com/gravitl/netmaker/mq"
	"github.com/gravitl/netmaker/servercfg"
	"golang.org/x/crypto/bcrypt"
	"golang.org/x/exp/slog"
)

func hostHandlers(r *mux.Router) {
	r.HandleFunc("/api/hosts", logic.SecurityCheck(true, http.HandlerFunc(getHosts))).
		Methods(http.MethodGet)
	r.HandleFunc("/api/hosts/keys", logic.SecurityCheck(true, http.HandlerFunc(updateAllKeys))).
		Methods(http.MethodPut)
	r.HandleFunc("/api/hosts/sync", logic.SecurityCheck(true, http.HandlerFunc(syncHosts))).
		Methods(http.MethodPost)
	r.HandleFunc("/api/hosts/upgrade", logic.SecurityCheck(true, http.HandlerFunc(upgradeHosts))).
		Methods(http.MethodPost)
	r.HandleFunc("/api/hosts/{hostid}/keys", logic.SecurityCheck(true, http.HandlerFunc(updateKeys))).
		Methods(http.MethodPut)
	r.HandleFunc("/api/hosts/{hostid}/sync", logic.SecurityCheck(true, http.HandlerFunc(syncHost))).
		Methods(http.MethodPost)
	r.HandleFunc("/api/hosts/{hostid}", logic.SecurityCheck(true, http.HandlerFunc(updateHost))).
		Methods(http.MethodPut)
	r.HandleFunc("/api/hosts/{hostid}", Authorize(true, false, "all", http.HandlerFunc(deleteHost))).
		Methods(http.MethodDelete)
	r.HandleFunc("/api/hosts/{hostid}/upgrade", logic.SecurityCheck(true, http.HandlerFunc(upgradeHost))).
		Methods(http.MethodPut)
	r.HandleFunc("/api/hosts/{hostid}/networks/{network}", logic.SecurityCheck(true, http.HandlerFunc(addHostToNetwork))).
		Methods(http.MethodPost)
	r.HandleFunc("/api/hosts/{hostid}/networks/{network}", logic.SecurityCheck(true, http.HandlerFunc(deleteHostFromNetwork))).
		Methods(http.MethodDelete)
	r.HandleFunc("/api/hosts/adm/authenticate", authenticateHost).Methods(http.MethodPost)
	r.HandleFunc("/api/v1/host", Authorize(true, false, "host", http.HandlerFunc(pull))).
		Methods(http.MethodGet)
	r.HandleFunc("/api/v1/host/{hostid}/signalpeer", Authorize(true, false, "host", http.HandlerFunc(signalPeer))).
		Methods(http.MethodPost)
	r.HandleFunc("/api/v1/fallback/host/{hostid}", Authorize(true, false, "host", http.HandlerFunc(hostUpdateFallback))).
		Methods(http.MethodPut)
	r.HandleFunc("/api/v1/host/{hostid}/peer_info", Authorize(true, false, "host", http.HandlerFunc(getHostPeerInfo))).
		Methods(http.MethodGet)
	r.HandleFunc("/api/emqx/hosts", logic.SecurityCheck(true, http.HandlerFunc(delEmqxHosts))).
		Methods(http.MethodDelete)
	r.HandleFunc("/api/v1/auth-register/host", socketHandler)
}

// @Summary     Requests all the hosts to upgrade their version
// @Router      /api/hosts/upgrade [post]
// @Tags        Hosts
// @Security    oauth
// @Param       force query bool false "Force upgrade"
// @Success     200 {string} string "upgrade all hosts request received"
func upgradeHosts(w http.ResponseWriter, r *http.Request) {
	w.Header().Set("Content-Type", "application/json")

	action := models.Upgrade

	if r.URL.Query().Get("force") == "true" {
		action = models.ForceUpgrade
	}

	user := r.Header.Get("user")

	go func() {
		slog.Info("requesting all hosts to upgrade", "user", user)

		hosts, err := logic.GetAllHosts()
		if err != nil {
			slog.Error("failed to retrieve all hosts", "user", user, "error", err)
			return
		}

		for _, host := range hosts {
			go func(host models.Host) {
				hostUpdate := models.HostUpdate{
					Action: action,
					Host:   host,
				}
				if err = mq.HostUpdate(&hostUpdate); err != nil {
					slog.Error("failed to request host to upgrade", "user", user, "host", host.ID.String(), "error", err)
				} else {
					slog.Info("host upgrade requested", "user", user, "host", host.ID.String())
				}
			}(host)
		}
	}()

	slog.Info("upgrade all hosts request received", "user", user)
	logic.ReturnSuccessResponse(w, r, "upgrade all hosts request received")
}

// @Summary     Upgrade a host
// @Router      /api/hosts/{hostid}/upgrade [put]
// @Tags        Hosts
// @Security    oauth
// @Param       hostid path string true "Host ID"
// @Param       force query bool false "Force upgrade"
// @Success     200 {string} string "passed message to upgrade host"
// @Failure     500 {object} models.ErrorResponse
// upgrade host is a handler to send upgrade message to a host
func upgradeHost(w http.ResponseWriter, r *http.Request) {
	host, err := logic.GetHost(mux.Vars(r)["hostid"])
	if err != nil {
		slog.Error("failed to find host", "error", err)
		logic.ReturnErrorResponse(w, r, logic.FormatError(err, "notfound"))
		return
	}

	action := models.Upgrade

	if r.URL.Query().Get("force") == "true" {
		action = models.ForceUpgrade
	}

	if err := mq.HostUpdate(&models.HostUpdate{Action: action, Host: *host}); err != nil {
		slog.Error("failed to upgrade host", "error", err)
		logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
		return
	}
	logic.ReturnSuccessResponse(w, r, "passed message to upgrade host")
}

// @Summary     List all hosts
// @Router      /api/hosts [get]
// @Tags        Hosts
// @Security    oauth
// @Success     200 {array} models.ApiHost
// @Failure     500 {object} models.ErrorResponse
func getHosts(w http.ResponseWriter, r *http.Request) {
	w.Header().Set("Content-Type", "application/json")
	currentHosts := []models.Host{}
	var err error
	if r.Header.Get("ismaster") == "yes" {
		currentHosts, err = logic.GetAllHosts()
		if err != nil {
			logger.Log(0, r.Header.Get("user"), "failed to fetch hosts: ", err.Error())
			logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
			return
		}
	} else {
		username := r.Header.Get("user")
		user, err := logic.GetUser(username)
		if err != nil {
			return
		}
		userPlatformRole, err := logic.GetRole(user.PlatformRoleID)
		if err != nil {
			return
		}
		respHostsMap := make(map[string]struct{})
		if !userPlatformRole.FullAccess {
			nodes, err := logic.GetAllNodes()
			if err != nil {
				logger.Log(0, "error fetching all nodes info: ", err.Error())
				logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
				return
			}
			filteredNodes := logic.GetFilteredNodesByUserAccess(*user, nodes)
			if len(filteredNodes) > 0 {
				currentHostsMap, err := logic.GetHostsMap()
				if err != nil {
					logger.Log(0, r.Header.Get("user"), "failed to fetch hosts: ", err.Error())
					logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
					return
				}
				for _, node := range filteredNodes {
					if _, ok := respHostsMap[node.HostID.String()]; ok {
						continue
					}
					if host, ok := currentHostsMap[node.HostID.String()]; ok {
						currentHosts = append(currentHosts, host)
						respHostsMap[host.ID.String()] = struct{}{}
					}
				}

			}
		} else {
			currentHosts, err = logic.GetAllHosts()
			if err != nil {
				logger.Log(0, r.Header.Get("user"), "failed to fetch hosts: ", err.Error())
				logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
				return
			}
		}
	}

	apiHosts := logic.GetAllHostsAPI(currentHosts[:])
	logger.Log(2, r.Header.Get("user"), "fetched all hosts")
	logic.SortApiHosts(apiHosts[:])
	w.WriteHeader(http.StatusOK)
	json.NewEncoder(w).Encode(apiHosts)
}

// @Summary     Used by clients for "pull" command
// @Router      /api/v1/host [get]
// @Tags        Hosts
// @Security    oauth
// @Success     200 {object} models.HostPull
// @Failure     500 {object} models.ErrorResponse
func pull(w http.ResponseWriter, r *http.Request) {

	hostID := r.Header.Get(hostIDHeader) // return JSON/API formatted keys
	if len(hostID) == 0 {
		logger.Log(0, "no host authorized to pull")
		logic.ReturnErrorResponse(
			w,
			r,
			logic.FormatError(fmt.Errorf("no host authorized to pull"), "internal"),
		)
		return
	}
	host, err := logic.GetHost(hostID)
	if err != nil {
		logger.Log(0, "no host found during pull", hostID)
		logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
		return
	}

	sendPeerUpdate := false
	for _, nodeID := range host.Nodes {
		node, err := logic.GetNodeByID(nodeID)
		if err != nil {
			slog.Error("failed to get node:", "id", node.ID, "error", err)
			continue
		}
		if node.FailedOverBy != uuid.Nil && r.URL.Query().Get("reset_failovered") == "true" {
			logic.ResetFailedOverPeer(&node)
			sendPeerUpdate = true
		}
	}
	if sendPeerUpdate {
		if err := mq.PublishPeerUpdate(false); err != nil {
			logger.Log(0, "fail to publish peer update: ", err.Error())
		}
	}
	allNodes, err := logic.GetAllNodes()
	if err != nil {
		logger.Log(0, "failed to get nodes: ", hostID)
		logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
		return
	}
	hPU, err := logic.GetPeerUpdateForHost("", host, allNodes, nil, nil)
	if err != nil {
		logger.Log(0, "could not pull peers for host", hostID, err.Error())
		logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
		return
	}
	serverConf := servercfg.GetServerInfo()
	key, keyErr := logic.RetrievePublicTrafficKey()
	if keyErr != nil {
		logger.Log(0, "error retrieving key:", keyErr.Error())
		logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
		return
	}

	serverConf.TrafficKey = key
	response := models.HostPull{
		Host:              *host,
		Nodes:             logic.GetHostNodes(host),
		ServerConfig:      serverConf,
		Peers:             hPU.Peers,
		PeerIDs:           hPU.PeerIDs,
		HostNetworkInfo:   hPU.HostNetworkInfo,
		EgressRoutes:      hPU.EgressRoutes,
		FwUpdate:          hPU.FwUpdate,
		ChangeDefaultGw:   hPU.ChangeDefaultGw,
		DefaultGwIp:       hPU.DefaultGwIp,
		IsInternetGw:      hPU.IsInternetGw,
		EndpointDetection: servercfg.IsEndpointDetectionEnabled(),
	}

	logger.Log(1, hostID, "completed a pull")
	w.WriteHeader(http.StatusOK)
	json.NewEncoder(w).Encode(&response)
}

// @Summary     Updates a Netclient host on Netmaker server
// @Router      /api/hosts/{hostid} [put]
// @Tags        Hosts
// @Security    oauth
// @Param       hostid path string true "Host ID"
// @Param       body body models.ApiHost true "New host data"
// @Success     200 {object} models.ApiHost
// @Failure     500 {object} models.ErrorResponse
func updateHost(w http.ResponseWriter, r *http.Request) {
	var newHostData models.ApiHost
	err := json.NewDecoder(r.Body).Decode(&newHostData)
	if err != nil {
		logger.Log(0, r.Header.Get("user"), "failed to update a host:", err.Error())
		logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
		return
	}

	// confirm host exists
	currHost, err := logic.GetHost(newHostData.ID)
	if err != nil {
		logger.Log(0, r.Header.Get("user"), "failed to update a host:", err.Error())
		logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
		return
	}

	newHost := newHostData.ConvertAPIHostToNMHost(currHost)

	logic.UpdateHost(newHost, currHost) // update the in memory struct values
	if err = logic.UpsertHost(newHost); err != nil {
		logger.Log(0, r.Header.Get("user"), "failed to update a host:", err.Error())
		logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
		return
	}
	// publish host update through MQ
	if err := mq.HostUpdate(&models.HostUpdate{
		Action: models.UpdateHost,
		Host:   *newHost,
	}); err != nil {
		logger.Log(
			0,
			r.Header.Get("user"),
			"failed to send host update: ",
			currHost.ID.String(),
			err.Error(),
		)
	}
	go func() {
		if err := mq.PublishPeerUpdate(false); err != nil {
			logger.Log(0, "fail to publish peer update: ", err.Error())
		}
		if newHost.Name != currHost.Name {
			if servercfg.IsDNSMode() {
				logic.SetDNS()
			}
		}
	}()

	apiHostData := newHost.ConvertNMHostToAPI()
	logger.Log(2, r.Header.Get("user"), "updated host", newHost.ID.String())
	w.WriteHeader(http.StatusOK)
	json.NewEncoder(w).Encode(apiHostData)
}

// @Summary     Updates a Netclient host on Netmaker server
// @Router      /api/v1/fallback/host/{hostid} [put]
// @Tags        Hosts
// @Security    oauth
// @Param       hostid path string true "Host ID"
// @Param       body body models.HostUpdate true "Host update data"
// @Success     200 {string} string "updated host data"
// @Failure     500 {object} models.ErrorResponse
func hostUpdateFallback(w http.ResponseWriter, r *http.Request) {
	var params = mux.Vars(r)
	hostid := params["hostid"]
	currentHost, err := logic.GetHost(hostid)
	if err != nil {
		slog.Error("error getting host", "id", hostid, "error", err)
		logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
		return
	}
	var sendPeerUpdate bool
	var replacePeers bool
	var hostUpdate models.HostUpdate
	err = json.NewDecoder(r.Body).Decode(&hostUpdate)
	if err != nil {
		slog.Error("failed to update a host:", "user", r.Header.Get("user"), "error", err.Error(), "host", currentHost.Name)
		logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
		return
	}
	slog.Info("recieved host update", "name", hostUpdate.Host.Name, "id", hostUpdate.Host.ID, "action", hostUpdate.Action)
	switch hostUpdate.Action {
	case models.CheckIn:
		sendPeerUpdate = mq.HandleHostCheckin(&hostUpdate.Host, currentHost)

	case models.UpdateHost:
		if hostUpdate.Host.PublicKey != currentHost.PublicKey {
			//remove old peer entry
			replacePeers = true
		}
		sendPeerUpdate = logic.UpdateHostFromClient(&hostUpdate.Host, currentHost)
		err := logic.UpsertHost(currentHost)
		if err != nil {
			slog.Error("failed to update host", "id", currentHost.ID, "error", err)
			logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
			return
		}

	case models.UpdateMetrics:
		mq.UpdateMetricsFallBack(hostUpdate.Node.ID.String(), hostUpdate.NewMetrics)
	}

	if sendPeerUpdate {
		err := mq.PublishPeerUpdate(replacePeers)
		if err != nil {
			slog.Error("failed to publish peer update", "error", err)
		}
	}
	logic.ReturnSuccessResponse(w, r, "updated host data")
}

// @Summary     Deletes a Netclient host from Netmaker server
// @Router      /api/hosts/{hostid} [delete]
// @Tags        Hosts
// @Security    oauth
// @Param       hostid path string true "Host ID"
// @Param       force query bool false "Force delete"
// @Success     200 {object} models.ApiHost
// @Failure     500 {object} models.ErrorResponse
func deleteHost(w http.ResponseWriter, r *http.Request) {
	var params = mux.Vars(r)
	hostid := params["hostid"]
	forceDelete := r.URL.Query().Get("force") == "true"

	// confirm host exists
	currHost, err := logic.GetHost(hostid)
	if err != nil {
		logger.Log(0, r.Header.Get("user"), "failed to delete a host:", err.Error())
		logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
		return
	}
	for _, nodeID := range currHost.Nodes {
		node, err := logic.GetNodeByID(nodeID)
		if err != nil {
			slog.Error("failed to get node", "nodeid", nodeID, "error", err)
			continue
		}
		var gwClients []models.ExtClient
		if node.IsIngressGateway {
			gwClients = logic.GetGwExtclients(node.ID.String(), node.Network)
		}
		go mq.PublishMqUpdatesForDeletedNode(node, false, gwClients)

	}
	if servercfg.GetBrokerType() == servercfg.EmqxBrokerType {
		// delete EMQX credentials for host
		if err := mq.GetEmqxHandler().DeleteEmqxUser(currHost.ID.String()); err != nil {
			slog.Error(
				"failed to remove host credentials from EMQX",
				"id",
				currHost.ID,
				"error",
				err,
			)
		}
	}
	if err = mq.HostUpdate(&models.HostUpdate{
		Action: models.DeleteHost,
		Host:   *currHost,
	}); err != nil {
		logger.Log(
			0,
			r.Header.Get("user"),
			"failed to send delete host update: ",
			currHost.ID.String(),
			err.Error(),
		)
	}
	if err = logic.RemoveHost(currHost, forceDelete); err != nil {
		logger.Log(0, r.Header.Get("user"), "failed to delete a host:", err.Error())
		logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
		return
	}

	apiHostData := currHost.ConvertNMHostToAPI()
	logger.Log(2, r.Header.Get("user"), "removed host", currHost.Name)
	w.WriteHeader(http.StatusOK)
	json.NewEncoder(w).Encode(apiHostData)
}

// @Summary     To Add Host To Network
// @Router      /api/hosts/{hostid}/networks/{network} [post]
// @Tags        Hosts
// @Security    oauth
// @Param       hostid path string true "Host ID"
// @Param       network path string true "Network name"
// @Success     200 {string} string "OK"
// @Failure     500 {object} models.ErrorResponse
func addHostToNetwork(w http.ResponseWriter, r *http.Request) {

	var params = mux.Vars(r)
	hostid := params["hostid"]
	network := params["network"]
	if hostid == "" || network == "" {
		logic.ReturnErrorResponse(
			w,
			r,
			logic.FormatError(errors.New("hostid or network cannot be empty"), "badrequest"),
		)
		return
	}
	// confirm host exists
	currHost, err := logic.GetHost(hostid)
	if err != nil {
		logger.Log(0, r.Header.Get("user"), "failed to find host:", hostid, err.Error())
		logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
		return
	}

	newNode, err := logic.UpdateHostNetwork(currHost, network, true)
	if err != nil {
		logger.Log(
			0,
			r.Header.Get("user"),
			"failed to add host to network:",
			hostid,
			network,
			err.Error(),
		)
		logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
		return
	}
	logger.Log(1, "added new node", newNode.ID.String(), "to host", currHost.Name)
	if currHost.IsDefault {
		// make  host failover
		logic.CreateFailOver(*newNode)
		// make host remote access gateway
		logic.CreateIngressGateway(network, newNode.ID.String(), models.IngressRequest{})
	}
	go func() {
		mq.HostUpdate(&models.HostUpdate{
			Action: models.JoinHostToNetwork,
			Host:   *currHost,
			Node:   *newNode,
		})
		mq.PublishPeerUpdate(false)
		if servercfg.IsDNSMode() {
			logic.SetDNS()
		}
	}()
	logger.Log(
		2,
		r.Header.Get("user"),
		fmt.Sprintf("added host %s to network %s", currHost.Name, network),
	)
	w.WriteHeader(http.StatusOK)
}

// @Summary     To Remove Host from Network
// @Router      /api/hosts/{hostid}/networks/{network} [delete]
// @Tags        Hosts
// @Security    oauth
// @Param       hostid path string true "Host ID"
// @Param       network path string true "Network name"
// @Param       force query bool false "Force delete"
// @Success     200 {string} string "OK"
// @Failure     500 {object} models.ErrorResponse
func deleteHostFromNetwork(w http.ResponseWriter, r *http.Request) {

	var params = mux.Vars(r)
	hostid := params["hostid"]
	network := params["network"]
	forceDelete := r.URL.Query().Get("force") == "true"
	if hostid == "" || network == "" {
		logic.ReturnErrorResponse(
			w,
			r,
			logic.FormatError(errors.New("hostid or network cannot be empty"), "badrequest"),
		)
		return
	}
	// confirm host exists
	currHost, err := logic.GetHost(hostid)
	if err != nil {
		if database.IsEmptyRecord(err) {
			// check if there is any daemon nodes that needs to be deleted
			node, err := logic.GetNodeByHostRef(hostid, network)
			if err != nil {
				slog.Error(
					"couldn't get node for host",
					"hostid",
					hostid,
					"network",
					network,
					"error",
					err,
				)
				logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
				return
			}
			if err = logic.DeleteNodeByID(&node); err != nil {
				slog.Error("failed to force delete daemon node",
					"nodeid", node.ID.String(), "hostid", hostid, "network", network, "error", err)
				logic.ReturnErrorResponse(
					w,
					r,
					logic.FormatError(
						fmt.Errorf("failed to force delete daemon node: %s", err.Error()),
						"internal",
					),
				)
				return
			}
			logic.ReturnSuccessResponse(w, r, "force deleted daemon node successfully")
			return
		}

		logger.Log(0, r.Header.Get("user"), "failed to find host:", err.Error())
		logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
		return
	}

	node, err := logic.UpdateHostNetwork(currHost, network, false)
	if err != nil {
		if node == nil && forceDelete {
			// force cleanup the node
			node, err := logic.GetNodeByHostRef(hostid, network)
			if err != nil {
				slog.Error(
					"couldn't get node for host",
					"hostid",
					hostid,
					"network",
					network,
					"error",
					err,
				)
				logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
				return
			}
			if err = logic.DeleteNodeByID(&node); err != nil {
				slog.Error("failed to force delete daemon node",
					"nodeid", node.ID.String(), "hostid", hostid, "network", network, "error", err)
				logic.ReturnErrorResponse(
					w,
					r,
					logic.FormatError(
						fmt.Errorf("failed to force delete daemon node: %s", err.Error()),
						"internal",
					),
				)
				return
			}
			logic.ReturnSuccessResponse(w, r, "force deleted daemon node successfully")
			return
		}
		logger.Log(
			0,
			r.Header.Get("user"),
			"failed to remove host from network:",
			hostid,
			network,
			err.Error(),
		)
		logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
		return
	}
	var gwClients []models.ExtClient
	if node.IsIngressGateway {
		gwClients = logic.GetGwExtclients(node.ID.String(), node.Network)
	}
	logger.Log(1, "deleting node", node.ID.String(), "from host", currHost.Name)
	if err := logic.DeleteNode(node, forceDelete); err != nil {
		logic.ReturnErrorResponse(
			w,
			r,
			logic.FormatError(fmt.Errorf("failed to delete node"), "internal"),
		)
		return
	}
	go func() {
		mq.PublishMqUpdatesForDeletedNode(*node, true, gwClients)
		if servercfg.IsDNSMode() {
			logic.SetDNS()
		}
	}()
	logger.Log(
		2,
		r.Header.Get("user"),
		fmt.Sprintf("removed host %s from network %s", currHost.Name, network),
	)
	w.WriteHeader(http.StatusOK)
}

// @Summary     To Fetch Auth Token for a Host
// @Router      /api/hosts/adm/authenticate [post]
// @Tags        Auth
// @Accept      json
// @Param       body body models.AuthParams true "Authentication parameters"
// @Success     200 {object} models.SuccessResponse
// @Failure     400 {object} models.ErrorResponse
// @Failure     401 {object} models.ErrorResponse
// @Failure     500 {object} models.ErrorResponse
func authenticateHost(response http.ResponseWriter, request *http.Request) {
	var authRequest models.AuthParams
	var errorResponse = models.ErrorResponse{
		Code: http.StatusInternalServerError, Message: "W1R3: It's not you it's me.",
	}

	decoder := json.NewDecoder(request.Body)
	decoderErr := decoder.Decode(&authRequest)
	defer request.Body.Close()

	if decoderErr != nil {
		errorResponse.Code = http.StatusBadRequest
		errorResponse.Message = decoderErr.Error()
		logger.Log(0, request.Header.Get("user"), "error decoding request body: ",
			decoderErr.Error())
		logic.ReturnErrorResponse(response, request, errorResponse)
		return
	}
	errorResponse.Code = http.StatusBadRequest
	if authRequest.ID == "" {
		errorResponse.Message = "W1R3: ID can't be empty"
		logger.Log(0, request.Header.Get("user"), errorResponse.Message)
		logic.ReturnErrorResponse(response, request, errorResponse)
		return
	} else if authRequest.Password == "" {
		errorResponse.Message = "W1R3: Password can't be empty"
		logger.Log(0, request.Header.Get("user"), errorResponse.Message)
		logic.ReturnErrorResponse(response, request, errorResponse)
		return
	}
	host, err := logic.GetHost(authRequest.ID)
	if err != nil {
		errorResponse.Code = http.StatusBadRequest
		errorResponse.Message = err.Error()
		logger.Log(0, request.Header.Get("user"),
			"error retrieving host: ", authRequest.ID, err.Error())
		logic.ReturnErrorResponse(response, request, errorResponse)
		return
	}
	err = bcrypt.CompareHashAndPassword([]byte(host.HostPass), []byte(authRequest.Password))
	if err != nil {
		errorResponse.Code = http.StatusUnauthorized
		errorResponse.Message = "unauthorized"
		logger.Log(0, request.Header.Get("user"),
			"error validating user password: ", err.Error())
		logic.ReturnErrorResponse(response, request, errorResponse)
		return
	}

	tokenString, err := logic.CreateJWT(authRequest.ID, authRequest.MacAddress, "")
	if tokenString == "" {
		errorResponse.Code = http.StatusUnauthorized
		errorResponse.Message = "unauthorized"
		logger.Log(0, request.Header.Get("user"),
			fmt.Sprintf("%s: %v", errorResponse.Message, err))
		logic.ReturnErrorResponse(response, request, errorResponse)
		return
	}

	var successResponse = models.SuccessResponse{
		Code:    http.StatusOK,
		Message: "W1R3: Host " + authRequest.ID + " Authorized",
		Response: models.SuccessfulLoginResponse{
			AuthToken: tokenString,
			ID:        authRequest.ID,
		},
	}
	successJSONResponse, jsonError := json.Marshal(successResponse)

	if jsonError != nil {
		errorResponse.Code = http.StatusBadRequest
		errorResponse.Message = err.Error()
		logger.Log(0, request.Header.Get("user"),
			"error marshalling resp: ", err.Error())
		logic.ReturnErrorResponse(response, request, errorResponse)
		return
	}
	go func() {
		// Create EMQX creds
		if servercfg.GetBrokerType() == servercfg.EmqxBrokerType {
			if err := mq.GetEmqxHandler().CreateEmqxUser(host.ID.String(), authRequest.Password); err != nil {
				slog.Error("failed to create host credentials for EMQX: ", err.Error())
			}
		}
	}()

	response.WriteHeader(http.StatusOK)
	response.Header().Set("Content-Type", "application/json")
	response.Write(successJSONResponse)
}

// @Summary     Send signal to peer
// @Router      /api/v1/host/{hostid}/signalpeer [post]
// @Tags        Hosts
// @Security    oauth
// @Param       hostid path string true "Host ID"
// @Param       body body models.Signal true "Signal data"
// @Success     200 {object} models.Signal
// @Failure     400 {object} models.ErrorResponse
func signalPeer(w http.ResponseWriter, r *http.Request) {
	var params = mux.Vars(r)
	hostid := params["hostid"]
	// confirm host exists
	_, err := logic.GetHost(hostid)
	if err != nil {
		logger.Log(0, r.Header.Get("user"), "failed to get host:", err.Error())
		logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
		return
	}
	var signal models.Signal
	w.Header().Set("Content-Type", "application/json")
	err = json.NewDecoder(r.Body).Decode(&signal)
	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
	}
	if signal.ToHostPubKey == "" {
		msg := "insufficient data to signal peer"
		logger.Log(0, r.Header.Get("user"), msg)
		logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New(msg), "badrequest"))
		return
	}
	signal.IsPro = servercfg.IsPro
	peerHost, err := logic.GetHost(signal.ToHostID)
	if err != nil {
		logic.ReturnErrorResponse(
			w,
			r,
			logic.FormatError(errors.New("failed to signal, peer not found"), "badrequest"),
		)
		return
	}
	err = mq.HostUpdate(&models.HostUpdate{
		Action: models.SignalHost,
		Host:   *peerHost,
		Signal: signal,
	})
	if err != nil {
		logic.ReturnErrorResponse(
			w,
			r,
			logic.FormatError(
				errors.New("failed to publish signal to peer: "+err.Error()),
				"badrequest",
			),
		)
		return
	}

	w.WriteHeader(http.StatusOK)
	json.NewEncoder(w).Encode(signal)
}

// @Summary     Update keys for all hosts
// @Router      /api/hosts/keys [put]
// @Tags        Hosts
// @Security    oauth
// @Success     200 {string} string "OK"
// @Failure     400 {object} models.ErrorResponse
func updateAllKeys(w http.ResponseWriter, r *http.Request) {
	var errorResponse = models.ErrorResponse{}
	w.Header().Set("Content-Type", "application/json")
	hosts, err := logic.GetAllHosts()
	if err != nil {
		errorResponse.Code = http.StatusBadRequest
		errorResponse.Message = err.Error()
		logger.Log(0, r.Header.Get("user"),
			"error retrieving hosts ", err.Error())
		logic.ReturnErrorResponse(w, r, errorResponse)
		return
	}
	go func() {
		hostUpdate := models.HostUpdate{}
		hostUpdate.Action = models.UpdateKeys
		for _, host := range hosts {
			hostUpdate.Host = host
			logger.Log(2, "updating host", host.ID.String(), " for a key update")
			if err = mq.HostUpdate(&hostUpdate); err != nil {
				logger.Log(
					0,
					"failed to send update to node during a network wide key update",
					host.ID.String(),
					err.Error(),
				)
			}
		}
	}()
	logger.Log(2, r.Header.Get("user"), "updated keys for all hosts")
	w.WriteHeader(http.StatusOK)
}

// @Summary     Update keys for a host
// @Router      /api/hosts/{hostid}/keys [put]
// @Tags        Hosts
// @Security    oauth
// @Param       hostid path string true "Host ID"
// @Success     200 {string} string "OK"
// @Failure     400 {object} models.ErrorResponse
func updateKeys(w http.ResponseWriter, r *http.Request) {
	var errorResponse = models.ErrorResponse{}
	w.Header().Set("Content-Type", "application/json")
	var params = mux.Vars(r)
	hostid := params["hostid"]
	host, err := logic.GetHost(hostid)
	if err != nil {
		logger.Log(0, "failed to retrieve host", hostid, err.Error())
		errorResponse.Code = http.StatusBadRequest
		errorResponse.Message = err.Error()
		logger.Log(0, r.Header.Get("user"),
			"error retrieving hosts ", err.Error())
		logic.ReturnErrorResponse(w, r, errorResponse)
		return
	}
	go func() {
		hostUpdate := models.HostUpdate{
			Action: models.UpdateKeys,
			Host:   *host,
		}
		if err = mq.HostUpdate(&hostUpdate); err != nil {
			logger.Log(0, "failed to send host key update", host.ID.String(), err.Error())
		}
	}()
	logger.Log(2, r.Header.Get("user"), "updated key on host", host.Name)
	w.WriteHeader(http.StatusOK)
}

// @Summary     Requests all the hosts to pull
// @Router      /api/hosts/sync [post]
// @Tags        Hosts
// @Security    oauth
// @Success     200 {string} string "sync all hosts request received"
func syncHosts(w http.ResponseWriter, r *http.Request) {
	w.Header().Set("Content-Type", "application/json")

	user := r.Header.Get("user")

	go func() {
		slog.Info("requesting all hosts to sync", "user", user)

		hosts, err := logic.GetAllHosts()
		if err != nil {
			slog.Error("failed to retrieve all hosts", "user", user, "error", err)
			return
		}

		for _, host := range hosts {
			go func(host models.Host) {
				hostUpdate := models.HostUpdate{
					Action: models.RequestPull,
					Host:   host,
				}
				if err = mq.HostUpdate(&hostUpdate); err != nil {
					slog.Error("failed to request host to sync", "user", user, "host", host.ID.String(), "error", err)
				} else {
					slog.Info("host sync requested", "user", user, "host", host.ID.String())
				}
			}(host)
			time.Sleep(time.Millisecond * 100)
		}
	}()

	slog.Info("sync all hosts request received", "user", user)
	logic.ReturnSuccessResponse(w, r, "sync all hosts request received")
}

// @Summary     Requests a host to pull
// @Router      /api/hosts/{hostid}/sync [post]
// @Tags        Hosts
// @Security    oauth
// @Param       hostid path string true "Host ID"
// @Success     200 {string} string "OK"
// @Failure     400 {object} models.ErrorResponse
func syncHost(w http.ResponseWriter, r *http.Request) {
	hostId := mux.Vars(r)["hostid"]

	var errorResponse = models.ErrorResponse{}
	w.Header().Set("Content-Type", "application/json")

	host, err := logic.GetHost(hostId)
	if err != nil {
		slog.Error("failed to retrieve host", "user", r.Header.Get("user"), "error", err)
		errorResponse.Code = http.StatusBadRequest
		errorResponse.Message = err.Error()
		logic.ReturnErrorResponse(w, r, errorResponse)
		return
	}

	go func() {
		hostUpdate := models.HostUpdate{
			Action: models.RequestPull,
			Host:   *host,
		}
		if err = mq.HostUpdate(&hostUpdate); err != nil {
			slog.Error("failed to send host pull request", "host", host.ID.String(), "error", err)
		}
	}()

	slog.Info("requested host pull", "user", r.Header.Get("user"), "host", host.ID.String())
	w.WriteHeader(http.StatusOK)
}

// @Summary     Deletes all EMQX hosts
// @Router      /api/emqx/hosts [delete]
// @Tags        Hosts
// @Security    oauth
// @Success     200 {string} string "deleted hosts data on emqx"
// @Failure     500 {object} models.ErrorResponse
func delEmqxHosts(w http.ResponseWriter, r *http.Request) {
	currentHosts, err := logic.GetAllHosts()
	if err != nil {
		logger.Log(0, r.Header.Get("user"), "failed to fetch hosts: ", err.Error())
		logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
		return
	}
	for _, host := range currentHosts {
		// delete EMQX credentials for host
		if err := mq.GetEmqxHandler().DeleteEmqxUser(host.ID.String()); err != nil {
			slog.Error("failed to remove host credentials from EMQX", "id", host.ID, "error", err)
		}
	}
	err = mq.GetEmqxHandler().DeleteEmqxUser(servercfg.GetMqUserName())
	if err != nil {
		slog.Error(
			"failed to remove server credentials from EMQX",
			"user",
			servercfg.GetMqUserName(),
			"error",
			err,
		)
	}
	logic.ReturnSuccessResponse(w, r, "deleted hosts data on emqx")
}

// @Summary     Fetches host peerinfo
// @Router      /api/host/{hostid}/peer_info [get]
// @Tags        Hosts
// @Security    oauth
// @Param       hostid path string true "Host ID"
// @Success     200 {object} models.SuccessResponse
// @Failure     500 {object} models.ErrorResponse
func getHostPeerInfo(w http.ResponseWriter, r *http.Request) {
	hostId := mux.Vars(r)["hostid"]
	var errorResponse = models.ErrorResponse{}

	host, err := logic.GetHost(hostId)
	if err != nil {
		slog.Error("failed to retrieve host", "error", err)
		errorResponse.Code = http.StatusBadRequest
		errorResponse.Message = err.Error()
		logic.ReturnErrorResponse(w, r, errorResponse)
		return
	}
	peerInfo, err := logic.GetHostPeerInfo(host)
	if err != nil {
		slog.Error("failed to retrieve host peerinfo", "error", err)
		errorResponse.Code = http.StatusBadRequest
		errorResponse.Message = err.Error()
		logic.ReturnErrorResponse(w, r, errorResponse)
		return
	}
	logic.ReturnSuccessResponseWithJson(w, r, peerInfo, "fetched host peer info")
}