mirror of https://github.com/asciimoo/wuzz
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
30 lines
788 B
Go
30 lines
788 B
Go
package formatter
|
|
|
|
import (
|
|
"io"
|
|
"mime"
|
|
"strings"
|
|
|
|
"github.com/asciimoo/wuzz/config"
|
|
)
|
|
|
|
type ResponseFormatter interface {
|
|
Format(writer io.Writer, data []byte) error
|
|
Title() string
|
|
Searchable() bool
|
|
Search(string, []byte) ([]string, error)
|
|
}
|
|
|
|
func New(appConfig *config.Config, contentType string) ResponseFormatter {
|
|
ctype, _, err := mime.ParseMediaType(contentType)
|
|
if err == nil && appConfig.General.FormatJSON && (ctype == config.ContentTypes["json"] || strings.HasSuffix(ctype, "+json")) {
|
|
return &jsonFormatter{}
|
|
} else if strings.Contains(contentType, "text/html") {
|
|
return &htmlFormatter{}
|
|
} else if strings.Index(contentType, "text") == -1 && strings.Index(contentType, "application") == -1 {
|
|
return &binaryFormatter{}
|
|
} else {
|
|
return &TextFormatter{}
|
|
}
|
|
}
|