Socket connection and token fetching

3 files changed, 26 insertions, 2 deletions

diff --git a/lib/rook_web/channels/share_channel.ex b/lib/rook_web/channels/share_channel.ex
new file mode 100644
index 0000000..fde0096
--- /dev/null
+++ b/lib/rook_web/channels/share_channel.ex
@@ -0,0 +1,11 @@
+defmodule RookWeb.ShareChannel do
+  use Phoenix.Channel
+
+  def join("share:" <> token, _params, socket) do
+    if token == socket.assigns[:token] do
+      {:ok, socket}
+    else
+      {:error, %{reason: "Wrong token."}}
+    end
+  end
+end
diff --git a/lib/rook_web/channels/token_channel.ex b/lib/rook_web/channels/token_channel.ex
new file mode 100644
index 0000000..72cabb1
--- /dev/null
+++ b/lib/rook_web/channels/token_channel.ex
@@ -0,0 +1,11 @@
+defmodule RookWeb.TokenChannel do
+  use Phoenix.Channel
+
+  def join("token", _params, socket) do
+    {:ok, socket}
+  end
+
+  def handle_in("get_token", _attrs, socket) do
+    {:reply, {:ok, %{token: socket.assigns[:token]}}, socket}
+  end
+end
diff --git a/lib/rook_web/channels/user_socket.ex b/lib/rook_web/channels/user_socket.ex
index 66637af..a036c4a 100644
--- a/lib/rook_web/channels/user_socket.ex
+++ b/lib/rook_web/channels/user_socket.ex
@@ -2,7 +2,8 @@ defmodule RookWeb.UserSocket do
   use Phoenix.Socket
 
   ## Channels
-  # channel "room:*", RookWeb.RoomChannel
+  channel "token", RookWeb.TokenChannel
+  channel "share:*", RookWeb.ShareChannel
 
   # Socket params are passed from the client and can
   # be used to verify and authenticate a user. After
@@ -17,7 +18,8 @@ defmodule RookWeb.UserSocket do
   # performing token verification on connect.
   @impl true
   def connect(_params, socket, _connect_info) do
-    {:ok, socket}
+    token = Rook.Utils.Token.token()
+    {:ok, assign(socket, :token, token)}
   end
 
   # Socket id's are topics that allow you to identify all sockets for a given user: