add accntcli

pkg/repocli/referer.go

@@ -0,0 +1,100 @@
+package repocli
+
+import (
+	"net/url"
+	"strings"
+)
+
+type Referer struct {
+	urlobj     *url.URL
+	user, pass string
+	base       string
+}
+
+func NewReferer(rawrepo string) (*Referer, error) {
+	repo := &Referer{}
+	if !strings.Contains(rawrepo, "://") {
+		rawrepo = "https://" + rawrepo
+	}
+	urlobj, err := url.Parse(rawrepo)
+	if err != nil {
+		return repo, err
+	}
+	if urlobj.User != nil {
+		repo.user = urlobj.User.Username()
+		repo.pass, _ = urlobj.User.Password()
+		urlobj.User = nil
+	}
+	repo.urlobj = urlobj
+	repo.base = repo.urlobj.Path
+	repo.urlobj.Path = "/"
+	repo.urlobj = urlobj
+
+	return repo, err
+}
+
+func (repo *Referer) String() string {
+	curl := repo.urlobj.JoinPath(repo.base)
+	return curl.String()
+}
+
+func (repo *Referer) Manifest(tag string) string {
+	curl := repo.urlobj.JoinPath("/v2", repo.base, "/manifests", tag)
+	return curl.String()
+}
+
+func (repo *Referer) Blob(digest string) string {
+	curl := repo.urlobj.JoinPath("/v2", repo.base, "/blobs", digest)
+	return curl.String()
+}
+
+func (repo *Referer) Upload() string {
+	curl := repo.urlobj.JoinPath("/v2", repo.base, "/blobs/uploads/")
+	return curl.String()
+}
+
+func (repo *Referer) Patch(loc string) (string, error) {
+	var curl *url.URL
+	var out string
+	var err error
+	if isUUID(loc) {
+		curl = repo.urlobj.JoinPath("/v2/", repo.base, "/blobs/uploads/", loc)
+		return curl.String(), nil
+	}
+	if strings.Contains(loc, "://") {
+		curl, err = url.Parse(loc)
+		if err != nil {
+			return out, err
+		}
+	} else {
+		curl = repo.urlobj.JoinPath(loc)
+	}
+	out = curl.String()
+	return out, err
+}
+
+func (repo *Referer) Put(loc, digest string) (string, error) {
+	var curl *url.URL
+	var out string
+	var err error
+
+	if isUUID(loc) {
+		curl = repo.urlobj.JoinPath("/v2/", repo.base, "/blobs/uploads/", loc)
+	} else if strings.Contains(loc, "://") {
+		curl, err = url.Parse(loc)
+		if err != nil {
+			return out, err
+		}
+	} else {
+		curl = repo.urlobj.JoinPath(loc)
+	}
+	query := curl.Query()
+	query.Set("digest", digest)
+	curl.RawQuery = query.Encode()
+	out = curl.String()
+	return out, err
+}
+
+func (repo *Referer) Userinfo() (string, string) {
+	return repo.user, repo.pass
+}