package opts import ( "encoding/csv" "fmt" "regexp" "strconv" "strings" "github.com/docker/docker/api/types/swarm" "github.com/docker/go-connections/nat" ) const ( portOptTargetPort = "target" portOptPublishedPort = "published" portOptProtocol = "protocol" portOptMode = "mode" ) // PortOpt represents a port config in swarm mode. type PortOpt struct { ports []swarm.PortConfig } // Set a new port value func (p *PortOpt) Set(value string) error { longSyntax, err := regexp.MatchString(`\w+=\w+(,\w+=\w+)*`, value) if err != nil { return err } if longSyntax { csvReader := csv.NewReader(strings.NewReader(value)) fields, err := csvReader.Read() if err != nil { return err } pConfig := swarm.PortConfig{} for _, field := range fields { parts := strings.SplitN(field, "=", 2) if len(parts) != 2 { return fmt.Errorf("invalid field %s", field) } key := strings.ToLower(parts[0]) value := strings.ToLower(parts[1]) switch key { case portOptProtocol: if value != string(swarm.PortConfigProtocolTCP) && value != string(swarm.PortConfigProtocolUDP) { return fmt.Errorf("invalid protocol value %s", value) } pConfig.Protocol = swarm.PortConfigProtocol(value) case portOptMode: if value != string(swarm.PortConfigPublishModeIngress) && value != string(swarm.PortConfigPublishModeHost) { return fmt.Errorf("invalid publish mode value %s", value) } pConfig.PublishMode = swarm.PortConfigPublishMode(value) case portOptTargetPort: tPort, err := strconv.ParseUint(value, 10, 16) if err != nil { return err } pConfig.TargetPort = uint32(tPort) case portOptPublishedPort: pPort, err := strconv.ParseUint(value, 10, 16) if err != nil { return err } pConfig.PublishedPort = uint32(pPort) default: return fmt.Errorf("invalid field key %s", key) } } if pConfig.TargetPort == 0 { return fmt.Errorf("missing mandatory field %q", portOptTargetPort) } if pConfig.PublishMode == "" { pConfig.PublishMode = swarm.PortConfigPublishModeIngress } if pConfig.Protocol == "" { pConfig.Protocol = swarm.PortConfigProtocolTCP } p.ports = append(p.ports, pConfig) } else { // short syntax portConfigs := []swarm.PortConfig{} ports, portBindingMap, err := nat.ParsePortSpecs([]string{value}) if err != nil { return err } for _, portBindings := range portBindingMap { for _, portBinding := range portBindings { if portBinding.HostIP != "" { return fmt.Errorf("HostIP is not supported.") } } } for port := range ports { portConfig, err := ConvertPortToPortConfig(port, portBindingMap) if err != nil { return err } portConfigs = append(portConfigs, portConfig...) } p.ports = append(p.ports, portConfigs...) } return nil } // Type returns the type of this option func (p *PortOpt) Type() string { return "port" } // String returns a string repr of this option func (p *PortOpt) String() string { ports := []string{} for _, port := range p.ports { repr := fmt.Sprintf("%v:%v/%s/%s", port.PublishedPort, port.TargetPort, port.Protocol, port.PublishMode) ports = append(ports, repr) } return strings.Join(ports, ", ") } // Value returns the ports func (p *PortOpt) Value() []swarm.PortConfig { return p.ports } // ConvertPortToPortConfig converts ports to the swarm type func ConvertPortToPortConfig( port nat.Port, portBindings map[nat.Port][]nat.PortBinding, ) ([]swarm.PortConfig, error) { ports := []swarm.PortConfig{} for _, binding := range portBindings[port] { hostPort, err := strconv.ParseUint(binding.HostPort, 10, 16) if err != nil && binding.HostPort != "" { return nil, fmt.Errorf("invalid hostport binding (%s) for port (%s)", binding.HostPort, port.Port()) } ports = append(ports, swarm.PortConfig{ //TODO Name: ? Protocol: swarm.PortConfigProtocol(strings.ToLower(port.Proto())), TargetPort: uint32(port.Int()), PublishedPort: uint32(hostPort), PublishMode: swarm.PortConfigPublishModeIngress, }) } return ports, nil }