KevinColemanInc/yeet_dba

namespace :yeet_dba do
  desc 'Add foreign keys in a rake migration'
  task add_foreign_keys: :environment do
    foreign_keys = YeetDba::MissingForeignKeys.foreign_keys

    puts "Trying to add #{foreign_keys.length}"
    puts
    foreign_keys.each do |foreign_key|
      
      begin
        ActiveRecord::Migration.add_foreign_key(foreign_key.table_a,
                                                foreign_key.table_b,
                                                column: foreign_key.column)

      rescue ActiveRecord::InvalidForeignKey
        puts "ERROR - #{foreign_key.table_a} . #{foreign_key.column} failed to add key"
      end
    end
  end
end