inject new parameter for image resolution into kubectl set image

Author: deads2k

pkg/oc/cli/cmd/set/set.go

@@ -96,9 +96,6 @@ func NewCmdImage(fullName string, f *clientcmd.Factory, out, err io.Writer) *cob
 	cmd.Long = setImageLong
 	cmd.Example = fmt.Sprintf(setImageExample, fullName)
 
-	flags := cmd.Flags()
-	f.ImageResolutionOptions().Bind(flags)
-
 	return cmd
 }
 

pkg/oc/cli/util/clientcmd/factory_client_access.go

@@ -24,13 +24,14 @@ import (
 	kapi "k8s.io/kubernetes/pkg/apis/core"
 	kclientset "k8s.io/kubernetes/pkg/client/clientset_generated/internalclientset"
 	"k8s.io/kubernetes/pkg/kubectl"
+	"k8s.io/kubernetes/pkg/kubectl/cmd/set"
 	kcmdutil "k8s.io/kubernetes/pkg/kubectl/cmd/util"
 	"k8s.io/kubernetes/pkg/kubectl/resource"
 	"k8s.io/kubernetes/pkg/kubectl/util/transport"
 
 	appsapiv1 "github.com/openshift/api/apps/v1"
 	appsapi "github.com/openshift/origin/pkg/apps/apis/apps"
-	imageclient "github.com/openshift/origin/pkg/image/generated/internalclientset"
+	imageapi "github.com/openshift/origin/pkg/image/apis/image"
 	deploymentcmd "github.com/openshift/origin/pkg/oc/cli/deploymentconfigs"
 	routegen "github.com/openshift/origin/pkg/route/generator"
 )
@@ -39,7 +40,6 @@ type ring0Factory struct {
 	*OpenshiftCLIClientBuilder
 
 	clientConfig            kclientcmd.ClientConfig
-	imageResolutionOptions  FlagBinder
 	kubeClientAccessFactory kcmdutil.ClientAccessFactory
 }
 
@@ -48,22 +48,21 @@ type ClientAccessFactory interface {
 	CLIClientBuilder
 
 	OpenShiftClientConfig() kclientcmd.ClientConfig
-	ImageResolutionOptions() FlagBinder
 }
 
 func NewClientAccessFactory(optionalClientConfig kclientcmd.ClientConfig) ClientAccessFactory {
 	// if we call this factory construction method, we want the openshift style config loading
 	kclientcmd.UseOpenShiftKubeConfigValues = true
 	kclientcmd.ErrEmptyConfig = kclientcmd.NewErrConfigurationMissing()
+	set.ParseDockerImageReferenceToStringFunc = ParseDockerImageReferenceToStringFunc
 
 	flags := pflag.NewFlagSet("", pflag.ContinueOnError)
 	clientConfig := optionalClientConfig
 	if optionalClientConfig == nil {
 		clientConfig = kcmdutil.DefaultClientConfig(flags)
 	}
 	factory := &ring0Factory{
-		clientConfig:           clientConfig,
-		imageResolutionOptions: &imageResolutionOptions{},
+		clientConfig: clientConfig,
 	}
 	factory.kubeClientAccessFactory = kcmdutil.NewClientAccessFactoryFromDiscovery(
 		flags,
@@ -276,49 +275,8 @@ func (f *ring0Factory) Pauser(info *resource.Info) ([]byte, error) {
 	}
 }
 
-// ImageResolutionOptions provides the "--source" flag to commands that deal with images
-// and need to provide extra capabilities for working with ImageStreamTags and
-// ImageStreamImages.
-type imageResolutionOptions struct {
-	bound  bool
-	Source string
-}
-
-func (o *imageResolutionOptions) Bound() bool {
-	return o.bound
-}
-
-func (o *imageResolutionOptions) Bind(f *pflag.FlagSet) {
-	if o.Bound() {
-		return
-	}
-	f.StringVarP(&o.Source, "source", "", "docker", "The image source type; valid types are 'imagestreamtag', 'istag', 'imagestreamimage', 'isimage', and 'docker'")
-	o.bound = true
-}
-
-func (f *ring0Factory) ImageResolutionOptions() FlagBinder {
-	return f.imageResolutionOptions
-}
-
 func (f *ring0Factory) ResolveImage(image string) (string, error) {
-	options := f.imageResolutionOptions.(*imageResolutionOptions)
-	if isDockerImageSource(options.Source) {
-		return f.kubeClientAccessFactory.ResolveImage(image)
-	}
-	config, err := f.OpenShiftClientConfig().ClientConfig()
-	if err != nil {
-		return "", err
-	}
-	imageClient, err := imageclient.NewForConfig(config)
-	if err != nil {
-		return "", err
-	}
-	namespace, _, err := f.DefaultNamespace()
-	if err != nil {
-		return "", err
-	}
-
-	return resolveImagePullSpec(imageClient.Image(), options.Source, image, namespace)
+	return f.kubeClientAccessFactory.ResolveImage(image)
 }
 
 func (f *ring0Factory) Resumer(info *resource.Info) ([]byte, error) {
@@ -416,3 +374,11 @@ func getProtocols(spec kapi.PodSpec) map[string]string {
 	}
 	return result
 }
+
+func ParseDockerImageReferenceToStringFunc(spec string) (string, error) {
+	ret, err := imageapi.ParseDockerImageReference(spec)
+	if err != nil {
+		return "", err
+	}
+	return ret.String(), nil
+}