114 lines
2.4 KiB
Go
114 lines
2.4 KiB
Go
package main
|
|
|
|
import (
|
|
"os"
|
|
|
|
"github.com/containers/storage"
|
|
"github.com/kubernetes-incubator/cri-o/libpod/images"
|
|
"github.com/pkg/errors"
|
|
"github.com/sirupsen/logrus"
|
|
"github.com/urfave/cli"
|
|
)
|
|
|
|
type saveOptions struct {
|
|
output string
|
|
quiet bool
|
|
format string
|
|
images []string
|
|
}
|
|
|
|
var (
|
|
saveFlags = []cli.Flag{
|
|
cli.StringFlag{
|
|
Name: "output, o",
|
|
Usage: "Write to a file, default is STDOUT",
|
|
Value: "/dev/stdout",
|
|
},
|
|
cli.BoolFlag{
|
|
Name: "quiet, q",
|
|
Usage: "Suppress the output",
|
|
},
|
|
cli.StringFlag{
|
|
Name: "format",
|
|
Usage: "Save image to oci-archive",
|
|
},
|
|
}
|
|
saveDescription = `
|
|
Save an image to docker-archive or oci-archive on the local machine.
|
|
Default is docker-archive`
|
|
|
|
saveCommand = cli.Command{
|
|
Name: "save",
|
|
Usage: "Save image to an archive",
|
|
Description: saveDescription,
|
|
Flags: saveFlags,
|
|
Action: saveCmd,
|
|
ArgsUsage: "",
|
|
}
|
|
)
|
|
|
|
// saveCmd saves the image to either docker-archive or oci
|
|
func saveCmd(c *cli.Context) error {
|
|
args := c.Args()
|
|
if len(args) == 0 {
|
|
return errors.Errorf("need at least 1 argument")
|
|
}
|
|
|
|
config, err := getConfig(c)
|
|
if err != nil {
|
|
return errors.Wrapf(err, "could not get config")
|
|
}
|
|
store, err := getStore(config)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
output := c.String("output")
|
|
|
|
if output == "/dev/stdout" {
|
|
fi := os.Stdout
|
|
if logrus.IsTerminal(fi) {
|
|
return errors.Errorf("refusing to save to terminal. Use -o flag or redirect")
|
|
}
|
|
}
|
|
|
|
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
|
|
switch opts.format {
|
|
case images.OCIArchive:
|
|
dst = images.OCIArchive + ":" + opts.output
|
|
case images.DockerArchive:
|
|
fallthrough
|
|
case "":
|
|
dst = images.DockerArchive + ":" + opts.output
|
|
default:
|
|
return errors.Errorf("unknown format option %q", opts.format)
|
|
}
|
|
|
|
saveOpts := images.CopyOptions{
|
|
SignaturePolicyPath: "",
|
|
Store: store,
|
|
}
|
|
|
|
// only one image is supported for now
|
|
// future pull requests will fix this
|
|
for _, image := range opts.images {
|
|
dest := dst + ":" + image
|
|
if err := images.PushImage(image, dest, saveOpts); err != nil {
|
|
return errors.Wrapf(err, "unable to save %q", image)
|
|
}
|
|
}
|
|
return nil
|
|
}
|