164 lines
		
	
	
	
		
			5.2 KiB
		
	
	
	
		
			Elixir
		
	
	
	
	
	
			
		
		
	
	
			164 lines
		
	
	
	
		
			5.2 KiB
		
	
	
	
		
			Elixir
		
	
	
	
	
	
# Pleroma: A lightweight social networking server
 | 
						|
# Copyright © 2017-2020 Pleroma Authors <https://pleroma.social/>
 | 
						|
# SPDX-License-Identifier: AGPL-3.0-only
 | 
						|
 | 
						|
defmodule Pleroma.Web.ActivityPub.ObjectValidator do
 | 
						|
  @moduledoc """
 | 
						|
  This module is responsible for validating an object (which can be an activity)
 | 
						|
  and checking if it is both well formed and also compatible with our view of
 | 
						|
  the system.
 | 
						|
  """
 | 
						|
 | 
						|
  alias Pleroma.EctoType.ActivityPub.ObjectValidators
 | 
						|
  alias Pleroma.Object
 | 
						|
  alias Pleroma.User
 | 
						|
  alias Pleroma.Web.ActivityPub.ObjectValidators.AnnounceValidator
 | 
						|
  alias Pleroma.Web.ActivityPub.ObjectValidators.BlockValidator
 | 
						|
  alias Pleroma.Web.ActivityPub.ObjectValidators.ChatMessageValidator
 | 
						|
  alias Pleroma.Web.ActivityPub.ObjectValidators.CreateChatMessageValidator
 | 
						|
  alias Pleroma.Web.ActivityPub.ObjectValidators.DeleteValidator
 | 
						|
  alias Pleroma.Web.ActivityPub.ObjectValidators.EmojiReactValidator
 | 
						|
  alias Pleroma.Web.ActivityPub.ObjectValidators.LikeValidator
 | 
						|
  alias Pleroma.Web.ActivityPub.ObjectValidators.UndoValidator
 | 
						|
  alias Pleroma.Web.ActivityPub.ObjectValidators.UpdateValidator
 | 
						|
 | 
						|
  @spec validate(map(), keyword()) :: {:ok, map(), keyword()} | {:error, any()}
 | 
						|
  def validate(object, meta)
 | 
						|
 | 
						|
  def validate(%{"type" => "Block"} = block_activity, meta) do
 | 
						|
    with {:ok, block_activity} <-
 | 
						|
           block_activity
 | 
						|
           |> BlockValidator.cast_and_validate()
 | 
						|
           |> Ecto.Changeset.apply_action(:insert) do
 | 
						|
      block_activity = stringify_keys(block_activity)
 | 
						|
      outgoing_blocks = Pleroma.Config.get([:activitypub, :outgoing_blocks])
 | 
						|
 | 
						|
      meta =
 | 
						|
        if !outgoing_blocks do
 | 
						|
          Keyword.put(meta, :do_not_federate, true)
 | 
						|
        else
 | 
						|
          meta
 | 
						|
        end
 | 
						|
 | 
						|
      {:ok, block_activity, meta}
 | 
						|
    end
 | 
						|
  end
 | 
						|
 | 
						|
  def validate(%{"type" => "Update"} = update_activity, meta) do
 | 
						|
    with {:ok, update_activity} <-
 | 
						|
           update_activity
 | 
						|
           |> UpdateValidator.cast_and_validate()
 | 
						|
           |> Ecto.Changeset.apply_action(:insert) do
 | 
						|
      update_activity = stringify_keys(update_activity)
 | 
						|
      {:ok, update_activity, meta}
 | 
						|
    end
 | 
						|
  end
 | 
						|
 | 
						|
  def validate(%{"type" => "Undo"} = object, meta) do
 | 
						|
    with {:ok, object} <-
 | 
						|
           object
 | 
						|
           |> UndoValidator.cast_and_validate()
 | 
						|
           |> Ecto.Changeset.apply_action(:insert) do
 | 
						|
      object = stringify_keys(object)
 | 
						|
      {:ok, object, meta}
 | 
						|
    end
 | 
						|
  end
 | 
						|
 | 
						|
  def validate(%{"type" => "Delete"} = object, meta) do
 | 
						|
    with cng <- DeleteValidator.cast_and_validate(object),
 | 
						|
         do_not_federate <- DeleteValidator.do_not_federate?(cng),
 | 
						|
         {:ok, object} <- Ecto.Changeset.apply_action(cng, :insert) do
 | 
						|
      object = stringify_keys(object)
 | 
						|
      meta = Keyword.put(meta, :do_not_federate, do_not_federate)
 | 
						|
      {:ok, object, meta}
 | 
						|
    end
 | 
						|
  end
 | 
						|
 | 
						|
  def validate(%{"type" => "Like"} = object, meta) do
 | 
						|
    with {:ok, object} <-
 | 
						|
           object
 | 
						|
           |> LikeValidator.cast_and_validate()
 | 
						|
           |> Ecto.Changeset.apply_action(:insert) do
 | 
						|
      object = stringify_keys(object)
 | 
						|
      {:ok, object, meta}
 | 
						|
    end
 | 
						|
  end
 | 
						|
 | 
						|
  def validate(%{"type" => "ChatMessage"} = object, meta) do
 | 
						|
    with {:ok, object} <-
 | 
						|
           object
 | 
						|
           |> ChatMessageValidator.cast_and_validate()
 | 
						|
           |> Ecto.Changeset.apply_action(:insert) do
 | 
						|
      object = stringify_keys(object)
 | 
						|
      {:ok, object, meta}
 | 
						|
    end
 | 
						|
  end
 | 
						|
 | 
						|
  def validate(%{"type" => "EmojiReact"} = object, meta) do
 | 
						|
    with {:ok, object} <-
 | 
						|
           object
 | 
						|
           |> EmojiReactValidator.cast_and_validate()
 | 
						|
           |> Ecto.Changeset.apply_action(:insert) do
 | 
						|
      object = stringify_keys(object |> Map.from_struct())
 | 
						|
      {:ok, object, meta}
 | 
						|
    end
 | 
						|
  end
 | 
						|
 | 
						|
  def validate(%{"type" => "Create", "object" => object} = create_activity, meta) do
 | 
						|
    with {:ok, object_data} <- cast_and_apply(object),
 | 
						|
         meta = Keyword.put(meta, :object_data, object_data |> stringify_keys),
 | 
						|
         {:ok, create_activity} <-
 | 
						|
           create_activity
 | 
						|
           |> CreateChatMessageValidator.cast_and_validate(meta)
 | 
						|
           |> Ecto.Changeset.apply_action(:insert) do
 | 
						|
      create_activity = stringify_keys(create_activity)
 | 
						|
      {:ok, create_activity, meta}
 | 
						|
    end
 | 
						|
  end
 | 
						|
 | 
						|
  def validate(%{"type" => "Announce"} = object, meta) do
 | 
						|
    with {:ok, object} <-
 | 
						|
           object
 | 
						|
           |> AnnounceValidator.cast_and_validate()
 | 
						|
           |> Ecto.Changeset.apply_action(:insert) do
 | 
						|
      object = stringify_keys(object |> Map.from_struct())
 | 
						|
      {:ok, object, meta}
 | 
						|
    end
 | 
						|
  end
 | 
						|
 | 
						|
  def cast_and_apply(%{"type" => "ChatMessage"} = object) do
 | 
						|
    ChatMessageValidator.cast_and_apply(object)
 | 
						|
  end
 | 
						|
 | 
						|
  def cast_and_apply(o), do: {:error, {:validator_not_set, o}}
 | 
						|
 | 
						|
  def stringify_keys(%{__struct__: _} = object) do
 | 
						|
    object
 | 
						|
    |> Map.from_struct()
 | 
						|
    |> stringify_keys
 | 
						|
  end
 | 
						|
 | 
						|
  def stringify_keys(object) when is_map(object) do
 | 
						|
    object
 | 
						|
    |> Map.new(fn {key, val} -> {to_string(key), stringify_keys(val)} end)
 | 
						|
  end
 | 
						|
 | 
						|
  def stringify_keys(object) when is_list(object) do
 | 
						|
    object
 | 
						|
    |> Enum.map(&stringify_keys/1)
 | 
						|
  end
 | 
						|
 | 
						|
  def stringify_keys(object), do: object
 | 
						|
 | 
						|
  def fetch_actor(object) do
 | 
						|
    with {:ok, actor} <- ObjectValidators.ObjectID.cast(object["actor"]) do
 | 
						|
      User.get_or_fetch_by_ap_id(actor)
 | 
						|
    end
 | 
						|
  end
 | 
						|
 | 
						|
  def fetch_actor_and_object(object) do
 | 
						|
    fetch_actor(object)
 | 
						|
    Object.normalize(object["object"], true)
 | 
						|
    :ok
 | 
						|
  end
 | 
						|
end
 |