mirror of
				https://github.com/go-gitea/gitea
				synced 2025-11-04 05:18:25 +00:00 
			
		
		
		
	Refactor: Move login out of models (#16199)
`models` does far too much. In particular it handles all `UserSignin`. It shouldn't be responsible for calling LDAP, SMTP or PAM for signing in. Therefore we should move this code out of `models`. This code has to depend on `models` - therefore it belongs in `services`. There is a package in `services` called `auth` and clearly this functionality belongs in there. Plan: - [x] Change `auth.Auth` to `auth.Method` - as they represent methods of authentication. - [x] Move `models.UserSignIn` into `auth` - [x] Move `models.ExternalUserLogin` - [x] Move most of the `LoginVia*` methods to `auth` or subpackages - [x] Move Resynchronize functionality to `auth` - Involved some restructuring of `models/ssh_key.go` to reduce the size of this massive file and simplify its files. - [x] Move the rest of the LDAP functionality in to the ldap subpackage - [x] Re-factor the login sources to express an interfaces `auth.Source`? - I've done this through some smaller interfaces Authenticator and Synchronizable - which would allow us to extend things in future - [x] Now LDAP is out of models - need to think about modules/auth/ldap and I think all of that functionality might just be moveable - [x] Similarly a lot Oauth2 functionality need not be in models too and should be moved to services/auth/source/oauth2 - [x] modules/auth/oauth2/oauth2.go uses xorm... This is naughty - probably need to move this into models. - [x] models/oauth2.go - mostly should be in modules/auth/oauth2 or services/auth/source/oauth2 - [x] More simplifications of login_source.go may need to be done - Allow wiring in of notify registration - *this can now easily be done - but I think we should do it in another PR* - see #16178 - More refactors...? - OpenID should probably become an auth Method but I think that can be left for another PR - Methods should also probably be cleaned up - again another PR I think. - SSPI still needs more refactors.* Rename auth.Auth auth.Method * Restructure ssh_key.go - move functions from models/user.go that relate to ssh_key to ssh_key - split ssh_key.go to try create clearer function domains for allow for future refactors here. Signed-off-by: Andrew Thornton <art27@cantab.net>
This commit is contained in:
		
							
								
								
									
										154
									
								
								models/oauth2.go
									
									
									
									
									
								
							
							
						
						
									
										154
									
								
								models/oauth2.go
									
									
									
									
									
								
							@@ -4,89 +4,10 @@
 | 
			
		||||
 | 
			
		||||
package models
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"sort"
 | 
			
		||||
 | 
			
		||||
	"code.gitea.io/gitea/modules/auth/oauth2"
 | 
			
		||||
	"code.gitea.io/gitea/modules/log"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
// OAuth2Provider describes the display values of a single OAuth2 provider
 | 
			
		||||
type OAuth2Provider struct {
 | 
			
		||||
	Name             string
 | 
			
		||||
	DisplayName      string
 | 
			
		||||
	Image            string
 | 
			
		||||
	CustomURLMapping *oauth2.CustomURLMapping
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// OAuth2Providers contains the map of registered OAuth2 providers in Gitea (based on goth)
 | 
			
		||||
// key is used to map the OAuth2Provider with the goth provider type (also in LoginSource.OAuth2Config.Provider)
 | 
			
		||||
// value is used to store display data
 | 
			
		||||
var OAuth2Providers = map[string]OAuth2Provider{
 | 
			
		||||
	"bitbucket": {Name: "bitbucket", DisplayName: "Bitbucket", Image: "/assets/img/auth/bitbucket.png"},
 | 
			
		||||
	"dropbox":   {Name: "dropbox", DisplayName: "Dropbox", Image: "/assets/img/auth/dropbox.png"},
 | 
			
		||||
	"facebook":  {Name: "facebook", DisplayName: "Facebook", Image: "/assets/img/auth/facebook.png"},
 | 
			
		||||
	"github": {
 | 
			
		||||
		Name: "github", DisplayName: "GitHub", Image: "/assets/img/auth/github.png",
 | 
			
		||||
		CustomURLMapping: &oauth2.CustomURLMapping{
 | 
			
		||||
			TokenURL:   oauth2.GetDefaultTokenURL("github"),
 | 
			
		||||
			AuthURL:    oauth2.GetDefaultAuthURL("github"),
 | 
			
		||||
			ProfileURL: oauth2.GetDefaultProfileURL("github"),
 | 
			
		||||
			EmailURL:   oauth2.GetDefaultEmailURL("github"),
 | 
			
		||||
		},
 | 
			
		||||
	},
 | 
			
		||||
	"gitlab": {
 | 
			
		||||
		Name: "gitlab", DisplayName: "GitLab", Image: "/assets/img/auth/gitlab.png",
 | 
			
		||||
		CustomURLMapping: &oauth2.CustomURLMapping{
 | 
			
		||||
			TokenURL:   oauth2.GetDefaultTokenURL("gitlab"),
 | 
			
		||||
			AuthURL:    oauth2.GetDefaultAuthURL("gitlab"),
 | 
			
		||||
			ProfileURL: oauth2.GetDefaultProfileURL("gitlab"),
 | 
			
		||||
		},
 | 
			
		||||
	},
 | 
			
		||||
	"gplus":         {Name: "gplus", DisplayName: "Google", Image: "/assets/img/auth/google.png"},
 | 
			
		||||
	"openidConnect": {Name: "openidConnect", DisplayName: "OpenID Connect", Image: "/assets/img/auth/openid_connect.svg"},
 | 
			
		||||
	"twitter":       {Name: "twitter", DisplayName: "Twitter", Image: "/assets/img/auth/twitter.png"},
 | 
			
		||||
	"discord":       {Name: "discord", DisplayName: "Discord", Image: "/assets/img/auth/discord.png"},
 | 
			
		||||
	"gitea": {
 | 
			
		||||
		Name: "gitea", DisplayName: "Gitea", Image: "/assets/img/auth/gitea.png",
 | 
			
		||||
		CustomURLMapping: &oauth2.CustomURLMapping{
 | 
			
		||||
			TokenURL:   oauth2.GetDefaultTokenURL("gitea"),
 | 
			
		||||
			AuthURL:    oauth2.GetDefaultAuthURL("gitea"),
 | 
			
		||||
			ProfileURL: oauth2.GetDefaultProfileURL("gitea"),
 | 
			
		||||
		},
 | 
			
		||||
	},
 | 
			
		||||
	"nextcloud": {
 | 
			
		||||
		Name: "nextcloud", DisplayName: "Nextcloud", Image: "/assets/img/auth/nextcloud.png",
 | 
			
		||||
		CustomURLMapping: &oauth2.CustomURLMapping{
 | 
			
		||||
			TokenURL:   oauth2.GetDefaultTokenURL("nextcloud"),
 | 
			
		||||
			AuthURL:    oauth2.GetDefaultAuthURL("nextcloud"),
 | 
			
		||||
			ProfileURL: oauth2.GetDefaultProfileURL("nextcloud"),
 | 
			
		||||
		},
 | 
			
		||||
	},
 | 
			
		||||
	"yandex": {Name: "yandex", DisplayName: "Yandex", Image: "/assets/img/auth/yandex.png"},
 | 
			
		||||
	"mastodon": {
 | 
			
		||||
		Name: "mastodon", DisplayName: "Mastodon", Image: "/assets/img/auth/mastodon.png",
 | 
			
		||||
		CustomURLMapping: &oauth2.CustomURLMapping{
 | 
			
		||||
			AuthURL: oauth2.GetDefaultAuthURL("mastodon"),
 | 
			
		||||
		},
 | 
			
		||||
	},
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// OAuth2DefaultCustomURLMappings contains the map of default URL's for OAuth2 providers that are allowed to have custom urls
 | 
			
		||||
// key is used to map the OAuth2Provider
 | 
			
		||||
// value is the mapping as defined for the OAuth2Provider
 | 
			
		||||
var OAuth2DefaultCustomURLMappings = map[string]*oauth2.CustomURLMapping{
 | 
			
		||||
	"github":    OAuth2Providers["github"].CustomURLMapping,
 | 
			
		||||
	"gitlab":    OAuth2Providers["gitlab"].CustomURLMapping,
 | 
			
		||||
	"gitea":     OAuth2Providers["gitea"].CustomURLMapping,
 | 
			
		||||
	"nextcloud": OAuth2Providers["nextcloud"].CustomURLMapping,
 | 
			
		||||
	"mastodon":  OAuth2Providers["mastodon"].CustomURLMapping,
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// GetActiveOAuth2ProviderLoginSources returns all actived LoginOAuth2 sources
 | 
			
		||||
func GetActiveOAuth2ProviderLoginSources() ([]*LoginSource, error) {
 | 
			
		||||
	sources := make([]*LoginSource, 0, 1)
 | 
			
		||||
	if err := x.Where("is_actived = ? and type = ?", true, LoginOAuth2).Find(&sources); err != nil {
 | 
			
		||||
	if err := x.Where("is_active = ? and type = ?", true, LoginOAuth2).Find(&sources); err != nil {
 | 
			
		||||
		return nil, err
 | 
			
		||||
	}
 | 
			
		||||
	return sources, nil
 | 
			
		||||
@@ -95,81 +16,10 @@ func GetActiveOAuth2ProviderLoginSources() ([]*LoginSource, error) {
 | 
			
		||||
// GetActiveOAuth2LoginSourceByName returns a OAuth2 LoginSource based on the given name
 | 
			
		||||
func GetActiveOAuth2LoginSourceByName(name string) (*LoginSource, error) {
 | 
			
		||||
	loginSource := new(LoginSource)
 | 
			
		||||
	has, err := x.Where("name = ? and type = ? and is_actived = ?", name, LoginOAuth2, true).Get(loginSource)
 | 
			
		||||
	has, err := x.Where("name = ? and type = ? and is_active = ?", name, LoginOAuth2, true).Get(loginSource)
 | 
			
		||||
	if !has || err != nil {
 | 
			
		||||
		return nil, err
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	return loginSource, nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// GetActiveOAuth2Providers returns the map of configured active OAuth2 providers
 | 
			
		||||
// key is used as technical name (like in the callbackURL)
 | 
			
		||||
// values to display
 | 
			
		||||
func GetActiveOAuth2Providers() ([]string, map[string]OAuth2Provider, error) {
 | 
			
		||||
	// Maybe also separate used and unused providers so we can force the registration of only 1 active provider for each type
 | 
			
		||||
 | 
			
		||||
	loginSources, err := GetActiveOAuth2ProviderLoginSources()
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return nil, nil, err
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	var orderedKeys []string
 | 
			
		||||
	providers := make(map[string]OAuth2Provider)
 | 
			
		||||
	for _, source := range loginSources {
 | 
			
		||||
		prov := OAuth2Providers[source.OAuth2().Provider]
 | 
			
		||||
		if source.OAuth2().IconURL != "" {
 | 
			
		||||
			prov.Image = source.OAuth2().IconURL
 | 
			
		||||
		}
 | 
			
		||||
		providers[source.Name] = prov
 | 
			
		||||
		orderedKeys = append(orderedKeys, source.Name)
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	sort.Strings(orderedKeys)
 | 
			
		||||
 | 
			
		||||
	return orderedKeys, providers, nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// InitOAuth2 initialize the OAuth2 lib and register all active OAuth2 providers in the library
 | 
			
		||||
func InitOAuth2() error {
 | 
			
		||||
	if err := oauth2.InitSigningKey(); err != nil {
 | 
			
		||||
		return err
 | 
			
		||||
	}
 | 
			
		||||
	if err := oauth2.Init(x); err != nil {
 | 
			
		||||
		return err
 | 
			
		||||
	}
 | 
			
		||||
	return initOAuth2LoginSources()
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// ResetOAuth2 clears existing OAuth2 providers and loads them from DB
 | 
			
		||||
func ResetOAuth2() error {
 | 
			
		||||
	oauth2.ClearProviders()
 | 
			
		||||
	return initOAuth2LoginSources()
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// initOAuth2LoginSources is used to load and register all active OAuth2 providers
 | 
			
		||||
func initOAuth2LoginSources() error {
 | 
			
		||||
	loginSources, _ := GetActiveOAuth2ProviderLoginSources()
 | 
			
		||||
	for _, source := range loginSources {
 | 
			
		||||
		oAuth2Config := source.OAuth2()
 | 
			
		||||
		err := oauth2.RegisterProvider(source.Name, oAuth2Config.Provider, oAuth2Config.ClientID, oAuth2Config.ClientSecret, oAuth2Config.OpenIDConnectAutoDiscoveryURL, oAuth2Config.CustomURLMapping)
 | 
			
		||||
		if err != nil {
 | 
			
		||||
			log.Critical("Unable to register source: %s due to Error: %v. This source will be disabled.", source.Name, err)
 | 
			
		||||
			source.IsActived = false
 | 
			
		||||
			if err = UpdateSource(source); err != nil {
 | 
			
		||||
				log.Critical("Unable to update source %s to disable it. Error: %v", err)
 | 
			
		||||
				return err
 | 
			
		||||
			}
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// wrapOpenIDConnectInitializeError is used to wrap the error but this cannot be done in modules/auth/oauth2
 | 
			
		||||
// inside oauth2: import cycle not allowed models -> modules/auth/oauth2 -> models
 | 
			
		||||
func wrapOpenIDConnectInitializeError(err error, providerName string, oAuth2Config *OAuth2Config) error {
 | 
			
		||||
	if err != nil && "openidConnect" == oAuth2Config.Provider {
 | 
			
		||||
		err = ErrOpenIDConnectInitialize{ProviderName: providerName, OpenIDConnectAutoDiscoveryURL: oAuth2Config.OpenIDConnectAutoDiscoveryURL, Cause: err}
 | 
			
		||||
	}
 | 
			
		||||
	return err
 | 
			
		||||
}
 | 
			
		||||
 
 | 
			
		||||
		Reference in New Issue
	
	Block a user