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,
+ timeout: :infinity # We never receive data.
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
+ def handle(:text, message, _state) do
IO.inspect message
#| :ok
#| state
{: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