51 lines
1.5 KiB
Ruby
51 lines
1.5 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
module DiscourseTopicVoting
|
|
module TopicExtension
|
|
def self.prepended(base)
|
|
base.has_one :topic_vote_count, class_name: 'DiscourseTopicVoting::TopicVoteCount', dependent: :destroy
|
|
base.has_many :votes, class_name: 'DiscourseTopicVoting::Vote', dependent: :destroy
|
|
base.attribute :current_user_voted
|
|
end
|
|
|
|
def can_topic_vote?
|
|
@can_topic_vote ||= SiteSetting.voting_enabled && regular? && Category.can_vote?(category_id) && category && category.topic_id != id
|
|
end
|
|
|
|
# a little verbose but we have to call it this way as
|
|
# a topic has post votes, and
|
|
# a topic has_one :topic_vote_count
|
|
def topic_topic_vote_count
|
|
self.topic_vote_count&.votes_count.to_i
|
|
end
|
|
|
|
def user_topic_voted?(user)
|
|
if self.current_user_voted
|
|
self.current_user_voted == 1
|
|
else
|
|
votes.map(&:user_id).include?(user.id)
|
|
end
|
|
end
|
|
|
|
def update_vote_count
|
|
count = self.votes.count
|
|
|
|
DB.exec(<<~SQL, topic_id: self.id, votes_count: count)
|
|
INSERT INTO discourse_voting_topic_vote_count
|
|
(topic_id, votes_count, created_at, updated_at)
|
|
VALUES
|
|
(:topic_id, :votes_count, CURRENT_TIMESTAMP, CURRENT_TIMESTAMP)
|
|
ON CONFLICT (topic_id) DO UPDATE SET
|
|
votes_count = :votes_count,
|
|
updated_at = CURRENT_TIMESTAMP
|
|
WHERE discourse_voting_topic_vote_count.topic_id = :topic_id
|
|
SQL
|
|
end
|
|
|
|
def who_voted
|
|
return nil unless SiteSetting.voting_show_who_voted
|
|
self.votes.map(&:user)
|
|
end
|
|
end
|
|
end
|