2016-06-13 22:56:23 -04:00
|
|
|
package swarm
|
|
|
|
|
|
|
|
import (
|
|
|
|
"fmt"
|
|
|
|
"strings"
|
|
|
|
|
|
|
|
"github.com/docker/engine-api/types/swarm"
|
|
|
|
)
|
|
|
|
|
|
|
|
const (
|
|
|
|
defaultListenAddr = "0.0.0.0:2377"
|
|
|
|
// WORKER constant for worker name
|
|
|
|
WORKER = "WORKER"
|
|
|
|
// MANAGER constant for manager name
|
|
|
|
MANAGER = "MANAGER"
|
|
|
|
)
|
|
|
|
|
|
|
|
var (
|
|
|
|
defaultPolicies = []swarm.Policy{
|
|
|
|
{Role: WORKER, Autoaccept: true},
|
|
|
|
{Role: MANAGER, Autoaccept: false},
|
|
|
|
}
|
|
|
|
)
|
|
|
|
|
|
|
|
// NodeAddrOption is a pflag.Value for listen and remote addresses
|
|
|
|
type NodeAddrOption struct {
|
|
|
|
addr string
|
|
|
|
}
|
|
|
|
|
|
|
|
// String prints the representation of this flag
|
|
|
|
func (a *NodeAddrOption) String() string {
|
|
|
|
return a.addr
|
|
|
|
}
|
|
|
|
|
|
|
|
// Set the value for this flag
|
|
|
|
func (a *NodeAddrOption) Set(value string) error {
|
|
|
|
if !strings.Contains(value, ":") {
|
2016-06-17 10:27:54 -04:00
|
|
|
return fmt.Errorf("Invalid url, a host and port are required")
|
2016-06-13 22:56:23 -04:00
|
|
|
}
|
|
|
|
|
|
|
|
parts := strings.Split(value, ":")
|
|
|
|
if len(parts) != 2 {
|
2016-06-17 10:27:54 -04:00
|
|
|
return fmt.Errorf("Invalid url, too many colons")
|
2016-06-13 22:56:23 -04:00
|
|
|
}
|
|
|
|
|
|
|
|
a.addr = value
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
|
|
|
// Type returns the type of this flag
|
|
|
|
func (a *NodeAddrOption) Type() string {
|
|
|
|
return "node-addr"
|
|
|
|
}
|
|
|
|
|
|
|
|
// NewNodeAddrOption returns a new node address option
|
|
|
|
func NewNodeAddrOption() NodeAddrOption {
|
|
|
|
return NodeAddrOption{addr: defaultListenAddr}
|
|
|
|
}
|
|
|
|
|
|
|
|
// AutoAcceptOption is a value type for auto-accept policy
|
|
|
|
type AutoAcceptOption struct {
|
|
|
|
values map[string]bool
|
|
|
|
}
|
|
|
|
|
|
|
|
// String prints a string representation of this option
|
|
|
|
func (o *AutoAcceptOption) String() string {
|
|
|
|
keys := []string{}
|
|
|
|
for key := range o.values {
|
|
|
|
keys = append(keys, key)
|
|
|
|
}
|
|
|
|
return strings.Join(keys, " ")
|
|
|
|
}
|
|
|
|
|
|
|
|
// Set sets a new value on this option
|
|
|
|
func (o *AutoAcceptOption) Set(value string) error {
|
|
|
|
value = strings.ToUpper(value)
|
|
|
|
switch value {
|
|
|
|
case "", "NONE":
|
|
|
|
if accept, ok := o.values[WORKER]; ok && accept {
|
|
|
|
return fmt.Errorf("value NONE is incompatible with %s", WORKER)
|
|
|
|
}
|
|
|
|
if accept, ok := o.values[MANAGER]; ok && accept {
|
|
|
|
return fmt.Errorf("value NONE is incompatible with %s", MANAGER)
|
|
|
|
}
|
|
|
|
o.values[WORKER] = false
|
|
|
|
o.values[MANAGER] = false
|
|
|
|
case WORKER, MANAGER:
|
|
|
|
if accept, ok := o.values[value]; ok && !accept {
|
|
|
|
return fmt.Errorf("value NONE is incompatible with %s", value)
|
|
|
|
}
|
|
|
|
o.values[value] = true
|
|
|
|
default:
|
|
|
|
return fmt.Errorf("must be one of %s, %s, NONE", WORKER, MANAGER)
|
|
|
|
}
|
|
|
|
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
|
|
|
// Type returns the type of this option
|
|
|
|
func (o *AutoAcceptOption) Type() string {
|
|
|
|
return "auto-accept"
|
|
|
|
}
|
|
|
|
|
|
|
|
// Policies returns a representation of this option for the api
|
2016-06-15 17:30:54 -04:00
|
|
|
func (o *AutoAcceptOption) Policies(secret *string) []swarm.Policy {
|
2016-06-13 22:56:23 -04:00
|
|
|
policies := []swarm.Policy{}
|
|
|
|
for _, p := range defaultPolicies {
|
|
|
|
if len(o.values) != 0 {
|
|
|
|
p.Autoaccept = o.values[string(p.Role)]
|
|
|
|
}
|
|
|
|
p.Secret = secret
|
|
|
|
policies = append(policies, p)
|
|
|
|
}
|
|
|
|
return policies
|
|
|
|
}
|
|
|
|
|
|
|
|
// NewAutoAcceptOption returns a new auto-accept option
|
|
|
|
func NewAutoAcceptOption() AutoAcceptOption {
|
|
|
|
return AutoAcceptOption{values: make(map[string]bool)}
|
|
|
|
}
|