diff --git a/lib/livebook/hubs/team_client.ex b/lib/livebook/hubs/team_client.ex index 51070f5cd..144bbe5af 100644 --- a/lib/livebook/hubs/team_client.ex +++ b/lib/livebook/hubs/team_client.ex @@ -166,6 +166,14 @@ defmodule Livebook.Hubs.TeamClient do GenServer.call(registry_name(id), :authorization_groups_enabled?) end + @doc """ + Returns if the Team client is an app server for given deployment group. + """ + @spec current_app_server_deployment_group?(String.t(), String.t()) :: boolean() + def current_app_server_deployment_group?(id, deployment_group_id) do + GenServer.call(registry_name(id), {:current_deployment_group?, deployment_group_id}) + end + @doc """ Returns if the Team client is connected. """ @@ -346,6 +354,10 @@ defmodule Livebook.Hubs.TeamClient do end end + def handle_call({:current_deployment_group?, id}, _caller, state) do + {:reply, state.deployment_group_id == id, state} + end + @impl true def handle_info(:connected, state) do Hubs.Broadcasts.hub_connected(state.hub.id)