39 lines
730 B
Go
39 lines
730 B
Go
|
package yaml
|
||
|
|
|
||
|
|
import (
|
||
|
|
"bytes"
|
||
|
|
"io"
|
||
|
|
"net/url"
|
||
|
|
"path"
|
||
|
|
"path/filepath"
|
||
|
|
|
||
|
|
"forge.cadoles.com/wpetit/formidable/internal/data/format"
|
||
|
|
"github.com/pkg/errors"
|
||
|
|
"gopkg.in/yaml.v3"
|
||
|
|
)
|
||
|
|
|
||
|
|
type EncoderHandler struct{}
|
||
|
|
|
||
|
|
func (d *EncoderHandler) Match(url *url.URL) bool {
|
||
|
|
ext := filepath.Ext(path.Join(url.Host, url.Path))
|
||
|
|
|
||
|
|
return ExtensionYAML.MatchString(ext) ||
|
||
|
|
format.MatchURLQueryFormat(url, FormatYAML)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (d *EncoderHandler) Encode(url *url.URL, data interface{}) (io.Reader, error) {
|
||
|
|
var buf bytes.Buffer
|
||
|
|
|
||
|
|
encoder := yaml.NewEncoder(&buf)
|
||
|
|
|
||
|
|
if err := encoder.Encode(data); err != nil {
|
||
|
|
return nil, errors.WithStack(err)
|
||
|
|
}
|
||
|
|
|
||
|
|
return &buf, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func NewEncoderHandler() *EncoderHandler {
|
||
|
|
return &EncoderHandler{}
|
||
|
|
}
|