From d75bbcf9a7fe589f119b103d357ffa874008ab07 Mon Sep 17 00:00:00 2001
From: René 'Necoro' Neumann <necoro@necoro.eu>
Date: Sat, 2 May 2020 16:48:51 +0200
Subject: Option 'body'

---
 pkg/config/body.go | 35 +++++++++++++++++++++++++++++++++++
 1 file changed, 35 insertions(+)
 create mode 100644 pkg/config/body.go

(limited to 'pkg/config/body.go')

diff --git a/pkg/config/body.go b/pkg/config/body.go
new file mode 100644
index 0000000..d9957d5
--- /dev/null
+++ b/pkg/config/body.go
@@ -0,0 +1,35 @@
+package config
+
+import (
+	"fmt"
+
+	"gopkg.in/yaml.v3"
+
+	"github.com/Necoro/feed2imap-go/pkg/util"
+)
+
+type Body string
+
+var validBody = []string{"default", "both", "content", "description"}
+
+func (b *Body) UnmarshalYAML(node *yaml.Node) error {
+	var val string
+	if err := node.Decode(&val); err != nil {
+		return err
+	}
+
+	if val == "" {
+		val = "default"
+	}
+
+	if !util.StrContains(validBody, val) {
+		return TypeError("line %d: Invalid value for 'body': %q", node.Line, val)
+	}
+
+	*b = Body(val)
+	return nil
+}
+
+func TypeError(format string, v ...interface{}) *yaml.TypeError {
+	return &yaml.TypeError{Errors: []string{fmt.Sprintf(format, v...)}}
+}
-- 
cgit v1.2.3-54-g00ecf