Fix poll updates being saved as status edits (#17373)

Fix #17344
This commit is contained in:
Eugen Rochko 2022-01-26 18:05:39 +01:00 committed by GitHub
parent bebf9bf33f
commit 6505b39e5d
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 98 additions and 36 deletions

View file

@ -10,6 +10,7 @@ class ActivityPub::ProcessStatusUpdateService < BaseService
@status = status @status = status
@account = status.account @account = status.account
@media_attachments_changed = false @media_attachments_changed = false
@poll_changed = false
# Only native types can be updated at the moment # Only native types can be updated at the moment
return if !expected_type? || already_updated_more_recently? return if !expected_type? || already_updated_more_recently?
@ -27,6 +28,9 @@ class ActivityPub::ProcessStatusUpdateService < BaseService
end end
queue_poll_notifications! queue_poll_notifications!
next unless significant_changes?
reset_preview_card! reset_preview_card!
broadcast_updates! broadcast_updates!
else else
@ -92,7 +96,7 @@ class ActivityPub::ProcessStatusUpdateService < BaseService
# If for some reasons the options were changed, it invalidates all previous # If for some reasons the options were changed, it invalidates all previous
# votes, so we need to remove them # votes, so we need to remove them
if poll_parser.significantly_changes?(poll) if poll_parser.significantly_changes?(poll)
@media_attachments_changed = true @poll_changed = true
poll.votes.delete_all unless poll.new_record? poll.votes.delete_all unless poll.new_record?
end end
@ -107,7 +111,7 @@ class ActivityPub::ProcessStatusUpdateService < BaseService
@status.poll_id = poll.id @status.poll_id = poll.id
elsif previous_poll.present? elsif previous_poll.present?
previous_poll.destroy! previous_poll.destroy!
@media_attachments_changed = true @poll_changed = true
@status.poll_id = nil @status.poll_id = nil
end end
end end
@ -117,7 +121,7 @@ class ActivityPub::ProcessStatusUpdateService < BaseService
@status.spoiler_text = @status_parser.spoiler_text || '' @status.spoiler_text = @status_parser.spoiler_text || ''
@status.sensitive = @account.sensitized? || @status_parser.sensitive || false @status.sensitive = @account.sensitized? || @status_parser.sensitive || false
@status.language = @status_parser.language || detected_language @status.language = @status_parser.language || detected_language
@status.edited_at = @status_parser.edited_at || Time.now.utc @status.edited_at = @status_parser.edited_at || Time.now.utc if significant_changes?
@status.save! @status.save!
end end
@ -227,12 +231,12 @@ class ActivityPub::ProcessStatusUpdateService < BaseService
end end
def create_edit! def create_edit!
return unless @status.text_previously_changed? || @status.spoiler_text_previously_changed? || @media_attachments_changed return unless significant_changes?
@status_edit = @status.edits.create( @status_edit = @status.edits.create(
text: @status.text, text: @status.text,
spoiler_text: @status.spoiler_text, spoiler_text: @status.spoiler_text,
media_attachments_changed: @media_attachments_changed, media_attachments_changed: @media_attachments_changed || @poll_changed,
account_id: @account.id, account_id: @account.id,
created_at: @status.edited_at created_at: @status.edited_at
) )
@ -248,13 +252,17 @@ class ActivityPub::ProcessStatusUpdateService < BaseService
mime_type.present? && !MediaAttachment.supported_mime_types.include?(mime_type) mime_type.present? && !MediaAttachment.supported_mime_types.include?(mime_type)
end end
def significant_changes?
@status.text_changed? || @status.text_previously_changed? || @status.spoiler_text_changed? || @status.spoiler_text_previously_changed? || @media_attachments_changed || @poll_changed
end
def already_updated_more_recently? def already_updated_more_recently?
@status.edited_at.present? && @status_parser.edited_at.present? && @status.edited_at > @status_parser.edited_at @status.edited_at.present? && @status_parser.edited_at.present? && @status.edited_at > @status_parser.edited_at
end end
def reset_preview_card! def reset_preview_card!
@status.preview_cards.clear if @status.text_previously_changed? || @status.spoiler_text.present? @status.preview_cards.clear
LinkCrawlWorker.perform_in(rand(1..59).seconds, @status.id) if @status.spoiler_text.blank? LinkCrawlWorker.perform_in(rand(1..59).seconds, @status.id)
end end
def broadcast_updates! def broadcast_updates!

View file

@ -4,14 +4,13 @@ RSpec.describe ActivityPub::Activity::Update do
let!(:sender) { Fabricate(:account) } let!(:sender) { Fabricate(:account) }
before do before do
stub_request(:get, actor_json[:outbox]).to_return(status: 404)
stub_request(:get, actor_json[:followers]).to_return(status: 404)
stub_request(:get, actor_json[:following]).to_return(status: 404)
stub_request(:get, actor_json[:featured]).to_return(status: 404)
sender.update!(uri: ActivityPub::TagManager.instance.uri_for(sender)) sender.update!(uri: ActivityPub::TagManager.instance.uri_for(sender))
end end
subject { described_class.new(json, sender) }
describe '#perform' do
context 'with an Actor object' do
let(:modified_sender) do let(:modified_sender) do
sender.tap do |modified_sender| sender.tap do |modified_sender|
modified_sender.display_name = 'Totally modified now' modified_sender.display_name = 'Totally modified now'
@ -32,10 +31,12 @@ RSpec.describe ActivityPub::Activity::Update do
}.with_indifferent_access }.with_indifferent_access
end end
describe '#perform' do
subject { described_class.new(json, sender) }
before do before do
stub_request(:get, actor_json[:outbox]).to_return(status: 404)
stub_request(:get, actor_json[:followers]).to_return(status: 404)
stub_request(:get, actor_json[:following]).to_return(status: 404)
stub_request(:get, actor_json[:featured]).to_return(status: 404)
subject.perform subject.perform
end end
@ -43,4 +44,57 @@ RSpec.describe ActivityPub::Activity::Update do
expect(sender.reload.display_name).to eq 'Totally modified now' expect(sender.reload.display_name).to eq 'Totally modified now'
end end
end end
context 'with a Question object' do
let!(:at_time) { Time.now.utc }
let!(:status) { Fabricate(:status, account: sender, poll: Poll.new(account: sender, options: %w(Bar Baz), cached_tallies: [0, 0], expires_at: at_time + 5.days)) }
let(:json) do
{
'@context': 'https://www.w3.org/ns/activitystreams',
id: 'foo',
type: 'Update',
actor: ActivityPub::TagManager.instance.uri_for(sender),
object: {
type: 'Question',
id: ActivityPub::TagManager.instance.uri_for(status),
content: 'Foo',
endTime: (at_time + 5.days).iso8601,
oneOf: [
{
type: 'Note',
name: 'Bar',
replies: {
type: 'Collection',
totalItems: 0,
},
},
{
type: 'Note',
name: 'Baz',
replies: {
type: 'Collection',
totalItems: 12,
},
},
],
},
}.with_indifferent_access
end
before do
status.update!(uri: ActivityPub::TagManager.instance.uri_for(status))
subject.perform
end
it 'updates poll numbers' do
expect(status.preloadable_poll.cached_tallies).to eq [0, 12]
end
it 'does not set status as edited' do
expect(status.edited_at).to be_nil
end
end
end
end end