Migration generator

burden_web/lib/burden_web.rb

@@ -1,4 +1,5 @@
-require "burden_web/engine"
+require 'burden'
+require 'burden_web/engine'
 
 module BurdenWeb
 end

burden_web/spec/dummy/db/schema.rb

@@ -11,7 +11,7 @@
 #
 # It's strongly recommended to check this file into your version control system.
 
-ActiveRecord::Schema.define(:version => 20130606131532) do
+ActiveRecord::Schema.define(:version => 20130607070108) do
 
   create_table "burden_runs", :force => true do |t|
     t.string   "name"

dummy

@@ -0,0 +1 @@
+burden_web/spec/dummy

lib/burden.rb

@@ -8,6 +8,8 @@ require 'burden/storage'
 require 'burden/version'
 require 'burden/wrapper'
 
+# require 'generators/burden/install_generator'
+
 module Burden
   include Config::Helper
 

lib/generators/active_record/burden_generator.rb

@@ -1,22 +0,0 @@
-require 'rails/generators'
-require 'rails/generators/migration'
-
-class BurdenGenerator < Rails::Generators::Base
-  include Rails::Generators::Migration
-
-  def self.source_root
-    @source_root ||= File.dirname(__FILE__)
-  end
-
-  def self.next_migration_number(dirname)
-    if ActiveRecord::Base.timestamped_migrations
-      Time.new.utc.strftime('%Y%m%d%H%M%S')
-    else
-      '%.3d' % (current_migration_number(dirname) + 1)
-    end
-  end
-
-  def create_migration_file
-    migration_template 'create_burden_runs.rb', 'db/migrate/create_burden_runs.rb'
-  end
-end

lib/rails/generators/burden/install/USAGE

@@ -0,0 +1,8 @@
+Description:
+    Explain the generator
+
+Example:
+    rails generate install Thing
+
+    This will create:
+        what/will/it/create

lib/rails/generators/burden/install/install_generator.rb

@@ -0,0 +1,20 @@
+require 'rails/generators'
+require 'rails/generators/migration'
+
+module Burden
+  module Generators
+    class InstallGenerator < Rails::Generators::Base
+      include Rails::Generators::Migration
+
+      source_root File.expand_path('../templates', __FILE__)
+
+      def self.next_migration_number(path)
+        Time.now.utc.strftime("%Y%m%d%H%M%S")
+      end
+
+      def create_migration
+        migration_template 'create_burden_runs.rb', 'db/migrate/create_burden_runs.rb'
+      end
+    end
+  end
+end

lib/rails/generators/burden/install/templates/create_burden_runs.rb

@@ -1,5 +1,5 @@
 class CreateBurdenRuns < ActiveRecord::Migration
-  def up
+  def self.up
     create_table :burden_runs do |t|
       t.string :name
       t.boolean :success
@@ -8,7 +8,7 @@ class CreateBurdenRuns < ActiveRecord::Migration
     end
   end
 
-  def down
+  def self.down
     drop_table :burden_runs
   end
 end