0adaf6be3b
This replaces the visitAll recursive function with a test that verifies that the option is set for all commands and subcommands, so that it doesn't have to be modified at runtime. We currently still have to loop over all functions for the setValidateArgs call, but that can be looked at separately. Signed-off-by: Sebastiaan van Stijn <github@gone.nl>
88 lines
2.3 KiB
Go
88 lines
2.3 KiB
Go
package context
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"io"
|
|
"os"
|
|
|
|
"github.com/docker/cli/cli"
|
|
"github.com/docker/cli/cli/command"
|
|
"github.com/docker/cli/cli/context/store"
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
// ExportOptions are the options used for exporting a context
|
|
//
|
|
// Deprecated: this type was for internal use and will be removed in the next release.
|
|
type ExportOptions struct {
|
|
ContextName string
|
|
Dest string
|
|
}
|
|
|
|
func newExportCommand(dockerCLI command.Cli) *cobra.Command {
|
|
return &cobra.Command{
|
|
Use: "export [OPTIONS] CONTEXT [FILE|-]",
|
|
Short: "Export a context to a tar archive FILE or a tar stream on STDOUT.",
|
|
Args: cli.RequiresRangeArgs(1, 2),
|
|
RunE: func(cmd *cobra.Command, args []string) error {
|
|
contextName := args[0]
|
|
var dest string
|
|
if len(args) == 2 {
|
|
dest = args[1]
|
|
} else {
|
|
dest = contextName + ".dockercontext"
|
|
}
|
|
return runExport(dockerCLI, contextName, dest)
|
|
},
|
|
ValidArgsFunction: completeContextNames(dockerCLI, 1, true),
|
|
DisableFlagsInUseLine: true,
|
|
}
|
|
}
|
|
|
|
func writeTo(dockerCli command.Cli, reader io.Reader, dest string) error {
|
|
var writer io.Writer
|
|
var printDest bool
|
|
if dest == "-" {
|
|
if dockerCli.Out().IsTerminal() {
|
|
return errors.New("cowardly refusing to export to a terminal, specify a file path")
|
|
}
|
|
writer = dockerCli.Out()
|
|
} else {
|
|
f, err := os.OpenFile(dest, os.O_RDWR|os.O_CREATE|os.O_EXCL, 0o600)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer f.Close()
|
|
writer = f
|
|
printDest = true
|
|
}
|
|
if _, err := io.Copy(writer, reader); err != nil {
|
|
return err
|
|
}
|
|
if printDest {
|
|
fmt.Fprintf(dockerCli.Err(), "Written file %q\n", dest)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// RunExport exports a Docker context
|
|
//
|
|
// Deprecated: this function was for internal use and will be removed in the next release.
|
|
func RunExport(dockerCli command.Cli, opts *ExportOptions) error {
|
|
if opts == nil {
|
|
opts = &ExportOptions{}
|
|
}
|
|
return runExport(dockerCli, opts.ContextName, opts.Dest)
|
|
}
|
|
|
|
// runExport exports a Docker context.
|
|
func runExport(dockerCLI command.Cli, contextName string, dest string) error {
|
|
if err := store.ValidateContextName(contextName); err != nil && contextName != command.DefaultContextName {
|
|
return err
|
|
}
|
|
reader := store.Export(contextName, dockerCLI.ContextStore())
|
|
defer reader.Close()
|
|
return writeTo(dockerCLI, reader, dest)
|
|
}
|