acts_as_ordinalized 0.1.2 → 0.1.3

data/VERSION

@@ -1 +1 @@
-0.1.2
+0.1.3

data/acts_as_ordinalized.gemspec

@@ -5,7 +5,7 @@
 
 Gem::Specification.new do |s|
   s.name = %q{acts_as_ordinalized}
-  s.version = "0.1.2"
+  s.version = "0.1.3"
 
   s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
   s.authors = ["mandaryn"]

data/lib/acts_as_ordinalized.rb

@@ -7,18 +7,23 @@ module ActsAsOrdinalized
   # each collection gets new ordinal numbers starting from one
   # paginated results should also get correct ordinal numbers on subsequent pages
   module ClassMethods
-    def acts_as_ordinalized
+    # Configuration options are:
+    #
+    # * +wrapped_methods+ - specifies additional classmethods that should be wrapped with functionality (basic are find and paginate)
+    def acts_as_ordinalized(options = {})
       attr_accessor :ordinal_number
+      wrapped_methods = ([:paginate, :find] << options[:wrapped_methods]).flatten.compact
 
-      #wrap find with function adding ordinal numbers to all returned models
-      if self.respond_to?(:paginate)
-        def self.paginate(*args)
-          ActsAsOrdinalized.ordinalize(super(*args))
+      #wrap various find methods with adding ordinal numbers to all returned models
+      wrapped_methods.each do |method_name|
+        if self.respond_to?(method_name)
+          class_eval <<-EOV
+            def self.#{method_name}(*args)
+              ActsAsOrdinalized.ordinalize(super(*args))
+            end
+          EOV
         end
       end
-      def self.find(*args)
-        ActsAsOrdinalized.ordinalize(super(*args))
-      end
     end
   end
 

metadata

@@ -1,13 +1,13 @@
 --- !ruby/object:Gem::Specification 
 name: acts_as_ordinalized
 version: !ruby/object:Gem::Version 
-  hash: 31
+  hash: 29
   prerelease: false
   segments: 
   - 0
   - 1
-  - 2
-  version: 0.1.2
+  - 3
+  version: 0.1.3
 platform: ruby
 authors: 
 - mandaryn