citrusbyte-schemer 0.0.7 → 0.0.8

data/README.markdown

@@ -50,6 +50,28 @@ Works just fine, and you can drop it at any time!
 
 Removes the `user_id` column the next time the `Widget` class is loaded.
 
+Feeling more confident and ready to make that big leap to migrations? Just run:
+
+    rake schemer:migration
+    
+To get:
+
+    Migration from schema declarations in User, Widget
+
+    create_table :users do |t|
+      t.string :username
+      t.string :password
+      t.string :email
+    end
+
+    create_table :widgets do |t|
+      t.string :name
+    end
+
+Then you can paste the output into a migration file, setup your columns (don't
+forget your indexes!) and get rid of your `schema` calls in the listed classes
+and you're on your way to the big leagues!
+
 **NOTE:** All columns are created as string columns.
 
 Installation

data/lib/schemer/migrator.rb

@@ -0,0 +1,18 @@
+module Schemer
+  class Migrator
+    class << self
+      # Outputs the Rails migration for the schema defined in the given class.
+      # 
+      # Outputs an empty string if no schema is defined on the class.
+      def migration(klass)
+        return nil unless klass.respond_to?(:schema_columns)
+        
+        "create_table :#{klass.table_name} do |t|\n" +
+        (klass.schema_columns - klass.protected_columns).collect do |column| 
+          "  t.string :#{column}"
+        end.join("\n") +
+        "\nend"
+      end
+    end
+  end
+end

data/lib/schemer.rb

@@ -35,7 +35,7 @@ module Schemer
       create_table unless table_exists?
       (schema_columns - column_names).each { |column| ActiveRecord::Migration.add_column(table_name, column, :string) }
       (column_names - protected_columns - schema_columns).each { |column| ActiveRecord::Migration.remove_column(table_name, column) }
-    end    
+    end
   end
 end
 

data/tasks/schemer.rake

@@ -0,0 +1,22 @@
+namespace :schemer do
+  desc "Outputs the basis of a Rails migration from your schema declarations"
+  task :migration => :environment do
+    klasses = []
+
+    Dir.glob(File.join(Rails.root, "app", "models", "**.rb")).each do |file|
+      lineno  = 0
+      result = File.readlines(file).each do |line|
+        lineno += 1
+        next unless line =~ /^class (.*) \<.*$/
+        klasses << $1.constantize
+      end
+    end
+    
+    migration = klasses.collect do |klass|
+      Schemer::Migrator.migration(klass)
+    end.join("\n\n")
+    
+    puts "\nMigration from schema declarations in #{klasses.join(', ')}"
+    puts "\n#{migration}\n\n"
+  end
+end

data/test/schemer_test.rb

@@ -2,12 +2,15 @@ require 'rubygems'
 require 'contest'
 require 'override'
 require File.join(File.dirname(__FILE__), '..', 'lib', 'schemer')
+require File.join(File.dirname(__FILE__), '..', 'lib', 'schemer', 'migrator')
 
 ActiveRecord::Base.establish_connection :adapter => 'sqlite3', :database => 'test/schemer_test.db'
 ActiveRecord::Base.logger = Logger.new(File.join(File.dirname(__FILE__), "debug.log"))
 
 class Foo < ActiveRecord::Base;end;
 ActiveRecord::Migration.drop_table(Foo.table_name) if Foo.table_exists?
+class Bar < ActiveRecord::Base;end;
+ActiveRecord::Migration.drop_table(Bar.table_name) if Bar.table_exists?
 
 class FooTest < Test::Unit::TestCase  
   context "defining the schema" do
@@ -31,7 +34,7 @@ class FooTest < Test::Unit::TestCase
       should "create the bar column" do
         assert @foo.respond_to?(:bar)
       end
-    end    
+    end
   end
   
   context "updating the schema" do
@@ -45,4 +48,19 @@ class FooTest < Test::Unit::TestCase
       assert !@foo.respond_to?(:bar)
     end
   end
+  
+  context "building a Rails migration" do
+    setup do
+      Foo.schema :foo, :bar
+    end
+    
+    should "output the migration for Foo" do
+      assert_equal(
+%Q{create_table :foos do |t|
+  t.string :foo
+  t.string :bar
+end}, Schemer::Migrator.migration(Foo)
+      )
+    end
+  end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification 
 name: citrusbyte-schemer
 version: !ruby/object:Gem::Version 
-  version: 0.0.7
+  version: 0.0.8
 platform: ruby
 authors: 
 - Ben Alavi
@@ -25,9 +25,11 @@ files:
 - README.markdown
 - LICENSE
 - Rakefile
+- lib/schemer/migrator.rb
 - lib/schemer.rb
 - rails/init.rb
 - test/schemer_test.rb
+- tasks/schemer.rake
 has_rdoc: false
 homepage: http://github.com/citrusbyte/schemer
 post_install_message: