identity_provider.go 1.4 KB
Newer Older
H
update  
hongming 已提交
1
/*
H
hongming 已提交
2 3 4 5 6 7 8 9 10 11 12 13 14 15
Copyright 2020 The KubeSphere Authors.

Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at

    http://www.apache.org/licenses/LICENSE-2.0

Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
*/
H
update  
hongming 已提交
16

17
package identityprovider
H
update  
hongming 已提交
18

H
update  
hongming 已提交
19 20 21 22
import (
	"errors"
	"kubesphere.io/kubesphere/pkg/apiserver/authentication/oauth"
)
H
update  
hongming 已提交
23

H
update  
hongming 已提交
24 25
var (
	ErrorIdentityProviderNotFound = errors.New("the identity provider was not found")
H
hongming 已提交
26
	oauthProviders                = make(map[string]OAuthProvider, 0)
H
update  
hongming 已提交
27 28 29 30
)

type OAuthProvider interface {
	Type() string
H
hongming 已提交
31 32
	Setup(options *oauth.DynamicOptions) (OAuthProvider, error)
	IdentityExchange(code string) (Identity, error)
H
update  
hongming 已提交
33
}
H
hongming 已提交
34 35 36
type Identity interface {
	GetName() string
	GetEmail() string
H
update  
hongming 已提交
37 38
}

H
hongming 已提交
39 40 41
func GetOAuthProvider(providerType string, options *oauth.DynamicOptions) (OAuthProvider, error) {
	if provider, ok := oauthProviders[providerType]; ok {
		return provider.Setup(options)
H
update  
hongming 已提交
42 43 44 45
	}
	return nil, ErrorIdentityProviderNotFound
}

H
hongming 已提交
46 47
func RegisterOAuthProviderCodec(provider OAuthProvider) {
	oauthProviders[provider.Type()] = provider
H
update  
hongming 已提交
48
}