Add 'ocs/' from commit '7ca52baa61c4370a1bad0e2f74e85073798bdde9'

git-subtree-dir: ocs
git-subtree-mainline: b274cac8c2
git-subtree-split: 7ca52baa61

ocs/pkg/middleware/accesstoken.go

@@ -0,0 +1,40 @@
+package middleware
+
+import (
+	"net/http"
+
+	"github.com/cs3org/reva/pkg/token/manager/jwt"
+	"github.com/cs3org/reva/pkg/user"
+)
+
+// AccessToken middleware is used to set the user from an x-access-token to the context
+func AccessToken(opts ...Option) func(next http.Handler) http.Handler {
+	opt := newOptions(opts...)
+
+	return func(next http.Handler) http.Handler {
+		// TODO: handle error
+		tokenManager, err := jwt.New(map[string]interface{}{
+			"secret":  opt.TokenManagerConfig.JWTSecret,
+			"expires": int64(60),
+		})
+		if err != nil {
+			opt.Logger.Fatal().Err(err).Msgf("Could not initialize token-manager")
+		}
+
+		return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+			token := r.Header.Get("x-access-token")
+			if token != "" {
+				u, err := tokenManager.DismantleToken(r.Context(), token)
+				if err != nil {
+					opt.Logger.Error().Err(err).Msg("could not dismantle token")
+					w.WriteHeader(http.StatusInternalServerError)
+					return
+				}
+				// store user in context for request
+				r = r.WithContext(user.ContextSetUser(r.Context(), u))
+			}
+
+			next.ServeHTTP(w, r)
+		})
+	}
+}

ocs/pkg/middleware/format.go

@@ -0,0 +1,24 @@
+package middleware
+
+import (
+	"context"
+	"net/http"
+
+	"github.com/go-chi/render"
+)
+
+// OCSFormatCtx middleware is used to determine the content type from
+// the format URL parameter passed in an ocs request. Defaults to XML
+func OCSFormatCtx(next http.Handler) http.Handler {
+	return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		switch r.URL.Query().Get("format") {
+		case "", "xml":
+			r.Header.Set("Accept", "application/xml")
+			r = r.WithContext(context.WithValue(r.Context(), render.ContentTypeCtxKey, render.ContentTypeXML))
+		case "json":
+			r.Header.Set("Accept", "application/json")
+			r = r.WithContext(context.WithValue(r.Context(), render.ContentTypeCtxKey, render.ContentTypeJSON))
+		}
+		next.ServeHTTP(w, r)
+	})
+}

ocs/pkg/middleware/options.go

@@ -0,0 +1,42 @@
+package middleware
+
+import (
+	"github.com/owncloud/ocis-ocs/pkg/config"
+	"github.com/owncloud/ocis-pkg/v2/log"
+)
+
+// Option defines a single option function.
+type Option func(o *Options)
+
+// Options defines the available options for this package.
+type Options struct {
+	// Logger to use for logging, must be set
+	Logger log.Logger
+	// TokenManagerConfig for communicating with the reva token manager
+	TokenManagerConfig config.TokenManager
+}
+
+// newOptions initializes the available default options.
+func newOptions(opts ...Option) Options {
+	opt := Options{}
+
+	for _, o := range opts {
+		o(&opt)
+	}
+
+	return opt
+}
+
+// Logger provides a function to set the logger option.
+func Logger(l log.Logger) Option {
+	return func(o *Options) {
+		o.Logger = l
+	}
+}
+
+// TokenManagerConfig provides a function to set the token manger config option.
+func TokenManagerConfig(cfg config.TokenManager) Option {
+	return func(o *Options) {
+		o.TokenManagerConfig = cfg
+	}
+}