package registry

import (
	"strings"

	"github.com/docker/distribution/manifest/schema1"
	"github.com/docker/distribution/manifest/schema2"
)

// Manifest returns the manifest for a specific repository:tag.
func (r *Registry) Manifest(repository, ref string) (interface{}, error) {
	uri := r.url("/v2/%s/manifests/%s", repository, ref)
	r.Logf("registry.manifests uri=%s repository=%s ref=%s", uri, repository, ref)

	var m schema2.Manifest
	h, err := r.getJSON(uri, &m, true)
	if err != nil {
		return m, err
	}

	if !strings.Contains(ref, ":") {
		// we got a tag, get the manifest for the ref
		r.Logf("ref: %s", h.Get("Docker-Content-Digest"))
	}

	if m.Versioned.SchemaVersion == 1 {
		return r.ManifestV1(repository, ref)
	}

	return m, nil
}

// ManifestV1 gets the registry v1 manifest.
func (r *Registry) ManifestV1(repository, ref string) (schema1.SignedManifest, error) {
	uri := r.url("/v2/%s/manifests/%s", repository, ref)
	r.Logf("registry.manifests uri=%s repository=%s ref=%s", uri, repository, ref)

	var m schema1.SignedManifest
	if _, err := r.getJSON(uri, &m, false); err != nil {
		return m, err
	}

	return m, nil
}