context

Author: krishpranav

context.go

@@ -1,6 +1,12 @@
 package godzilla
 
-import "github.com/valyala/fasthttp"
+import (
+	"fmt"
+	"strings"
+
+	jsoniter "github.com/json-iterator/go"
+	"github.com/valyala/fasthttp"
+)
 
 const (
 	MimeApplicationJSON = "application/json"
@@ -45,11 +51,72 @@ func (ctx *context) Param(key string) string {
 	return ctx.paramValues[key]
 }
 
-func (ctx *context) Context() *fasthttp.Request {
+func (ctx *context) Context() *fasthttp.RequestCtx {
 	return ctx.requestCtx
 }
 
 func (ctx *context) SendBytes(value []byte) Context {
 	ctx.requestCtx.Response.SetBodyRaw(value)
 	return ctx
 }
+
+func (ctx *context) SendString(value string) Context {
+	ctx.requestCtx.SetBodyString(value)
+	return ctx
+}
+
+func (ctx *context) SendJSON(in interface{}) error {
+	json := jsoniter.ConfigCompatibleWithStandardLibrary
+	raw, err := json.Marshal(in)
+
+	if err != nil {
+		return err
+	}
+
+	ctx.requestCtx.Response.Header.SetContentType(MimeApplicationJSON)
+	ctx.requestCtx.Response.SetBodyRaw(raw)
+
+	return nil
+}
+
+func (ctx *context) Status(status int) Context {
+	ctx.requestCtx.Response.SetStatusCode(status)
+	return ctx
+}
+
+func (ctx *context) Get(key string) string {
+	return GetString(ctx.requestCtx.Request.Header.Peek(key))
+}
+
+func (ctx *context) Set(key, value string) {
+	ctx.requestCtx.Response.Header.Set(key, value)
+}
+
+func (ctx *context) Query(key string) string {
+	return GetString(ctx.requestCtx.QueryArgs().Peek(key))
+}
+
+func (ctx *context) Body() string {
+	return GetString(ctx.requestCtx.Request.Body())
+}
+
+func (ctx *context) SetLocal(key string, value interface{}) {
+	ctx.requestCtx.SetUserValue(key, value)
+}
+
+func (ctx *context) GetLocal(key string) interface{} {
+	return ctx.requestCtx.UserValue(key)
+}
+
+func (ctx *context) ParseBody(out interface{}) error {
+	contentType := GetString(ctx.requestCtx.Request.Header.ContentType())
+	if strings.HasPrefix(contentType, MimeApplicationJSON) {
+		json := jsoniter.ConfigCompatibleWithStandardLibrary
+		return json.Unmarshal(ctx.requestCtx.Request.Body(), out)
+	}
+
+	return fmt.Errorf("content type '%s' is not supported, "+
+		"please open a request to support it "+
+		"(https://github.com/godzillaframework//godzilla/issues",
+		contentType)
+}