Monitor share channels

1 files changed, 5 insertions, 0 deletions

diff --git a/lib/rook_web/channels/share_channel.ex b/lib/rook_web/channels/share_channel.ex
index fde0096..7454986 100644
--- a/lib/rook_web/channels/share_channel.ex
+++ b/lib/rook_web/channels/share_channel.ex
@@ -3,9 +3,14 @@ defmodule RookWeb.ShareChannel do
 
   def join("share:" <> token, _params, socket) do
     if token == socket.assigns[:token] do
+      :ok = Rook.ShareMonitor.track(self(), token)
       {:ok, socket}
     else
       {:error, %{reason: "Wrong token."}}
     end
   end
+
+  def handle_close(_requests) do
+    # Notify all requests that share is gone.
+  end
 end