2 files changed, 39 insertions, 0 deletions
diff --git a/lib/aggiedit_web/channels/post_channel.ex b/lib/aggiedit_web/channels/post_channel.ex
new file mode 100644
index 0000000..308c6de
--- /dev/null
+++ b/lib/aggiedit_web/channels/post_channel.ex
@@ -0,0 +1,22 @@
+defmodule AggieditWeb.PostChannel do
+  use AggieditWeb, :channel
+
+  alias Aggiedit.Roles
+  alias Aggiedit.Rooms
+
+  @impl true
+  def join("post:" <> post_id, _payload, socket) do
+    post = Rooms.get_post!(post_id)
+    if Roles.guard?(socket.assigns.current_user, :show, post) do
+      {:ok, socket}
+    else
+      {:error, "You do not have permission to view this post."}
+    end
+  end
+
+  @impl true
+  def handle_in("send", body, socket) do
+    broadcast!(socket, "shout", body)
+    {:reply, :ok, socket}
+  end
+end
+\ No newline at end of file
diff --git a/lib/aggiedit_web/channels/user_socket.ex b/lib/aggiedit_web/channels/user_socket.ex
new file mode 100644
index 0000000..b7c0124
--- /dev/null
+++ b/lib/aggiedit_web/channels/user_socket.ex
@@ -0,0 +1,17 @@
+defmodule AggieditWeb.UserSocket do
+  alias Aggiedit.Accounts
+  use Phoenix.Socket
+
+  channel "post:*", AggieditWeb.PostChannel
+
+  @impl true
+  def connect(_params, socket, %{:session => %{"user_token" => token}}) do
+    case Accounts.get_user_by_session_token(token) do
+      user=%Accounts.User{} -> {:ok, assign(socket, %{:current_user => user})}
+      _ -> {:error, "Invalid user token."}
+    end
+  end
+
+  @impl true
+  def id(_socket), do: nil
+end
+\ No newline at end of file