mirror of
https://github.com/discourse/discourse.git
synced 2024-12-24 02:46:17 +08:00
7d484864fe
Ensure we escape the display names before passing it to the regexp used to update quotes whenever a user change their display name.
83 lines
2.6 KiB
Ruby
83 lines
2.6 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
class QuoteRewriter
|
|
def initialize(user_id)
|
|
@user_id = user_id
|
|
end
|
|
|
|
def rewrite_raw_username(raw, old_username, new_username)
|
|
escaped_old_username = Regexp.escape(old_username)
|
|
pattern =
|
|
Regexp.union(
|
|
/(?<pre>\[quote\s*=\s*["'']?.*username:)#{escaped_old_username}(?<post>\,?[^\]]*\])/i,
|
|
/(?<pre>\[quote\s*=\s*["'']?)#{escaped_old_username}(?<post>\,?[^\]]*\])/i,
|
|
)
|
|
|
|
raw.gsub(pattern, "\\k<pre>#{new_username}\\k<post>")
|
|
end
|
|
|
|
def rewrite_cooked_username(cooked, old_username, new_username, avatar_img)
|
|
formatted_old_username = PrettyText::Helpers.format_username(old_username)
|
|
escaped_old_username = Regexp.escape(formatted_old_username)
|
|
pattern = /(?<=\s)#{escaped_old_username}(?=:)/i
|
|
|
|
cooked
|
|
.css("aside.quote")
|
|
.each do |aside|
|
|
next unless div = aside.at_css("div.title")
|
|
|
|
username_replaced = false
|
|
|
|
aside["data-username"] = new_username if aside["data-username"] == old_username
|
|
|
|
div.children.each do |child|
|
|
if child.text?
|
|
content = child.content
|
|
username_replaced = content.gsub!(pattern, new_username).present?
|
|
child.content = content if username_replaced
|
|
end
|
|
end
|
|
|
|
if username_replaced || quotes_correct_user?(aside)
|
|
div.at_css("img.avatar")&.replace(avatar_img)
|
|
end
|
|
end
|
|
end
|
|
|
|
def rewrite_raw_display_name(raw, old_display_name, new_display_name)
|
|
escaped_old_display_name = Regexp.escape(old_display_name)
|
|
pattern =
|
|
/(?<pre>\[quote\s*=\s*["'']?)#{escaped_old_display_name}(?<post>\,[^\]]*username[^\]]*\])/i
|
|
|
|
raw.gsub(pattern, "\\k<pre>#{new_display_name}\\k<post>")
|
|
end
|
|
|
|
def rewrite_cooked_display_name(cooked, old_display_name, new_display_name)
|
|
formatted_old_display_name = PrettyText::Helpers.format_username(old_display_name)
|
|
escaped_old_display_name = Regexp.escape(formatted_old_display_name)
|
|
pattern = /(?<=\s)#{escaped_old_display_name}(?=:)/i
|
|
|
|
cooked
|
|
.css("aside.quote")
|
|
.each do |aside|
|
|
next unless div = aside.at_css("div.title")
|
|
|
|
div.children.each do |child|
|
|
if child.text?
|
|
content = child.content
|
|
display_name_replaced = content.gsub!(pattern, new_display_name).present?
|
|
child.content = content if display_name_replaced
|
|
end
|
|
end
|
|
end
|
|
end
|
|
|
|
private
|
|
|
|
attr_reader :user_id
|
|
|
|
def quotes_correct_user?(aside)
|
|
Post.exists?(topic_id: aside["data-topic"], post_number: aside["data-post"], user_id: user_id)
|
|
end
|
|
end
|