mirror of
https://github.com/Luzifer/gcr-clean.git
synced 2024-12-24 12:21:18 +00:00
107 lines
2.4 KiB
Go
107 lines
2.4 KiB
Go
|
package opts
|
||
|
|
||
|
import (
|
||
|
"encoding/csv"
|
||
|
"fmt"
|
||
|
"regexp"
|
||
|
"strings"
|
||
|
)
|
||
|
|
||
|
const (
|
||
|
networkOptName = "name"
|
||
|
networkOptAlias = "alias"
|
||
|
driverOpt = "driver-opt"
|
||
|
)
|
||
|
|
||
|
// NetworkAttachmentOpts represents the network options for endpoint creation
|
||
|
type NetworkAttachmentOpts struct {
|
||
|
Target string
|
||
|
Aliases []string
|
||
|
DriverOpts map[string]string
|
||
|
}
|
||
|
|
||
|
// NetworkOpt represents a network config in swarm mode.
|
||
|
type NetworkOpt struct {
|
||
|
options []NetworkAttachmentOpts
|
||
|
}
|
||
|
|
||
|
// Set networkopts value
|
||
|
func (n *NetworkOpt) Set(value string) error {
|
||
|
longSyntax, err := regexp.MatchString(`\w+=\w+(,\w+=\w+)*`, value)
|
||
|
if err != nil {
|
||
|
return err
|
||
|
}
|
||
|
|
||
|
var netOpt NetworkAttachmentOpts
|
||
|
if longSyntax {
|
||
|
csvReader := csv.NewReader(strings.NewReader(value))
|
||
|
fields, err := csvReader.Read()
|
||
|
if err != nil {
|
||
|
return err
|
||
|
}
|
||
|
|
||
|
netOpt.Aliases = []string{}
|
||
|
for _, field := range fields {
|
||
|
parts := strings.SplitN(field, "=", 2)
|
||
|
|
||
|
if len(parts) < 2 {
|
||
|
return fmt.Errorf("invalid field %s", field)
|
||
|
}
|
||
|
|
||
|
key := strings.TrimSpace(strings.ToLower(parts[0]))
|
||
|
value := strings.TrimSpace(strings.ToLower(parts[1]))
|
||
|
|
||
|
switch key {
|
||
|
case networkOptName:
|
||
|
netOpt.Target = value
|
||
|
case networkOptAlias:
|
||
|
netOpt.Aliases = append(netOpt.Aliases, value)
|
||
|
case driverOpt:
|
||
|
key, value, err = parseDriverOpt(value)
|
||
|
if err == nil {
|
||
|
if netOpt.DriverOpts == nil {
|
||
|
netOpt.DriverOpts = make(map[string]string)
|
||
|
}
|
||
|
netOpt.DriverOpts[key] = value
|
||
|
} else {
|
||
|
return err
|
||
|
}
|
||
|
default:
|
||
|
return fmt.Errorf("invalid field key %s", key)
|
||
|
}
|
||
|
}
|
||
|
if len(netOpt.Target) == 0 {
|
||
|
return fmt.Errorf("network name/id is not specified")
|
||
|
}
|
||
|
} else {
|
||
|
netOpt.Target = value
|
||
|
}
|
||
|
n.options = append(n.options, netOpt)
|
||
|
return nil
|
||
|
}
|
||
|
|
||
|
// Type returns the type of this option
|
||
|
func (n *NetworkOpt) Type() string {
|
||
|
return "network"
|
||
|
}
|
||
|
|
||
|
// Value returns the networkopts
|
||
|
func (n *NetworkOpt) Value() []NetworkAttachmentOpts {
|
||
|
return n.options
|
||
|
}
|
||
|
|
||
|
// String returns the network opts as a string
|
||
|
func (n *NetworkOpt) String() string {
|
||
|
return ""
|
||
|
}
|
||
|
|
||
|
func parseDriverOpt(driverOpt string) (string, string, error) {
|
||
|
parts := strings.SplitN(driverOpt, "=", 2)
|
||
|
if len(parts) != 2 {
|
||
|
return "", "", fmt.Errorf("invalid key value pair format in driver options")
|
||
|
}
|
||
|
key := strings.TrimSpace(strings.ToLower(parts[0]))
|
||
|
value := strings.TrimSpace(strings.ToLower(parts[1]))
|
||
|
return key, value, nil
|
||
|
}
|