feat: add support for sending string and byte data (#47)

Author: mojixcoder

context.go

@@ -162,19 +162,52 @@ func (c *Context) HTMLString(code int, tpl string) {
 	c.mustWrite([]byte(tpl))
 }
 
+// String sends bare string as a plain text response with the given status code.
+func (c *Context) String(code int, data string) {
+	c.writeContentType("text/plain")
+	c.response.WriteHeader(code)
+	c.mustWrite([]byte(data))
+}
+
+// Byte sends bare bytes as response with the given status code.
+func (c *Context) Byte(code int, data []byte) {
+	c.writeContentType("application/octet-stream")
+	c.response.WriteHeader(code)
+	c.mustWrite([]byte(data))
+}
+
 // NoContent returns an empty response with the given status code.
 func (c *Context) NoContent(code int) {
 	c.response.WriteHeader(code)
 	c.response.WriteHeaderNow()
 }
 
+// GetResponseHeader gets a response header.
+func (c *Context) GetResponseHeader(key string) string {
+	return c.response.Header().Get(key)
+}
+
+// SetResponseHeader sets a header to the response.
+func (c *Context) SetResponseHeader(key, value string) {
+	c.response.Header().Set(key, value)
+}
+
+// SetRequestHeader sets a header to the request.
+func (c *Context) SetRequestHeader(key, value string) {
+	c.request.Header.Set(key, value)
+}
+
+// GetRequestHeader gets a request header.
+func (c *Context) GetRequestHeader(key string) string {
+	return c.request.Header.Get(key)
+}
+
 // writeContentType sets content type header for response.
 // It won't overwrite content type if it's already set.
 func (c *Context) writeContentType(contentType string) {
 	contentTypeHeader := "Content-Type"
-	header := c.response.Header()
-	if header.Get(contentTypeHeader) == "" {
-		header.Set(contentTypeHeader, contentType)
+	if c.GetResponseHeader(contentTypeHeader) == "" {
+		c.SetResponseHeader(contentTypeHeader, contentType)
 	}
 }
 

context_test.go

@@ -466,3 +466,71 @@ func TestContext_Debug(t *testing.T) {
 
 	assert.True(t, ctx.Debug())
 }
+
+func TestContext_String(t *testing.T) {
+	ctx := newContext(New())
+
+	res := httptest.NewRecorder()
+	ctx.reset(nil, res)
+
+	ctx.String(http.StatusAccepted, "hello")
+
+	assert.Equal(t, http.StatusAccepted, res.Code)
+	assert.Equal(t, "hello", res.Body.String())
+	assert.Equal(t, "text/plain", res.Header().Get("Content-Type"))
+}
+
+func TestContext_Byte(t *testing.T) {
+	ctx := newContext(New())
+
+	res := httptest.NewRecorder()
+	ctx.reset(nil, res)
+
+	ctx.Byte(http.StatusOK, []byte("mojix"))
+
+	assert.Equal(t, http.StatusOK, res.Code)
+	assert.Equal(t, "mojix", res.Body.String())
+	assert.Equal(t, "application/octet-stream", res.Header().Get("Content-Type"))
+}
+
+func TestContext_GetResponseHeader(t *testing.T) {
+	ctx := newContext(New())
+
+	res := httptest.NewRecorder()
+	res.Header().Set("key", "value")
+
+	ctx.reset(nil, res)
+
+	assert.Equal(t, "value", ctx.GetResponseHeader("key"))
+}
+
+func TestContext_SetResponseHeader(t *testing.T) {
+	ctx := newContext(New())
+
+	res := httptest.NewRecorder()
+	ctx.reset(nil, res)
+
+	ctx.SetResponseHeader("key", "value")
+	assert.Equal(t, "value", ctx.GetResponseHeader("key"))
+}
+
+func TestContext_GetRequestHeader(t *testing.T) {
+	ctx := newContext(New())
+
+	req := httptest.NewRequest(http.MethodGet, "/path", nil)
+	req.Header.Set("key", "value")
+
+	ctx.reset(req, nil)
+
+	assert.Equal(t, "value", ctx.GetRequestHeader("key"))
+}
+
+func TestContext_SetRequestHeader(t *testing.T) {
+	ctx := newContext(New())
+
+	req := httptest.NewRequest(http.MethodGet, "/path", nil)
+	ctx.reset(req, nil)
+
+	ctx.SetRequestHeader("key", "value")
+	assert.Equal(t, "value", ctx.GetRequestHeader("key"))
+}