Merge pull request #639 from soulseen/improve/search_image

add docker image search api

pkg/apiserver/registries/registries.go

@@ -24,6 +24,9 @@ import (
 
 	"kubesphere.io/kubesphere/pkg/errors"
 	"kubesphere.io/kubesphere/pkg/models/registries"
+
+	k8serror "k8s.io/apimachinery/pkg/api/errors"
+	log "k8s.io/klog"
 )
 
 func RegistryVerify(request *restful.Request, response *restful.Response) {
@@ -46,3 +49,79 @@ func RegistryVerify(request *restful.Request, response *restful.Response) {
 
 	response.WriteAsJson(errors.None)
 }
+
+func RegistryImageBlob(request *restful.Request, response *restful.Response) {
+	imageName := request.QueryParameter("image")
+	namespace := request.QueryParameter("namespace")
+	secretName := request.QueryParameter("secret")
+
+	// get entry
+	entry, err := registries.GetEntryBySecret(namespace, secretName)
+	if err != nil {
+		log.Errorf("%+v", err)
+		if k8serror.IsNotFound(err) {
+			log.Errorf("%+v", err)
+			errors.ParseSvcErr(restful.NewError(http.StatusBadRequest, err.Error()), response)
+			return
+		}
+		response.WriteAsJson(&registries.ImageDetails{Status: registries.StatusFailed, Message: err.Error()})
+		return
+	}
+
+	// parse image
+	image, err := registries.ParseImage(imageName)
+	if err != nil {
+		log.Errorf("%+v", err)
+		errors.ParseSvcErr(restful.NewError(http.StatusBadRequest, err.Error()), response)
+		return
+	}
+
+	// Create the registry client.
+	r, err := registries.CreateRegistryClient(entry.Username, entry.Password, image.Domain)
+	if err != nil {
+		log.Errorf("%+v", err)
+		response.WriteAsJson(&registries.ImageDetails{Status: registries.StatusFailed, Message: err.Error()})
+		return
+	}
+
+	digestUrl := r.GetDigestUrl(image)
+
+	// Get token.
+	token, err := r.Token(digestUrl)
+	if err != nil {
+		log.Errorf("%+v", err)
+		response.WriteAsJson(&registries.ImageDetails{Status: registries.StatusFailed, Message: err.Error()})
+		return
+	}
+
+	// Get digest.
+	imageManifest, err := r.ImageManifest(image, token)
+	if err != nil {
+		if serviceError, ok := err.(restful.ServiceError); ok {
+			response.WriteAsJson(&registries.ImageDetails{Status: registries.StatusFailed, Message: serviceError.Message})
+			return
+		}
+		log.Errorf("%+v", err)
+		response.WriteAsJson(&registries.ImageDetails{Status: registries.StatusFailed, Message: err.Error()})
+		return
+	}
+	image.Digest = imageManifest.ManifestConfig.Digest
+
+	// Get blob.
+	imageBlob, err := r.ImageBlob(image, token)
+	if err != nil {
+		log.Errorf("%+v", err)
+		response.WriteAsJson(&registries.ImageDetails{Status: registries.StatusFailed, Message: err.Error()})
+		return
+	}
+
+	imageDetails := &registries.ImageDetails{
+		Status:        registries.StatusSuccess,
+		ImageManifest: imageManifest,
+		ImageBlob:     imageBlob,
+		ImageTag:      image.Tag,
+		Registry:      image.Domain,
+	}
+
+	response.WriteAsJson(imageDetails)
+}