super_migration 0.0.0.pre1 → 0.0.1

data/README.rdoc

@@ -1,6 +1,38 @@
 = super_migration
 
-Description goes here.
+Migrate legacy database content into a databse with a new structure.
+
+== Example
+
+  require 'super_migration'
+  include SM
+
+  SuperMigration.setup do |config|
+    # same options as in database.yml
+    config.from_database :database => "sm1",
+                         :adapter  => "mysql",
+                         :host     => "localhost",
+                         :username => "root",
+                         :password => ""
+
+    config.to_database   :database => "sm2",
+                         :adapter  => "mysql",
+                         :host     => "localhost",
+                         :username => "root",
+                         :password => ""
+  end
+
+  SuperMigration.migrate do 
+    table :from => :books, :to => :livres do
+
+      field :from => :author, :to => :autheur
+  
+      # apply a transformation to dob field
+      field :from => :title,  :to => :titre do |title|
+        Date.today.to_s + title
+      end
+    end
+  end
 
 == Note on Patches/Pull Requests
  

data/VERSION

@@ -1 +1 @@
-0.0.0.pre1
+0.0.1

data/lib/entities/fields_map.rb

@@ -0,0 +1,18 @@
+module SM
+  # map @from to @to aditionally applying a block @block to @to
+  # e.g.
+  #
+  # for the following fields
+  #
+  #    {:profile_name    => :name, :block => BLOCK},      # => applies the BLOCK on the :name field and put it in profile_name
+  #    {:profile_email   => :email, :block => BLOCK},
+  #    {:profile_website => :website}                 # => put the :profile_website data on the corresponding :website field 
+  class FieldsMap
+    attr_accessor :from, :to, :block
+    def initialize(from, to, block = nil)
+      @from = from
+      @to   = to
+      @block= block
+    end
+  end
+end

data/lib/entities/record.rb

@@ -0,0 +1,9 @@
+module SM
+  class Record
+    attr_accessor :fields_maps
+  
+    def initialize
+      @fields_maps = Array.new 
+    end
+  end
+end

data/lib/example.rb

@@ -0,0 +1,29 @@
+require 'super_migration'
+include SM
+
+SuperMigration.setup do |config|
+  # same options as in database.yml
+  config.from_database :database => "sm1",
+                       :adapter  => "mysql",
+                       :host     => "localhost",
+                       :username => "root",
+                       :password => ""
+  
+  config.to_database   :database => "sm2",
+                       :adapter  => "mysql",
+                       :host     => "localhost",
+                       :username => "root",
+                       :password => ""
+end
+
+SuperMigration.migrate do 
+  table :from => :books, :to => :livres do
+
+    field :from => :author, :to => :autheur
+    
+    # apply a transformation to dob field
+    field :from => :title,  :to => :titre do |title|
+      Date.today.to_s + title
+    end
+  end
+end

data/lib/super_migration.rb

@@ -1,41 +1,131 @@
-#require 'active-record'
+require 'rubygems'
+require 'active_record'
+require 'active_support'
 
-module SM
+Dir["#{File.dirname(__FILE__)}/entities/*.rb"].each { |file| require file }
 
+module SM
   class SuperMigration
-    attr_accessor :from_db_name, :from_host, :from_username, :from_password
-    attr_accessor :to_db_name, :to_host, :to_username, :to_password
+    @@from_db_options = nil
+    @@to_db_options   = nil
+    
+    # I have a feeling this may not be quite good
+    @@current_table_from = nil
+    @@current_table_to   = nil
+    
+    @@current_record = Record.new
+    # ---
     
     def self.setup(&block)
       block.call(self)
     end
     
-    private
+    def self.migrate(&block)
+      yield
+    end
+    
     def self.from_database(options = {})
-      puts "from"
       return if options.size < 4
-      from_db_name = options[:db_name]
-      from_host    = options[:host]
-      from_username = options[:username]
-      from_password = options[:password]
-      
+      @@from_db_options = options.dup
     end
     
     def self.to_database(options = {})
-      puts "to"
       return if options.size < 4
