lib/autostrip/extension.rb in autostrip-1.0.3 vs lib/autostrip/extension.rb in autostrip-1.0.4

@@ -1,19 +1,37 @@
 # encoding: UTF-8
 # frozen_string_literal: true
 
+require 'active_support/concern'
+
 module Autostrip
   module Extension
-    def autostrip(*attributes)
-      attributes.each do |attribute|
+    extend ActiveSupport::Concern
+
+    def attributes_for_autostrip(attrs)
+      attrs.each_with_object([]) do |attr, memo|
+        case attr
+          when Regexp
+            self.class.columns_hash.values.each { |col| memo << col.name if col.name =~ attr }
+          else
+            memo << attr
+        end
+      end
+    end
+
+    module ClassMethods
+      def autostrip(*attributes)
         # Use prepend to be sure this runs before all other "before_validation" callbacks.
-        before_validation prepend: true do |model|
-          value = model.send(attribute)
-          if Autostrip.performable?(value)
-            # http://www.davidverhasselt.com/set-attributes-in-activerecord/
-            model.send("#{attribute}=", Autostrip.perform(value))
+        before_validation prepend: true do
+          attributes_for_autostrip(attributes).each do |attribute|
+            value = send(attribute)
+            if Autostrip.performable?(value)
+              # http://www.davidverhasselt.com/set-attributes-in-activerecord/
+              send("#{attribute}=", Autostrip.perform(value))
+            end
           end
         end
+        nil
       end
     end
   end
 end