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>
95 lines
2.6 KiB
Go
95 lines
2.6 KiB
Go
package container
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"net"
|
|
"sort"
|
|
"strconv"
|
|
"strings"
|
|
|
|
"github.com/docker/cli/cli"
|
|
"github.com/docker/cli/cli/command"
|
|
"github.com/docker/cli/cli/command/completion"
|
|
"github.com/fvbommel/sortorder"
|
|
"github.com/moby/moby/api/types/container"
|
|
"github.com/pkg/errors"
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
type portOptions struct {
|
|
container string
|
|
|
|
port string
|
|
}
|
|
|
|
// newPortCommand creates a new cobra.Command for "docker container port".
|
|
func newPortCommand(dockerCLI command.Cli) *cobra.Command {
|
|
var opts portOptions
|
|
|
|
cmd := &cobra.Command{
|
|
Use: "port CONTAINER [PRIVATE_PORT[/PROTO]]",
|
|
Short: "List port mappings or a specific mapping for the container",
|
|
Args: cli.RequiresRangeArgs(1, 2),
|
|
RunE: func(cmd *cobra.Command, args []string) error {
|
|
opts.container = args[0]
|
|
if len(args) > 1 {
|
|
opts.port = args[1]
|
|
}
|
|
return runPort(cmd.Context(), dockerCLI, &opts)
|
|
},
|
|
Annotations: map[string]string{
|
|
"aliases": "docker container port, docker port",
|
|
},
|
|
ValidArgsFunction: completion.ContainerNames(dockerCLI, false),
|
|
DisableFlagsInUseLine: true,
|
|
}
|
|
return cmd
|
|
}
|
|
|
|
// runPort shows the port mapping for a given container. Optionally, it
|
|
// allows showing the mapping for a specific (container)port and proto.
|
|
//
|
|
// TODO(thaJeztah): currently this defaults to show the TCP port if no
|
|
// proto is specified. We should consider changing this to "any" protocol
|
|
// for the given private port.
|
|
func runPort(ctx context.Context, dockerCli command.Cli, opts *portOptions) error {
|
|
c, err := dockerCli.Client().ContainerInspect(ctx, opts.container)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
var out []string
|
|
if opts.port != "" {
|
|
port, proto, _ := strings.Cut(opts.port, "/")
|
|
if proto == "" {
|
|
proto = "tcp"
|
|
}
|
|
if _, err = strconv.ParseUint(port, 10, 16); err != nil {
|
|
return errors.Wrapf(err, "Error: invalid port (%s)", port)
|
|
}
|
|
frontends, exists := c.NetworkSettings.Ports[container.PortRangeProto(port+"/"+proto)]
|
|
if !exists || len(frontends) == 0 {
|
|
return errors.Errorf("Error: No public port '%s' published for %s", opts.port, opts.container)
|
|
}
|
|
for _, frontend := range frontends {
|
|
out = append(out, net.JoinHostPort(frontend.HostIP, frontend.HostPort))
|
|
}
|
|
} else {
|
|
for from, frontends := range c.NetworkSettings.Ports {
|
|
for _, frontend := range frontends {
|
|
out = append(out, fmt.Sprintf("%s -> %s", from, net.JoinHostPort(frontend.HostIP, frontend.HostPort)))
|
|
}
|
|
}
|
|
}
|
|
|
|
if len(out) > 0 {
|
|
sort.Slice(out, func(i, j int) bool {
|
|
return sortorder.NaturalLess(out[i], out[j])
|
|
})
|
|
_, _ = fmt.Fprintln(dockerCli.Out(), strings.Join(out, "\n"))
|
|
}
|
|
|
|
return nil
|
|
}
|