debian-mirror-gitlab/app/models/concerns/awardable.rb

136 lines
3.4 KiB
Ruby
Raw Normal View History

2018-11-20 20:47:30 +05:30
# frozen_string_literal: true
module Awardable
extend ActiveSupport::Concern
included do
2017-09-10 17:25:29 +05:30
has_many :award_emoji, -> { includes(:user).order(:id) }, as: :awardable, dependent: :destroy # rubocop:disable Cop/ActiveRecordDependent
if self < Participable
2016-08-24 12:49:21 +05:30
# By default we always load award_emoji user association
participant :award_emoji
end
end
2018-11-20 20:47:30 +05:30
class_methods do
2018-12-13 13:39:08 +05:30
def awarded(user, name = nil)
2018-03-17 18:26:18 +05:30
sql = <<~EOL
EXISTS (
SELECT TRUE
FROM award_emoji
WHERE user_id = :user_id AND
2018-12-13 13:39:08 +05:30
#{"name = :name AND" if name.present?}
2018-03-17 18:26:18 +05:30
awardable_type = :awardable_type AND
awardable_id = #{self.arel_table.name}.id
)
EOL
where(sql, user_id: user.id, name: name, awardable_type: self.name)
end
2018-12-13 13:39:08 +05:30
def not_awarded(user)
sql = <<~EOL
NOT EXISTS (
SELECT TRUE
FROM award_emoji
WHERE user_id = :user_id AND
awardable_type = :awardable_type AND
awardable_id = #{self.arel_table.name}.id
)
EOL
where(sql, user_id: user.id, awardable_type: self.name)
end
def order_upvotes_desc
2018-12-23 12:14:25 +05:30
order_votes(AwardEmoji::UPVOTE_NAME, 'DESC')
end
def order_upvotes_asc
order_votes(AwardEmoji::UPVOTE_NAME, 'ASC')
end
def order_downvotes_desc
2018-12-23 12:14:25 +05:30
order_votes(AwardEmoji::DOWNVOTE_NAME, 'DESC')
end
2018-12-23 12:14:25 +05:30
# Order votes by emoji, optional sort order param `descending` defaults to true
def order_votes(emoji_name, direction)
awardable_table = self.arel_table
awards_table = AwardEmoji.arel_table
join_clause = awardable_table.join(awards_table, Arel::Nodes::OuterJoin).on(
awards_table[:awardable_id].eq(awardable_table[:id]).and(
awards_table[:awardable_type].eq(self.name).and(
awards_table[:name].eq(emoji_name)
)
)
).join_sources
2018-12-23 12:14:25 +05:30
joins(join_clause).group(awardable_table[:id]).reorder("COUNT(award_emoji.id) #{direction}")
end
end
def grouped_awards(with_thumbs: true)
2016-08-24 12:49:21 +05:30
# By default we always load award_emoji user association
awards = award_emoji.group_by(&:name)
if with_thumbs
awards[AwardEmoji::UPVOTE_NAME] ||= []
awards[AwardEmoji::DOWNVOTE_NAME] ||= []
end
awards
end
def downvotes
award_emoji.downvotes.count
end
def upvotes
award_emoji.upvotes.count
end
def emoji_awardable?
true
end
2018-11-20 20:47:30 +05:30
def user_can_award?(current_user)
Ability.allowed?(current_user, :award_emoji, self)
2016-09-29 09:46:39 +05:30
end
def user_authored?(current_user)
author = self.respond_to?(:author) ? self.author : self.user
author == current_user
end
def awarded_emoji?(emoji_name, current_user)
award_emoji.where(name: emoji_name, user: current_user).exists?
end
def create_award_emoji(name, current_user)
return unless emoji_awardable?
2018-03-17 18:26:18 +05:30
2016-08-24 12:49:21 +05:30
award_emoji.create(name: normalize_name(name), user: current_user)
end
def remove_award_emoji(name, current_user)
2018-11-20 20:47:30 +05:30
award_emoji.where(name: name, user: current_user).destroy_all # rubocop: disable DestroyAll
end
def toggle_award_emoji(emoji_name, current_user)
if awarded_emoji?(emoji_name, current_user)
remove_award_emoji(emoji_name, current_user)
else
create_award_emoji(emoji_name, current_user)
end
end
2016-08-24 12:49:21 +05:30
private
def normalize_name(name)
2017-08-17 22:00:37 +05:30
Gitlab::Emoji.normalize_emoji_name(name)
2016-08-24 12:49:21 +05:30
end
end