FEATURE: only limit new registrations from that IP if all the users from that IP are TL1 or TL0

lib/spam_handler.rb

@@ -3,6 +3,12 @@ class SpamHandler
   def self.should_prevent_registration_from_ip?(ip_address)
     return false if SiteSetting.max_new_accounts_per_registration_ip <= 0
 
+    tl2_plus_accounts_with_same_ip = User.where("trust_level >= ?", TrustLevel[2])
+                                         .where("ip_address = ?", ip_address.to_s)
+                                         .count
+
+    return false if tl2_plus_accounts_with_same_ip > 0
+
     tl0_accounts_with_same_ip = User.unscoped
                                     .where(trust_level: TrustLevel[0])
                                     .where("ip_address = ?", ip_address.to_s)

lib/trust_level.rb

@@ -1,22 +1,20 @@
 require_dependency 'enum'
 
 class InvalidTrustLevel < StandardError; end
+
 class TrustLevel
+
   attr_reader :id, :name
 
   class << self
 
-    def[](level)
-      if !valid?(level)
-        raise InvalidTrustLevel
-      end
+    def [](level)
+      raise InvalidTrustLevel if !valid?(level)
       level
     end
 
     def levels
-      @levels ||= Enum.new(
-        :newuser, :basic, :regular, :leader, :elder, start: 0
-      )
+      @levels ||= Enum.new(:newuser, :basic, :regular, :leader, :elder, start: 0)
     end
 
     def all
@@ -36,6 +34,7 @@ class TrustLevel
     def compare(current_level, level)
       (current_level || 0) >= level
     end
+
   end
 
   def initialize(name_key, id)
@@ -46,4 +45,5 @@ class TrustLevel
   def serializable_hash
     { id: @id, name: @name }
   end
+
 end

spec/components/spam_handler_spec.rb

@@ -22,6 +22,17 @@ describe SpamHandler do
       -> { Fabricate(:user, ip_address: "42.42.42.42", trust_level: TrustLevel[0]) }.should raise_error(ActiveRecord::RecordInvalid)
     end
 
+    it "only limit new registrations from an IP if *all* the users from that IP are TL1 or TL0" do
+      # setup
+      SiteSetting.stubs(:max_new_accounts_per_registration_ip).returns(0)
+      Fabricate(:user, ip_address: "42.42.42.42", trust_level: TrustLevel[0])
+      Fabricate(:user, ip_address: "42.42.42.42", trust_level: TrustLevel[2])
+
+      # should not limit registrations since there is a TL2 user with that IP
+      SiteSetting.stubs(:max_new_accounts_per_registration_ip).returns(1)
+      Fabricate(:user, ip_address: "42.42.42.42", trust_level: TrustLevel[0])
+    end
+
   end
 
 end