1 defmodule Pleroma.Web.MastodonAPI.MastodonSocket do
4 alias Pleroma.Web.OAuth.Token
5 alias Pleroma.{User, Repo}
9 Phoenix.Transports.WebSocket.Raw,
10 # We never receive data.
14 def connect(params, socket) do
15 with token when not is_nil(token) <- params["access_token"],
16 %Token{user_id: user_id} <- Repo.get_by(Token, token: token),
17 %User{} = user <- Repo.get(User, user_id),
18 stream when stream in ["public", "public:local", "user", "direct", "list"] <-
20 topic = if stream == "list", do: "list:#{params["list"]}", else: stream
24 |> assign(:topic, topic)
25 |> assign(:user, user)
27 Pleroma.Web.Streamer.add_socket(params["stream"], socket)
36 def handle(:text, message, _state) do
40 # | {:text, message, state}
41 # | {:close, "Goodbye!"}
45 def handle(:closed, _, %{socket: socket}) do
46 topic = socket.assigns[:topic]
47 Pleroma.Web.Streamer.remove_socket(topic, socket)