builder.rb in pg_migrate-0.1.0

- old
+ new

@@ -6,14 +6,12 @@
 
 module PgMigrate
   # takes a unprocessed manifest directory, and adds before/after headers to each file
   class Builder
 
-
     attr_accessor :manifest_reader, :sql_reader
 
-
     def initialize(manifest_reader, sql_reader)
       @log = Logging.logger[self]
       @manifest_reader = manifest_reader
       @sql_reader = sql_reader
       @template_dir = File.join(File.dirname(__FILE__), 'templates')
@@ -24,21 +22,21 @@
     # output_dir will have a manifest and migrations folder, but processed
     # force will create the output dir if needed, and *delete an existing directory* if it's in the way
     def build(input_dir, output_dir, options={:force=>true})
       input_dir = File.expand_path(input_dir)
       output_dir = File.expand_path(output_dir)
-                                                      
+
       if input_dir == output_dir
         raise 'input_dir can not be same as output_dir: #{input_dir}'
       end
 
       @log.debug "building migration directory #{input_dir} and placing result at: #{output_dir}"
 
       output = Pathname.new(output_dir)
       if !output.exist?
         if !options[:force]
-          raise "Output directory '#{output_dir}' does not exist.  Create it or specify create_output_dir=true"
+          raise "Output directory '#{output_dir}' does not exist.  Create it or specify force=true"
         else
           output.mkpath
         end
       else
         # verify that it's is a directory
@@ -62,24 +60,40 @@
         IO.readlines(input_manifest).each do |input|
           fout.puts input
         end
       end
 
+      # if .pg_migrate file exists, copy it
+      input_pg_config = File.join(input_dir, PG_CONFIG)
+      if FileTest::exist? input_pg_config
+        output_pg_config = File.join(output_dir, PG_CONFIG)
+
+        File.open(output_pg_config, 'w') do |fout|
+          IO.readlines(input_pg_config).each do |input|
+            fout.puts input
+          end
+        end
+      end
+
       # in order array of manifest declarations
       loaded_manifest = @manifest_reader.load_input_manifest(input_dir)
       # hashed on migration name hash of manifest
 
       loaded_manifest_hash = @manifest_reader.hash_loaded_manifest(loaded_manifest)
       @manifest_reader.validate_migration_paths(input_dir, loaded_manifest)
 
-      build_up(input_dir, output_dir, loaded_manifest_hash)
+      build_up(input_dir, output_dir, loaded_manifest_hash, loaded_manifest)
     end
 
-    def build_up(input_dir, output_dir, loaded_manifest_hash)
+    def build_up(input_dir, output_dir, loaded_manifest_hash, loaded_manifest)
       migrations_input = File.join(input_dir, UP_DIRNAME)
       migrations_output = File.join(output_dir, UP_DIRNAME)
 
+      if(!FileTest::exist? migrations_input)
+        raise "'up' directory must exist at #{migrations_input}"
+      end
+
       # iterate through files in input migrations path, wrapping files with transactions and other required bits
 
       Find.find(migrations_input) do |path|
         if path == ".."
           Find.prune
@@ -93,11 +107,11 @@
           migration_in_path = path
 
           # create the filename correct for the output directory, for this file
           migration_out_path = File.join(migrations_output, relative_path)
 
-          process_and_copy_up(migration_in_path, migration_out_path, relative_path, loaded_manifest_hash)
+          process_and_copy_up(migration_in_path, migration_out_path, relative_path, loaded_manifest_hash, loaded_manifest)
         end
       end
 
       create_bootstrap_script(migrations_output)
     end
@@ -105,14 +119,15 @@
     # creates the 'pg_migrations table'
     def create_bootstrap_script(migration_out_path)
       run_template("bootstrap.erb", binding, File.join(migration_out_path, BOOTSTRAP_FILENAME))
     end
 
-    def create_wrapped_up_migration(migration_in_filepath, migration_out_filepath, migration_def)
-    builder_version="pg_migrate_ruby-#{PgMigrate::VERSION}"
+    def create_wrapped_up_migration(migration_in_filepath, migration_out_filepath, migration_def, loaded_manifest)
+      builder_version="pg_migrate_ruby-#{PgMigrate::VERSION}"
+      manifest_version=loaded_manifest[-1].ordinal
       migration_content = nil
-      File.open(migration_in_filepath, 'r') {|reader| migration_content = reader.read }
+      File.open(migration_in_filepath, 'r') { |reader| migration_content = reader.read }
       run_template("up.erb", binding, File.join(migration_out_filepath))
     end
 
     # given an input template and binding, writes to an output file
     def run_template(template, binding, output_filepath)
@@ -127,11 +142,11 @@
       File.open(output_filepath, 'w') do |writer|
         writer.syswrite(content)
       end
     end
 
-    def process_and_copy_up(migration_in_path, migration_out_path, relative_path, loaded_manifest_hash)
+    def process_and_copy_up(migration_in_path, migration_out_path, relative_path, loaded_manifest_hash, loaded_manifest)
 
       if FileTest.directory?(migration_in_path)
         # copy over directories
         # find relative-to-migrations dir this path
         FileUtils.mkdir(migration_out_path)
@@ -145,10 +160,10 @@
 
           @log.debug("retrieving manifest definition for #{manifest_name}")
 
           migration_def = loaded_manifest_hash[manifest_name]
 
-          create_wrapped_up_migration(migration_in_path, migration_out_path, migration_def)
+          create_wrapped_up_migration(migration_in_path, migration_out_path, migration_def, loaded_manifest)
         else
           # if not a .sql file, just copy it over
           FileUtils.cp(migration_in_path, migration_out_path)
         end
       end
\ No newline at end of file