mana-potion 0.1.0 → 0.2.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 80942d403a27cbfd662010a58bcfa152da5799ce
-  data.tar.gz: 47cd96ebb4f0f9ef77334e4f3fc7d8454539ae65
+  metadata.gz: 9c9af497a3a05274ce139631e4a087e51de2b326
+  data.tar.gz: 1ec2c03a3ce9bb2634d061d8960cbe8b57885f82
 SHA512:
-  metadata.gz: 0872d4cdd206a859d238b99e1ccca4e908a4eea47654aa75f5d46e2e6589b34265f1a98ef25e609ffbbfd8a5289f08f7be04120132ce8abfdbecb9082ec51960
-  data.tar.gz: 73004fe76622ecbbd22785e7bd6acfcb0d2dec66fdf185726d70a2e5f438686e3ce285475956adfd3ff8ee573e685b00f6827ed1d102cb33b7540da4157e807b
+  metadata.gz: c54e21e53dc8664a61520c34f990d601e9faa01d6109619b6b72a793164609d4f8789ac1d860a5d330ee5e2975cd58a020ee85d322c57db6ec3b6ca1cc035250
+  data.tar.gz: f9db133e5d0ce04186e48d4804b1bacff363861b380e7c808ce58394f1ff162ad92880d26ab966592fe95997457504c668f1b52f289d7da846069a0f19165302

data/README.md

@@ -28,6 +28,7 @@ You just need to include the `ManaPotion::Pool` module in your model's class and
 ```ruby
 class Post < ActiveRecord::Base
   include ManaPotion::Pool
+  belongs_to :user
 
   mana_pool_for :user
 end
@@ -46,11 +47,32 @@ If you want to configure the limit and/or period, just pass them as parameters t
 ```ruby
 class Post < ActiveRecord::Base
   include ManaPotion::Pool
+  belongs_to :user
 
   mana_pool_for :user, limit: 10, period: 1.hour
 end
 ```
 
+The `limit` and `period` options also accept procs, so you can dynamically define their values:
+
+```ruby
+class Post < ActiveRecord::Base
+  include ManaPotion::Pool
+  belongs_to :user
+
+  mana_pool_for :user, limit: -> { limit }, period: -> { period }
+
+  def limit
+    user.premium? ? 20 : 10
+  end
+
+  def period
+    user.premium? ? 1.hour : 1.day
+  end
+end
+```
+
+
 ## Development
 
 After checking out the repo, run `bin/setup` to install dependencies. Then, run `bin/console` for an interactive prompt that will allow you to experiment.

data/lib/mana-potion.rb

@@ -1,5 +1,5 @@
-require "mana-potion/version"
+require 'mana-potion/version'
+require 'mana-potion/pool'
 
 module ManaPotion
-  # Your code goes here...
 end

data/lib/mana-potion/check_usage.rb

@@ -0,0 +1,30 @@
+module ManaPotion
+  class CheckUsage
+    attr_reader :model, :owner, :limit, :period
+
+    def initialize(model, owner, limit, period)
+      @model = model
+      @owner = owner
+      @limit = limit
+      @period = period
+    end
+
+    def exceeded?
+      count >= limit
+    end
+
+    def association
+      owner
+      .class
+      .reflect_on_all_associations
+      .detect { |r| r.class_name == model.class.name }
+    end
+
+    def count
+      owner
+      .send(association.name)
+      .where(created_at: period.ago..Time.current)
+      .count
+    end
+  end
+end

data/lib/mana-potion/pool.rb

@@ -1,3 +1,5 @@
+require 'mana-potion/check_usage'
+
 module ManaPotion
   module Pool
     def self.included(other)
@@ -7,19 +9,12 @@ module ManaPotion
     module ClassMethods
       def mana_pool_for(association, limit: 1, period: 1.day)
         before_validation do
-          owner = send(association)
-          other_side_association = owner
-          .class
-          .reflect_on_all_associations
-          .detect { |r| r.class_name == self.class.name }
-
-          count = owner
-          .send(other_side_association.name)
-          .where(created_at: period.ago..Time.current)
-          .count
+          limit = instance_exec &limit if limit.respond_to?(:call)
+          period = instance_exec &period if period.respond_to?(:call)
 
-          if count >= limit
-            errors.add(association, :limit, limit: limit, count: count)
+          check_usage = ManaPotion::CheckUsage.new(self, send(association), limit, period)
+          if check_usage.exceeded?
+            errors.add(association, :limit, limit: limit, count: check_usage.count)
           end
         end
       end

data/lib/mana-potion/version.rb

@@ -1,3 +1,3 @@
 module ManaPotion
-  VERSION = "0.1.0"
+  VERSION = "0.2.1"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: mana-potion
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.2.1
 platform: ruby
 authors:
 - mrodrigues
@@ -129,6 +129,7 @@ files:
 - bin/console
 - bin/setup
 - lib/mana-potion.rb
+- lib/mana-potion/check_usage.rb
 - lib/mana-potion/pool.rb
 - lib/mana-potion/version.rb
 - mana-potion.gemspec