TwApi ActivityView: Add delete activities.

[akkoma] / lib / pleroma / web / mastodon_api / mastodon_socket.ex

diff --git a/lib/pleroma/web/mastodon_api/mastodon_socket.ex b/lib/pleroma/web/mastodon_api/mastodon_socket.ex
index c27d025c488f7348936275ab007558519634598e..f3e0629416d6dae4ba17d9b5998b3bd864f0d1bf 100644 (file)
--- a/lib/pleroma/web/mastodon_api/mastodon_socket.ex
+++ b/lib/pleroma/web/mastodon_api/mastodon_socket.ex
@@ -1,27 +1,46 @@
 defmodule Pleroma.Web.MastodonAPI.MastodonSocket do
   use Phoenix.Socket
 
-  transport :streaming, Phoenix.Transports.WebSocket.Raw
+  alias Pleroma.Web.OAuth.Token
+  alias Pleroma.{User, Repo}
+
+  transport(
+    :streaming,
+    Phoenix.Transports.WebSocket.Raw,
+    # We never receive data.
+    timeout: :infinity
+  )
 
   def connect(params, socket) do
-    IO.inspect(params)
-    Pleroma.Web.Streamer.add_socket(params["stream"], socket)
-    {:ok, socket}
+    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
+      _e -> :error
+    end
   end
 
-  def id(socket), do: nil
+  def id(_), do: nil
 
-  def handle(:text, message, state) do
-    IO.inspect message
-    #| :ok
-    #| state
-    #| {:text, message}
-    #| {:text, message, state}
-    #| {:close, "Goodbye!"}
+  def handle(:text, message, _state) do
+    # | :ok
+    # | state
+    # | {:text, message}
+    # | {:text, message, state}
+    # | {:close, "Goodbye!"}
     {:text, message}
   end
 
-  def handle(:closed, reason, _state) do
-    IO.inspect reason
+  def handle(:closed, _, %{socket: socket}) do
+    topic = socket.assigns[:topic]
+    Pleroma.Web.Streamer.remove_socket(topic, socket)
   end
 end