- if params["stream"] == "public" do
- socket = socket
- |> assign(:topic, params["stream"])
- Pleroma.Web.Streamer.add_socket(params["stream"], 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",
+ "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)