Pipeline: Always run common_pipeline in a transaction for now.

This commit is contained in:
lain 2020-04-29 13:45:50 +02:00
parent 67659afe48
commit e055b8d203
4 changed files with 36 additions and 40 deletions

View File

@ -5,6 +5,7 @@
defmodule Pleroma.Web.ActivityPub.Pipeline do defmodule Pleroma.Web.ActivityPub.Pipeline do
alias Pleroma.Activity alias Pleroma.Activity
alias Pleroma.Object alias Pleroma.Object
alias Pleroma.Repo
alias Pleroma.Web.ActivityPub.ActivityPub alias Pleroma.Web.ActivityPub.ActivityPub
alias Pleroma.Web.ActivityPub.MRF alias Pleroma.Web.ActivityPub.MRF
alias Pleroma.Web.ActivityPub.ObjectValidator alias Pleroma.Web.ActivityPub.ObjectValidator
@ -14,6 +15,16 @@ defmodule Pleroma.Web.ActivityPub.Pipeline do
@spec common_pipeline(map(), keyword()) :: @spec common_pipeline(map(), keyword()) ::
{:ok, Activity.t() | Object.t(), keyword()} | {:error, any()} {:ok, Activity.t() | Object.t(), keyword()} | {:error, any()}
def common_pipeline(object, meta) do def common_pipeline(object, meta) do
case Repo.transaction(fn -> do_common_pipeline(object, meta) end) do
{:ok, value} ->
value
{:error, e} ->
{:error, e}
end
end
def do_common_pipeline(object, meta) do
with {_, {:ok, validated_object, meta}} <- with {_, {:ok, validated_object, meta}} <-
{:validate_object, ObjectValidator.validate(object, meta)}, {:validate_object, ObjectValidator.validate(object, meta)},
{_, {:ok, mrfd_object}} <- {:mrf_object, MRF.filter(validated_object)}, {_, {:ok, mrfd_object}} <- {:mrf_object, MRF.filter(validated_object)},

View File

@ -3,24 +3,18 @@
# SPDX-License-Identifier: AGPL-3.0-only # SPDX-License-Identifier: AGPL-3.0-only
defmodule Pleroma.Web.ActivityPub.Transmogrifier.ChatMessageHandling do defmodule Pleroma.Web.ActivityPub.Transmogrifier.ChatMessageHandling do
alias Pleroma.Repo
alias Pleroma.Web.ActivityPub.Pipeline alias Pleroma.Web.ActivityPub.Pipeline
def handle_incoming( def handle_incoming(
%{"type" => "Create", "object" => %{"type" => "ChatMessage"}} = data, %{"type" => "Create", "object" => %{"type" => "ChatMessage"}} = data,
_options _options
) do ) do
# Create has to be run inside a transaction because the object is created as a side effect. case Pipeline.common_pipeline(data, local: false) do
# If this does not work, we need to roll back creating the activity. {:ok, activity, _} ->
case Repo.transaction(fn -> Pipeline.common_pipeline(data, local: false) end) do
{:ok, {:ok, activity, _}} ->
{:ok, activity} {:ok, activity}
{:ok, e} -> e ->
e e
{:error, e} ->
{:error, e}
end end
end end
end end

View File

@ -9,7 +9,6 @@ defmodule Pleroma.Web.CommonAPI do
alias Pleroma.FollowingRelationship alias Pleroma.FollowingRelationship
alias Pleroma.Formatter alias Pleroma.Formatter
alias Pleroma.Object alias Pleroma.Object
alias Pleroma.Repo
alias Pleroma.ThreadMute alias Pleroma.ThreadMute
alias Pleroma.User alias Pleroma.User
alias Pleroma.UserRelationship alias Pleroma.UserRelationship
@ -26,36 +25,27 @@ defmodule Pleroma.Web.CommonAPI do
require Logger require Logger
def post_chat_message(%User{} = user, %User{} = recipient, content) do def post_chat_message(%User{} = user, %User{} = recipient, content) do
transaction = with {_, true} <-
Repo.transaction(fn -> {:content_length,
with {_, true} <- String.length(content) <= Pleroma.Config.get([:instance, :chat_limit])},
{:content_length, {_, {:ok, chat_message_data, _meta}} <-
String.length(content) <= Pleroma.Config.get([:instance, :chat_limit])}, {:build_object,
{_, {:ok, chat_message_data, _meta}} <- Builder.chat_message(
{:build_object, user,
Builder.chat_message( recipient.ap_id,
user, content |> Formatter.html_escape("text/plain")
recipient.ap_id, )},
content |> Formatter.html_escape("text/plain") {_, {:ok, create_activity_data, _meta}} <-
)}, {:build_create_activity, Builder.create(user, chat_message_data, [recipient.ap_id])},
{_, {:ok, create_activity_data, _meta}} <- {_, {:ok, %Activity{} = activity, _meta}} <-
{:build_create_activity, {:common_pipeline,
Builder.create(user, chat_message_data, [recipient.ap_id])}, Pipeline.common_pipeline(create_activity_data,
{_, {:ok, %Activity{} = activity, _meta}} <- local: true
{:common_pipeline, )} do
Pipeline.common_pipeline(create_activity_data, {:ok, activity}
local: true else
)} do {:content_length, false} -> {:error, :content_too_long}
{:ok, activity} e -> e
else
{:content_length, false} -> {:error, :content_too_long}
e -> e
end
end)
case transaction do
{:ok, value} -> value
error -> error
end end
end end

View File

@ -68,6 +68,7 @@ test "it inserts it and creates a chat" do
recipient = insert(:user, ap_id: List.first(data["to"]), local: true) recipient = insert(:user, ap_id: List.first(data["to"]), local: true)
{:ok, %Activity{} = activity} = Transmogrifier.handle_incoming(data) {:ok, %Activity{} = activity} = Transmogrifier.handle_incoming(data)
assert activity.local == false
assert activity.actor == author.ap_id assert activity.actor == author.ap_id
assert activity.recipients == [recipient.ap_id, author.ap_id] assert activity.recipients == [recipient.ap_id, author.ap_id]