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

   1 defmodule Pleroma.Activity do
   2   use Ecto.Schema
   3   alias Pleroma.{Repo, Activity, Notification}
   4   import Ecto.Query
   5
   6   schema "activities" do
   7     field(:data, :map)
   8     field(:local, :boolean, default: true)
   9     field(:actor, :string)
  10     field(:recipients, {:array, :string})
  11     has_many(:notifications, Notification, on_delete: :delete_all)
  12
  13     timestamps()
  14   end
  15
  16   def get_by_ap_id(ap_id) do
  17     Repo.one(
  18       from(
  19         activity in Activity,
  20         where: fragment("(?)->>'id' = ?", activity.data, ^to_string(ap_id))
  21       )
  22     )
  23   end
  24
  25   # TODO:
  26   # Go through these and fix them everywhere.
  27   # Wrong name, only returns create activities
  28   def all_by_object_ap_id_q(ap_id) do
  29     from(
  30       activity in Activity,
  31       where:
  32         fragment(
  33           "coalesce((?)->'object'->>'id', (?)->>'object') = ?",
  34           activity.data,
  35           activity.data,
  36           ^to_string(ap_id)
  37         ),
  38       where: fragment("(?)->>'type' = 'Create'", activity.data)
  39     )
  40   end
  41
  42   # Wrong name, returns all.
  43   def all_non_create_by_object_ap_id_q(ap_id) do
  44     from(
  45       activity in Activity,
  46       where:
  47         fragment(
  48           "coalesce((?)->'object'->>'id', (?)->>'object') = ?",
  49           activity.data,
  50           activity.data,
  51           ^to_string(ap_id)
  52         )
  53     )
  54   end
  55
  56   # Wrong name plz fix thx
  57   def all_by_object_ap_id(ap_id) do
  58     Repo.all(all_by_object_ap_id_q(ap_id))
  59   end
  60
  61   def create_activity_by_object_id_query(ap_ids) do
  62     from(
  63       activity in Activity,
  64       where:
  65         fragment(
  66           "coalesce((?)->'object'->>'id', (?)->>'object') = ANY(?)",
  67           activity.data,
  68           activity.data,
  69           ^ap_ids
  70         ),
  71       where: fragment("(?)->>'type' = 'Create'", activity.data)
  72     )
  73   end
  74
  75   def get_create_activity_by_object_ap_id(ap_id) when is_binary(ap_id) do
  76     create_activity_by_object_id_query([ap_id])
  77     |> Repo.one()
  78   end
  79
  80   def get_create_activity_by_object_ap_id(_), do: nil
  81
  82   def normalize(obj) when is_map(obj), do: Activity.get_by_ap_id(obj["id"])
  83   def normalize(ap_id) when is_binary(ap_id), do: Activity.get_by_ap_id(ap_id)
  84   def normalize(_), do: nil
  85 end