consistency_fail 0.1.0 → 0.1.1

data/bin/consistency_fail

@@ -1,5 +1,16 @@
 #!/usr/bin/env ruby
 
+begin
+  require File.join(Dir.pwd, "config", "boot")
+rescue LoadError => e
+  puts "\nUh-oh! You must be in the root directory of a Rails project.\n"
+  raise
+end
+
+require 'active_record'
+require File.join(Dir.pwd, "config", "environment")
+
+$:<< File.join(File.dirname(__FILE__), "..", "lib")
 require "consistency_fail"
 
 def problems(models, introspector)

data/lib/consistency_fail.rb

@@ -1,14 +1,3 @@
-begin
-  require File.join(Dir.pwd, "config", "boot")
-rescue LoadError => e
-  puts "\nUh-oh! You must be in the root directory of a Rails project.\n"
-  raise
-end
-
-require 'active_record'
-require 'validation_reflection'
-require File.join(Dir.pwd, "config", "environment")
-
 require 'consistency_fail/models'
 require 'consistency_fail/introspectors/table_data'
 require 'consistency_fail/introspectors/validates_uniqueness_of'

data/lib/consistency_fail/enforcer.rb

@@ -0,0 +1,56 @@
+require 'consistency_fail'
+
+module ConsistencyFail
+  class Enforcer
+    def problems(models, introspector)
+      problems = models.map do |m|
+        [m, introspector.missing_indexes(m)]
+      end.reject do |m, indexes|
+        indexes.empty?
+      end
+    end
+
+    def self.enforce!
+      models = ConsistencyFail::Models.new($LOAD_PATH)
+      models.preload_all
+
+      introspectors = [ConsistencyFail::Introspectors::ValidatesUniquenessOf.new,
+                       ConsistencyFail::Introspectors::HasOne.new]
+
+      problem_models_exist = models.all.detect do |model|
+        introspectors.any? {|i| !i.missing_indexes(model).empty?}
+      end
+
+      if problem_models_exist
+        mega_fail!
+      end
+    end
+
+    def self.mega_fail!
+      ActiveRecord::Base.class_eval do
+        class << self
+          def panic
+            raise "You've got missing indexes! Run `consistency_fail` to find and fix them."
+          end
+
+          def find(*arguments)
+            panic
+          end
+
+          alias :first :find
+          alias :last :find
+          alias :count :find
+        end
+
+        def save
+          self.class.panic
+        end
+
+        def save!
+          self.class.panic
+        end
+      end
+    end
+
+  end
+end

data/lib/consistency_fail/introspectors/validates_uniqueness_of.rb

@@ -1,3 +1,4 @@
+require 'validation_reflection'
 require 'consistency_fail/index'
 
 module ConsistencyFail

data/lib/consistency_fail/version.rb

@@ -1,3 +1,3 @@
 module ConsistencyFail
-  VERSION = "0.1.0"
+  VERSION = "0.1.1"
 end

metadata

@@ -1,13 +1,13 @@
 --- !ruby/object:Gem::Specification 
 name: consistency_fail
 version: !ruby/object:Gem::Version 
-  hash: 27
+  hash: 25
   prerelease: 
   segments: 
   - 0
   - 1
-  - 0
-  version: 0.1.0
+  - 1
+  version: 0.1.1
 platform: ruby
 authors: 
 - Colin Jones
@@ -15,7 +15,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2011-06-10 00:00:00 -05:00
+date: 2011-06-17 00:00:00 -05:00
 default_executable: 
 dependencies: 
 - !ruby/object:Gem::Dependency 
@@ -89,6 +89,7 @@ files:
 - bin/consistency_fail
 - consistency_fail.gemspec
 - lib/consistency_fail.rb
+- lib/consistency_fail/enforcer.rb
 - lib/consistency_fail/index.rb
 - lib/consistency_fail/introspectors/has_one.rb
 - lib/consistency_fail/introspectors/table_data.rb