mirror of
https://github.com/discourse/discourse.git
synced 2024-11-25 22:43:49 +08:00
55 lines
1.5 KiB
Ruby
55 lines
1.5 KiB
Ruby
|
# frozen_string_literal: true
|
||
|
|
||
|
class QuoteRewriter
|
||
|
def initialize(user_id, old_username, new_username, avatar_img)
|
||
|
@user_id = user_id
|
||
|
@old_username = old_username
|
||
|
@new_username = new_username
|
||
|
@avatar_img = avatar_img
|
||
|
end
|
||
|
|
||
|
def rewrite_raw(raw)
|
||
|
pattern =
|
||
|
Regexp.union(
|
||
|
/(?<pre>\[quote\s*=\s*["'']?.*username:)#{old_username}(?<post>\,?[^\]]*\])/i,
|
||
|
/(?<pre>\[quote\s*=\s*["'']?)#{old_username}(?<post>\,?[^\]]*\])/i,
|
||
|
)
|
||
|
|
||
|
raw.gsub(pattern, "\\k<pre>#{new_username}\\k<post>")
|
||
|
end
|
||
|
|
||
|
def rewrite_cooked(cooked)
|
||
|
pattern = /(?<=\s)#{PrettyText::Helpers.format_username(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
|
||
|
|
||
|
private
|
||
|
|
||
|
attr_reader :user_id, :old_username, :new_username, :avatar_img
|
||
|
|
||
|
def quotes_correct_user?(aside)
|
||
|
Post.exists?(topic_id: aside["data-topic"], post_number: aside["data-post"], user_id: user_id)
|
||
|
end
|
||
|
end
|