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 @spec connect(params :: map(), Phoenix.Socket.t()) :: {:ok, Phoenix.Socket.t()} | :error
15 def connect(%{"access_token" => token} = params, socket) do
16 with %Token{user_id: user_id} <- Repo.get_by(Token, token: token),
17 %User{} = user <- Repo.get(User, user_id),
28 ] <- params["stream"] do
31 "hashtag" -> "hashtag:#{params["tag"]}"
32 "list" -> "list:#{params["list"]}"
38 |> assign(:topic, topic)
39 |> assign(:user, user)
41 Pleroma.Web.Streamer.add_socket(topic, socket)
48 def connect(%{"stream" => stream} = params, socket)
49 when stream in ["public", "public:local", "hashtag"] do
52 "hashtag" -> "hashtag:#{params["tag"]}"
58 |> assign(:topic, topic)
60 Pleroma.Web.Streamer.add_socket(topic, socket)
64 def connect(_params, _socket), do: :error
68 def handle(:text, message, _state) do
72 # | {:text, message, state}
73 # | {:close, "Goodbye!"}
77 def handle(:closed, _, %{socket: socket}) do
78 topic = socket.assigns[:topic]
79 Pleroma.Web.Streamer.remove_socket(topic, socket)