Don't show omniauth options in a space where they won't work

Author: fbacall

app/helpers/spaces_helper.rb

@@ -11,4 +11,8 @@ def space_feature_options
       [t("features.#{f}.short"), f]
     end
   end
+
+  def space_supports_omniauth?(space = current_space)
+    space.nil? || space.default? || space.is_subdomain?(TeSS::Config.base_uri.domain)
+  end
 end

app/views/layouts/_login_menu.html.erb

@@ -9,8 +9,10 @@
     <strong>Log In</strong> <span class="caret"></span>
   </a>
   <ul class="dropdown-menu dropdown-menu-right">
-    <% Devise.omniauth_configs.each do |provider, config| -%>
-      <li class="dropdown-item"><%= omniauth_login_link(provider, config) %></li>
+    <% if space_supports_omniauth? %>
+      <% Devise.omniauth_configs.each do |provider, config| -%>
+        <li class="dropdown-item"><%= omniauth_login_link(provider, config) %></li>
+      <% end %>
     <% end %>
 
     <li class="dropdown-item">

config/application.rb

@@ -170,6 +170,10 @@ def _sentry_dsn
     def sentry_enabled?
       _sentry_dsn.present? && Rails.env.production?
     end
+
+    def base_uri
+      @base_uri ||= Addressable::URI.parse(base_url)
+    end
   end
 
   Config = TessConfig.new(tess_config)

test/controllers/static_controller_test.rb

@@ -712,4 +712,26 @@ class StaticControllerTest < ActionController::TestCase
       end
     end
   end
+
+  test 'show omniauth login options when on subdomain space' do
+    space = spaces(:plants)
+    space.update!(host: 'plants.example.com')
+    with_host(space.host) do
+      get :home
+      assert_select 'ul.user-options.nav.navbar-nav.navbar-right' do
+        assert_select "a[href=\"/users/auth/oidc?space_id=#{space.id}\"]", count: 1
+      end
+    end
+  end
+
+  test 'do not show omniauth login options when on non-subdomain space' do
+    space = spaces(:plants)
+    space.update!(host: 'other-host.com')
+    with_host(space.host) do
+      get :home
+      assert_select 'ul.user-options.nav.navbar-nav.navbar-right' do
+        assert_select "a[href=\"/users/auth/oidc?space_id=#{space.id}\"]", count: 0
+      end
+    end
+  end
 end

test/unit/config_test.rb

@@ -27,4 +27,8 @@ class ConfigTest < ActiveSupport::TestCase
           end
     assert_equal exp, TeSS::Config.redis_url
   end
+
+  test 'base_uri' do
+    assert_equal 'example.com', TeSS::Config.base_uri.domain
+  end
 end