fix: return to original page after login (algora-io#101)

Signed-off-by: bryans-go <[email protected]>

lib/algora_web/controllers/oauth_login_controller.ex

@@ -1,6 +1,7 @@
 defmodule AlgoraWeb.OAuthLoginController do
   use AlgoraWeb, :controller
   require Logger
+  import AlgoraWeb.UserAuth, only: [maybe_store_return_to: 1]
 
   def new(conn, %{"provider" => "restream"} = params) do
     if conn.assigns.current_user do
@@ -11,7 +12,9 @@ defmodule AlgoraWeb.OAuthLoginController do
       |> put_session(:restream_state, state)
       |> redirect(external: Algora.Restream.authorize_url(state))
     else
-      conn |> redirect(to: ~p"/auth/login")
+      conn 
+      |> maybe_store_return_to()
+      |> redirect(to: ~p"/auth/login")
     end
   end
 end

lib/algora_web/controllers/redirect_controller.ex

@@ -1,15 +1,17 @@
 defmodule AlgoraWeb.RedirectController do
   use AlgoraWeb, :controller
 
-  import AlgoraWeb.UserAuth, only: [fetch_current_user: 2]
+  import AlgoraWeb.UserAuth, only: [fetch_current_user: 2, maybe_store_return_to: 1]
 
   plug :fetch_current_user
 
   def redirect_authenticated(conn, _) do
     if conn.assigns.current_user do
       AlgoraWeb.UserAuth.redirect_if_user_is_authenticated(conn, [])
     else
-      redirect(conn, to: ~p"/auth/login")
+      conn
+      |> maybe_store_return_to()
+      |> redirect(to: ~p"/auth/login")
     end
   end
 end

lib/algora_web/controllers/user_auth.ex

@@ -54,8 +54,12 @@ defmodule AlgoraWeb.UserAuth do
   end
 
   defp redirect_require_login(socket) do
-    socket
+
+  conn = Phoenix.LiveView.get_connect_info(socket).conn 
+
+    conn
     |> LiveView.put_flash(:error, "Please sign in")
+    |> maybe_store_return_to()
     |> LiveView.redirect(to: ~p"/auth/login")
   end
 
@@ -157,13 +161,13 @@ defmodule AlgoraWeb.UserAuth do
     end
   end
 
-  defp maybe_store_return_to(%{method: "GET"} = conn) do
+  def maybe_store_return_to(%{method: "GET"} = conn) do
     %{request_path: request_path, query_string: query_string} = conn
     return_to = if query_string == "", do: request_path, else: request_path <> "?" <> query_string
     put_session(conn, :user_return_to, return_to)
   end
 
-  defp maybe_store_return_to(conn), do: conn
+  def maybe_store_return_to(conn), do: conn
 
   def signed_in_path(_conn), do: "/"
 end

lib/algora_web/live/show_live/show.ex

@@ -1,5 +1,6 @@
 defmodule AlgoraWeb.ShowLive.Show do
   use AlgoraWeb, :live_view
+  import AlgoraWeb.UserAuth, only: [maybe_store_return_to: 1]
 
   alias Algora.{Shows, Library, Events}
   alias Algora.Accounts
@@ -344,9 +345,12 @@ defmodule AlgoraWeb.ShowLive.Show do
 
     show = Shows.get_show_by_fields!(slug: slug)
 
+    conn = Phoenix.LiveView.get_connect_info(socket).conn
+
     cond do
       current_user == nil ->
-        socket
+        conn
+        |> maybe_store_return_to()
         |> redirect(to: ~p"/auth/login")
 
       current_user.id != show.user_id && !Accounts.admin?(current_user) ->