# 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