123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191 |
- // 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.
- package render
- import (
- "bytes"
- "fmt"
- "html/template"
- "net/http"
- "github.com/gin-gonic/gin/internal/json"
- )
- // JSON contains the given interface object.
- type JSON struct {
- Data interface{}
- }
- // IndentedJSON contains the given interface object.
- type IndentedJSON struct {
- Data interface{}
- }
- // SecureJSON contains the given interface object and its prefix.
- type SecureJSON struct {
- Prefix string
- Data interface{}
- }
- // JsonpJSON contains the given interface object its callback.
- type JsonpJSON struct {
- Callback string
- Data interface{}
- }
- // AsciiJSON contains the given interface object.
- type AsciiJSON struct {
- Data interface{}
- }
- // SecureJSONPrefix is a string which represents SecureJSON prefix.
- type SecureJSONPrefix string
- // PureJSON contains the given interface object.
- type PureJSON struct {
- Data interface{}
- }
- var jsonContentType = []string{"application/json; charset=utf-8"}
- var jsonpContentType = []string{"application/javascript; charset=utf-8"}
- var jsonAsciiContentType = []string{"application/json"}
- // Render (JSON) writes data with custom ContentType.
- func (r JSON) Render(w http.ResponseWriter) (err error) {
- if err = WriteJSON(w, r.Data); err != nil {
- panic(err)
- }
- return
- }
- // WriteContentType (JSON) writes JSON ContentType.
- func (r JSON) WriteContentType(w http.ResponseWriter) {
- writeContentType(w, jsonContentType)
- }
- // WriteJSON marshals the given interface object and writes it with custom ContentType.
- func WriteJSON(w http.ResponseWriter, obj interface{}) error {
- writeContentType(w, jsonContentType)
- encoder := json.NewEncoder(w)
- err := encoder.Encode(&obj)
- return err
- }
- // Render (IndentedJSON) marshals the given interface object and writes it with custom ContentType.
- func (r IndentedJSON) Render(w http.ResponseWriter) error {
- r.WriteContentType(w)
- jsonBytes, err := json.MarshalIndent(r.Data, "", " ")
- if err != nil {
- return err
- }
- _, err = w.Write(jsonBytes)
- return err
- }
- // WriteContentType (IndentedJSON) writes JSON ContentType.
- func (r IndentedJSON) WriteContentType(w http.ResponseWriter) {
- writeContentType(w, jsonContentType)
- }
- // Render (SecureJSON) marshals the given interface object and writes it with custom ContentType.
- func (r SecureJSON) Render(w http.ResponseWriter) error {
- r.WriteContentType(w)
- jsonBytes, err := json.Marshal(r.Data)
- if err != nil {
- return err
- }
- // if the jsonBytes is array values
- if bytes.HasPrefix(jsonBytes, []byte("[")) && bytes.HasSuffix(jsonBytes, []byte("]")) {
- _, err = w.Write([]byte(r.Prefix))
- if err != nil {
- return err
- }
- }
- _, err = w.Write(jsonBytes)
- return err
- }
- // WriteContentType (SecureJSON) writes JSON ContentType.
- func (r SecureJSON) WriteContentType(w http.ResponseWriter) {
- writeContentType(w, jsonContentType)
- }
- // Render (JsonpJSON) marshals the given interface object and writes it and its callback with custom ContentType.
- func (r JsonpJSON) Render(w http.ResponseWriter) (err error) {
- r.WriteContentType(w)
- ret, err := json.Marshal(r.Data)
- if err != nil {
- return err
- }
- if r.Callback == "" {
- _, err = w.Write(ret)
- return err
- }
- callback := template.JSEscapeString(r.Callback)
- _, err = w.Write([]byte(callback))
- if err != nil {
- return err
- }
- _, err = w.Write([]byte("("))
- if err != nil {
- return err
- }
- _, err = w.Write(ret)
- if err != nil {
- return err
- }
- _, err = w.Write([]byte(");"))
- if err != nil {
- return err
- }
- return nil
- }
- // WriteContentType (JsonpJSON) writes Javascript ContentType.
- func (r JsonpJSON) WriteContentType(w http.ResponseWriter) {
- writeContentType(w, jsonpContentType)
- }
- // Render (AsciiJSON) marshals the given interface object and writes it with custom ContentType.
- func (r AsciiJSON) Render(w http.ResponseWriter) (err error) {
- r.WriteContentType(w)
- ret, err := json.Marshal(r.Data)
- if err != nil {
- return err
- }
- var buffer bytes.Buffer
- for _, r := range string(ret) {
- cvt := string(r)
- if r >= 128 {
- cvt = fmt.Sprintf("\\u%04x", int64(r))
- }
- buffer.WriteString(cvt)
- }
- _, err = w.Write(buffer.Bytes())
- return err
- }
- // WriteContentType (AsciiJSON) writes JSON ContentType.
- func (r AsciiJSON) WriteContentType(w http.ResponseWriter) {
- writeContentType(w, jsonAsciiContentType)
- }
- // Render (PureJSON) writes custom ContentType and encodes the given interface object.
- func (r PureJSON) Render(w http.ResponseWriter) error {
- r.WriteContentType(w)
- encoder := json.NewEncoder(w)
- encoder.SetEscapeHTML(false)
- return encoder.Encode(r.Data)
- }
- // WriteContentType (PureJSON) writes custom ContentType.
- func (r PureJSON) WriteContentType(w http.ResponseWriter) {
- writeContentType(w, jsonContentType)
- }
|