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} = user, %Object{data: %{ "id" => id}} = object) do
  23     cond do
  24       # There's already a like here, so return the original activity.
  25       ap_id in (object.data["likes"] || []) ->
  26         query = from activity in Activity,
  27           where: fragment("? @> ?", activity.data, ^%{actor: ap_id, object: id})
  28
  29         activity = Repo.one(query)
  30         {:ok, activity, object}
  31       true ->
  32         data = %{
  33           "type" => "Like",
  34           "actor" => ap_id,
  35           "object" => id,
  36           "to" => [User.ap_followers(user)]
  37         }
  38
  39         {:ok, activity} = insert(data)
  40
  41         likes = [ap_id | (object.data["likes"] || [])] |> Enum.uniq
  42
  43         new_data = object.data
  44         |> Map.put("like_count", length(likes))
  45         |> Map.put("likes", likes)
  46
  47         changeset = Ecto.Changeset.change(object, data: new_data)
  48         {:ok, object} = Repo.update(changeset)
  49
  50         # Update activities that already had this. Could be done in a seperate process.
  51         relevant_activities = Activity.all_by_object_ap_id(id)
  52         Enum.map(relevant_activities, fn (activity) ->
  53           new_activity_data = activity.data |> Map.put("object", new_data)
  54           changeset = Ecto.Changeset.change(activity, data: new_activity_data)
  55           Repo.update(changeset)
  56         end)
  57         {:ok, activity, object}
  58     end
  59   end
  60
  61   def generate_activity_id do
  62     generate_id("activities")
  63   end
  64
  65   def generate_context_id do
  66     generate_id("contexts")
  67   end
  68
  69   def generate_object_id do
  70     generate_id("objects")
  71   end
  72
  73   def generate_id(type) do
  74     "#{Pleroma.Web.base_url()}/#{type}/#{Ecto.UUID.generate}"
  75   end
  76
  77   def fetch_public_activities(opts \\ %{}) do
  78     public = ["https://www.w3.org/ns/activitystreams#Public"]
  79     fetch_activities(public, opts)
  80   end
  81
  82   def fetch_activities(recipients, opts \\ %{}) do
  83     since_id = opts["since_id"] || 0
  84
  85     query = from activity in Activity,
  86       limit: 20,
  87       order_by: [desc: :inserted_at]
  88
  89     query = Enum.reduce(recipients, query, fn (recipient, q) ->
  90       map = %{ to: [recipient] }
  91       from activity in q,
  92       or_where: fragment(~s(? @> ?), activity.data, ^map)
  93     end)
  94
  95     query = from activity in query,
  96       where: activity.id > ^since_id
  97
  98     query = if opts["max_id"] do
  99       from activity in query, where: activity.id < ^opts["max_id"]
 100     else
 101       query
 102     end
 103
 104     Repo.all(query)
 105     |> Enum.reverse
 106   end
 107
 108   def fetch_activities_for_context(context) do
 109     query = from activity in Activity,
 110       where: fragment("? @> ?", activity.data, ^%{ context: context })
 111     Repo.all(query)
 112   end
 113
 114   def upload(%Plug.Upload{} = file) do
 115     data = Upload.store(file)
 116     Repo.insert(%Object{data: data})
 117   end
 118
 119   defp make_date do
 120     DateTime.utc_now() |> DateTime.to_iso8601
 121   end
 122 end