livebook-dev/livebook
1
1
Fork 0
mirror of https://github.com/livebook-dev/livebook.git synced 2025-11-13 23:59:47 +08:00
Code Issues Projects Releases Packages Wiki Activity

Persist the hub id, not the struct

Browse source
This commit is contained in:
Alexandre de Souza 2025-08-20 14:15:03 -03:00
parent 8887ded506
commit 02deec7f13
No known key found for this signature in database
GPG key ID: E39228FFBA346545
1 changed files with 10 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
lib/livebook/zta/livebook_teams.ex
View file

@ -18,15 +18,20 @@ defmodule Livebook.ZTA.LivebookTeams do
def start_link(opts) do def start_link(opts) do
name = Keyword.fetch!(opts, :name) name = Keyword.fetch!(opts, :name)
identity_key = Keyword.fetch!(opts, :identity_key) identity_key = Keyword.fetch!(opts, :identity_key)
team = Livebook.Hubs.fetch_hub!(identity_key)
Livebook.ZTA.put(name, team) # we need to guarantee the hub exists
# otherwise, it should raise an exception
_ = Livebook.Hubs.fetch_hub!(identity_key)
Livebook.ZTA.put(name, identity_key)
:ignore :ignore
end end
@impl true @impl true
def authenticate(name, conn, _opts) do def authenticate(name, conn, _opts) do
team = Livebook.ZTA.get(name) id = Livebook.ZTA.get(name)
# fetch the hub everytime, so we will have the `billing_status` updated
team = Livebook.Hubs.fetch_hub!(id)
if Livebook.Hubs.TeamClient.identity_enabled?(team.id) do if Livebook.Hubs.TeamClient.identity_enabled?(team.id) do
handle_request(conn, team, conn.params) handle_request(conn, team, conn.params)
@ -38,7 +43,8 @@ defmodule Livebook.ZTA.LivebookTeams do
# Our extension to Livebook.ZTA to deal with logouts # Our extension to Livebook.ZTA to deal with logouts
def logout(name, conn) do def logout(name, conn) do
token = get_session(conn, :livebook_teams_access_token) token = get_session(conn, :livebook_teams_access_token)
team = Livebook.ZTA.get(name) id = Livebook.ZTA.get(name)
team = Livebook.Hubs.fetch_hub!(id)
url = url =
Livebook.Config.teams_url() Livebook.Config.teams_url()