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, User}
   4   import Ecto.Query
   5
   6   def insert(map) when is_map(map) do
   7     map = map
   8     |> Map.put_new_lazy("id", &generate_activity_id/0)
   9     |> Map.put_new_lazy("published", &make_date/0)
  10
  11     map = if is_map(map["object"]) do
  12       object = Map.put_new_lazy(map["object"], "id", &generate_object_id/0)
  13       Repo.insert!(%Object{data: object})
  14       Map.put(map, "object", object)
  15     else
  16       map
  17     end
  18
  19     Repo.insert(%Activity{data: map})
  20   end
  21
  22   def like(%User{ap_id: ap_id}, object = %Object{data: %{ "id" => id}}) do
  23     data = %{
  24       "type" => "Like",
  25       "actor" => ap_id,
  26       "object" => id
  27     }
  28
  29     {:ok, activity} = insert(data)
  30     like_count = (object.data["like_count"] || 0) + 1
  31     new_data = object.data |> Map.put("like_count", like_count)
  32     changeset = Ecto.Changeset.change(object, data: new_data)
  33     {:ok, object} = Repo.update(changeset)
  34
  35     # Update activities that already had this. Could be done in a seperate process.
  36     relevant_activities = Activity.all_by_object_ap_id(id)
  37     Enum.map(relevant_activities, fn (activity) ->
  38       new_activity_data = activity.data |> Map.put("object", new_data)
  39       changeset = Ecto.Changeset.change(activity, data: new_activity_data)
  40       Repo.update(changeset)
  41     end)
  42
  43     {:ok, activity, object}
  44   end
  45
  46   def generate_activity_id do
  47     generate_id("activities")
  48   end
  49
  50   def generate_context_id do
  51     generate_id("contexts")
  52   end
  53
  54   def generate_object_id do
  55     generate_id("objects")
  56   end
  57
  58   def generate_id(type) do
  59     "#{Pleroma.Web.base_url()}/#{type}/#{Ecto.UUID.generate}"
  60   end
  61
  62   def fetch_public_activities(opts \\ %{}) do
  63     public = ["https://www.w3.org/ns/activitystreams#Public"]
  64     fetch_activities(public, opts)
  65   end
  66
  67   def fetch_activities(recipients, opts \\ %{}) do
  68     since_id = opts["since_id"] || 0
  69
  70     query = from activity in Activity,
  71       limit: 20,
  72       order_by: [desc: :inserted_at]
  73
  74     query = Enum.reduce(recipients, query, fn (recipient, q) ->
  75       map = %{ to: [recipient] }
  76       from activity in q,
  77       or_where: fragment(~s(? @> ?), activity.data, ^map)
  78     end)
  79
  80     query = from activity in query,
  81       where: activity.id > ^since_id
  82
  83     query = if opts["max_id"] do
  84       from activity in query, where: activity.id < ^opts["max_id"]
  85     else
  86       query
  87     end
  88
  89     Repo.all(query)
  90     |> Enum.reverse
  91   end
  92
  93   def fetch_activities_for_context(context) do
  94     query = from activity in Activity,
  95       where: fragment("? @> ?", activity.data, ^%{ context: context })
  96     Repo.all(query)
  97   end
  98
  99   def upload(%Plug.Upload{} = file) do
 100     data = Upload.store(file)
 101     Repo.insert(%Object{data: data})
 102   end
 103
 104   defp make_date do
 105     DateTime.utc_now() |> DateTime.to_iso8601
 106   end
 107 end