mirror of
https://github.com/grafana/grafana.git
synced 2025-02-20 11:48:34 -06:00
d3a269ab3b
* Return data in camelCase from the OAuth fb strategy * changes * wip * Add defaults for oauth fb strategy * revert other changes * Add tests * Add Defaults to cfg and use it in OAuthStrategy * Return *OAuthInfo from OAuthStrategy * lint * Remove unnecessary Defaults * Introduce const for fields, fix import order * Align failing tests * clean up * Changes requested by @gamab * Update pkg/services/ssosettings/strategies/oauth_strategy_test.go Co-authored-by: Gabriel MABILLE <gamab@users.noreply.github.com> * Load data on startup * Rename + simplify --------- Co-authored-by: Gabriel MABILLE <gamab@users.noreply.github.com>
103 lines
4.2 KiB
Go
103 lines
4.2 KiB
Go
package strategies
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/grafana/grafana/pkg/login/social"
|
|
"github.com/grafana/grafana/pkg/services/ssosettings"
|
|
"github.com/grafana/grafana/pkg/setting"
|
|
"github.com/grafana/grafana/pkg/util"
|
|
)
|
|
|
|
type OAuthStrategy struct {
|
|
cfg *setting.Cfg
|
|
settingsByProvider map[string]*social.OAuthInfo
|
|
}
|
|
|
|
var extraKeysByProvider = map[string][]string{
|
|
social.AzureADProviderName: social.ExtraAzureADSettingKeys,
|
|
social.GenericOAuthProviderName: social.ExtraGenericOAuthSettingKeys,
|
|
social.GitHubProviderName: social.ExtraGithubSettingKeys,
|
|
social.GrafanaComProviderName: social.ExtraGrafanaComSettingKeys,
|
|
social.GrafanaNetProviderName: social.ExtraGrafanaComSettingKeys,
|
|
}
|
|
|
|
var _ ssosettings.FallbackStrategy = (*OAuthStrategy)(nil)
|
|
|
|
func NewOAuthStrategy(cfg *setting.Cfg) *OAuthStrategy {
|
|
oauthStrategy := &OAuthStrategy{
|
|
cfg: cfg,
|
|
settingsByProvider: make(map[string]*social.OAuthInfo),
|
|
}
|
|
|
|
oauthStrategy.loadAllSettings()
|
|
return oauthStrategy
|
|
}
|
|
|
|
func (s *OAuthStrategy) IsMatch(provider string) bool {
|
|
_, ok := s.settingsByProvider[provider]
|
|
return ok
|
|
}
|
|
|
|
func (s *OAuthStrategy) GetProviderConfig(_ context.Context, provider string) (any, error) {
|
|
return s.settingsByProvider[provider], nil
|
|
}
|
|
|
|
func (s *OAuthStrategy) loadAllSettings() {
|
|
allProviders := append(ssosettings.AllOAuthProviders, social.GrafanaNetProviderName)
|
|
for _, provider := range allProviders {
|
|
settings := s.loadSettingsForProvider(provider)
|
|
if provider == social.GrafanaNetProviderName {
|
|
provider = social.GrafanaComProviderName
|
|
}
|
|
s.settingsByProvider[provider] = settings
|
|
}
|
|
}
|
|
|
|
func (s *OAuthStrategy) loadSettingsForProvider(provider string) *social.OAuthInfo {
|
|
section := s.cfg.SectionWithEnvOverrides("auth." + provider)
|
|
|
|
result := &social.OAuthInfo{
|
|
AllowAssignGrafanaAdmin: section.Key("allow_assign_grafana_admin").MustBool(false),
|
|
AllowSignup: section.Key("allow_sign_up").MustBool(false),
|
|
AllowedDomains: util.SplitString(section.Key("allowed_domains").Value()),
|
|
AllowedGroups: util.SplitString(section.Key("allowed_groups").Value()),
|
|
ApiUrl: section.Key("api_url").Value(),
|
|
AuthStyle: section.Key("auth_style").Value(),
|
|
AuthUrl: section.Key("auth_url").Value(),
|
|
AutoLogin: section.Key("auto_login").MustBool(false),
|
|
ClientId: section.Key("client_id").Value(),
|
|
ClientSecret: section.Key("client_secret").Value(),
|
|
EmailAttributeName: section.Key("email_attribute_name").Value(),
|
|
EmailAttributePath: section.Key("email_attribute_path").Value(),
|
|
EmptyScopes: section.Key("empty_scopes").MustBool(false),
|
|
Enabled: section.Key("enabled").MustBool(false),
|
|
GroupsAttributePath: section.Key("groups_attribute_path").Value(),
|
|
HostedDomain: section.Key("hosted_domain").Value(),
|
|
Icon: section.Key("icon").Value(),
|
|
Name: section.Key("name").Value(),
|
|
RoleAttributePath: section.Key("role_attribute_path").Value(),
|
|
RoleAttributeStrict: section.Key("role_attribute_strict").MustBool(false),
|
|
Scopes: util.SplitString(section.Key("scopes").Value()),
|
|
SignoutRedirectUrl: section.Key("signout_redirect_url").Value(),
|
|
SkipOrgRoleSync: section.Key("skip_org_role_sync").MustBool(false),
|
|
TeamIdsAttributePath: section.Key("team_ids_attribute_path").Value(),
|
|
TeamsUrl: section.Key("teams_url").Value(),
|
|
TlsClientCa: section.Key("tls_client_ca").Value(),
|
|
TlsClientCert: section.Key("tls_client_cert").Value(),
|
|
TlsClientKey: section.Key("tls_client_key").Value(),
|
|
TlsSkipVerify: section.Key("tls_skip_verify_insecure").MustBool(false),
|
|
TokenUrl: section.Key("token_url").Value(),
|
|
UsePKCE: section.Key("use_pkce").MustBool(false),
|
|
UseRefreshToken: section.Key("use_refresh_token").MustBool(false),
|
|
Extra: map[string]string{},
|
|
}
|
|
|
|
extraFields := extraKeysByProvider[provider]
|
|
for _, key := range extraFields {
|
|
result.Extra[key] = section.Key(key).Value()
|
|
}
|
|
|
|
return result
|
|
}
|