activitypub: utils: use same object type list for mention extraction as insertion
authorWilliam Pitcock <nenolod@dereferenced.org>
Fri, 9 Nov 2018 13:39:44 +0000 (13:39 +0000)
committerWilliam Pitcock <nenolod@dereferenced.org>
Fri, 9 Nov 2018 13:40:39 +0000 (13:40 +0000)
lib/pleroma/web/activity_pub/utils.ex

index d438236c7f81fc8aabba97fa4a683701a5ec2704..fac91830ab86c8840c0b74f54ea35be8f7a47534 100644 (file)
@@ -6,6 +6,8 @@ defmodule Pleroma.Web.ActivityPub.Utils do
   import Ecto.Query
   require Logger
 
+  @supported_object_types ["Article", "Note", "Video", "Page"]
+
   # Some implementations send the actor URI as the actor field, others send the entire actor object,
   # so figure out what the actor's URI is based on what we have.
   def get_ap_id(object) do
@@ -95,7 +97,7 @@ defmodule Pleroma.Web.ActivityPub.Utils do
     "#{Web.base_url()}/#{type}/#{UUID.generate()}"
   end
 
-  def get_notified_from_object(%{"type" => type} = object) when type == "Note" do
+  def get_notified_from_object(%{"type" => type} = object) when type in @supported_object_types do
     fake_create_activity = %{
       "to" => object["to"],
       "cc" => object["cc"],
@@ -179,7 +181,7 @@ defmodule Pleroma.Web.ActivityPub.Utils do
   Inserts a full object if it is contained in an activity.
   """
   def insert_full_object(%{"object" => %{"type" => type} = object_data})
-      when is_map(object_data) and type in ["Article", "Note", "Video", "Page"] do
+      when is_map(object_data) and type in @supported_object_types do
     with {:ok, _} <- Object.create(object_data) do
       :ok
     end