talos/pkg/machinery/nethelpers/address_flags.go
Andrey Smirnov d02d944ec7
chore: provide umarshal from YAML methods for network resource specs
This will be required to preserve platform network configuration in a
cache across boots. Network configuration will be marshaled to YAML and
unmarshaled back.

Changes are pretty simple and most of the code is auto-generated:
replace `stringer` codegen with `enumer` which produces also methods
which convert from string back to enum values.

Added tests to verify YAML marshaling/unmarshaling.

Signed-off-by: Andrey Smirnov <andrey.smirnov@talos-systems.com>
2022-01-10 17:00:53 +03:00

76 lines
2.1 KiB
Go

// This Source Code Form is subject to the terms of the Mozilla Public
// License, v. 2.0. If a copy of the MPL was not distributed with this
// file, You can obtain one at http://mozilla.org/MPL/2.0/.
package nethelpers
//go:generate enumer -type=AddressFlag -linecomment -text
import (
"strings"
)
// AddressFlags is a bitmask of AddressFlag.
type AddressFlags uint32
func (flags AddressFlags) String() string {
var values []string
for flag := AddressTemporary; flag <= AddressStablePrivacy; flag <<= 1 {
if (AddressFlag(flags) & flag) == flag {
values = append(values, flag.String())
}
}
return strings.Join(values, ",")
}
// AddressFlagsString converts string representation of flags into AddressFlags.
func AddressFlagsString(s string) (AddressFlags, error) {
flags := AddressFlags(0)
for _, p := range strings.Split(s, ",") {
flag, err := AddressFlagString(p)
if err != nil {
return flags, err
}
flags |= AddressFlags(flag)
}
return flags, nil
}
// MarshalText implements text.Marshaler.
func (flags AddressFlags) MarshalText() ([]byte, error) {
return []byte(flags.String()), nil
}
// UnmarshalText implements text.Unmarshaler.
func (flags *AddressFlags) UnmarshalText(b []byte) error {
var err error
*flags, err = AddressFlagsString(string(b))
return err
}
// AddressFlag wraps IFF_* constants.
type AddressFlag uint32
// AddressFlag constants.
const (
AddressTemporary AddressFlag = 1 << iota // temporary
AddressNoDAD // nodad
AddressOptimistic // optimistic
AddressDADFailed // dadfailed
AddressHome // homeaddress
AddressDeprecated // deprecated
AddressTentative // tentative
AddressPermanent // permanent
AddressManagementTemp // mngmtmpaddr
AddressNoPrefixRoute // noprefixroute
AddressMCAutoJoin // mcautojoin
AddressStablePrivacy // stableprivacy
)