livebook-dev/livebook
1
1
Fork 0
mirror of https://github.com/livebook-dev/livebook.git synced 2025-09-07 13:34:55 +08:00
Code Issues Projects Releases Packages Wiki Activity

Do not perform requests for offline teams

Browse source
This commit is contained in:
José Valim 2024-03-25 22:19:41 +01:00
parent fdd2d2c2f2
commit b918f8ab47
1 changed files with 16 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

24
lib/livebook/teams/requests.ex
View file

@ -314,14 +314,22 @@ defmodule Livebook.Teams.Requests do
defp add_team_auth(req, nil), do: req
defp add_team_auth(req, team) do
token =
if team.user_id do
"#{team.user_id}:#{team.org_id}:#{team.org_key_id}:#{team.session_token}"
else
"#{team.session_token}:#{Livebook.Config.agent_name()}:#{team.org_id}:#{team.org_key_id}"
end
if team.offline do
Req.Request.append_request_steps(req,
unauthorized: fn req ->
{req, Req.Response.new(status: 401)}
end
)
else
token =
if team.user_id do
"#{team.user_id}:#{team.org_id}:#{team.org_key_id}:#{team.session_token}"
else
"#{team.session_token}:#{Livebook.Config.agent_name()}:#{team.org_id}:#{team.org_key_id}"
end
Req.Request.merge_options(req, auth: {:bearer, token})
Req.Request.merge_options(req, auth: {:bearer, token})
end
end
defp request(req, opts) do
@ -339,7 +347,7 @@ defmodule Livebook.Teams.Requests do
{:ok, %{status: 401}} ->
{:transport_error,
"You are not authorized to perform this action, make sure you have the access or you are not in a Livebook Agent instance"}
"You are not authorized to perform this action, make sure you have the access or you are not in a Livebook Agent/Offline instance"}
_otherwise ->
{:transport_error, @error_message}