meta.go

@@ -1,5 +1,7 @@
 package toml
 
+import "fmt"
+
 // MetaData allows access to meta information about TOML.
 //
 // It allows determining whether a key has been defined, the TOML type of a
@@ -28,6 +30,27 @@ func NewMetaData() MetaData {
 	return MetaData{}
 }
 
+type (
+	Doc     string
+	Comment string
+)
+
+func (enc *MetaData) Key(key string, args ...interface{}) *MetaData {
+	for _, a := range args {
+		switch aa := a.(type) {
+		default:
+			panic(fmt.Sprintf("toml.MetaData.Key: unsupported type: %T", a))
+		case tomlType:
+			enc.SetType(key, aa)
+		case Doc:
+			enc.Doc(key, string(aa))
+		case Comment:
+			enc.Comment(key, string(aa))
+		}
+	}
+	return enc
+}
+
 func (enc *MetaData) SetType(key string, t tomlType) *MetaData {
 	enc.types[key] = t
 	return enc