Stream function to handle direct messages.

This commit is contained in:
csaurus 2018-05-10 22:15:42 -04:00
parent c60a5405db
commit 392bd9ef56
2 changed files with 21 additions and 10 deletions

View file

@ -42,18 +42,16 @@ defmodule Pleroma.Web.ActivityPub.ActivityPub do
def stream_out(activity) do
if activity.data["type"] in ["Create", "Announce"] do
Pleroma.Web.Streamer.stream("user", activity)
direct? = activity.data["object"]["visibility"] == "direct"
cond do
direct? ->
Pleroma.Web.Streamer.stream("direct", activity)
visibility = Pleroma.Web.MastodonAPI.StatusView.get_visibility(activity.data["object"])
Enum.member?(activity.data["to"], "https://www.w3.org/ns/activitystreams#Public") ->
case visibility do
"public" ->
Pleroma.Web.Streamer.stream("public", activity)
if activity.local, do: Pleroma.Web.Streamer.stream("public:local", activity)
if activity.local do
Pleroma.Web.Streamer.stream("public:local", activity)
end
"direct" ->
Pleroma.Web.Streamer.stream("direct", activity)
end
end
end

View file

@ -46,6 +46,19 @@ defmodule Pleroma.Web.Streamer do
{:noreply, topics}
end
def handle_cast(%{action: :stream, topic: "direct", item: item}, topics) do
recipient_topics =
User.get_recipients_from_activity(item)
|> Enum.map(fn %{id: id} -> "direct:#{id}" end)
Enum.each(recipient_topics || [], fn user_topic ->
Logger.debug("Trying to push direct message to #{user_topic}\n\n")
push_to_socket(topics, user_topic, item)
end)
{:noreply, topics}
end
def handle_cast(%{action: :stream, topic: "user", item: %Notification{} = item}, topics) do
topic = "user:#{item.user_id}"
@ -137,8 +150,8 @@ defmodule Pleroma.Web.Streamer do
end)
end
defp internal_topic("user", socket) do
"user:#{socket.assigns[:user].id}"
defp internal_topic(topic, socket) when topic in ~w[user, direct] do
"#{topic}:#{socket.assigns[:user].id}"
end
defp internal_topic(topic, _), do: topic