Merge pull request #23 from wizarddevelopment/rake-task

A rake task that checks for missing keys and exits with an error code.

lib/immigrant/railtie.rb

@@ -6,8 +6,13 @@ class Railtie < Rails::Railtie
         Immigrant.load
       end
     end
+
     generators do
-      require "generators/immigration_generator"
+      require 'generators/immigration_generator'
+    end
+
+    rake_tasks do
+      require 'immigrant/task'
     end
   end
-end
+end

lib/immigrant/task.rb

@@ -0,0 +1,20 @@
+namespace :immigrant do
+  desc 'Checks for missing foreign key relationships in the database'
+  task check_keys: :environment do
+    Rails.application.eager_load!
+
+    keys, warnings = Immigrant::KeyFinder.new.infer_keys
+    warnings.values.each { |warning| $stderr.puts "WARNING: #{warning}" }
+
+    keys.each do |key|
+      column = key.options[:column]
+      pk = key.options[:primary_key]
+      $stderr.puts "Missing foreign key relationship on '#{key.from_table}.#{column}' to '#{key.to_table}.#{pk}'"
+    end
+
+    if keys.any?
+      puts 'Found missing foreign keys, run `rails generate immigration MigrationName` to create a migration to add them.'
+      exit keys.count
+    end
+  end
+end