set_schema_migrations.rb

require 'optparse'

ENV['RAILS_ENV'] = ENV['RAILS_ENV'] || 'development'
require File.expand_path(File.dirname(__FILE__) + "/../config/environment")


opts = {}
ARGV.options {|opt|
  opt.on('-s', '--save', 'save') {|v| opts[:save] = v }
  opt.parse!
}

Dir::entries("#{RAILS_ROOT}/db/migrate/").sort.each do |v|
  next if v == "." or v == ".." or v == ".svn"
  version, name = v.split('_', 2)
  begin
    if opts[:save]
      ActiveRecord::Base.connection.execute("INSERT INTO schema_migrations values ('#{version}')")
    end
  rescue => e
    p e
    exit
  end
end