Move assignment to live_helpers, fix bug when user updates email

1 files changed, 18 insertions, 0 deletions

diff --git a/lib/aggiedit_web/live/live_helpers.ex b/lib/aggiedit_web/live/live_helpers.ex
index e31fedc..a8d067e 100644
--- a/lib/aggiedit_web/live/live_helpers.ex
+++ b/lib/aggiedit_web/live/live_helpers.ex
@@ -7,6 +7,11 @@ defmodule AggieditWeb.LiveHelpers do
   alias Aggiedit.Accounts
   alias Aggiedit.Accounts.User
 
+  alias Aggiedit.Rooms
+  alias Aggiedit.Roles
+
+  alias AggieditWeb.Router.Helpers, as: Routes
+
   @doc """
   Renders a live component inside a modal.
 
@@ -69,4 +74,17 @@ defmodule AggieditWeb.LiveHelpers do
       _ -> socket
     end
   end
+
+  def assign_socket_room_and_user_or_error(%{"room_id" => room_id}, session, socket) do
+    socket = assign_socket_user(session, socket)
+    case socket.assigns do
+      %{:current_user => user} -> 
+        room = Rooms.get_room!(room_id)
+        case Roles.guard?(user, :index, room) do
+          true -> {:ok, assign(socket, %{:room => room})}
+          _ -> {:ok, socket |> put_flash(:error, "You cannot view that room") |> redirect(to: Routes.page_path(socket, :index))}
+        end
+      _ -> {:ok, socket |> put_flash(:error, "You must log in to access this page.") |> redirect(to: Routes.user_session_path(socket, :new))}
+    end
+  end
 end