mokolabs · orangewolf · Nov 29, 2017 · Nov 29, 2017 · Nov 29, 2017 · Nov 29, 2017
diff --git a/README.md b/README.md
@@ -3,19 +3,19 @@ Use Trucker to migrate legacy data into your Rails app.
 
 ## Installation
 
-1. Install the trucker gem
-  ```bash
-  sudo gem install trucker
+1. Add the trucker gem in to your Gemfile
+      ```ruby
+  gem 'trucker'
   ```
 
-2. Add trucker to your `config.gem` block in `environment.rb`.
-  ```ruby  
-  config.gem "trucker"
+2. Bundle 
+  ```bash
+  bundle install
   ```
 
 3. Generate the basic trucker files
   ```bash
-  script/generate truck
+  rails g truck
   ```
 
   This will do the following things:

diff --git a/VERSION.yml b/VERSION.yml
@@ -1,5 +1,5 @@
 --- 
 :patch: 1
 :build: 
-:major: 0
-:minor: 5
+:major: 1
+:minor: 0
diff --git a/generators/truck/templates/legacy_model.erb b/generators/truck/templates/legacy_model.erb
diff --git a/generators/truck/truck_generator.rb b/generators/truck/truck_generator.rb
diff --git a/generators/truck/USAGE → lib/generators/truck/USAGE b/generators/truck/USAGE → lib/generators/truck/USAGE
diff --git a/generators/truck/lib/insert_commands.rb → lib/generators/truck/lib/insert_commands.rb b/generators/truck/lib/insert_commands.rb → lib/generators/truck/lib/insert_commands.rb
@@ -1,12 +1,10 @@
 # Mostly pinched from http://github.com/ryanb/nifty-generators/tree/master
 
-Rails::Generator::Commands::Base.class_eval do
+Rails::Generators::Actions.class_eval do
   def file_contains?(relative_destination, line)
     File.read(destination_path(relative_destination)).include?(line)
   end
-end
 
-Rails::Generator::Commands::Create.class_eval do
   def insert_before(file, line, stop='^(class|module) .+$')
     logger.insert "#{line} into #{file}"
     unless options[:pretend] || file_contains?(file, line)
@@ -37,9 +35,6 @@ def append(file, line)
     end
   end
 
-end
-
-Rails::Generator::Commands::Destroy.class_eval do
   def insert_before(file, line, stop='')
     logger.remove "#{line} from #{file}"
     unless options[:pretend]
@@ -58,9 +53,6 @@ def append(file, line)
     logger.insert "added legacy adapter to end of database.yml"
   end
 
-end
-
-Rails::Generator::Commands::List.class_eval do
   def insert_before(file, line, stop='')
     logger.insert "#{line} into #{file}"
   end

diff --git a/generators/truck/templates/legacy_base.rb → ...generators/truck/templates/legacy_base.rb b/generators/truck/templates/legacy_base.rb → ...generators/truck/templates/legacy_base.rb
diff --git a/lib/generators/truck/templates/legacy_model.erb b/lib/generators/truck/templates/legacy_model.erb
@@ -0,0 +1,11 @@
+class Legacy<%= config[:model_name].classify %> < LegacyBase
+  self.table_name =  "<%= config[:model_name].pluralize %>"
+
+  def map
+    {
+      :field_one => self.old_field_one.squish,
+      :field_two => self.old_field_two.squish
+    }
+  end
+
+end
diff --git a/generators/truck/templates/legacy_task.erb → ...enerators/truck/templates/legacy_task.erb b/generators/truck/templates/legacy_task.erb → ...enerators/truck/templates/legacy_task.erb
@@ -4,14 +4,14 @@ include Trucker
 namespace :db do
   namespace :migrate do
 
-    <%- unless legacy_models.size > 0 -%>
+    <%- unless config[:legacy_models].size > 0 -%>
     # desc 'Migrates products'
     # task :products => :environment do
     #   migrate :products
     # end
     <%- end -%>
 
-    <%- legacy_models.each do |model_name| -%>
+    <%- config[:legacy_models].each do |model_name| -%>
     desc 'Migrates <%= model_name.pluralize.downcase %>'
     task :<%= model_name.pluralize.downcase %> => :environment do
       Trucker.migrate :<%= model_name.pluralize.downcase %>
@@ -20,4 +20,4 @@ namespace :db do
     <%- end -%>
 
   end
-end
+end
diff --git a/lib/generators/truck/truck_generator.rb b/lib/generators/truck/truck_generator.rb
@@ -0,0 +1,33 @@
+require File.expand_path(File.dirname(__FILE__) + "/lib/insert_commands.rb")
+
+class TruckGenerator < Rails::Generators::Base
+  source_root File.expand_path("../templates", __FILE__)
+  def manifest
+    @legacy_models = Dir.glob(Rails.root.join('app/models/*.rb')).collect { |model_path| File.basename(model_path).gsub('.rb', '') }
+    empty_directory 'app/models/legacy'
+    copy_file 'legacy_base.rb', 'app/models/legacy/legacy_base.rb'
+
+    @legacy_models.each do |model_name|
+      template 'legacy_model.erb', "app/models/legacy/legacy_#{model_name.downcase}.rb", { :model_name => model_name }
+    end
+
+    empty_directory 'lib/tasks'
+    template 'legacy_task.erb', 'lib/tasks/legacy.rake', { :legacy_models => @legacy_models }
+
+    snippet = <<EOS
+legacy:
+  adapter: mysql2
+  database: #{Rails.root.to_s.split('/').last}_legacy
+  encoding: utf8
+  username:
+  password:
+EOS
+
+    append_to_file "config/database.yml", snippet
+
+    snippet = ' config.autoload_paths << "#{Rails.root}/app/models/legacy"'
+    insert_after "config/application.rb", snippet, "    # -- all .rb files in that directory are automatically loaded.\n"
+
+  end
+
+end
diff --git a/lib/trucker.rb b/lib/trucker.rb
@@ -20,10 +20,10 @@ def self.migrate(name, options={})
       # Set import counter
       counter = 0
       counter += offset_for_records if offset_for_records
-      total_records = "Legacy#{model}".constantize.find(:all).size
+      total_records = "Legacy#{model}".constantize.count
 
       # Start import
-      "Legacy#{model}".constantize.find(:all, with(options)).each do |record|
+      "Legacy#{model}".constantize.limit(number_of_records).offset(offset_for_records).each do |record|
         counter += 1
         puts status + " (#{counter}/#{total_records})"
         record.migrate
@@ -35,10 +35,6 @@ def self.migrate(name, options={})
 
   protected
 
-    def self.with(options={})
-      {:limit => number_of_records, :offset => offset_for_records}.merge(options)
-    end
-
     def self.number_of_records
       nil || ENV['limit'].to_i if ENV['limit'].to_i > 0
     end