Changed: DB Params

templ/examples/counter/session/session.go

@@ -0,0 +1,56 @@
+package session
+
+import (
+	"net/http"
+
+	"github.com/segmentio/ksuid"
+)
+
+type MiddlewareOpts func(*Middleware)
+
+func NewMiddleware(next http.Handler, opts ...MiddlewareOpts) http.Handler {
+	mw := Middleware{
+		Next:     next,
+		Secure:   true,
+		HTTPOnly: true,
+	}
+	for _, opt := range opts {
+		opt(&mw)
+	}
+	return mw
+}
+
+func WithSecure(secure bool) MiddlewareOpts {
+	return func(m *Middleware) {
+		m.Secure = secure
+	}
+}
+
+func WithHTTPOnly(httpOnly bool) MiddlewareOpts {
+	return func(m *Middleware) {
+		m.HTTPOnly = httpOnly
+	}
+}
+
+type Middleware struct {
+	Next     http.Handler
+	Secure   bool
+	HTTPOnly bool
+}
+
+func ID(r *http.Request) (id string) {
+	cookie, err := r.Cookie("sessionID")
+	if err != nil {
+		return
+	}
+	return cookie.Value
+}
+
+func (mw Middleware) ServeHTTP(w http.ResponseWriter, r *http.Request) {
+	id := ID(r)
+	if id == "" {
+		id = ksuid.New().String()
+		http.SetCookie(w, &http.Cookie{Name: "sessionID", Value: id, Secure: mw.Secure, HttpOnly: mw.HTTPOnly})
+	}
+	mw.Next.ServeHTTP(w, r)
+}