Refactor emoji reactions

Instead of processing tag and then look for the custom emoji, let the processing return an emoji.

Add `name` to `process_emoji_tags` to check if it matches the shortcode.

Removed `process_single_emoji` and added its code to `process_emoji_tags`

Removed arg from `maybe_process_misskey_reaction`.
Ideally, `original_status` should be a global object, but I wanted to modify vanilla code as little as possible.

Signed-off-by: Plastikmensch <plastikmensch@users.noreply.github.com>
This commit is contained in:
Plastikmensch 2023-05-17 14:53:11 +02:00 committed by Jeremy Kescher
parent f4f1cc9eab
commit 1a42ad69f7
No known key found for this signature in database
GPG Key ID: 80A419A7A613DFA4
4 changed files with 20 additions and 30 deletions

View File

@ -182,19 +182,15 @@ class ActivityPub::Activity
# Ensure emoji declared in the activity's tags are # Ensure emoji declared in the activity's tags are
# present in the database and downloaded to the local cache. # present in the database and downloaded to the local cache.
# Required by EmojiReact and Like for emoji reactions. # Required by EmojiReact and Like for emoji reactions.
def process_emoji_tags(tags) def process_emoji_tags(name, tags)
emoji_tag = as_array(tags).find { |tag| tag['type'] == 'Emoji' } tag = as_array(tags).find { |item| item['type'] == 'Emoji' }
return if emoji_tag.nil? return if tag.nil?
process_single_emoji emoji_tag
end
def process_single_emoji(tag)
custom_emoji_parser = ActivityPub::Parser::CustomEmojiParser.new(tag) custom_emoji_parser = ActivityPub::Parser::CustomEmojiParser.new(tag)
return if custom_emoji_parser.shortcode.blank? || custom_emoji_parser.image_remote_url.blank? return if custom_emoji_parser.shortcode.blank? || custom_emoji_parser.image_remote_url.blank? || !name.eql?(custom_emoji_parser.shortcode)
emoji = CustomEmoji.find_by(shortcode: custom_emoji_parser.shortcode, domain: @account.domain) emoji = CustomEmoji.find_by(shortcode: custom_emoji_parser.shortcode, domain: @account.domain)
return unless emoji.nil? || return emoji unless emoji.nil? ||
custom_emoji_parser.image_remote_url != emoji.image_remote_url || custom_emoji_parser.image_remote_url != emoji.image_remote_url ||
(custom_emoji_parser.updated_at && custom_emoji_parser.updated_at >= emoji.updated_at) (custom_emoji_parser.updated_at && custom_emoji_parser.updated_at >= emoji.updated_at)
@ -206,6 +202,8 @@ class ActivityPub::Activity
emoji.save emoji.save
rescue Seahorse::Client::NetworkingError => e rescue Seahorse::Client::NetworkingError => e
Rails.logger.warn "Error fetching emoji: #{e}" Rails.logger.warn "Error fetching emoji: #{e}"
return
end end
emoji
end end
end end

View File

@ -8,12 +8,10 @@ class ActivityPub::Activity::EmojiReact < ActivityPub::Activity
!original_status.account.local? || !original_status.account.local? ||
delete_arrived_first?(@json['id']) delete_arrived_first?(@json['id'])
custom_emoji = nil
if /^:.*:$/.match?(name) if /^:.*:$/.match?(name)
process_emoji_tags(@json['tag'])
name.delete! ':' name.delete! ':'
custom_emoji = CustomEmoji.find_by(shortcode: name, domain: @account.domain) custom_emoji = process_emoji_tags(@json['tag'])
return if custom_emoji.nil? return if custom_emoji.nil?
end end

View File

@ -5,7 +5,7 @@ class ActivityPub::Activity::Like < ActivityPub::Activity
original_status = status_from_uri(object_uri) original_status = status_from_uri(object_uri)
return if original_status.nil? || !original_status.account.local? || delete_arrived_first?(@json['id']) return if original_status.nil? || !original_status.account.local? || delete_arrived_first?(@json['id'])
return if maybe_process_misskey_reaction(original_status) return if maybe_process_misskey_reaction
return if @account.favourited?(original_status) return if @account.favourited?(original_status)
@ -17,16 +17,15 @@ class ActivityPub::Activity::Like < ActivityPub::Activity
# Misskey delivers reactions as likes with the emoji in _misskey_reaction # Misskey delivers reactions as likes with the emoji in _misskey_reaction
# see https://misskey-hub.net/ns.html#misskey-reaction for details # see https://misskey-hub.net/ns.html#misskey-reaction for details
def maybe_process_misskey_reaction(original_status) def maybe_process_misskey_reaction
original_status = status_from_uri(object_uri)
name = @json['_misskey_reaction'] name = @json['_misskey_reaction']
return false if name.nil? return false if name.nil?
custom_emoji = nil
if /^:.*:$/.match?(name) if /^:.*:$/.match?(name)
process_emoji_tags(@json['tag'])
name.delete! ':' name.delete! ':'
custom_emoji = CustomEmoji.find_by(shortcode: name, domain: @account.domain) custom_emoji = process_emoji_tags(@json['tag'])
return false if custom_emoji.nil? # invalid custom emoji, treat it as a regular like return false if custom_emoji.nil? # invalid custom emoji, treat it as a regular like
end end
return true if @account.reacted?(original_status, name, custom_emoji) return true if @account.reacted?(original_status, name, custom_emoji)

View File

@ -119,22 +119,17 @@ class ActivityPub::Activity::Undo < ActivityPub::Activity
def undo_emoji_react def undo_emoji_react
name = @object['content'] || @object['_misskey_reaction'] name = @object['content'] || @object['_misskey_reaction']
tags = @object['tag']
return if name.nil? return if name.nil?
status = status_from_uri(target_uri) status = status_from_uri(target_uri)
name.delete! ':'
return if status.nil? || !status.account.local? return if status.nil? || !status.account.local?
custom_emoji = nil if /^:.*:$/.match?(name)
emoji_tag = as_array(tags).find { |tag| tag['type'] == 'Emoji' } name.delete! ':'
custom_emoji = process_emoji_tags(name, @object['tag'])
if emoji_tag return if custom_emoji.nil?
custom_emoji_parser = ActivityPub::Parser::CustomEmojiParser.new(emoji_tag)
return if custom_emoji_parser.shortcode.blank? || custom_emoji_parser.image_remote_url.blank?
custom_emoji = CustomEmoji.find_by(shortcode: custom_emoji_parser.shortcode, domain: @account.domain)
end end
if @account.reacted?(status, name, custom_emoji) if @account.reacted?(status, name, custom_emoji)