Switch kpod load/push/save to use libpod runtime

Since this is the last use of libpod/images/copy.go,
removing that code

Signed-off-by: Daniel J Walsh <dwalsh@redhat.com>
This commit is contained in:
Daniel J Walsh 2017-09-09 11:14:40 +00:00
parent 11f537759f
commit 1392b29a5d
4 changed files with 45 additions and 91 deletions

View file

@ -2,22 +2,14 @@ package main
import ( import (
"io" "io"
"io/ioutil"
"os" "os"
"io/ioutil"
"github.com/containers/storage"
"github.com/kubernetes-incubator/cri-o/libpod/images" "github.com/kubernetes-incubator/cri-o/libpod/images"
"github.com/pkg/errors" "github.com/pkg/errors"
"github.com/urfave/cli" "github.com/urfave/cli"
) )
type loadOptions struct {
input string
quiet bool
image string
}
var ( var (
loadFlags = []cli.Flag{ loadFlags = []cli.Flag{
cli.StringFlag{ cli.StringFlag{
@ -44,14 +36,6 @@ var (
// loadCmd gets the image/file to be loaded from the command line // loadCmd gets the image/file to be loaded from the command line
// and calls loadImage to load the image to containers-storage // and calls loadImage to load the image to containers-storage
func loadCmd(c *cli.Context) error { func loadCmd(c *cli.Context) error {
config, err := getConfig(c)
if err != nil {
return errors.Wrapf(err, "could not get config")
}
store, err := getStore(config)
if err != nil {
return err
}
args := c.Args() args := c.Args()
var image string var image string
@ -62,6 +46,11 @@ func loadCmd(c *cli.Context) error {
return errors.New("too many arguments. Requires exactly 1") return errors.New("too many arguments. Requires exactly 1")
} }
runtime, err := getRuntime(c)
if err != nil {
return errors.Wrapf(err, "could not get runtime")
}
input := c.String("input") input := c.String("input")
if input == "/dev/stdin" { if input == "/dev/stdin" {
@ -93,33 +82,21 @@ func loadCmd(c *cli.Context) error {
} }
} }
opts := loadOptions{ var output io.Writer
input: input, if !c.Bool("quiet") {
quiet: c.Bool("quiet"), output = os.Stdout
image: image,
} }
src := images.DockerArchive + ":" + input
return loadImage(store, opts) if err := runtime.PullImage(src, false, output); err != nil {
} src = images.OCIArchive + ":" + input
// loadImage loads the image from docker-archive or oci to containers-storage
// using the pullImage function
func loadImage(store storage.Store, opts loadOptions) error {
loadOpts := images.CopyOptions{
Quiet: opts.quiet,
Store: store,
}
src := images.DockerArchive + ":" + opts.input
if err := images.PullImage(src, false, loadOpts); err != nil {
src = images.OCIArchive + ":" + opts.input
// generate full src name with specified image:tag // generate full src name with specified image:tag
if opts.image != "" { if image != "" {
src = src + ":" + opts.image src = src + ":" + image
} }
if err := images.PullImage(src, false, loadOpts); err != nil { if err := runtime.PullImage(src, false, output); err != nil {
return errors.Wrapf(err, "error pulling from %q", opts.input) return errors.Wrapf(err, "error pulling %q", src)
} }
} }
return nil return nil
} }

View file

@ -2,12 +2,13 @@ package main
import ( import (
"fmt" "fmt"
"io"
"os" "os"
"github.com/containers/image/types" "github.com/containers/image/types"
"github.com/containers/storage/pkg/archive" "github.com/containers/storage/pkg/archive"
"github.com/kubernetes-incubator/cri-o/libpod"
"github.com/kubernetes-incubator/cri-o/libpod/common" "github.com/kubernetes-incubator/cri-o/libpod/common"
"github.com/kubernetes-incubator/cri-o/libpod/images"
"github.com/pkg/errors" "github.com/pkg/errors"
"github.com/urfave/cli" "github.com/urfave/cli"
"golang.org/x/crypto/ssh/terminal" "golang.org/x/crypto/ssh/terminal"
@ -70,7 +71,6 @@ func pushCmd(c *cli.Context) error {
srcName := c.Args().Get(0) srcName := c.Args().Get(0)
destName := c.Args().Get(1) destName := c.Args().Get(1)
signaturePolicy := c.String("signature-policy")
registryCredsString := c.String("creds") registryCredsString := c.String("creds")
certPath := c.String("cert-dir") certPath := c.String("cert-dir")
skipVerify := !c.BoolT("tls-verify") skipVerify := !c.BoolT("tls-verify")
@ -94,19 +94,19 @@ func pushCmd(c *cli.Context) error {
registryCreds = creds registryCreds = creds
} }
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)
if err != nil {
return err
} }
options := images.CopyOptions{ var writer io.Writer
if !c.Bool("quiet") {
writer = os.Stdout
}
options := libpod.CopyOptions{
Compression: archive.Uncompressed, Compression: archive.Uncompressed,
SignaturePolicyPath: signaturePolicy, SignaturePolicyPath: c.String("signature-policy"),
Store: store,
DockerRegistryOptions: common.DockerRegistryOptions{ DockerRegistryOptions: common.DockerRegistryOptions{
DockerRegistryCreds: registryCreds, DockerRegistryCreds: registryCreds,
DockerCertPath: certPath, DockerCertPath: certPath,
@ -117,8 +117,6 @@ func pushCmd(c *cli.Context) error {
SignBy: signBy, SignBy: signBy,
}, },
} }
if !c.Bool("quiet") {
options.ReportWriter = os.Stderr return runtime.PushImage(srcName, destName, options, writer)
}
return images.PushImage(srcName, destName, options)
} }

View file

@ -1,22 +1,16 @@
package main package main
import ( import (
"io"
"os" "os"
"github.com/containers/storage" "github.com/kubernetes-incubator/cri-o/libpod"
"github.com/kubernetes-incubator/cri-o/libpod/images" "github.com/kubernetes-incubator/cri-o/libpod/images"
"github.com/pkg/errors" "github.com/pkg/errors"
"github.com/sirupsen/logrus" "github.com/sirupsen/logrus"
"github.com/urfave/cli" "github.com/urfave/cli"
) )
type saveOptions struct {
output string
quiet bool
format string
images []string
}
var ( var (
saveFlags = []cli.Flag{ saveFlags = []cli.Flag{
cli.StringFlag{ cli.StringFlag{
@ -54,17 +48,16 @@ func saveCmd(c *cli.Context) error {
return errors.Errorf("need at least 1 argument") return errors.Errorf("need at least 1 argument")
} }
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) var writer io.Writer
if err != nil { if !c.Bool("quiet") {
return err writer = os.Stdout
} }
output := c.String("output") output := c.String("output")
if output == "/dev/stdout" { if output == "/dev/stdout" {
fi := os.Stdout fi := os.Stdout
if logrus.IsTerminal(fi) { if logrus.IsTerminal(fi) {
@ -72,41 +65,27 @@ func saveCmd(c *cli.Context) error {
} }
} }
opts := saveOptions{
output: output,
quiet: c.Bool("quiet"),
format: c.String("format"),
images: args,
}
return saveImage(store, opts)
}
// saveImage pushes the image to docker-archive or oci by
// calling pushImage
func saveImage(store storage.Store, opts saveOptions) error {
var dst string var dst string
switch opts.format { switch c.String("format") {
case images.OCIArchive: case images.OCIArchive:
dst = images.OCIArchive + ":" + opts.output dst = images.OCIArchive + ":" + output
case images.DockerArchive: case images.DockerArchive:
fallthrough fallthrough
case "": case "":
dst = images.DockerArchive + ":" + opts.output dst = images.DockerArchive + ":" + output
default: default:
return errors.Errorf("unknown format option %q", opts.format) return errors.Errorf("unknown format option %q", c.String("format"))
} }
saveOpts := images.CopyOptions{ saveOpts := libpod.CopyOptions{
SignaturePolicyPath: "", SignaturePolicyPath: "",
Store: store,
} }
// only one image is supported for now // only one image is supported for now
// future pull requests will fix this // future pull requests will fix this
for _, image := range opts.images { for _, image := range args {
dest := dst + ":" + image dest := dst + ":" + image
if err := images.PushImage(image, dest, saveOpts); err != nil { if err := runtime.PushImage(image, dest, saveOpts, writer); err != nil {
return errors.Wrapf(err, "unable to save %q", image) return errors.Wrapf(err, "unable to save %q", image)
} }
} }

View file

@ -50,7 +50,7 @@ func addImageNames(runtime *libpod.Runtime, image *storage.Image, addNames []str
} }
for _, name := range names { for _, name := range names {
if err := runtime.TagImage(image, name); err != nil { if err := runtime.TagImage(image, name); err != nil {
return errors.Wrapf(err, "error adding names (%v) to image %q", name, image.ID) return errors.Wrapf(err, "error adding name (%v) to image %q", name, image.ID)
} }
} }
return nil return nil