docker-ls/lib/registry_api.go

package lib

import (
	"net/http"
	"net/url"
	"strconv"
)

type registryApi struct {
	cfg       Config
	connector *registryConnector
}

func (r *registryApi) endpointUrl(path string) *url.URL {
	url := r.cfg.registryUrl

	url.Path = path

	return &url
}

func (r *registryApi) paginatedRequestEndpointUrl(path string, lastApiResponse *http.Response) (url *url.URL, err error) {
	url = r.endpointUrl(path)

	if lastApiResponse != nil {
		linkHeader := lastApiResponse.Header.Get("link")

		if linkHeader != "" {
			// This is a hack to work around what looks like a bug in the registry:
			// the supplied link URL currently lacks scheme and host
			scheme, host := url.Scheme, url.Host

			url, err = parseLinkToNextHeader(linkHeader)

			if err != nil {
				return
			}

			if url.Scheme == "" {
				url.Scheme = scheme
			}

			if url.Host == "" {
				url.Host = host
			}
		}
	} else {
		queryParams := url.Query()
		queryParams.Set("n", strconv.Itoa(int(r.pageSize())))
		url.RawQuery = queryParams.Encode()
	}

	return
}

func (r *registryApi) pageSize() uint {
	return r.cfg.pageSize
}

func NewRegistryApi(cfg Config) (registry RegistryApi) {
	registry = &registryApi{
		cfg:       cfg,
		connector: NewRegistryConnector(cfg),
	}

	return
}