alias Pleroma.Web.OAuth.Token
alias Pleroma.{User, Repo}
- transport :streaming, Phoenix.Transports.WebSocket.Raw,
- timeout: :infinity # We never receive data.
+ transport(
+ :streaming,
+ Phoenix.Transports.WebSocket.Raw,
+ # We never receive data.
+ timeout: :infinity
+ )
def connect(params, socket) 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"] <- params["stream"] do
- socket = socket
- |> assign(:topic, params["stream"])
- |> assign(:user, user)
- Pleroma.Web.Streamer.add_socket(params["stream"], socket)
+ stream
+ when stream in [
+ "public",
+ "public:local",
+ "public:media",
+ "public:local:media",
+ "user",
+ "direct",
+ "list",
+ "hashtag"
+ ] <- params["stream"] do
+ topic = if stream == "list", do: "list:#{params["list"]}", else: stream
+ socket_stream = if stream == "hashtag", do: "hashtag:#{params["tag"]}", else: stream
+
+ socket =
+ socket
+ |> assign(:topic, topic)
+ |> assign(:user, user)
+
+ Pleroma.Web.Streamer.add_socket(socket_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, %{socket: socket}) do
+ def handle(:closed, _, %{socket: socket}) do
topic = socket.assigns[:topic]
Pleroma.Web.Streamer.remove_socket(topic, socket)
end