mirror of https://github.com/portainer/portainer
				
				
				
			
		
			
				
	
	
		
			70 lines
		
	
	
		
			2.7 KiB
		
	
	
	
		
			Go
		
	
	
			
		
		
	
	
			70 lines
		
	
	
		
			2.7 KiB
		
	
	
	
		
			Go
		
	
	
package openamt
 | 
						|
 | 
						|
import (
 | 
						|
	"errors"
 | 
						|
	"net/http"
 | 
						|
 | 
						|
	portainer "github.com/portainer/portainer/api"
 | 
						|
	"github.com/portainer/portainer/api/internal/endpointutils"
 | 
						|
	httperror "github.com/portainer/portainer/pkg/libhttp/error"
 | 
						|
	"github.com/portainer/portainer/pkg/libhttp/request"
 | 
						|
	"github.com/portainer/portainer/pkg/libhttp/response"
 | 
						|
)
 | 
						|
 | 
						|
// @id openAMTActivate
 | 
						|
// @summary Activate OpenAMT device and associate to agent endpoint
 | 
						|
// @description Activate OpenAMT device and associate to agent endpoint
 | 
						|
// @description **Access policy**: administrator
 | 
						|
// @tags intel
 | 
						|
// @security jwt
 | 
						|
// @produce json
 | 
						|
// @param id path int true "Environment identifier"
 | 
						|
// @success 200 "Success"
 | 
						|
// @failure 400 "Invalid request"
 | 
						|
// @failure 403 "Permission denied to access settings"
 | 
						|
// @failure 500 "Server error"
 | 
						|
// @router /open_amt/{id}/activate [post]
 | 
						|
func (handler *Handler) openAMTActivate(w http.ResponseWriter, r *http.Request) *httperror.HandlerError {
 | 
						|
	endpointID, err := request.RetrieveNumericRouteVariableValue(r, "id")
 | 
						|
	if err != nil {
 | 
						|
		return httperror.BadRequest("Invalid environment identifier route variable", err)
 | 
						|
	}
 | 
						|
 | 
						|
	endpoint, err := handler.DataStore.Endpoint().Endpoint(portainer.EndpointID(endpointID))
 | 
						|
	if handler.DataStore.IsErrObjectNotFound(err) {
 | 
						|
		return httperror.NotFound("Unable to find an endpoint with the specified identifier inside the database", err)
 | 
						|
	} else if err != nil {
 | 
						|
		return httperror.InternalServerError("Unable to find an endpoint with the specified identifier inside the database", err)
 | 
						|
	} else if !endpointutils.IsAgentEndpoint(endpoint) {
 | 
						|
		errMsg := endpoint.Name + " is not an agent environment"
 | 
						|
		return httperror.BadRequest(errMsg, errors.New(errMsg))
 | 
						|
	}
 | 
						|
 | 
						|
	settings, err := handler.DataStore.Settings().Settings()
 | 
						|
	if err != nil {
 | 
						|
		return httperror.InternalServerError("Unable to retrieve settings from the database", err)
 | 
						|
	}
 | 
						|
 | 
						|
	if err := handler.activateDevice(endpoint, *settings); err != nil {
 | 
						|
		return httperror.InternalServerError("Unable to activate device", err)
 | 
						|
	}
 | 
						|
 | 
						|
	hostInfo, _, err := handler.getEndpointAMTInfo(endpoint)
 | 
						|
	if err != nil {
 | 
						|
		return httperror.InternalServerError("Unable to retrieve AMT information", err)
 | 
						|
	}
 | 
						|
	if hostInfo.ControlModeRaw < 1 {
 | 
						|
		return httperror.InternalServerError("Failed to activate device", errors.New("failed to activate device"))
 | 
						|
	}
 | 
						|
	if hostInfo.UUID == "" {
 | 
						|
		return httperror.InternalServerError("Unable to retrieve device UUID", errors.New("unable to retrieve device UUID"))
 | 
						|
	}
 | 
						|
 | 
						|
	endpoint.AMTDeviceGUID = hostInfo.UUID
 | 
						|
	if err := handler.DataStore.Endpoint().UpdateEndpoint(endpoint.ID, endpoint); err != nil {
 | 
						|
		return httperror.InternalServerError("Unable to persist environment changes inside the database", err)
 | 
						|
	}
 | 
						|
 | 
						|
	return response.Empty(w)
 | 
						|
}
 |