-      to_db_name = options[:db_name]
-      to_host    = options[:host]
-      to_username = options[:username]
-      to_password = options[:password]
+      @@to_db_options = options.dup
+    end
+    
+    def self.from_db_options
+      @@from_db_options
+    end
+    
+    def self.to_db_options
+      @@to_db_options
+    end
+    
+    def self.current_table_from
+      @@current_table_from
+    end
+    
+    def self.current_table_from=(table_from)
+      @@current_table_from = table_from
+    end
+    
+    def self.current_table_to
+      @@current_table_to
+    end
+    
+    def self.current_table_to=(table_to)
+      @@current_table_to = table_to
+    end
+    
+    def self.current_record
+      @@current_record
+    end
+    
+    def self.current_record=(record_hash)
+      @@current_record = record_hash
     end
   end
-
-  def table(options = {})
-    return if options.empty?
+  
+  def define_init_classes(from, to)
+    # create a class with the name table[:from] in the SM module
+    # which inherits from ActiveRecord::Base
+    # call the establish_connection method for this class
+    
+    SuperMigration.current_table_from = from.to_s.singularize.capitalize
+    SM.class_eval("class #{SuperMigration.current_table_from} < ActiveRecord::Base ; establish_connection(SuperMigration.from_db_options) ; end")
+    
+    # create a class with the name table[:from] in the SM module
+    # which inherits from ActiveRecord::Base
+    # call the establish_connection nethod for this class
+    
+    SuperMigration.current_table_to = to.to_s.singularize.capitalize
+    SM.class_eval("class #{SuperMigration.current_table_to} < ActiveRecord::Base ; establish_connection(SuperMigration.to_db_options); end")
   end
   
-  def field(options = {})
-    return if options.empty?
+  def table(table = {}, &block)
+    return unless block_given?
+    
+    define_init_classes(table[:from], table[:to])
+    
+    puts "Migrating data from table #{table[:from]} to table #{table[:to]}"
+    puts "-----------------------------------------------------------------"
+    
+    yield
+    
+    from_table = SuperMigration.current_table_from
+    to_table   = SuperMigration.current_table_to
+    
+    Kernel.const_get(from_table).all.collect do |record| 
+      rc = SuperMigration.current_record.dup
+      new_record_hash = Hash.new
+      
+      rc.fields_maps.each do |field_map|
+        unless record.respond_to?(field_map.from)
+          raise "Undefined attribute name \"#{field_map.from}\" for Model #{from_table}."
+        end
+        
+        if field_map.block
+          new_record_hash[field_map.to] = field_map.block.call(record.send(field_map.from))
+        else
+          new_record_hash[field_map.to] = record.send(field_map.from)
+        end
+      end
+      new_record = Kernel.const_get(to_table).new(new_record_hash)
+      new_record.save!
+    end
+    
+    # reinitialize current_record
+    SuperMigration.current_record = Hash.new
+  end
+
+  def field(options = {}, &block)
+    return if options.size < 2
+    if block_given?
+      raise "Arity for block is not good." if block.arity != 1
+    end
+    
+    fields_map = FieldsMap.new(options[:from], options[:to], block)
+    SuperMigration.current_record.fields_maps << fields_map
   end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification 
 name: super_migration
 version: !ruby/object:Gem::Version 
-  version: 0.0.0.pre1
+  version: 0.0.1
 platform: ruby
 authors: 
 - Cristian Prodan
@@ -9,7 +9,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2010-02-18 00:00:00 +02:00
+date: 2010-02-22 00:00:00 +02:00
 default_executable: 
 dependencies: 
 - !ruby/object:Gem::Dependency 
@@ -38,6 +38,9 @@ files:
 - README.rdoc
 - Rakefile
 - VERSION
+- lib/entities/fields_map.rb
+- lib/entities/record.rb
+- lib/example.rb
 - lib/super_migration.rb
 - test/helper.rb
 - test/test_super_migration.rb
@@ -58,9 +61,9 @@ required_ruby_version: !ruby/object:Gem::Requirement
   version: 
 required_rubygems_version: !ruby/object:Gem::Requirement 
   requirements: 
-  - - ">"
+  - - ">="
     - !ruby/object:Gem::Version 
-      version: 1.3.1
+      version: "0"
   version: 
 requirements: []