X-Git-Url: http://git.squeep.com/?a=blobdiff_plain;f=lib%2Fpleroma%2Fweb%2Fmastodon_api%2Fmastodon_socket.ex;h=c3bae59354688223c7e91175b72c9df712e147b1;hb=83d4a1495ce530b6b308bb47668268d68a8fb3e2;hp=f9c8cec321e6e4def2ef35d04d19118346550f71;hpb=08bc31674218cd7ce634b008b7766b48e49c52e3;p=akkoma

diff --git a/lib/pleroma/web/mastodon_api/mastodon_socket.ex b/lib/pleroma/web/mastodon_api/mastodon_socket.ex
index f9c8cec32..c3bae5935 100644
--- a/lib/pleroma/web/mastodon_api/mastodon_socket.ex
+++ b/lib/pleroma/web/mastodon_api/mastodon_socket.ex
@@ -1,24 +1,30 @@
 defmodule Pleroma.Web.MastodonAPI.MastodonSocket do
   use Phoenix.Socket
 
+  alias Pleroma.Web.OAuth.Token
+  alias Pleroma.{User, Repo}
+
   transport :streaming, Phoenix.Transports.WebSocket.Raw,
     timeout: :infinity # We never receive data.
 
   def connect(params, socket) do
-    if params["stream"] == "public" do
+    with token when not is_nil(token) <- params["access_token"],
+         %Token{user_id: user_id} <- Repo.get_by(Token, token: token),
+         %User{} = user <- Repo.get(User, user_id),
+         stream when stream in ["public", "public:local", "user"] <- params["stream"] do
       socket = socket
       |> assign(:topic, params["stream"])
+      |> assign(:user, user)
       Pleroma.Web.Streamer.add_socket(params["stream"], socket)
       {:ok, socket}
     else
-      :error
+      _e -> :error
     end
   end
 
-  def id(socket), do: nil
+  def id(_), do: nil
 
-  def handle(:text, message, state) do
-    IO.inspect message
+  def handle(:text, message, _state) do
     #| :ok
     #| state
     #| {:text, message}
@@ -27,7 +33,7 @@ defmodule Pleroma.Web.MastodonAPI.MastodonSocket do
     {:text, message}
   end
 
-  def handle(:closed, reason, %{socket: socket}) do
+  def handle(:closed, _, %{socket: socket}) do
     topic = socket.assigns[:topic]
     Pleroma.Web.Streamer.remove_socket(topic, socket)
   end