0
@@ -4,12 +4,14 @@ module AutoMigrations
0
     # Turn off schema_info code for auto-migration
0
     class << ActiveRecord::Schema
0
       alias :old_define :define
0
-      def define(info={}, &block) instance_eval(&block) end
0
+      attr_accessor :version
0
+      def define(info={}, &block) @version=info[:version]; instance_eval(&block) end
0
     end
0
   
0
     load(File.join(RAILS_ROOT, 'db', 'schema.rb'))
0
     ActiveRecord::Migration.drop_unused_tables
0
     ActiveRecord::Migration.drop_unused_indexes
0
+    ActiveRecord::Migration.update_schema_version(ActiveRecord::Schema.version) if ActiveRecord::Schema.version
0
   
0
     class << ActiveRecord::Schema
0
       alias :define :old_define
0
@@ -130,6 +132,10 @@ module AutoMigrations
0
         end
0
       end
0
     end
0
+    
0
+    def update_schema_version(version)
0
+      ActiveRecord::Base.connection.update("UPDATE #{ActiveRecord::Migrator.schema_info_table_name} SET version = #{version}")
0
+    end
0
   
0
   end
0