dotcloud/docker

package network

import (
    "github.com/docker/docker/api/types/network"
)

// WithDriver sets the driver of the network
func WithDriver(driver string) func(*network.CreateOptions) {
    return func(n *network.CreateOptions) {
        n.Driver = driver
    }
}

// WithIPv4 enables/disables IPv4 on the network
func WithIPv4(enable bool) func(*network.CreateOptions) {
    return func(n *network.CreateOptions) {
        enableIPv4 := enable
        n.EnableIPv4 = &enableIPv4
    }
}

// WithIPv6 Enables IPv6 on the network
func WithIPv6() func(*network.CreateOptions) {
    return func(n *network.CreateOptions) {
        enableIPv6 := true
        n.EnableIPv6 = &enableIPv6
    }
}

// WithInternal enables Internal flag on the create network request
func WithInternal() func(*network.CreateOptions) {
    return func(n *network.CreateOptions) {
        n.Internal = true
    }
}

// WithAttachable sets Attachable flag on the create network request
func WithAttachable() func(*network.CreateOptions) {
    return func(n *network.CreateOptions) {
        n.Attachable = true
    }
}

// WithMacvlan sets the network as macvlan with the specified parent
func WithMacvlan(parent string) func(*network.CreateOptions) {
    return func(n *network.CreateOptions) {
        n.Driver = "macvlan"
        if parent != "" {
            n.Options = map[string]string{
                "parent": parent,
            }
        }
    }
}

// WithMacvlanPassthru sets the network as macvlan with the specified parent in passthru mode
func WithMacvlanPassthru(parent string) func(options *network.CreateOptions) {
    return func(n *network.CreateOptions) {
        n.Driver = "macvlan"
        n.Options = map[string]string{
            "macvlan_mode": "passthru",
        }
        if parent != "" {
            n.Options["parent"] = parent
        }
    }
}

// WithIPvlan sets the network as ipvlan with the specified parent and mode
func WithIPvlan(parent, mode string) func(*network.CreateOptions) {
    return func(n *network.CreateOptions) {
        n.Driver = "ipvlan"
        if n.Options == nil {
            n.Options = map[string]string{}
        }
        if parent != "" {
            n.Options["parent"] = parent
        }
        if mode != "" {
            n.Options["ipvlan_mode"] = mode
        }
    }
}

// WithOption adds the specified key/value pair to network's options
func WithOption(key, value string) func(*network.CreateOptions) {
    return func(n *network.CreateOptions) {
        if n.Options == nil {
            n.Options = map[string]string{}
        }
        n.Options[key] = value
    }
}

// WithIPAM adds an IPAM with the specified Subnet and Gateway to the network
func WithIPAM(subnet, gateway string) func(*network.CreateOptions) {
    return WithIPAMRange(subnet, "", gateway)
}

// WithIPAM adds an IPAM with the specified Subnet, IPRange and Gateway to the network
func WithIPAMRange(subnet, iprange, gateway string) func(*network.CreateOptions) {
    return func(n *network.CreateOptions) {
        if n.IPAM == nil {
            n.IPAM = &network.IPAM{}
        }

        n.IPAM.Config = append(n.IPAM.Config, network.IPAMConfig{
            Subnet:     subnet,
            IPRange:    iprange,
            Gateway:    gateway,
            AuxAddress: map[string]string{},
        })
    }
}