livebook/lib/livebook_web/controllers/user_controller.ex
2025-03-06 14:55:29 -03:00

31 lines
722 B
Elixir

defmodule LivebookWeb.UserController do
use LivebookWeb, :controller
def logout(conn, _params) do
if get_session(conn, :user_id) do
if Livebook.Config.logout_enabled?() do
do_zta_logout(conn)
else
do_logout(conn)
end
else
redirect(conn, to: ~p"/")
end
end
defp do_logout(conn) do
conn
|> configure_session(renew: true)
|> clear_session()
|> render("logout.html")
end
defp do_zta_logout(conn) do
{_type, module, _key} = Livebook.Config.identity_provider()
case module.logout(LivebookWeb.ZTA, conn) do
:ok -> do_logout(conn)
{:error, reason} -> conn |> redirect(to: ~p"/") |> put_flash(:error, reason)
end
end
end