aboutsummaryrefslogtreecommitdiff
path: root/pkg/config/body.go
blob: 3b2f676e9102dc6b745e253adffa21b43a2f7351 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
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", "fetch"}

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...)}}
}