mirror of https://github.com/beego/bee.git
90 lines
2.2 KiB
Go
90 lines
2.2 KiB
Go
package beeParser
|
|
|
|
import (
|
|
"encoding/json"
|
|
"io/ioutil"
|
|
"strings"
|
|
|
|
"github.com/talos-systems/talos/pkg/machinery/config/encoder"
|
|
)
|
|
|
|
type JsonFormatter struct {
|
|
}
|
|
|
|
func (f *JsonFormatter) FieldFormatFunc(field *StructField) ([]byte, error) {
|
|
annotation := NewAnnotation(field.Doc+field.Comment, field.Name, field.Type)
|
|
res := map[string]interface{}{}
|
|
if field.NestedType != nil {
|
|
res[annotation.Key] = field.NestedType
|
|
} else {
|
|
res[annotation.Key] = annotation.Default
|
|
}
|
|
return json.Marshal(res)
|
|
}
|
|
|
|
func (f *JsonFormatter) StructFormatFunc(node *StructNode) ([]byte, error) {
|
|
return json.Marshal(node.Fields)
|
|
}
|
|
|
|
func (f *JsonFormatter) Marshal(node *StructNode) ([]byte, error) {
|
|
return json.MarshalIndent(node, "", " ")
|
|
}
|
|
|
|
type YamlFormatter struct {
|
|
}
|
|
|
|
var result encoder.Doc
|
|
|
|
type Result map[string]interface{}
|
|
|
|
func (c Result) Doc() *encoder.Doc {
|
|
return &result
|
|
}
|
|
|
|
func (f *YamlFormatter) FieldFormatFunc(field *StructField) ([]byte, error) {
|
|
annotation := NewAnnotation(field.Doc+field.Comment, field.Name, field.Type)
|
|
res := Result{}
|
|
// add head comment for this field
|
|
res.Doc().Comments[encoder.HeadComment] = annotation.Description
|
|
if field.NestedType != nil {
|
|
// nestedType format result as this field value
|
|
b, err := field.NestedType.FormatFunc(field.NestedType)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
res[annotation.Key] = string(b)
|
|
} else {
|
|
res[annotation.Key] = annotation.Default
|
|
}
|
|
|
|
encoder := encoder.NewEncoder(&res, []encoder.Option{
|
|
encoder.WithComments(encoder.CommentsAll),
|
|
}...)
|
|
encodeByte, err := encoder.Encode()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
// when field.NestedType != nil, the key and nested value strings are encoded with "|"
|
|
// remove "|" by string replace
|
|
encodeByte = []byte(strings.Replace(string(encodeByte), annotation.Key+": |", annotation.Key+":", 1))
|
|
return encodeByte, nil
|
|
}
|
|
|
|
func (f *YamlFormatter) StructFormatFunc(node *StructNode) ([]byte, error) {
|
|
res := make([]byte, 0)
|
|
for _, f := range node.Fields {
|
|
b, _ := f.FormatFunc(f)
|
|
res = append(res, b...)
|
|
}
|
|
return res, nil
|
|
}
|
|
|
|
func (f *YamlFormatter) Marshal(node *StructNode) ([]byte, error) {
|
|
res, err := node.FormatFunc(node)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
ioutil.WriteFile(node.Name+".yaml", res, 0667)
|
|
return res, nil
|
|
}
|