Repos / bykclk/echo-scs-session / 1b2eee47df6b / github.com/wainuiomata/echo-scs-session/session.go

github.com/wainuiomata/echo-scs-session/session.go

29 of 38 statements covered · → +0.0% · commit 1b2eee47df6b

File coverage

76.3%

⋯30 lines, fully covered · expand
1package session
2
3import (
4	"net/http"
5	"time"
6
7	"github.com/alexedwards/scs/v2"
8	"github.com/labstack/echo/v4"
9	"github.com/labstack/echo/v4/middleware"
10)
11
12type SessionConfig struct {
13	Skipper        middleware.Skipper
14	SessionManager *scs.SessionManager
15}
16
17var DefaultSessionConfig = SessionConfig{
18	Skipper: middleware.DefaultSkipper,
19}
20
211×func LoadAndSave(sessionManager *scs.SessionManager) echo.MiddlewareFunc {
221×	c := DefaultSessionConfig
231×	c.SessionManager = sessionManager
241×
251×	return LoadAndSaveWithConfig(c)
261×}
27
281×func LoadAndSaveWithConfig(config SessionConfig) echo.MiddlewareFunc {
291×
301×	if config.Skipper == nil {
31✗		config.Skipper = DefaultSessionConfig.Skipper
32✗	}
33
341×	if config.SessionManager == nil {
35✗		panic("Session middleware requires a session manager")
36	}
37
382×	return func(next echo.HandlerFunc) echo.HandlerFunc {
392×		return func(c echo.Context) error {
402×			if config.Skipper(c) {
41✗				return next(c)
42✗			}
⋯11 lines, fully covered · expand
43
442×			ctx := c.Request().Context()
452×
462×			var token string
472×			cookie, err := c.Cookie(config.SessionManager.Cookie.Name)
482×			if err == nil {
491×				token = cookie.Value
501×			}
51
522×			ctx, err = config.SessionManager.Load(ctx, token)
532×			if err != nil {
54✗				return err
55✗			}
⋯18 lines, fully covered · expand
56
572×			c.SetRequest(c.Request().WithContext(ctx))
582×
592×			c.Response().Before(func() {
602×				if config.SessionManager.Status(ctx) != scs.Unmodified {
611×					responseCookie := &http.Cookie{
621×						Name:     config.SessionManager.Cookie.Name,
631×						Path:     config.SessionManager.Cookie.Path,
641×						Domain:   config.SessionManager.Cookie.Domain,
651×						Secure:   config.SessionManager.Cookie.Secure,
661×						HttpOnly: config.SessionManager.Cookie.HttpOnly,
671×						SameSite: config.SessionManager.Cookie.SameSite,
681×					}
691×
701×					switch config.SessionManager.Status(ctx) {
711×					case scs.Modified:
721×						token, _, err := config.SessionManager.Commit(ctx)
731×						if err != nil {
74✗							panic(err)
75						}
76
771×						responseCookie.Value = token
78
79✗					case scs.Destroyed:
80✗						responseCookie.Expires = time.Unix(1, 0)
81✗						responseCookie.MaxAge = -1
⋯15 lines, fully covered · expand
82					}
83
841×					c.SetCookie(responseCookie)
851×					addHeaderIfMissing(c.Response(), "Cache-Control", `no-cache="Set-Cookie"`)
861×					addHeaderIfMissing(c.Response(), "Vary", "Cookie")
87				}
88			})
89
902×			return next(c)
91		}
92	}
93}
94
952×func addHeaderIfMissing(w http.ResponseWriter, key, value string) {
962×	for _, h := range w.Header()[key] {
97✗		if h == value {
98✗			return
99✗		}
100	}
1012×	w.Header().Add(key, value)
102}

Executed, with hit count Never executed Not a statement The rail maps all 102 lines of the file.