Removed echo-contrib dependency.

pkg/middleware/session.go

@@ -0,0 +1,19 @@
+package middleware
+
+import (
+	"github.com/gorilla/context"
+	"github.com/gorilla/sessions"
+	"github.com/labstack/echo/v4"
+	"github.com/mikestefanello/pagoda/pkg/session"
+)
+
+// Session sets the session storage in the request context
+func Session(store sessions.Store) echo.MiddlewareFunc {
+	return func(next echo.HandlerFunc) echo.HandlerFunc {
+		return func(ctx echo.Context) error {
+			defer context.Clear(ctx.Request())
+			session.Store(ctx, store)
+			return next(ctx)
+		}
+	}
+}

pkg/middleware/session_test.go

@@ -0,0 +1,24 @@
+package middleware
+
+import (
+	"testing"
+
+	"github.com/gorilla/sessions"
+	"github.com/mikestefanello/pagoda/pkg/session"
+	"github.com/mikestefanello/pagoda/pkg/tests"
+	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/require"
+)
+
+func TestSession(t *testing.T) {
+	ctx, _ := tests.NewContext(c.Web, "/")
+	_, err := session.Get(ctx, "test")
+	assert.Equal(t, session.ErrStoreNotFound, err)
+
+	store := sessions.NewCookieStore([]byte("secret"))
+	err = tests.ExecuteMiddleware(ctx, Session(store))
+	require.NoError(t, err)
+
+	_, err = session.Get(ctx, "test")
+	assert.NotEqual(t, session.ErrStoreNotFound, err)
+}