git.squeep.com Git - akkoma/blob - lib/pleroma/web/activity_pub/object_validators/emoji_react_validator.ex

   1 # Pleroma: A lightweight social networking server
   2 # Copyright © 2017-2021 Pleroma Authors <https://pleroma.social/>
   3 # SPDX-License-Identifier: AGPL-3.0-only
   4
   5 defmodule Pleroma.Web.ActivityPub.ObjectValidators.EmojiReactValidator do
   6   use Ecto.Schema
   7
   8   alias Pleroma.EctoType.ActivityPub.ObjectValidators
   9   alias Pleroma.Object
  10
  11   import Ecto.Changeset
  12   import Pleroma.Web.ActivityPub.ObjectValidators.CommonValidations
  13
  14   @primary_key false
  15
  16   embedded_schema do
  17     field(:id, ObjectValidators.ObjectID, primary_key: true)
  18     field(:type, :string)
  19     field(:object, ObjectValidators.ObjectID)
  20     field(:actor, ObjectValidators.ObjectID)
  21     field(:context, :string)
  22     field(:content, :string)
  23     field(:to, ObjectValidators.Recipients, default: [])
  24     field(:cc, ObjectValidators.Recipients, default: [])
  25   end
  26
  27   def cast_and_validate(data) do
  28     data
  29     |> cast_data()
  30     |> validate_data()
  31   end
  32
  33   def cast_data(data) do
  34     %__MODULE__{}
  35     |> changeset(data)
  36   end
  37
  38   def changeset(struct, data) do
  39     struct
  40     |> cast(data, __schema__(:fields))
  41     |> fix_after_cast()
  42   end
  43
  44   def fix_after_cast(cng) do
  45     cng
  46     |> fix_context()
  47   end
  48
  49   def fix_context(cng) do
  50     object = get_field(cng, :object)
  51
  52     with nil <- get_field(cng, :context),
  53          %Object{data: %{"context" => context}} <- Object.get_cached_by_ap_id(object) do
  54       cng
  55       |> put_change(:context, context)
  56     else
  57       _ ->
  58         cng
  59     end
  60   end
  61
  62   def validate_emoji(cng) do
  63     content = get_field(cng, :content)
  64
  65     if Pleroma.Emoji.is_unicode_emoji?(content) do
  66       cng
  67     else
  68       cng
  69       |> add_error(:content, "must be a single character emoji")
  70     end
  71   end
  72
  73   defp validate_data(data_cng) do
  74     data_cng
  75     |> validate_inclusion(:type, ["EmojiReact"])
  76     |> validate_required([:id, :type, :object, :actor, :context, :to, :cc, :content])
  77     |> validate_actor_presence()
  78     |> validate_object_presence()
  79     |> validate_emoji()
  80   end
  81 end