git.squeep.com Git - akkoma/blob - lib/pleroma/web/activity_pub/activity_pub.ex

   1 defmodule Pleroma.Web.ActivityPub.ActivityPub do
   2   alias Pleroma.Repo
   3   alias Pleroma.{Activity, Object, Upload}
   4   import Ecto.Query
   5
   6   def insert(map) when is_map(map) do
   7     map = Map.put_new_lazy(map, "id", &generate_activity_id/0)
   8
   9     map = if is_map(map["object"]) do
  10       object = Map.put_new_lazy(map["object"], "id", &generate_object_id/0)
  11       Repo.insert!(%Object{data: object})
  12       Map.put(map, "object", object)
  13     else
  14       map
  15     end
  16
  17     Repo.insert(%Activity{data: map})
  18   end
  19
  20   def generate_activity_id do
  21     generate_id("activities")
  22   end
  23
  24   def generate_context_id do
  25     generate_id("contexts")
  26   end
  27
  28   def generate_object_id do
  29     generate_id("objects")
  30   end
  31
  32   def generate_id(type) do
  33     "#{Pleroma.Web.base_url()}/#{type}/#{Ecto.UUID.generate}"
  34   end
  35
  36   def fetch_public_activities(opts \\ %{}) do
  37     public = ["https://www.w3.org/ns/activitystreams#Public"]
  38     fetch_activities(public, opts)
  39   end
  40
  41   def fetch_activities(recipients, opts \\ %{}) do
  42     since_id = opts["since_id"] || 0
  43
  44     query = from activity in Activity,
  45       limit: 20,
  46       order_by: [desc: :inserted_at]
  47
  48     query = Enum.reduce(recipients, query, fn (recipient, q) ->
  49       map = %{ to: [recipient] }
  50       from activity in q,
  51       or_where: fragment(~s(? @> ?), activity.data, ^map)
  52     end)
  53
  54     query = from activity in query,
  55       where: activity.id > ^since_id
  56
  57     Repo.all(query)
  58     |> Enum.reverse
  59   end
  60
  61   def fetch_activities_for_context(context) do
  62     query = from activity in Activity,
  63       where: fragment("? @> ?", activity.data, ^%{ context: context })
  64     Repo.all(query)
  65   end
  66
  67   def upload(%Plug.Upload{} = file) do
  68     data = Upload.store(file)
  69     Repo.insert(%Object{data: data})
  70   end
  71 end