Modify kpod tag to use libpod runtime interface

Signed-off-by: Daniel J Walsh <dwalsh@redhat.com>
This commit is contained in:
Daniel J Walsh 2017-09-06 16:51:29 +00:00 committed by umohnani8
parent 0ff3580f05
commit 8949e669c9
2 changed files with 73 additions and 18 deletions

View file

@ -3,7 +3,7 @@ package main
import ( import (
"github.com/containers/image/docker/reference" "github.com/containers/image/docker/reference"
"github.com/containers/storage" "github.com/containers/storage"
"github.com/kubernetes-incubator/cri-o/libpod/images" "github.com/kubernetes-incubator/cri-o/libpod"
"github.com/pkg/errors" "github.com/pkg/errors"
"github.com/urfave/cli" "github.com/urfave/cli"
) )
@ -24,37 +24,34 @@ func tagCmd(c *cli.Context) error {
if len(args) < 2 { if len(args) < 2 {
return errors.Errorf("image name and at least one new name must be specified") return errors.Errorf("image name and at least one new name must be specified")
} }
config, err := getConfig(c) runtime, err := getRuntime(c)
if err != nil { if err != nil {
return errors.Wrapf(err, "Could not get config") return errors.Wrapf(err, "could not create runtime")
} }
store, err := getStore(config) img, err := runtime.GetImage(args[0])
if err != nil {
return err
}
img, err := images.FindImage(store, args[0])
if err != nil { if err != nil {
return err return err
} }
if img == nil { if img == nil {
return errors.New("null image") return errors.New("null image")
} }
err = addImageNames(store, img, args[1:]) err = addImageNames(runtime, img, args[1:])
if err != nil { if err != nil {
return errors.Wrapf(err, "error adding names %v to image %q", args[1:], args[0]) return errors.Wrapf(err, "error adding names %v to image %q", args[1:], args[0])
} }
return nil return nil
} }
func addImageNames(store storage.Store, image *storage.Image, addNames []string) error { func addImageNames(runtime *libpod.Runtime, image *storage.Image, addNames []string) error {
// Add tags to the names if applicable // Add tags to the names if applicable
names, err := expandedTags(addNames) names, err := expandedTags(addNames)
if err != nil { if err != nil {
return err return err
} }
err = store.SetNames(image.ID, append(image.Names, names...)) for _, name := range names {
if err != nil { if err := runtime.TagImage(image, name); err != nil {
return errors.Wrapf(err, "error adding names (%v) to image %q", names, image.ID) return errors.Wrapf(err, "error adding names (%v) to image %q", name, image.ID)
}
} }
return nil return nil
} }

View file

@ -11,6 +11,7 @@ import (
"github.com/containers/image/signature" "github.com/containers/image/signature"
is "github.com/containers/image/storage" is "github.com/containers/image/storage"
"github.com/containers/image/transports/alltransports" "github.com/containers/image/transports/alltransports"
"github.com/containers/image/types"
"github.com/containers/storage" "github.com/containers/storage"
"github.com/containers/storage/pkg/archive" "github.com/containers/storage/pkg/archive"
"github.com/kubernetes-incubator/cri-o/libpod/common" "github.com/kubernetes-incubator/cri-o/libpod/common"
@ -41,6 +42,9 @@ type CopyOptions struct {
// strip or add signatures to the image when pushing (uploading) the // strip or add signatures to the image when pushing (uploading) the
// image to a registry. // image to a registry.
common.SigningOptions common.SigningOptions
// SigningPolicyPath this points to a alternative signature policy file, used mainly for testing
SignaturePolicyPath string
} }
// Image API // Image API
@ -153,7 +157,7 @@ func (r *Runtime) PushImage(source string, destination string, options CopyOptio
defer policyContext.Destroy() defer policyContext.Destroy()
// Look up the image name and its layer, then build the imagePushData from // Look up the image name and its layer, then build the imagePushData from
// the image // the image
img, err := images.FindImage(r.store, source) img, err := r.GetImage(source)
if err != nil { if err != nil {
return errors.Wrapf(err, "error locating image %q for importing settings", source) return errors.Wrapf(err, "error locating image %q for importing settings", source)
} }
@ -183,25 +187,79 @@ func (r *Runtime) PushImage(source string, destination string, options CopyOptio
// TagImage adds a tag to the given image // TagImage adds a tag to the given image
func (r *Runtime) TagImage(image *storage.Image, tag string) error { func (r *Runtime) TagImage(image *storage.Image, tag string) error {
return ctr.ErrNotImplemented tags, err := r.store.Names(image.ID)
if err != nil {
return err
}
for _, key := range tags {
if key == tag {
return nil
}
}
tags = append(tags, tag)
return r.store.SetNames(image.ID, tags)
} }
// UntagImage removes a tag from the given image // UntagImage removes a tag from the given image
func (r *Runtime) UntagImage(image *storage.Image, tag string) error { func (r *Runtime) UntagImage(image *storage.Image, tag string) error {
return ctr.ErrNotImplemented tags, err := r.store.Names(image.ID)
if err != nil {
return err
}
for i, key := range tags {
if key == tag {
tags[i] = tags[len(tags)-1]
tags = tags[:len(tags)-1]
break
}
}
return r.store.SetNames(image.ID, tags)
} }
// RemoveImage deletes an image from local storage // RemoveImage deletes an image from local storage
// Images being used by running containers cannot be removed // Images being used by running containers cannot be removed
func (r *Runtime) RemoveImage(image *storage.Image) error { func (r *Runtime) RemoveImage(image *storage.Image) error {
return ctr.ErrNotImplemented _, err := r.store.DeleteImage(image.ID, false)
return err
} }
// GetImage retrieves an image matching the given name or hash from system // GetImage retrieves an image matching the given name or hash from system
// storage // storage
// If no matching image can be found, an error is returned // If no matching image can be found, an error is returned
func (r *Runtime) GetImage(image string) (*storage.Image, error) { func (r *Runtime) GetImage(image string) (*storage.Image, error) {
return nil, ctr.ErrNotImplemented var img *storage.Image
ref, err := is.Transport.ParseStoreReference(r.store, image)
if err == nil {
img, err = is.Transport.GetStoreImage(r.store, ref)
}
if err != nil {
img2, err2 := r.store.Image(image)
if err2 != nil {
if ref == nil {
return nil, errors.Wrapf(err, "error parsing reference to image %q", image)
}
return nil, errors.Wrapf(err, "unable to locate image %q", image)
}
img = img2
}
return img, nil
}
// GetImageRef searches for and returns a new types.Image matching the given name or ID in the given store.
func (r *Runtime) GetImageRef(image string) (types.Image, error) {
img, err := r.GetImage(image)
if err != nil {
return nil, errors.Wrapf(err, "unable to locate image %q", image)
}
ref, err := is.Transport.ParseStoreReference(r.store, "@"+img.ID)
if err != nil {
return nil, errors.Wrapf(err, "error parsing reference to image %q", img.ID)
}
imgRef, err := ref.NewImage(nil)
if err != nil {
return nil, errors.Wrapf(err, "error reading image %q", img.ID)
}
return imgRef, nil
} }
// GetImages retrieves all images present in storage // GetImages retrieves all images present in storage