mirror of
https://github.com/discourse/discourse.git
synced 2024-11-24 01:58:12 +08:00
77 lines
2.5 KiB
Ruby
77 lines
2.5 KiB
Ruby
class QuotedPost < ActiveRecord::Base
|
|
belongs_to :post
|
|
belongs_to :quoted_post, class_name: 'Post'
|
|
|
|
# NOTE we already have a path that does this for topic links,
|
|
# however topic links exclude quotes and links within a topic
|
|
# we are double parsing this fragment, this may be worth optimising later
|
|
def self.extract_from(post)
|
|
|
|
doc = Nokogiri::HTML.fragment(post.cooked)
|
|
|
|
uniq = {}
|
|
ids = []
|
|
|
|
doc.css("aside.quote[data-topic]").each do |a|
|
|
topic_id = a['data-topic'].to_i
|
|
post_number = a['data-post'].to_i
|
|
|
|
next if topic_id == 0 || post_number == 0
|
|
next if uniq[[topic_id, post_number]]
|
|
uniq[[topic_id, post_number]] = true
|
|
|
|
begin
|
|
# It would be so much nicer if we used post_id in quotes
|
|
results = exec_sql "INSERT INTO quoted_posts(post_id, quoted_post_id, created_at, updated_at)
|
|
SELECT :post_id, p.id, current_timestamp, current_timestamp
|
|
FROM posts p
|
|
LEFT JOIN quoted_posts q on q.post_id = :post_id AND q.quoted_post_id = p.id
|
|
WHERE post_number = :post_number AND
|
|
topic_id = :topic_id AND
|
|
q.id IS NULL
|
|
RETURNING quoted_post_id
|
|
", post_id: post.id, post_number: post_number, topic_id: topic_id
|
|
|
|
results = results.to_a
|
|
ids << results[0]["quoted_post_id"].to_i if results.length > 0
|
|
rescue ActiveRecord::RecordNotUnique, PG::UniqueViolation
|
|
# it's fine
|
|
end
|
|
end
|
|
|
|
if ids.length > 0
|
|
exec_sql "DELETE FROM quoted_posts WHERE post_id = :post_id AND quoted_post_id NOT IN (:ids)",
|
|
post_id: post.id, ids: ids
|
|
end
|
|
|
|
# simplest place to add this code
|
|
reply_quoted = false
|
|
|
|
if post.reply_to_post_number
|
|
reply_post_id = Post.where(topic_id: post.topic_id, post_number: post.reply_to_post_number).pluck(:id).first
|
|
reply_quoted = !!(reply_post_id && ids.include?(reply_post_id))
|
|
end
|
|
|
|
if reply_quoted != post.reply_quoted
|
|
post.update_columns(reply_quoted: reply_quoted)
|
|
end
|
|
|
|
end
|
|
end
|
|
|
|
# == Schema Information
|
|
#
|
|
# Table name: quoted_posts
|
|
#
|
|
# id :integer not null, primary key
|
|
# post_id :integer not null
|
|
# quoted_post_id :integer not null
|
|
# created_at :datetime not null
|
|
# updated_at :datetime not null
|
|
#
|
|
# Indexes
|
|
#
|
|
# index_quoted_posts_on_post_id_and_quoted_post_id (post_id,quoted_post_id) UNIQUE
|
|
# index_quoted_posts_on_quoted_post_id_and_post_id (quoted_post_id,post_id) UNIQUE
|
|
#
|