PageRenderTime 33ms CodeModel.GetById 28ms RepoModel.GetById 1ms app.codeStats 0ms

/app/helpers/auth_helper.rb

https://gitlab.com/alexcern/gitlab-ee
Ruby | 118 lines | 85 code | 28 blank | 5 comment | 3 complexity | 849184fedbe9aaaa23e63364317e203f MD5 | raw file
  1. # frozen_string_literal: true
  2. module AuthHelper
  3. PROVIDERS_WITH_ICONS = %w(twitter github gitlab bitbucket google_oauth2 facebook azure_oauth2 authentiq).freeze
  4. LDAP_PROVIDER = /\Aldap/.freeze
  5. def ldap_enabled?
  6. Gitlab::Auth::LDAP::Config.enabled?
  7. end
  8. def omniauth_enabled?
  9. Gitlab::Auth.omniauth_enabled?
  10. end
  11. def provider_has_icon?(name)
  12. PROVIDERS_WITH_ICONS.include?(name.to_s)
  13. end
  14. def qa_class_for_provider(provider)
  15. {
  16. saml: 'qa-saml-login-button',
  17. github: 'qa-github-login-button'
  18. }[provider.to_sym]
  19. end
  20. def auth_providers
  21. Gitlab::Auth::OAuth::Provider.providers
  22. end
  23. def label_for_provider(name)
  24. Gitlab::Auth::OAuth::Provider.label_for(name)
  25. end
  26. def form_based_provider_priority
  27. ['crowd', /^ldap/, 'kerberos']
  28. end
  29. def form_based_provider_with_highest_priority
  30. @form_based_provider_with_highest_priority ||= begin
  31. form_based_provider_priority.each do |provider_regexp|
  32. highest_priority = form_based_providers.find { |provider| provider.match?(provider_regexp) }
  33. break highest_priority unless highest_priority.nil?
  34. end
  35. end
  36. end
  37. def form_based_auth_provider_has_active_class?(provider)
  38. form_based_provider_with_highest_priority == provider
  39. end
  40. def form_based_provider?(name)
  41. [LDAP_PROVIDER, 'crowd'].any? { |pattern| pattern === name.to_s }
  42. end
  43. def form_based_providers
  44. auth_providers.select { |provider| form_based_provider?(provider) }
  45. end
  46. def crowd_enabled?
  47. auth_providers.include? :crowd
  48. end
  49. def button_based_providers
  50. auth_providers.reject { |provider| form_based_provider?(provider) }
  51. end
  52. def display_providers_on_profile?
  53. button_based_providers.any?
  54. end
  55. def providers_for_base_controller
  56. auth_providers.reject { |provider| LDAP_PROVIDER === provider }
  57. end
  58. def enabled_button_based_providers
  59. disabled_providers = Gitlab::CurrentSettings.disabled_oauth_sign_in_sources || []
  60. button_based_providers.map(&:to_s) - disabled_providers
  61. end
  62. def button_based_providers_enabled?
  63. enabled_button_based_providers.any?
  64. end
  65. def provider_image_tag(provider, size = 64)
  66. label = label_for_provider(provider)
  67. if provider_has_icon?(provider)
  68. file_name = "#{provider.to_s.split('_').first}_#{size}.png"
  69. image_tag("auth_buttons/#{file_name}", alt: label, title: "Sign in with #{label}")
  70. else
  71. label
  72. end
  73. end
  74. # rubocop: disable CodeReuse/ActiveRecord
  75. def auth_active?(provider)
  76. current_user.identities.exists?(provider: provider.to_s)
  77. end
  78. # rubocop: enable CodeReuse/ActiveRecord
  79. def unlink_provider_allowed?(provider)
  80. IdentityProviderPolicy.new(current_user, provider).can?(:unlink)
  81. end
  82. def link_provider_allowed?(provider)
  83. IdentityProviderPolicy.new(current_user, provider).can?(:link)
  84. end
  85. extend self
  86. end
  87. AuthHelper.prepend(EE::AuthHelper) # rubocop: disable Cop/InjectEnterpriseEditionModule
  88. # The methods added in EE should be available as both class and instance
  89. # methods, just like the methods provided by `AuthHelper` itself.
  90. AuthHelper.extend(EE::AuthHelper)