api_agent_registrations.go

/*
StackState API

This API documentation page describes the StackState server API. The StackState UI and CLI use the StackState server API to configure and query StackState.  You can use the API for similar purposes.  Each request sent to the StackState server API must be authenticated using one of the available authentication methods.   *Note that the StackState receiver API, used to send topology, telemetry and traces to StackState, is not described on this page and requires a different authentication method.*  For more information on StackState, refer to the [StackState documentation](https://docs.stackstate.com).

API version: 5.2.0
Contact: info@stackstate.com
*/

// Code generated by OpenAPI Generator (https://openapi-generator.tech); DO NOT EDIT.

package stackstate_api

import (
	"bytes"
	"context"
	"io/ioutil"
	"net/http"
	"net/url"
)

type AgentRegistrationsApi interface {

	/*
		AllAgentRegistrations Overview of registered agents

		Give an overview of all registered agents

		@param ctx context.Context - for authentication, logging, cancellation, deadlines, tracing, etc. Passed from http.Request or context.Background().
		@return ApiAllAgentRegistrationsRequest
	*/
	AllAgentRegistrations(ctx context.Context) ApiAllAgentRegistrationsRequest

	// AllAgentRegistrationsExecute executes the request
	//  @return AgentRegistrations
	AllAgentRegistrationsExecute(r ApiAllAgentRegistrationsRequest) (*AgentRegistrations, *http.Response, error)
}

// AgentRegistrationsApiService AgentRegistrationsApi service
type AgentRegistrationsApiService service

type ApiAllAgentRegistrationsRequest struct {
	ctx        context.Context
	ApiService AgentRegistrationsApi
}

func (r ApiAllAgentRegistrationsRequest) Execute() (*AgentRegistrations, *http.Response, error) {
	return r.ApiService.AllAgentRegistrationsExecute(r)
}

/*
AllAgentRegistrations Overview of registered agents

Give an overview of all registered agents

	@param ctx context.Context - for authentication, logging, cancellation, deadlines, tracing, etc. Passed from http.Request or context.Background().
	@return ApiAllAgentRegistrationsRequest
*/
func (a *AgentRegistrationsApiService) AllAgentRegistrations(ctx context.Context) ApiAllAgentRegistrationsRequest {
	return ApiAllAgentRegistrationsRequest{
		ApiService: a,
		ctx:        ctx,
	}
}

