1 files changed, 90 insertions, 0 deletions
diff --git a/session.go b/session.go
new file mode 100644
index 0000000..ea5bbc7
--- /dev/null
+++ b/session.go
@@ -0,0 +1,90 @@
+package echosession
+
+import (
+	"context"
+	"sync"
+
+	"github.com/labstack/echo"
+	"gopkg.in/session.v2"
+)
+
+type (
+	// Skipper defines a function to skip middleware. Returning true skips processing
+	// the middleware.
+	Skipper func(c echo.Context) bool
+
+	// Config defines the config for Session middleware.
+	Config struct {
+		// Skipper defines a function to skip middleware.
+		Skipper Skipper
+		// StoreKey keys stored in the context
+		StoreKey string
+	}
+)
+
+var (
+	// DefaultConfig is the default Recover middleware config.
+	DefaultConfig = Config{
+		Skipper:  func(_ echo.Context) bool { return false },
+		StoreKey: "github.com/go-session/echo-session",
+	}
+
+	once            sync.Once
+	internalManager *session.Manager
+	storeKey        string
+)
+
+// get a session manager
+func manager(opt ...session.Option) *session.Manager {
+	once.Do(func() {
+		internalManager = session.NewManager(opt...)
+	})
+	return internalManager
+}
+
+// New create a session middleware
+func New(opt ...session.Option) echo.MiddlewareFunc {
+	return NewWithConfig(DefaultConfig, opt...)
+}
+
+// NewWithConfig create a session middleware
+func NewWithConfig(config Config, opt ...session.Option) echo.MiddlewareFunc {
+	if config.Skipper == nil {
+		config.Skipper = DefaultConfig.Skipper
+	}
+
+	storeKey = config.StoreKey
+	if storeKey == "" {
+		storeKey = DefaultConfig.StoreKey
+	}
+
+	return func(next echo.HandlerFunc) echo.HandlerFunc {
+		return func(c echo.Context) error {
+			if config.Skipper(c) {
+				return next(c)
+			}
+
+			store, err := manager(opt...).Start(context.Background(), c.Response(), c.Request())
+			if err != nil {
+				return err
+			}
+			c.Set(storeKey, store)
+			return next(c)
+		}
+	}
+}
+
+// FromContext get session storage from context
+func FromContext(ctx echo.Context) session.Store {
+	return ctx.Get(storeKey).(session.Store)
+}
+
+// Destroy a session
+func Destroy(ctx echo.Context) error {
+	return manager().Destroy(context.Background(), ctx.Response(), ctx.Request())
+}
+
+// Refresh a session and return to session storage
+func Refresh(ctx echo.Context) (session.Store, error) {
+	return manager().Refresh(context.Background(), ctx.Response(), ctx.Request())
+}