feat: add context methods that write responses fast

1 files changed, 40 insertions, 0 deletions

diff --git a/context.go b/context.go
index 628cb4f..98f52c7 100644
--- a/context.go
+++ b/context.go
@@ -1,7 +1,9 @@
 package growel
 
 import (
+	"encoding/json"
 	"net/http"
+	"strconv"
 )
 
 type Context struct {
@@ -10,4 +12,42 @@ type Context struct {
 	Params map[string]string
 }
 
+func (c *Context) Plain(code int, data string) {
+	c.W.Header().Set("Content-Type", "text/plain; charset=utf-8")
+	c.W.WriteHeader(code)
+	c.W.Write([]byte(data))
+}
+
+func (c *Context) JSON(code int, data any) {
+	c.W.Header().Set("Content-Type", "application/json; charset=utf-8")
+	c.W.WriteHeader(code)
+
+	enc := json.NewEncoder(c.W)
+	_ = enc.Encode(data)
+}
+
+func (c *Context) Error(code int, msg string) {
+	c.JSON(code, map[string]string{
+		"error": http.StatusText(code),
+		"message": msg,
+		"status": strconv.Itoa(code),
+	})
+}
+
+func (c *Context) NotFound(msg string) {
+	c.Error(http.StatusNotFound, msg)
+}
+
+func (c *Context) BadRequest(msg string) {
+	c.Error(http.StatusBadRequest, msg)
+}
+
+func (c *Context) Unauthorized(msg string) {
+	c.Error(http.StatusUnauthorized, msg)
+}
+
+func (c *Context) InternalError(msg string) {
+	c.Error(http.StatusInternalServerError, msg)
+}
+
 type Handler func(c *Context)