107 lines
2.5 KiB
Ruby
107 lines
2.5 KiB
Ruby
module Awardable
|
|
extend ActiveSupport::Concern
|
|
|
|
included do
|
|
has_many :award_emoji, -> { includes(:user).order(:id) }, as: :awardable, dependent: :destroy
|
|
|
|
if self < Participable
|
|
# By default we always load award_emoji user association
|
|
participant :award_emoji
|
|
end
|
|
end
|
|
|
|
module ClassMethods
|
|
def order_upvotes_desc
|
|
order_votes_desc(AwardEmoji::UPVOTE_NAME)
|
|
end
|
|
|
|
def order_downvotes_desc
|
|
order_votes_desc(AwardEmoji::DOWNVOTE_NAME)
|
|
end
|
|
|
|
def order_votes_desc(emoji_name)
|
|
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
|
|
|
|
joins(join_clause).group(awardable_table[:id]).reorder("COUNT(award_emoji.id) DESC")
|
|
end
|
|
end
|
|
|
|
def grouped_awards(with_thumbs: true)
|
|
# 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
|
|
|
|
def awardable_votes?(name)
|
|
AwardEmoji::UPVOTE_NAME == name || AwardEmoji::DOWNVOTE_NAME == name
|
|
end
|
|
|
|
def user_can_award?(current_user, name)
|
|
if user_authored?(current_user)
|
|
!awardable_votes?(normalize_name(name))
|
|
else
|
|
true
|
|
end
|
|
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?
|
|
award_emoji.create(name: normalize_name(name), user: current_user)
|
|
end
|
|
|
|
def remove_award_emoji(name, current_user)
|
|
award_emoji.where(name: name, user: current_user).destroy_all
|
|
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
|
|
|
|
private
|
|
|
|
def normalize_name(name)
|
|
Gitlab::Emoji.normalize_emoji_name(name)
|
|
end
|
|
end
|