storj/pkg/satellite/satelliteweb/satelliteql/token.go

// Copyright (C) 2018 Storj Labs, Inc.
// See LICENSE for copying information.

package satelliteql

import (
	"github.com/graphql-go/graphql"

	"storj.io/storj/pkg/auth"
	"storj.io/storj/pkg/satellite"
)

const (
	tokenType = "token"
)

// graphqlToken creates *graphql.Object type that encapsulates user and token string
func graphqlToken(service *satellite.Service, types Types) *graphql.Object {
	return graphql.NewObject(graphql.ObjectConfig{
		Name: tokenType,
		Fields: graphql.Fields{
			tokenType: &graphql.Field{
				Type: graphql.String,
			},
			userType: &graphql.Field{
				Type: types.User(),
				Resolve: func(p graphql.ResolveParams) (interface{}, error) {
					wrapper, _ := p.Source.(tokenWrapper)

					ctx := auth.WithAPIKey(p.Context, []byte(wrapper.Token))

					auth, err := service.Authorize(ctx)
					if err != nil {
						return nil, err
					}

					// pass context to root value so child resolvers could get auth auth
					rootValue := p.Info.RootValue.(map[string]interface{})
					rootValue["context"] = satellite.WithAuth(ctx, auth)

					return &auth.User, nil
				},
			},
		},
	})
}

// tokenWrapper holds token string value so it can be parsed by graphql pkg
type tokenWrapper struct {
	Token string
}