context_render.go

package xingyun

import (
	"encoding/json"
	"mime"
	"strings"
)

func (ctx *Context) SetContentType(val string) string {
	var ctype string
	if strings.ContainsRune(val, '/') {
		ctype = val
	} else {
		if !strings.HasPrefix(val, ".") {
			val = "." + val
		}
		ctype = mime.TypeByExtension(val)
	}
	if ctype != "" {
		ctx.ResponseWriter.Header().Set("Content-Type", ctype)
	}
	return ctype
}

func (ctx *Context) WriteString(s string) {
	_, err := ctx.Write([]byte(s))
	if err != nil {
		ctx.Logger.Errorf(err.Error())
		return
	}
}

func (ctx *Context) WriteJSON(obj interface{}) {
	bs, err := json.Marshal(obj)
	if err != nil {
		panic(err)
	}
	ctx.Write(bs)
}

func (ctx *Context) NotModified() {
	ctx.WriteHeader(304)
}

func (ctx *Context) NotFound() {
	ctx.WriteHeader(404)
}

func (ctx *Context) Unauthorized() {
	ctx.WriteHeader(401)
}

func (ctx *Context) Forbidden() {
	ctx.WriteHeader(403)
}

func (ctx *Context) Redirect(url_ string) {
	ctx.ResponseWriter.Header().Set("Location", url_)
	ctx.ResponseWriter.WriteHeader(302)
	ctx.ResponseWriter.Write([]byte("Redirecting to: " + url_))
}