service_client.go

package gophercloud

import (
	"io"
	"net/http"
	"strings"
	"sync"
)

// ServiceClient stores details required to interact with a specific service API implemented by a provider.
// Generally, you'll acquire these by calling the appropriate `New` method on a ProviderClient.
type ServiceClient struct {
	// ProviderClient is a reference to the provider that implements this service.
	*ProviderClient

	// Endpoint is the base URL of the service's API, acquired from a service catalog.
	// It MUST end with a /.
	Endpoint string

	// ResourceBase is the base URL shared by the resources within a service's API. It should include
	// the API version and, like Endpoint, MUST end with a / if set. If not set, the Endpoint is used
	// as-is, instead.
	ResourceBase string

	// This is the service client type (e.g. compute, sharev2).
	// NOTE: FOR INTERNAL USE ONLY. DO NOT SET. GOPHERCLOUD WILL SET THIS.
	// It is only exported because it gets set in a different package.
	Type string

	// The microversion of the service to use. Set this to use a particular microversion.
	Microversion string

	// lock,The service needs to be locked when setting the micro version.
	lock sync.Mutex

	// MoreHeaders allows users (or Gophercloud) to set service-wide headers on requests. Put another way,
	// values set in this field will be set on all the HTTP requests the service client sends.
	MoreHeaders map[string]string

	HandleError func(httpStatus int, responseContent string) error
}

// SetMicroversion,Set the service micro version field.
func (client *ServiceClient) SetMicroversion(microversion string) {
	client.lock.Lock()
	client.Microversion = microversion
	client.lock.Unlock()
	return
}

// UnsetMicroversion,Unset the service micro version field.
func (client *ServiceClient) UnsetMicroversion() {
	client.lock.Lock()
	client.Microversion = ""
	client.lock.Unlock()
	return
}

// ResourceBaseURL returns the base URL of any resources used by this service. It MUST end with a /.
func (client *ServiceClient) ResourceBaseURL() string {
	if client.ResourceBase != "" {
		return client.ResourceBase
	}
	return client.Endpoint
}

// ServiceURL constructs a URL for a resource belonging to this provider.
func (client *ServiceClient) ServiceURL(parts ...string) string {
	return client.ResourceBaseURL() + strings.Join(parts, "/")
}

func (client *ServiceClient) initReqOpts(url string, JSONBody interface{}, JSONResponse interface{}, opts *RequestOpts) {
	if v, ok := (JSONBody).(io.Reader); ok {
		opts.RawBody = v
	} else if JSONBody != nil {
		opts.JSONBody = JSONBody
	}

	if JSONResponse != nil {
		opts.JSONResponse = JSONResponse
	}

	if opts.MoreHeaders == nil {
		opts.MoreHeaders = make(map[string]string)
	}

	if client.Microversion != "" {
		client.setMicroversionHeader(opts)
	}

	if &client.ProviderClient.AKSKOptions != nil {
		if client.ProviderClient.AKSKOptions.SecurityToken != "" {
			opts.MoreHeaders["X-Security-Token"] = client.ProviderClient.AKSKOptions.SecurityToken
		}
		if client.ProviderClient.AKSKOptions.ProjectID != "" {
			opts.MoreHeaders["X-Project-Id"] = client.ProjectID
		} else if client.ProviderClient.AKSKOptions.DomainID != "" {
			opts.MoreHeaders["X-Domain-Id"] = client.DomainID
		}
	}
}

// Get calls `Request` with the "GET" HTTP verb.
func (client *ServiceClient) Get(url string, JSONResponse interface{}, opts *RequestOpts) (*http.Response, error) {
	if opts == nil {
		opts = new(RequestOpts)
	}
	client.initReqOpts(url, nil, JSONResponse, opts)
	return client.Request("GET", url, opts)
}

// Post calls `Request` with the "POST" HTTP verb.
func (client *ServiceClient) Post(url string, JSONBody interface{}, JSONResponse interface{}, opts *RequestOpts) (*http.Response, error) {
	if opts == nil {
		opts = new(RequestOpts)
	}
	client.initReqOpts(url, JSONBody, JSONResponse, opts)
	return client.Request("POST", url, opts)
}

// Put calls `Request` with the "PUT" HTTP verb.
func (client *ServiceClient) Put(url string, JSONBody interface{}, JSONResponse interface{}, opts *RequestOpts) (*http.Response, error) {
	if opts == nil {
		opts = new(RequestOpts)
	}
	client.initReqOpts(url, JSONBody, JSONResponse, opts)
	return client.Request("PUT", url, opts)
}

// Patch calls `Request` with the "PATCH" HTTP verb.
func (client *ServiceClient) Patch(url string, JSONBody interface{}, JSONResponse interface{}, opts *RequestOpts) (*http.Response, error) {
	if opts == nil {
		opts = new(RequestOpts)
	}
	client.initReqOpts(url, JSONBody, JSONResponse, opts)
	return client.Request("PATCH", url, opts)
}

// Delete calls `Request` with the "DELETE" HTTP verb.
func (client *ServiceClient) Delete(url string, opts *RequestOpts) (*http.Response, error) {
	if opts == nil {
		opts = new(RequestOpts)
	}
	client.initReqOpts(url, nil, nil, opts)
	return client.Request("DELETE", url, opts)
}

// Head calls `Request` with the "HEAD" HTTP verb.
func (client *ServiceClient) Head(url string, opts *RequestOpts) (*http.Response, error) {
	if opts == nil {
		opts = new(RequestOpts)
	}
	client.initReqOpts(url, nil, nil, opts)
	return client.Request("HEAD", url, opts)
}

func (client *ServiceClient) setMicroversionHeader(opts *RequestOpts) {

	switch client.Type {
	case "compute":

		if client.Microversion <= "2.26" {
			opts.MoreHeaders["X-OpenStack-Nova-API-Version"] = client.Microversion
		} else {
			opts.MoreHeaders["OpenStack-API-Version"] = client.Type + " " + client.Microversion
		}
		//case "sharev2":
		//	opts.MoreHeaders["X-OpenStack-Manila-API-Version"] = client.Microversion
		//case "volume":
		//	opts.MoreHeaders["X-OpenStack-Volume-API-Version"] = client.Microversion
	}
	//
	//if client.Type != "" {
	//	opts.MoreHeaders["OpenStack-API-Version"] = client.Type + " " + client.Microversion
	//}
}

// Request carries out the HTTP operation for the service client
func (client *ServiceClient) Request(method, url string, options *RequestOpts) (*http.Response, error) {
	if len(client.MoreHeaders) > 0 {
		if options == nil {
			options = new(RequestOpts)
		}
		for k, v := range client.MoreHeaders {
			options.MoreHeaders[k] = v
			//if _, ok := options.MoreHeaders[k]; !ok {
				//options.MoreHeaders[k] = v
			//}
		}
	}

	if client.HandleError != nil {
		options.HandleError = client.HandleError
	}
	return client.ProviderClient.Request(method, url, options)
}