fix(yaml): add encoding.Text(Un)marshaler to the interface whitelist (#119)

Un1oR · web-flow · commit 7d096c24efc2 · 2025-04-24T19:22:44.000+03:00
diff --git a/builtins.go b/builtins.go
@@ -19,10 +19,10 @@ var builtins = []Func{
 	{"(*encoding/xml.Decoder).DecodeElement", "xml", 0, []string{"encoding/xml.Unmarshaler", "encoding.TextUnmarshaler"}},
 
 	// https://pkg.go.dev/gopkg.in/yaml.v3
-	{"gopkg.in/yaml.v3.Marshal", "yaml", 0, []string{"gopkg.in/yaml.v3.Marshaler"}},
-	{"gopkg.in/yaml.v3.Unmarshal", "yaml", 1, []string{"gopkg.in/yaml.v3.Unmarshaler"}},
-	{"(*gopkg.in/yaml.v3.Encoder).Encode", "yaml", 0, []string{"gopkg.in/yaml.v3.Marshaler"}},
-	{"(*gopkg.in/yaml.v3.Decoder).Decode", "yaml", 0, []string{"gopkg.in/yaml.v3.Unmarshaler"}},
+	{"gopkg.in/yaml.v3.Marshal", "yaml", 0, []string{"gopkg.in/yaml.v3.Marshaler", "encoding.TextMarshaler"}},
+	{"gopkg.in/yaml.v3.Unmarshal", "yaml", 1, []string{"gopkg.in/yaml.v3.Unmarshaler", "encoding.TextUnmarshaler"}},
+	{"(*gopkg.in/yaml.v3.Encoder).Encode", "yaml", 0, []string{"gopkg.in/yaml.v3.Marshaler", "encoding.TextMarshaler"}},
+	{"(*gopkg.in/yaml.v3.Decoder).Decode", "yaml", 0, []string{"gopkg.in/yaml.v3.Unmarshaler", "encoding.TextUnmarshaler"}},
 
 	// https://pkg.go.dev/github.com/BurntSushi/toml
 	{"github.com/BurntSushi/toml.Unmarshal", "toml", 1, []string{"github.com/BurntSushi/toml.Unmarshaler", "encoding.TextUnmarshaler"}},
diff --git a/testdata/src/tests/builtins.go b/testdata/src/tests/builtins.go
@@ -96,6 +96,12 @@ func testYAML() {
 	yaml.Unmarshal(nil, &m)
 	yaml.NewEncoder(nil).Encode(m)
 	yaml.NewDecoder(nil).Decode(&m)
+
+	var tm TextMarshaler
+	yaml.Marshal(tm)
+	yaml.Unmarshal(nil, &tm)
+	yaml.NewEncoder(nil).Encode(tm)
+	yaml.NewDecoder(nil).Decode(&tm)
 }
 
 func testTOML() {
