userconversion.go

package registry

import (
	"errors"

	"github.com/openshift/origin/pkg/auth/api"

	oapi "github.com/openshift/origin/pkg/oauth/api"
)

type UserConversion struct{}

// NewUserConversion creates an object that can convert the UserInfo object to and from
// an oauth access/authorize token object.
func NewUserConversion() *UserConversion {
	return &UserConversion{}
}

func (s *UserConversion) ConvertToAuthorizeToken(user interface{}, token *oapi.AuthorizeToken) error {
	info, ok := user.(api.UserInfo)
	if !ok {
		return errors.New("did not receive UserInfo")
	}
	token.UserName = info.GetName()
	if token.UserName == "" {
		return errors.New("user name is empty")
	}
	token.UserUID = info.GetUID()
	return nil
}

func (s *UserConversion) ConvertToAccessToken(user interface{}, token *oapi.AccessToken) error {
	return s.ConvertToAuthorizeToken(user, &token.AuthorizeToken)
}

func (s *UserConversion) ConvertFromAuthorizeToken(token *oapi.AuthorizeToken) (interface{}, error) {
	if token.UserName == "" {
		return nil, errors.New("token has no user name stored")
	}
	return &api.DefaultUserInfo{
		Name: token.UserName,
		UID:  token.UserUID,
	}, nil
}

func (s *UserConversion) ConvertFromAccessToken(token *oapi.AccessToken) (interface{}, error) {
	return s.ConvertFromAuthorizeToken(&token.AuthorizeToken)
}