portainer/portainer

package endpoints

import (
    "errors"
    "net/http"

    portainer "github.com/portainer/portainer/api"
    httperror "github.com/portainer/portainer/pkg/libhttp/error"
    "github.com/portainer/portainer/pkg/libhttp/response"
)

type endpointCreateGlobalKeyResponse struct {
    EndpointID portainer.EndpointID `json:"endpointID"`
}

// @id EndpointCreateGlobalKey
// @summary Create or retrieve the endpoint for an EdgeID
// @tags endpoints
// @success 200 {object} endpointCreateGlobalKeyResponse "Success"
// @failure 400 "Invalid request"
// @failure 500 "Server error"
// @router /endpoints/global-key [post]
func (handler *Handler) endpointCreateGlobalKey(w http.ResponseWriter, r *http.Request) *httperror.HandlerError {
    edgeID := r.Header.Get(portainer.PortainerAgentEdgeIDHeader)
    if edgeID == "" {
        return httperror.BadRequest("Invalid Edge ID", errors.New("the Edge ID cannot be empty"))
    }

    // Search for existing endpoints for the given edgeID

    endpointID, ok := handler.DataStore.Endpoint().EndpointIDByEdgeID(edgeID)
    if ok {
        return response.JSON(w, endpointCreateGlobalKeyResponse{endpointID})
    }

    return httperror.NotFound("Unable to find the endpoint in the database", nil)
}