Add: #591 リモート保留中アカウントからメンションが来た場合にuriを記録し、承認時にフェッチしに行く処理

app/lib/activitypub/activity/create.rb

@@ -50,6 +50,11 @@ def message_franking
   def create_status
     return reject_payload! if unsupported_object_type? || non_matching_uri_hosts?(@account.uri, object_uri) || tombstone_exists? || !related_to_local_activity?
 
+    if @account.suspended?
+      process_pending_status if @account.remote_pending?
+      return
+    end
+
     with_redis_lock("create:#{object_uri}") do
       return if delete_arrived_first?(object_uri) || poll_vote?
 
@@ -405,6 +410,18 @@ def poll_vote!
     ActivityPub::DistributePollUpdateWorker.perform_in(3.minutes, replied_to_status.id) unless replied_to_status.preloadable_poll.hide_totals?
   end
 
+  def process_pending_status
+    return if PendingStatus.exists?(uri: @object['id'])
+
+    fetch_account = as_array(@object['tag'])
+                    .filter_map { |tag| equals_or_includes?(tag['type'], 'Mention') && tag['href'] && ActivityPub::TagManager.instance.local_uri?(tag['href']) && ActivityPub::TagManager.instance.uri_to_resource(tag['href'], Account) }
+                    .first
+    fetch_account ||= (audience_to + audience_cc).filter_map { |uri| ActivityPub::TagManager.instance.local_uri?(uri) && ActivityPub::TagManager.instance.uri_to_resource(uri, Account) }.first
+    fetch_account ||= Account.representative
+
+    PendingStatus.create!(account: @account, uri: @object['id'], fetch_account: fetch_account)
+  end
+
   def resolve_thread(status)
     return unless status.reply? && status.thread.nil? && Request.valid_url?(in_reply_to_uri)
 

app/lib/activitypub/activity/follow.rb

@@ -20,6 +20,11 @@ def perform
       return
     end
 
+    if @account.suspended?
+      PendingFollowRequest.create!(account: @account, target_account: target_account, uri: @json['id']) if @account.remote_pending?
+      return
+    end
+
     if target_account.blocking?(@account) || target_account.domain_blocking?(@account.domain) || target_account.moved? || target_account.instance_actor? || block_new_follow?
       reject_follow_request!(target_account)
       return
@@ -33,13 +38,6 @@ def perform
       return
     end
 
-    if @account.suspended? && @account.remote_pending?
-      PendingFollowRequest.create!(account: @account, target_account: target_account, uri: @json['id'])
-      return
-    elsif @account.suspended?
-      return
-    end
-
     follow_request = FollowRequest.create!(account: @account, target_account: target_account, uri: @json['id'])
 
     if request_pending_follow?(@account, target_account)

app/models/account.rb

@@ -300,7 +300,7 @@ def unsuspend!
   def approve_remote!
     update!(remote_pending: false)
     unsuspend!
-    EnableFollowRequestsWorker.perform_async(id)
+    ActivateRemoteAccountWorker.perform_async(id)
   end
 
   def reject_remote!

app/models/pending_status.rb

@@ -0,0 +1,18 @@
+# frozen_string_literal: true
+
+# == Schema Information
+#
+# Table name: pending_statuses
+#
+#  id               :bigint(8)        not null, primary key
+#  account_id       :bigint(8)        not null
+#  fetch_account_id :bigint(8)        not null
+#  uri              :string           not null
+#  created_at       :datetime         not null
+#  updated_at       :datetime         not null
+#
+
+class PendingStatus < ApplicationRecord
+  belongs_to :account
+  belongs_to :fetch_account, class_name: 'Account'
+end

app/services/enable_follow_requests_service.rb → app/services/activate_follow_requests_service.rb

@@ -1,6 +1,6 @@
 # frozen_string_literal: true
 
-class EnableFollowRequestsService < BaseService
+class ActivateFollowRequestsService < BaseService
   include Payloadable
   include FollowHelper
 

app/services/activate_remote_statuses_service.rb

@@ -0,0 +1,29 @@
+# frozen_string_literal: true
+
+class ActivateRemoteStatusesService < BaseService
+  include Payloadable
+  include FollowHelper
+
+  def call(account)
+    @account = account
+
+    PendingStatus.transaction do
+      PendingStatus.where(account: account).find_each do |status_info|
+        approve_status!(status_info)
+      end
+    end
+  end
+
+  private
+
+  def approve_status!(pending)
+    account_id       = pending.account_id
+    fetch_account_id = pending.fetch_account_id
+    uri              = pending.uri
+    pending.destroy!
+
+    return if ActivityPub::TagManager.instance.uri_to_resource(uri, Status).present?
+
+    FetchRemoteStatusWorker.perform_async(uri, account_id, fetch_account_id)
+  end
+end

app/services/activitypub/process_collection_service.rb

@@ -56,7 +56,7 @@ def activity_allowed_while_suspended?
   end
 
   def activity_allowed_while_remote_pending?
-    %w(Follow).include?(@json['type']) || activity_allowed_while_suspended?
+    %w(Follow Create).include?(@json['type']) || activity_allowed_while_suspended?
   end
 
   def process_items(items)

app/workers/enable_follow_requests_worker.rb → app/workers/activate_remote_account_worker.rb

@@ -1,13 +1,14 @@
 # frozen_string_literal: true
 
-class EnableFollowRequestsWorker
+class ActivateRemoteAccountWorker
   include Sidekiq::Worker
 
   def perform(account_id)
     account = Account.find_by(id: account_id)
     return true if account.nil?
     return true if account.suspended?
 
-    EnableFollowRequestsService.new.call(account)
+    ActivateFollowRequestsService.new.call(account)
+    ActivateRemoteStatusesService.new.call(account)
   end
 end

