gex/srv/res.go

package srv

import (
	"fmt"
	"net/http"
	"time"
)

type Res struct {
	w      http.ResponseWriter
	r      *http.Request
	status int
}

func (res *Res) Construct(w http.ResponseWriter, r *http.Request) {
	res.w = w
	res.r = r
	res.status = 200
}

func (res *Res) Send(txt string) {
	res.w.WriteHeader(res.status)
	fmt.Fprint(res.w, txt)
}

func (res *Res) Redirect(url string) {
	http.Redirect(res.w, res.r, url, http.StatusTemporaryRedirect)
}

func (res *Res) Status(code int) *Res {
	res.status = code
	return res
}

func (res *Res) Header(k string, v string) *Res {
	res.w.Header().Add(k, v)
	return res
}

func (res *Res) JSON(json string) {
	res.Header("Content-type", "application/json").Send(json)
	res.w.WriteHeader(res.status)
}

func (res *Res) Cookie(name string, val string, expires ...time.Time) *Res {
	Expires := time.Now().Add(365 * 24 * time.Hour)
	for _, v := range expires {
		Expires = v
	}
	cookie := &http.Cookie{
		Name:     name,
		Value:    val,
		Expires:  Expires,
		Path:     "/",
		Secure:   true,
		HttpOnly: true,
		SameSite: http.SameSiteStrictMode,
	}

	http.SetCookie(res.w, cookie)
	return res
}