validates_schema 1.0.0 → 1.1.0

data/lib/validates_schema/validates_schema.rb

@@ -4,7 +4,46 @@ class ActiveRecord::Base
     
     def inherited(klass)
       super(klass)
+      begin
+        add_schema_based_validations(klass)
+        return # it worked, so just return
+      rescue ActiveRecord::StatementInvalid => e
+        # puts "e.message: #{e.message}"
+
+        begin
+          # The ActiveRecord::StatementInvalid could be a result
+          # of an abstract superclass, so let's reset the table name
+          # and try again:
+          klass.reset_table_name
+          add_schema_based_validations(klass)
+          return # it worked, so just return
+        rescue ActiveRecord::StatementInvalid => e
+          # puts "e.message: #{e.message}"
+        end
+      end # rescue ActiveRecord::StatementInvalid
+      
+      # Ok, so it failed, let's add a hook around setting
+      # the table name and try that. Hopefully all should be well then.
+      unless klass.respond_to?(:set_table_name_with_schema_validations)
+        klass.class_eval do
+          class << self
+            def set_table_name_with_schema_validations(*args)
+              set_table_name_without_schema_validations(*args)
+              return if abstract_class
+              add_schema_based_validations(self)
+            end
+            alias_method_chain :set_table_name, :schema_validations
+          end
+        end
+      end
+      
+    end # inherited
+    
+    def add_schema_based_validations(klass = self)
+      klass.reset_column_information
       # puts klass
+      # puts "klass.arel_table: #{klass.arel_table.inspect}"
+      validations = []
       klass.arel_table.columns.each do |c|
         detail = c.column
         next if detail.name == 'id'
@@ -19,15 +58,21 @@ class ActiveRecord::Base
         options[:presence] = true unless detail.null
 
         unless options.empty?
-          # puts "validates :#{detail.name.to_sym}, #{options.inspect}"
+          validations << "validates :#{detail.name.to_sym}, #{options.inspect}"
           klass.class_eval do
             validates detail.name.to_sym, options || {}
           end
         end
-        
       end
-    end
+      # uncomment to see a list of generated validations:
+      # unless validations.empty?
+      #   puts "#{klass}:"
+      #   validations.each do |line|
+      #     puts "  #{line}"
+      #   end
+      # end
+    end # add_schema_based_validations
     
-  end
+  end # class << self
   
-end
+end # ActiveRecord::Base

metadata

@@ -4,9 +4,9 @@ version: !ruby/object:Gem::Version
   prerelease: false
   segments: 
   - 1
+  - 1
   - 0
-  - 0
-  version: 1.0.0
+  version: 1.1.0
 platform: ruby
 authors: 
 - markbates
@@ -60,7 +60,7 @@ required_ruby_version: !ruby/object:Gem::Requirement
   requirements: 
   - - ">="
     - !ruby/object:Gem::Version 
-      hash: 1478591478662090228
+      hash: 372854992706096450
       segments: 
       - 0
       version: "0"