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

   1 # Pleroma: A lightweight social networking server
   2 # Copyright © 2017-2019 Pleroma Authors <https://pleroma.social/>
   3 # SPDX-License-Identifier: AGPL-3.0-only
   4
   5 defmodule Pleroma.Activity do
   6   use Ecto.Schema
   7   alias Pleroma.{Repo, Activity, Notification}
   8   import Ecto.Query
   9
  10   @type t :: %__MODULE__{}
  11
  12   # https://github.com/tootsuite/mastodon/blob/master/app/models/notification.rb#L19
  13   @mastodon_notification_types %{
  14     "Create" => "mention",
  15     "Follow" => "follow",
  16     "Announce" => "reblog",
  17     "Like" => "favourite"
  18   }
  19
  20   schema "activities" do
  21     field(:data, :map)
  22     field(:local, :boolean, default: true)
  23     field(:actor, :string)
  24     field(:recipients, {:array, :string})
  25     has_many(:notifications, Notification, on_delete: :delete_all)
  26
  27     timestamps()
  28   end
  29
  30   def get_by_ap_id(ap_id) do
  31     Repo.one(
  32       from(
  33         activity in Activity,
  34         where: fragment("(?)->>'id' = ?", activity.data, ^to_string(ap_id))
  35       )
  36     )
  37   end
  38
  39   # TODO:
  40   # Go through these and fix them everywhere.
  41   # Wrong name, only returns create activities
  42   def all_by_object_ap_id_q(ap_id) do
  43     from(
  44       activity in Activity,
  45       where:
  46         fragment(
  47           "coalesce((?)->'object'->>'id', (?)->>'object') = ?",
  48           activity.data,
  49           activity.data,
  50           ^to_string(ap_id)
  51         ),
  52       where: fragment("(?)->>'type' = 'Create'", activity.data)
  53     )
  54   end
  55
  56   def by_object_ap_id(ap_id) do
  57     from(
  58       activity in Activity,
  59       where:
  60         fragment(
  61           "coalesce((?)->'object'->>'id', (?)->>'object') = ?",
  62           activity.data,
  63           activity.data,
  64           ^to_string(ap_id)
  65         )
  66     )
  67   end
  68
  69   def get_all_by_object_ap_id(ap_id) do
  70     Repo.all(all_by_object_ap_id_q(ap_id))
  71   end
  72
  73   def create_activity_by_object_id_query(ap_ids) do
  74     from(
  75       activity in Activity,
  76       where:
  77         fragment(
  78           "coalesce((?)->'object'->>'id', (?)->>'object') = ANY(?)",
  79           activity.data,
  80           activity.data,
  81           ^ap_ids
  82         ),
  83       where: fragment("(?)->>'type' = 'Create'", activity.data)
  84     )
  85   end
  86
  87   def get_create_activity_by_object_ap_id(ap_id) when is_binary(ap_id) do
  88     create_activity_by_object_id_query([ap_id])
  89     |> Repo.one()
  90   end
  91
  92   def get_create_activity_by_object_ap_id(_), do: nil
  93
  94   def normalize(obj) when is_map(obj), do: Activity.get_by_ap_id(obj["id"])
  95   def normalize(ap_id) when is_binary(ap_id), do: Activity.get_by_ap_id(ap_id)
  96   def normalize(_), do: nil
  97
  98   def get_in_reply_to_activity(%Activity{data: %{"object" => %{"inReplyTo" => ap_id}}}) do
  99     get_create_activity_by_object_ap_id(ap_id)
 100   end
 101
 102   def get_in_reply_to_activity(_), do: nil
 103
 104   for {ap_type, type} <- @mastodon_notification_types do
 105     def mastodon_notification_type(%Activity{data: %{"type" => unquote(ap_type)}}),
 106       do: unquote(type)
 107   end
 108
 109   def mastodon_notification_type(%Activity{}), do: nil
 110 end