Added cube middleware

Signed-off-by: Vishal Rana <[email protected]>

Author: vishr

Gopkg.lock

@@ -1,83 +1,38 @@
 
-## Gopkg.toml example (these lines may be deleted)
-
-## "metadata" defines metadata about the project that could be used by other independent
-## systems. The metadata defined here will be ignored by dep.
-# [metadata]
-# key1 = "value that convey data to other systems"
-# system1-data = "value that is used by a system"
-# system2-data = "value that is used by another system"
-
-## "required" lists a set of packages (not projects) that must be included in
-## Gopkg.lock. This list is merged with the set of packages imported by the current
-## project. Use it when your project needs a package it doesn't explicitly import -
-## including "main" packages.
-# required = ["github.com/user/thing/cmd/thing"]
-
-## "ignored" lists a set of packages (not projects) that are ignored when
-## dep statically analyzes source code. Ignored packages can be in this project,
-## or in a dependency.
-# ignored = ["github.com/user/project/badpkg"]
-
-## Constraints are rules for how directly imported projects
-## may be incorporated into the depgraph. They are respected by
-## dep whether coming from the Gopkg.toml of the current project or a dependency.
-# [[constraint]]
-## Required: the root import path of the project being constrained.
-# name = "github.com/user/project"
+# Gopkg.toml example
 #
-## Recommended: the version constraint to enforce for the project.
-## Only one of "branch", "version" or "revision" can be specified.
-# version = "1.0.0"
-# branch = "master"
-# revision = "abc123"
+# Refer to https://github.com/golang/dep/blob/master/docs/Gopkg.toml.md
+# for detailed Gopkg.toml documentation.
 #
-## Optional: an alternate location (URL or import path) for the project's source.
-# source = "https://github.com/myfork/package.git"
+# required = ["github.com/user/thing/cmd/thing"]
+# ignored = ["github.com/user/project/pkgX", "bitbucket.org/user/project/pkgA/pkgY"]
 #
-## "metadata" defines metadata about the dependency or override that could be used
-## by other independent systems. The metadata defined here will be ignored by dep.
-# [metadata]
-# key1 = "value that convey data to other systems"
-# system1-data = "value that is used by a system"
-# system2-data = "value that is used by another system"
-
-## Overrides have the same structure as [[constraint]], but supersede all
-## [[constraint]] declarations from all projects. Only [[override]] from
-## the current project's are applied.
-##
-## Overrides are a sledgehammer. Use them only as a last resort.
-# [[override]]
-## Required: the root import path of the project being constrained.
-# name = "github.com/user/project"
+# [[constraint]]
+#   name = "github.com/user/project"
+#   version = "1.0.0"
 #
-## Optional: specifying a version constraint override will cause all other
-## constraints on this project to be ignored; only the overridden constraint
-## need be satisfied.
-## Again, only one of "branch", "version" or "revision" can be specified.
-# version = "1.0.0"
-# branch = "master"
-# revision = "abc123"
+# [[constraint]]
+#   name = "github.com/user/project2"
+#   branch = "dev"
+#   source = "github.com/myfork/project2"
 #
-## Optional: specifying an alternate source location as an override will
-## enforce that the alternate location is used for that project, regardless of
-## what source location any dependent projects specify.
-# source = "https://github.com/myfork/package.git"
-
+# [[override]]
+#  name = "github.com/x/y"
+#  version = "2.4.0"
 
 
 [[constraint]]
   name = "github.com/casbin/casbin"
-  version = "0.2.0"
+  version = "0.8.0"
 
 [[constraint]]
-  branch = "master"
-  name = "github.com/labstack/echo"
+  name = "github.com/gorilla/sessions"
+  version = "1.1.0"
 
 [[constraint]]
-  name = "github.com/labstack/gommon"
-  version = "0.2.1"
+  name = "github.com/labstack/echo"
+  version = "3.2.1"
 
 [[constraint]]
   branch = "master"
-  name = "golang.org/x/crypto"
+  name = "github.com/labstack/labstack-go"

Gopkg.toml

@@ -0,0 +1,92 @@
+package cube
+
+import (
+	"time"
+
+	"github.com/labstack/echo"
+	"github.com/labstack/echo/middleware"
+	"github.com/labstack/labstack-go"
+)
+
+type (
+	// Config defines the config for Cube middleware.
+	Config struct {
+		// Skipper defines a function to skip middleware.
+		Skipper middleware.Skipper
+
+		// Node name
+		Node string `json:"node"`
+
+		// Node group
+		Group string `json:"group"`
+
+		// LabStack API key
+		APIKey string `json:"api_key"`
+
+		// Number of requests in a batch
+		BatchSize int `json:"batch_size"`
+
+		// Interval in seconds to dispatch the batch
+		DispatchInterval time.Duration `json:"dispatch_interval"`
+
+		// TODO: To be implemented
+		ClientLookup string `json:"client_lookup"`
+	}
+)
+
+var (
+	// DefaultConfig is the default Cube middleware config.
+	DefaultConfig = Config{
+		Skipper:          middleware.DefaultSkipper,
+		BatchSize:        60,
+		DispatchInterval: 60,
+	}
+)
+
+// Middleware implements Cube middleware.
+func Middleware(apiKey string) echo.MiddlewareFunc {
+	c := DefaultConfig
+	c.APIKey = apiKey
+	return MiddlewareWithConfig(c)
+}
+
+// MiddlewareWithConfig returns a Cube middleware with config.
+// See: `Middleware()`.
+func MiddlewareWithConfig(config Config) echo.MiddlewareFunc {
+	// Defaults
+	if config.APIKey == "" {
+		panic("echo: cube middleware requires an api key")
+	}
+	if config.Skipper == nil {
+		config.Skipper = DefaultConfig.Skipper
+	}
+	if config.BatchSize == 0 {
+		config.BatchSize = DefaultConfig.BatchSize
+	}
+	if config.DispatchInterval == 0 {
+		config.DispatchInterval = DefaultConfig.DispatchInterval
+	}
+
+	// Initialize
+	cube := labstack.NewClient(config.APIKey).Cube()
+	cube.Node = config.Node
+	cube.Group = config.Group
+	cube.APIKey = config.APIKey
+	cube.BatchSize = config.BatchSize
+	cube.DispatchInterval = config.DispatchInterval
+	cube.ClientLookup = config.ClientLookup
+
+	return func(next echo.HandlerFunc) echo.HandlerFunc {
+		return func(c echo.Context) (err error) {
+			if config.Skipper(c) {
+				return next(c)
+			}
+			request := cube.Start(c.Request(), c.Response())
+			if err = next(c); err != nil {
+				c.Error(err)
+			}
+			cube.Stop(request, c.Response().Status, c.Response().Size)
+			return
+		}
+	}
+}