constellation/internal/role/role.go
2022-10-13 15:29:29 +02:00

54 lines
917 B
Go

/*
Copyright (c) Edgeless Systems GmbH
SPDX-License-Identifier: AGPL-3.0-only
*/
package role
import (
"encoding/json"
"strings"
)
//go:generate stringer -type=Role
// Role is a peer's role.
type Role uint
const (
Unknown Role = iota
ControlPlane
Worker
Admin
)
// MarshalJSON marshals the Role to JSON string.
func (r Role) MarshalJSON() ([]byte, error) {
return json.Marshal(r.String())
}
// UnmarshalJSON unmarshals the Role from JSON string.
func (r *Role) UnmarshalJSON(b []byte) error {
var roleString string
if err := json.Unmarshal(b, &roleString); err != nil {
return err
}
*r = FromString(roleString)
return nil
}
// FromString returns the Role for the given string.
func FromString(s string) Role {
switch strings.ToLower(s) {
case "controlplane", "control-plane":
return ControlPlane
case "worker":
return Worker
case "admin":
return Admin
default:
return Unknown
}
}