opencloud/proxy/pkg/middleware/create_home.go

package middleware

import (
	"net/http"

	rpc "github.com/cs3org/go-cs3apis/cs3/rpc/v1beta1"
	provider "github.com/cs3org/go-cs3apis/cs3/storage/provider/v1beta1"
	"github.com/cs3org/reva/pkg/rgrpc/status"
	tokenpkg "github.com/cs3org/reva/pkg/token"
	"github.com/cs3org/reva/pkg/token/manager/jwt"
	"github.com/micro/go-micro/v2/errors"
	"github.com/owncloud/ocis/accounts/pkg/proto/v0"
	"google.golang.org/grpc/metadata"
)

// CreateHome provides a middleware which sends a CreateHome request to the reva gateway
func CreateHome(opts ...Option) func(next http.Handler) http.Handler {
	opt := newOptions(opts...)

	return func(next http.Handler) http.Handler {
		return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
			accounts := opt.AccountsClient

			tokenManager, err := jwt.New(map[string]interface{}{
				"secret": opt.TokenManagerConfig.JWTSecret,
			})
			if err != nil {
				opt.Logger.Error().Err(err).Msg("error creating a token manager")
				w.WriteHeader(http.StatusInternalServerError)
				return
			}

			token := r.Header.Get("x-access-token")
			if token == "" {
				next.ServeHTTP(w, r)
				return
			}

			user, err := tokenManager.DismantleToken(r.Context(), token)
			if err != nil {
				opt.Logger.Err(err).Msg("error getting user from access token")
				w.WriteHeader(http.StatusInternalServerError)
				return
			}
			_, err = accounts.GetAccount(r.Context(), &proto.GetAccountRequest{
				Id: user.Id.OpaqueId,
			})
			if err != nil {
				e := errors.Parse(err.Error())
				if e.Code == http.StatusNotFound {
					opt.Logger.Debug().Msgf("account with id %s not found", user.Id.OpaqueId)
					next.ServeHTTP(w, r)
					return
				}
				opt.Logger.Err(err).Msgf("error getting user with id %s from accounts service", user.Id.OpaqueId)
				w.WriteHeader(http.StatusInternalServerError)
				return
			}

			// we need to pass the token to authenticate the CreateHome request.
			//ctx := tokenpkg.ContextSetToken(r.Context(), token)
			ctx := metadata.AppendToOutgoingContext(r.Context(), tokenpkg.TokenHeader, token)

			createHomeReq := &provider.CreateHomeRequest{}
			createHomeRes, err := opt.RevaGatewayClient.CreateHome(ctx, createHomeReq)

			if err != nil {
				opt.Logger.Err(err).Msg("error calling CreateHome")
			} else if createHomeRes.Status.Code != rpc.Code_CODE_OK {
				err := status.NewErrorFromCode(createHomeRes.Status.Code, "gateway")
				opt.Logger.Err(err).Msg("error when calling Createhome")
			}

			next.ServeHTTP(w, r)
		})
	}
}