diff --git a/lib/plausible/teams.ex b/lib/plausible/teams.ex index ed94c16c62..07e26b1d55 100644 --- a/lib/plausible/teams.ex +++ b/lib/plausible/teams.ex @@ -25,13 +25,13 @@ defmodule Plausible.Teams do def setup?(nil), do: false def setup?(%{setup_complete: setup_complete}), do: setup_complete - @spec enabled?(nil | Teams.Team.t()) :: boolean() + @spec enabled?(nil | Auth.User.t()) :: boolean() def enabled?(nil) do FunWithFlags.enabled?(:teams) end - def enabled?(team) do - FunWithFlags.enabled?(:teams, for: team) + def enabled?(user) do + FunWithFlags.enabled?(:teams, for: user) end @spec get(pos_integer() | binary() | nil) :: Teams.Team.t() | nil diff --git a/lib/plausible_web/live/sites.ex b/lib/plausible_web/live/sites.ex index 943e04ec9d..dc5eaa4ba4 100644 --- a/lib/plausible_web/live/sites.ex +++ b/lib/plausible_web/live/sites.ex @@ -59,7 +59,7 @@ defmodule PlausibleWeb.Live.Sites do

- + {Teams.name(@current_team)} <.unstyled_link :if={Teams.setup?(@current_team)} @@ -69,7 +69,7 @@ defmodule PlausibleWeb.Live.Sites do - My Sites + My Sites

diff --git a/lib/plausible_web/live/team_setup.ex b/lib/plausible_web/live/team_setup.ex index c6438cd67a..47126bf287 100644 --- a/lib/plausible_web/live/team_setup.ex +++ b/lib/plausible_web/live/team_setup.ex @@ -11,8 +11,9 @@ defmodule PlausibleWeb.Live.TeamSetup do alias PlausibleWeb.Router.Helpers, as: Routes def mount(_params, _session, socket) do + current_user = socket.assigns.current_user current_team = socket.assigns.current_team - enabled? = Teams.enabled?(current_team) + enabled? = Teams.enabled?(current_user) socket = case {enabled?, current_team} do diff --git a/lib/plausible_web/plugs/authorize_team_access.ex b/lib/plausible_web/plugs/authorize_team_access.ex index 7ec4257033..2ce086b2d7 100644 --- a/lib/plausible_web/plugs/authorize_team_access.ex +++ b/lib/plausible_web/plugs/authorize_team_access.ex @@ -21,9 +21,10 @@ defmodule Plausible.Plugs.AuthorizeTeamAccess do end def call(conn, roles \\ @all_roles) do + current_user = conn.assigns[:current_user] current_team = conn.assigns[:current_team] - if current_team && Plausible.Teams.enabled?(current_team) do + if current_team && Plausible.Teams.enabled?(current_user) do current_team_role = conn.assigns[:current_team_role] if current_team_role in roles do diff --git a/lib/plausible_web/templates/layout/_header.html.heex b/lib/plausible_web/templates/layout/_header.html.heex index f2ff9c00ec..52e2d45dfc 100644 --- a/lib/plausible_web/templates/layout/_header.html.heex +++ b/lib/plausible_web/templates/layout/_header.html.heex @@ -73,7 +73,7 @@

<.team_switcher - :if={Plausible.Teams.enabled?(@current_team)} + :if={Plausible.Teams.enabled?(@current_user)} conn={@conn} teams={@teams} my_team={@my_team} @@ -86,7 +86,7 @@
<.dropdown_item class="flex" href={Routes.team_setup_path(@conn, :setup)}> @@ -101,7 +101,7 @@
<.dropdown_item diff --git a/lib/plausible_web/templates/layout/settings.html.heex b/lib/plausible_web/templates/layout/settings.html.heex index 00959bfdd0..cb3a045769 100644 --- a/lib/plausible_web/templates/layout/settings.html.heex +++ b/lib/plausible_web/templates/layout/settings.html.heex @@ -37,7 +37,7 @@