gin/render/render.go

115 lines
2.5 KiB
Go
Raw Normal View History

2014-08-29 17:49:50 +00:00
// Copyright 2014 Manu Martinez-Almeida. All rights reserved.
// Use of this source code is governed by a MIT style
// license that can be found in the LICENSE file.
2014-07-15 15:41:56 +00:00
package render
import (
"encoding/json"
"encoding/xml"
"fmt"
"html/template"
"net/http"
)
type (
Render interface {
Render(http.ResponseWriter, int, ...interface{}) error
}
jsonRender struct{}
xmlRender struct{}
2015-03-23 03:42:05 +00:00
plainTextRender struct{}
2014-07-15 15:41:56 +00:00
htmlPlainRender struct{}
2014-07-28 22:48:02 +00:00
redirectRender struct{}
2014-07-15 15:41:56 +00:00
HTMLRender struct {
Template *template.Template
}
)
var (
JSON = jsonRender{}
XML = xmlRender{}
HTMLPlain = htmlPlainRender{}
2015-03-23 03:42:05 +00:00
Plain = plainTextRender{}
Redirect = redirectRender{}
2014-07-15 15:41:56 +00:00
)
2014-07-28 22:48:02 +00:00
func (_ redirectRender) Render(w http.ResponseWriter, code int, data ...interface{}) error {
w.Header().Set("Location", data[0].(string))
2014-07-28 22:53:56 +00:00
w.WriteHeader(code)
2014-07-28 22:48:02 +00:00
return nil
}
func (_ jsonRender) Render(w http.ResponseWriter, code int, data ...interface{}) error {
WriteHeader(w, code, "application/json")
return json.NewEncoder(w).Encode(data[0])
}
2014-07-15 15:41:56 +00:00
func (_ xmlRender) Render(w http.ResponseWriter, code int, data ...interface{}) error {
WriteHeader(w, code, "application/xml")
return xml.NewEncoder(w).Encode(data[0])
2014-07-15 15:41:56 +00:00
}
2015-03-23 03:42:05 +00:00
func (_ plainTextRender) Render(w http.ResponseWriter, code int, data ...interface{}) (err error) {
WriteHeader(w, code, "text/plain")
2014-07-15 15:41:56 +00:00
format := data[0].(string)
args := data[1].([]interface{})
if len(args) > 0 {
_, err = w.Write([]byte(fmt.Sprintf(format, args...)))
} else {
_, err = w.Write([]byte(format))
}
2015-03-23 03:42:05 +00:00
return
}
func (_ htmlPlainRender) Render(w http.ResponseWriter, code int, data ...interface{}) (err error) {
WriteHeader(w, code, "text/html")
format := data[0].(string)
args := data[1].([]interface{})
2014-07-17 23:56:59 +00:00
if len(args) > 0 {
_, err = w.Write([]byte(fmt.Sprintf(format, args...)))
} else {
_, err = w.Write([]byte(format))
}
return
2014-07-15 15:41:56 +00:00
}
func (html HTMLRender) Render(w http.ResponseWriter, code int, data ...interface{}) error {
WriteHeader(w, code, "text/html")
file := data[0].(string)
args := data[1]
return html.Template.ExecuteTemplate(w, file, args)
}
func WriteHeader(w http.ResponseWriter, code int, contentType string) {
contentType = joinStrings(contentType, "; charset=utf-8")
w.Header().Set("Content-Type", contentType)
w.WriteHeader(code)
}
func joinStrings(a ...string) string {
if len(a) == 0 {
return ""
2014-08-20 23:04:35 +00:00
}
if len(a) == 1 {
return a[0]
}
n := 0
for i := 0; i < len(a); i++ {
n += len(a[i])
}
b := make([]byte, n)
n = 0
for _, s := range a {
n += copy(b[n:], s)
}
return string(b)
}