Merge pull request #8 from coder/f0ssel/enum-yaml

coder · Mar 18, 2024 · c842d75 · c842d75
2 parents ae1178c + 0d2ed47
commit c842d75
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/values.go b/values.go
@@ -517,6 +517,17 @@ func (e *Enum) String() string {
 	return *e.Value
 }
 
+func (e *Enum) MarshalYAML() (interface{}, error) {
+	return yaml.Node{
+		Kind:  yaml.ScalarNode,
+		Value: e.String(),
+	}, nil
+}
+
+func (e *Enum) UnmarshalYAML(n *yaml.Node) error {
+	return e.Set(n.Value)
+}
+
 type Regexp regexp.Regexp
 
 func (r *Regexp) MarshalJSON() ([]byte, error) {