app/workers/activitypub/fetch_remove_status_worker.rb

@@ -0,0 +1,17 @@
+# frozen_string_literal: true
+
+class ActivityPub::FetchRemoteStatusWorker
+  include Sidekiq::Worker
+  include Redisable
+
+  sidekiq_options queue: 'pull', retry: 3
+
+  def perform(uri, author_account_id, on_behalf_of_account_id)
+    author       = Account.find(author_account_id)
+    on_behalf_of = on_behalf_of_account_id.present? ? Account.find(on_behalf_of_account_id) : nil
+
+    ActivityPub::FetchRemoteStatusService.new.call(uri, on_behalf_of: on_behalf_of, expected_actor_uri: author, request_id: uri)
+  rescue ActiveRecord::RecordNotFound, Mastodon::RaceConditionError
+    true
+  end
+end

db/migrate/20240227225017_create_pending_statuses.rb

@@ -0,0 +1,13 @@
+# frozen_string_literal: true
+
+class CreatePendingStatuses < ActiveRecord::Migration[7.1]
+  def change
+    create_table :pending_statuses do |t|
+      t.references :account, null: false, foreign_key: { on_delete: :cascade }
+      t.references :fetch_account, null: false, foreign_key: { to_table: 'accounts', on_delete: :cascade }
+      t.string :uri, null: false, index: { unique: true }
+
+      t.timestamps
+    end
+  end
+end

db/schema.rb

@@ -10,7 +10,7 @@
 #
 # It's strongly recommended that you check this file into your version control system.
 
-ActiveRecord::Schema[7.1].define(version: 2024_02_27_222450) do
+ActiveRecord::Schema[7.1].define(version: 2024_02_27_225017) do
   # These are extensions that must be enabled in order to support this database
   enable_extension "plpgsql"
 
@@ -1023,6 +1023,17 @@
     t.index ["uri"], name: "index_pending_follow_requests_on_uri", unique: true
   end
 
+  create_table "pending_statuses", force: :cascade do |t|
+    t.bigint "account_id", null: false
+    t.bigint "fetch_account_id", null: false
+    t.string "uri", null: false
+    t.datetime "created_at", null: false
+    t.datetime "updated_at", null: false
+    t.index ["account_id"], name: "index_pending_statuses_on_account_id"
+    t.index ["fetch_account_id"], name: "index_pending_statuses_on_fetch_account_id"
+    t.index ["uri"], name: "index_pending_statuses_on_uri", unique: true
+  end
+
   create_table "pghero_space_stats", force: :cascade do |t|
     t.text "database"
     t.text "schema"
@@ -1617,6 +1628,8 @@
   add_foreign_key "one_time_keys", "devices", on_delete: :cascade
   add_foreign_key "pending_follow_requests", "accounts", column: "target_account_id", on_delete: :cascade
   add_foreign_key "pending_follow_requests", "accounts", on_delete: :cascade
+  add_foreign_key "pending_statuses", "accounts", column: "fetch_account_id", on_delete: :cascade
+  add_foreign_key "pending_statuses", "accounts", on_delete: :cascade
   add_foreign_key "poll_votes", "accounts", on_delete: :cascade
   add_foreign_key "poll_votes", "polls", on_delete: :cascade
   add_foreign_key "polls", "accounts", on_delete: :cascade

lib/tasks/dangerous.rake

@@ -91,6 +91,7 @@ namespace :dangerous do
       20240218233621
       20240227033337
       20240227222450
+      20240227225017
     )
     # Removed: account_groups
     target_tables = %w(
@@ -111,6 +112,7 @@ namespace :dangerous do
       ng_rule_histories
       ngword_histories
       pending_follow_requests
+      pending_statuses
       scheduled_expiration_statuses
       status_capability_tokens
       status_references

spec/lib/activitypub/activity/create_spec.rb

@@ -2468,6 +2468,40 @@ def activity_for_object(json)
       end
     end
 
+    context 'when sender is in remote pending' do
+      subject { described_class.new(json, sender, delivery: true) }
+
+      let!(:local_account) { Fabricate(:account) }
+      let(:object_json) do
+        {
+          id: [ActivityPub::TagManager.instance.uri_for(sender), '#bar'].join,
+          type: 'Note',
+          content: 'Lorem ipsum',
+          to: local_account ? ActivityPub::TagManager.instance.uri_for(local_account) : 'https://www.w3.org/ns/activitystreams#Public',
+        }
+      end
+
+      before do
+        sender.update(suspended_at: Time.now.utc, suspension_origin: :local, remote_pending: true)
+        subject.perform
+      end
+
+      it 'does not create a status' do
+        status = sender.statuses.first
+
+        expect(status).to be_nil
+      end
+
+      it 'pending data is created' do
+        pending = PendingStatus.find_by(account: sender)
+
+        expect(pending).to_not be_nil
+        expect(pending.uri).to eq object_json[:id]
+        expect(pending.account_id).to eq sender.id
+        expect(pending.fetch_account_id).to eq local_account.id
+      end
+    end
+
     context 'when sender is followed by local users' do
       subject { described_class.new(json, sender, delivery: true) }
 

spec/services/enable_follow_requests_service_spec.rb → spec/services/activate_follow_requests_service_spec.rb

@@ -2,7 +2,7 @@
 
 require 'rails_helper'
 
-RSpec.describe EnableFollowRequestsService, type: :service do
+RSpec.describe ActivateFollowRequestsService, type: :service do
   subject { described_class.new.call(sender) }
 
   let(:sender) { Fabricate(:account, domain: 'example.com', uri: 'https://example.com/actor') }