knova/akkoma
1
0
Fork 0
forked from AkkomaGang/akkoma
Code Issues Pull requests Projects Releases Packages Wiki Activity

Add type restriction to activitypub fetcher

Browse source 
Mainly because Mastodon only returns notes, not the other activities.
This commit is contained in:
Roger Braun 2017-09-09 12:05:17 +02:00
parent 95cedd6000
commit a22f2e683b
1 changed files with 7 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
lib/pleroma/web/activity_pub/activity_pub.ex
View file

@ -133,6 +133,12 @@ defp restrict_actor(query, %{"actor_id" => actor_id}) do
end end
defp restrict_actor(query, _), do: query defp restrict_actor(query, _), do: query
defp restrict_type(query, %{"type" => type}) do
from activity in query,
where: fragment("?->>'type' = ?", activity.data, ^type)
end
defp restrict_type(query, _), do: query
def fetch_activities(recipients, opts \\ %{}) do def fetch_activities(recipients, opts \\ %{}) do
base_query = from activity in Activity, base_query = from activity in Activity,
limit: 20, limit: 20,
@ -144,6 +150,7 @@ def fetch_activities(recipients, opts \\ %{}) do
|> restrict_local(opts) |> restrict_local(opts)
|> restrict_max(opts) |> restrict_max(opts)
|> restrict_actor(opts) |> restrict_actor(opts)
|> restrict_type(opts)
|> Repo.all |> Repo.all
|> Enum.reverse |> Enum.reverse
end end