Add like-based system badges.

app/jobs/regular/update_badges.rb

@@ -21,5 +21,22 @@ module Jobs
       end
     end
 
+    def post_like(args)
+      post = Post.find(args[:post_id])
+      user = post.user
+
+      # Grant "Welcome" badge to the user if they do not already have it.
+      BadgeGranter.grant(Badge.find(5), user)
+
+      [{id: 6, count: 10}, {id: 7, count: 25}, {id: 8, count: 100}].each do |b|
+        if post.like_count >= b[:count]
+          BadgeGranter.grant(Badge.find(b[:id]), user, post_id: post.id)
+        else
+          user_badge = UserBadge.find_by(badge_id: b[:id], user_id: user.id, post_id: post.id)
+          user_badge && BadgeGranter.revoke(user_badge)
+        end
+      end
+    end
+
   end
 end

app/models/post_action.rb

@@ -131,13 +131,19 @@ class PostAction < ActiveRecord::Base
       post.topic.posts_count != 1
     end
 
-    create( post_id: post.id,
+    post_action = create( post_id: post.id,
-            user_id: user.id,
+                          user_id: user.id,
-            post_action_type_id: post_action_type_id,
+                          post_action_type_id: post_action_type_id,
-            message: opts[:message],
+                          message: opts[:message],
-            staff_took_action: opts[:take_action] || false,
+                          staff_took_action: opts[:take_action] || false,
-            related_post_id: related_post_id,
+                          related_post_id: related_post_id,
-            targets_topic: !!targets_topic )
+                          targets_topic: !!targets_topic )
+
+    if post_action && post_action.is_like?
+      BadgeGranter.update_badges(action: :post_like, post_id: post.id)
+    end
+
+    post_action
 
   rescue ActiveRecord::RecordNotUnique
     # can happen despite being .create

app/services/badge_granter.rb

@@ -3,6 +3,7 @@ class BadgeGranter
   def initialize(badge, user, opts={})
     @badge, @user, @opts = badge, user, opts
     @granted_by = opts[:granted_by] || Discourse.system_user
+    @post_id = opts[:post_id]
   end
 
   def self.grant(badge, user, opts={})
@@ -12,12 +13,14 @@ class BadgeGranter
   def grant
     return if @granted_by and !Guardian.new(@granted_by).can_grant_badges?(@user)
 
-    user_badge = UserBadge.find_by(badge_id: @badge.id, user_id: @user.id)
+    user_badge = UserBadge.find_by(badge_id: @badge.id, user_id: @user.id, post_id: @post_id)
 
     if user_badge.nil? || @badge.multiple_grant?
       UserBadge.transaction do
         user_badge = UserBadge.create!(badge: @badge, user: @user,
-                                       granted_by: @granted_by, granted_at: Time.now)
+                                       granted_by: @granted_by,
+                                       granted_at: Time.now,
+                                       post_id: @post_id)
 
         if @granted_by != Discourse.system_user
           StaffActionLogger.new(@granted_by).log_badge_grant(user_badge)

config/locales/client.en.yml

@@ -1927,3 +1927,15 @@ en:
         elder:
           name: Elder
           description: Granted global edit, pin, close, archive, split and merge.
+        welcome:
+          name: Welcome
+          description: Received a like.
+        nice_post:
+          name: Nice Post
+          description: Received 10 likes on a post.
+        good_post:
+          name: Good Post
+          description: Received 25 likes on a post.
+        great_post:
+          name: Great Post
+          description: Received 100 likes on a post.

db/fixtures/601_badges.rb

@@ -1,3 +1,4 @@
+# Trust level system badges.
 trust_level_badges = [
   {id: 1, name: "Basic User", type: 3},
   {id: 2, name: "Regular User", type: 3},
@@ -44,3 +45,29 @@ SQL
 
   Badge.where(id: Badge.trust_level_badge_ids).each {|badge| badge.reset_grant_count! }
 end
+#
+# Like system badges.
+like_badges = [
+  {id: 5, name: "Welcome", type: 3, multiple: false},
+  {id: 6, name: "Nice Post", type: 3, multiple: true},
+  {id: 7, name: "Good Post", type: 2, multiple: true},
+  {id: 8, name: "Great Post", type: 1, multiple: true}
+]
+
+like_badges.each do |spec|
+  Badge.seed do |b|
+    b.id = spec[:id]
+    b.name = spec[:name]
+    b.badge_type_id = spec[:type]
+    b.multiple_grant = spec[:multiple]
+  end
+end
+
+# Create an example badge if one does not already exist.
+if Badge.find_by(id: 101).nil?
+  Badge.seed do |b|
+    b.id = 101
+    b.name = "Example Badge"
+    b.badge_type_id = 3
+  end
+end

spec/services/badge_granter_spec.rb

@@ -72,6 +72,7 @@ describe BadgeGranter do
 
   context "update_badges" do
     let(:user) { Fabricate(:user) }
+    let(:liker) { Fabricate(:user) }
 
     it "grants and revokes trust level badges" do
       user.change_trust_level!(:elder)
@@ -80,6 +81,29 @@ describe BadgeGranter do
       UserBadge.where(user_id: user.id, badge_id: 1).first.should_not be_nil
       UserBadge.where(user_id: user.id, badge_id: 2).first.should be_nil
     end
+
+    it "grants system like badges" do
+      post = create_post(user: user)
+      # Welcome badge
+      PostAction.act(liker, post, PostActionType.types[:like])
+      UserBadge.find_by(user_id: user.id, badge_id: 5).should_not be_nil
+      # Nice post badge
+      post.update_attributes like_count: 10
+      BadgeGranter.update_badges(action: :post_like, post_id: post.id)
+      UserBadge.find_by(user_id: user.id, badge_id: 6).should_not be_nil
+      # Good post badge
+      post.update_attributes like_count: 25
+      BadgeGranter.update_badges(action: :post_like, post_id: post.id)
+      UserBadge.find_by(user_id: user.id, badge_id: 7).should_not be_nil
+      # Great post badge
+      post.update_attributes like_count: 100
+      BadgeGranter.update_badges(action: :post_like, post_id: post.id)
+      UserBadge.find_by(user_id: user.id, badge_id: 8).should_not be_nil
+      # Revoke badges on unlike
+      post.update_attributes like_count: 99
+      BadgeGranter.update_badges(action: :post_like, post_id: post.id)
+      UserBadge.find_by(user_id: user.id, badge_id: 8).should be_nil
+    end
   end
 
 end