mirror of
https://github.com/moby/moby.git
synced 2022-11-09 12:21:53 -05:00
5ee5e2452f
Also supporting name, id & partial-id lookups for all the network commands Signed-off-by: Madhu Venugopal <madhu@docker.com>
189 lines
4.9 KiB
Go
189 lines
4.9 KiB
Go
package client
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/json"
|
|
"fmt"
|
|
"io"
|
|
"net/http"
|
|
|
|
flag "github.com/docker/docker/pkg/mflag"
|
|
)
|
|
|
|
const (
|
|
nullNetType = "null"
|
|
)
|
|
|
|
type command struct {
|
|
name string
|
|
description string
|
|
}
|
|
|
|
var (
|
|
networkCommands = []command{
|
|
{"create", "Create a network"},
|
|
{"rm", "Remove a network"},
|
|
{"ls", "List all networks"},
|
|
{"info", "Display information of a network"},
|
|
}
|
|
)
|
|
|
|
// CmdNetwork handles the root Network UI
|
|
func (cli *NetworkCli) CmdNetwork(chain string, args ...string) error {
|
|
cmd := cli.Subcmd(chain, "network", "COMMAND [OPTIONS] [arg...]", networkUsage(chain), false)
|
|
cmd.Require(flag.Min, 1)
|
|
err := cmd.ParseFlags(args, true)
|
|
if err == nil {
|
|
cmd.Usage()
|
|
return fmt.Errorf("invalid command : %v", args)
|
|
}
|
|
return err
|
|
}
|
|
|
|
// CmdNetworkCreate handles Network Create UI
|
|
func (cli *NetworkCli) CmdNetworkCreate(chain string, args ...string) error {
|
|
cmd := cli.Subcmd(chain, "create", "NETWORK-NAME", "Creates a new network with a name specified by the user", false)
|
|
flDriver := cmd.String([]string{"d", "-driver"}, "null", "Driver to manage the Network")
|
|
cmd.Require(flag.Min, 1)
|
|
err := cmd.ParseFlags(args, true)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if *flDriver == "" {
|
|
*flDriver = nullNetType
|
|
}
|
|
|
|
nc := networkCreate{Name: cmd.Arg(0), NetworkType: *flDriver}
|
|
|
|
obj, _, err := readBody(cli.call("POST", "/networks", nc, nil))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if _, err := io.Copy(cli.out, bytes.NewReader(obj)); err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// CmdNetworkRm handles Network Delete UI
|
|
func (cli *NetworkCli) CmdNetworkRm(chain string, args ...string) error {
|
|
cmd := cli.Subcmd(chain, "rm", "NETWORK", "Deletes a network", false)
|
|
cmd.Require(flag.Min, 1)
|
|
err := cmd.ParseFlags(args, true)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
id, err := lookupNetworkID(cli, cmd.Arg(0))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
obj, _, err := readBody(cli.call("DELETE", "/networks/"+id, nil, nil))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if _, err := io.Copy(cli.out, bytes.NewReader(obj)); err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// CmdNetworkLs handles Network List UI
|
|
func (cli *NetworkCli) CmdNetworkLs(chain string, args ...string) error {
|
|
cmd := cli.Subcmd(chain, "ls", "", "Lists all the networks created by the user", false)
|
|
err := cmd.ParseFlags(args, true)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
obj, _, err := readBody(cli.call("GET", "/networks", nil, nil))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if _, err := io.Copy(cli.out, bytes.NewReader(obj)); err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// CmdNetworkInfo handles Network Info UI
|
|
func (cli *NetworkCli) CmdNetworkInfo(chain string, args ...string) error {
|
|
cmd := cli.Subcmd(chain, "info", "NETWORK", "Displays detailed information on a network", false)
|
|
cmd.Require(flag.Min, 1)
|
|
err := cmd.ParseFlags(args, true)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
id, err := lookupNetworkID(cli, cmd.Arg(0))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
obj, _, err := readBody(cli.call("GET", "/networks/"+id, nil, nil))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if _, err := io.Copy(cli.out, bytes.NewReader(obj)); err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// Helper function to predict if a string is a name or id or partial-id
|
|
// This provides a best-effort mechanism to identify a id with the help of GET Filter APIs
|
|
// Being a UI, its most likely that name will be used by the user, which is used to lookup
|
|
// the corresponding ID. If ID is not found, this function will assume that the passed string
|
|
// is an ID by itself.
|
|
|
|
func lookupNetworkID(cli *NetworkCli, nameID string) (string, error) {
|
|
obj, statusCode, err := readBody(cli.call("GET", "/networks?name="+nameID, nil, nil))
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
if statusCode != http.StatusOK {
|
|
return "", fmt.Errorf("name query failed for %s due to : statuscode(%d) %v", nameID, statusCode, string(obj))
|
|
}
|
|
|
|
var list []*networkResource
|
|
err = json.Unmarshal(obj, &list)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
if len(list) > 0 {
|
|
// name query filter will always return a single-element collection
|
|
return list[0].ID, nil
|
|
}
|
|
|
|
// Check for Partial-id
|
|
obj, statusCode, err = readBody(cli.call("GET", "/networks?partial-id="+nameID, nil, nil))
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
if statusCode != http.StatusOK {
|
|
return "", fmt.Errorf("partial-id match query failed for %s due to : statuscode(%d) %v", nameID, statusCode, string(obj))
|
|
}
|
|
|
|
err = json.Unmarshal(obj, &list)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
if len(list) == 0 {
|
|
return "", fmt.Errorf("resource not found %s", nameID)
|
|
}
|
|
if len(list) > 1 {
|
|
return "", fmt.Errorf("multiple Networks matching the partial identifier (%s). Please use full identifier", nameID)
|
|
}
|
|
return list[0].ID, nil
|
|
}
|
|
|
|
func networkUsage(chain string) string {
|
|
help := "Commands:\n"
|
|
|
|
for _, cmd := range networkCommands {
|
|
help += fmt.Sprintf(" %-10.10s%s\n", cmd.name, cmd.description)
|
|
}
|
|
|
|
help += fmt.Sprintf("\nRun '%s network COMMAND --help' for more information on a command.", chain)
|
|
return help
|
|
}
|