// Execute executes the request
//
//	@return AgentRegistrations
func (a *AgentRegistrationsApiService) AllAgentRegistrationsExecute(r ApiAllAgentRegistrationsRequest) (*AgentRegistrations, *http.Response, error) {
	var (
		localVarHTTPMethod  = http.MethodGet
		localVarPostBody    interface{}
		formFiles           []formFile
		localVarReturnValue *AgentRegistrations
	)

	localBasePath, err := a.client.cfg.ServerURLWithContext(r.ctx, "AgentRegistrationsApiService.AllAgentRegistrations")
	if err != nil {
		return localVarReturnValue, nil, &GenericOpenAPIError{error: err.Error()}
	}

	localVarPath := localBasePath + "/agents"

	localVarHeaderParams := make(map[string]string)
	localVarQueryParams := url.Values{}
	localVarFormParams := url.Values{}

	// to determine the Content-Type header
	localVarHTTPContentTypes := []string{}

	// set Content-Type header
	localVarHTTPContentType := selectHeaderContentType(localVarHTTPContentTypes)
	if localVarHTTPContentType != "" {
		localVarHeaderParams["Content-Type"] = localVarHTTPContentType
	}

	// to determine the Accept header
	localVarHTTPHeaderAccepts := []string{"application/json"}

	// set Accept header
	localVarHTTPHeaderAccept := selectHeaderAccept(localVarHTTPHeaderAccepts)
	if localVarHTTPHeaderAccept != "" {
		localVarHeaderParams["Accept"] = localVarHTTPHeaderAccept
	}
	if r.ctx != nil {
		// API Key Authentication
		if auth, ok := r.ctx.Value(ContextAPIKeys).(map[string]APIKey); ok {
			if apiKey, ok := auth["ApiToken"]; ok {
				var key string
				if apiKey.Prefix != "" {
					key = apiKey.Prefix + " " + apiKey.Key
				} else {
					key = apiKey.Key
				}
				localVarHeaderParams["X-API-Token"] = key
			}
		}
	}
	if r.ctx != nil {
		// API Key Authentication
		if auth, ok := r.ctx.Value(ContextAPIKeys).(map[string]APIKey); ok {
			if apiKey, ok := auth["ServiceBearer"]; ok {
				var key string
				if apiKey.Prefix != "" {
					key = apiKey.Prefix + " " + apiKey.Key
				} else {
					key = apiKey.Key
				}
				localVarHeaderParams["X-API-ServiceBearer"] = key
			}
		}
	}
	if r.ctx != nil {
		// API Key Authentication
		if auth, ok := r.ctx.Value(ContextAPIKeys).(map[string]APIKey); ok {
			if apiKey, ok := auth["ServiceToken"]; ok {
				var key string
				if apiKey.Prefix != "" {
					key = apiKey.Prefix + " " + apiKey.Key
				} else {
					key = apiKey.Key
				}
				localVarHeaderParams["X-API-Key"] = key
			}
		}
	}
	req, err := a.client.prepareRequest(r.ctx, localVarPath, localVarHTTPMethod, localVarPostBody, localVarHeaderParams, localVarQueryParams, localVarFormParams, formFiles)
	if err != nil {
		return localVarReturnValue, nil, err
	}

	localVarHTTPResponse, err := a.client.callAPI(req)
	if err != nil || localVarHTTPResponse == nil {
		return localVarReturnValue, localVarHTTPResponse, err
	}

	localVarBody, err := ioutil.ReadAll(localVarHTTPResponse.Body)
	localVarHTTPResponse.Body.Close()
	localVarHTTPResponse.Body = ioutil.NopCloser(bytes.NewBuffer(localVarBody))
	if err != nil {
		return localVarReturnValue, localVarHTTPResponse, err
	}

	if localVarHTTPResponse.StatusCode >= 300 {
		newErr := &GenericOpenAPIError{
			body:  localVarBody,
			error: localVarHTTPResponse.Status,
		}
		if localVarHTTPResponse.StatusCode == 500 {
			var v GenericErrorsResponse
			err = a.client.decode(&v, localVarBody, localVarHTTPResponse.Header.Get("Content-Type"))
			if err != nil {
				newErr.error = err.Error()
				return localVarReturnValue, localVarHTTPResponse, newErr
			}
			newErr.model = v
		}
		return localVarReturnValue, localVarHTTPResponse, newErr
	}

	err = a.client.decode(&localVarReturnValue, localVarBody, localVarHTTPResponse.Header.Get("Content-Type"))
	if err != nil {
		newErr := &GenericOpenAPIError{
			body:  localVarBody,
			error: err.Error(),
		}
		return localVarReturnValue, localVarHTTPResponse, newErr
	}

	return localVarReturnValue, localVarHTTPResponse, nil
}

// ---------------------------------------------
// ------------------ MOCKS --------------------
// ---------------------------------------------

type AgentRegistrationsApiMock struct {
	AllAgentRegistrationsCalls    *[]AllAgentRegistrationsCall
	AllAgentRegistrationsResponse AllAgentRegistrationsMockResponse
}

func NewAgentRegistrationsApiMock() AgentRegistrationsApiMock {
	xAllAgentRegistrationsCalls := make([]AllAgentRegistrationsCall, 0)
	return AgentRegistrationsApiMock{
		AllAgentRegistrationsCalls: &xAllAgentRegistrationsCalls,
	}
}

type AllAgentRegistrationsMockResponse struct {
	Result   AgentRegistrations
	Response *http.Response
	Error    error
}

type AllAgentRegistrationsCall struct {
}

func (mock AgentRegistrationsApiMock) AllAgentRegistrations(ctx context.Context) ApiAllAgentRegistrationsRequest {
	return ApiAllAgentRegistrationsRequest{
		ApiService: mock,
		ctx:        ctx,
	}
}

func (mock AgentRegistrationsApiMock) AllAgentRegistrationsExecute(r ApiAllAgentRegistrationsRequest) (*AgentRegistrations, *http.Response, error) {
	p := AllAgentRegistrationsCall{}
	*mock.AllAgentRegistrationsCalls = append(*mock.AllAgentRegistrationsCalls, p)
	return &mock.AllAgentRegistrationsResponse.Result, mock.AllAgentRegistrationsResponse.Response, mock.AllAgentRegistrationsResponse.Error
}