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