feat: support Cookies

1 files changed, 34 insertions, 0 deletions

diff --git a/context.go b/context.go
index 271b886..e0de660 100644
--- a/context.go
+++ b/context.go
@@ -6,6 +6,8 @@ import (
 	"strconv"
 )
 
+/* === Writer Functions */
+
 // Plain writes plain text to ResponseWriter
 // with code and string passed as parameters
 // this automatically sets Content-Type to "text/plain; charset=utf-8"
@@ -77,3 +79,35 @@ func (c *Context) Forbidden(msg string) {
 func (c *Context) InternalError(msg string) {
 	c.Error(http.StatusInternalServerError, msg)
 }
+
+/* Cookies */
+
+func (c *Context) Cookies() []*http.Cookie {
+	return c.R.Cookies()
+}
+
+func (c *Context) Cookie(name string) *http.Cookie {
+	for _, cookie := range c.R.Cookies() {
+		if cookie.Name == name {
+			return cookie
+		}
+	}
+	return nil
+}
+
+func (c *Context) SetCompleteCookie(cookie *http.Cookie) {
+	http.SetCookie(c.W, cookie)
+}
+
+func (c *Context) SetCookie(name string, value string) {
+	cookie := http.Cookie{
+        Name:     name,
+        Value:    value,
+        Path:     "/",
+        MaxAge:   3600,
+        HttpOnly: true,
+        Secure:   true,
+        SameSite: http.SameSiteLaxMode,
+    }
+    c.SetCompleteCookie(&cookie)
+}