citrusbyte-schemer 0.0.8 → 0.0.9

data/README.markdown

@@ -11,6 +11,9 @@ updated for you without worrying about migrations. Useful for when you want to
 play around with real data handling during prototyping but you really don't 
 care about keeping the data or how it's defined.
 
+This isn't meant to be a replacement for migrations, it is simply a way to get
+started on a project quickly while you iron out your data definitions.
+
 **WARNING:** This will create and delete data definitions on the fly with no
 warning! Only use this with volatile data! Never attach it to an existing model
 you care about as it will start adding and dropping columns without your
@@ -50,6 +53,15 @@ Works just fine, and you can drop it at any time!
 
 Removes the `user_id` column the next time the `Widget` class is loaded.
 
+Have a need for a particular type of column?
+
+    class Widget < ActiveRecord::Base
+      schema :name, { :size => :integer }, :description
+    end
+    
+Will create `size` as an `:integer` column so you can take advantage of Rails'
+casting.
+
 Feeling more confident and ready to make that big leap to migrations? Just run:
 
     rake schemer:migration
@@ -66,13 +78,15 @@ To get:
 
     create_table :widgets do |t|
       t.string :name
+      t.integer :size
+      t.string :description
     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.
+**NOTE:** All columns are created as string columns unless type is given.
 
 Installation
 ------------

data/lib/schemer/migrator.rb

@@ -8,8 +8,8 @@ module Schemer
         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}"
+        (klass.schema_columns.keys - klass.protected_columns).collect do |column| 
+          "  t.#{klass.schema_columns[column]} :#{column}"
         end.join("\n") +
         "\nend"
       end

data/lib/schemer.rb

@@ -5,7 +5,10 @@ module Schemer
     extend ClassMethods
     
     class_inheritable_accessor :schema_columns
-    self.schema_columns = args.collect(&:to_s)
+    self.schema_columns = {}
+    args.collect{ |a| a.is_a?(Hash) ? a.stringify_keys : { a.to_s => :string } }.each do |column|
+      self.schema_columns.merge!(column)
+    end 
 
     update_schema
     update_methods
@@ -17,7 +20,7 @@ module Schemer
     def protected_columns
       %w( id )
     end
-
+    
     # Create the underlying table for this class
     def create_table
       ActiveRecord::Migration.create_table(table_name) do |t|;end;
@@ -33,8 +36,14 @@ module Schemer
     # Update the underlying schema as defined by schema call
     def update_schema
       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) }
+
+      (schema_columns.keys - column_names).each do |column|
+        ActiveRecord::Migration.add_column(table_name, column, schema_columns[column])
+      end
+      
+      (column_names - protected_columns - schema_columns.keys).each do |column|
+        ActiveRecord::Migration.remove_column(table_name, column)
+      end
     end
   end
 end

data/test/schemer_test.rb

@@ -1,6 +1,7 @@
 require 'rubygems'
 require 'contest'
 require 'override'
+require 'ruby-debug'
 require File.join(File.dirname(__FILE__), '..', 'lib', 'schemer')
 require File.join(File.dirname(__FILE__), '..', 'lib', 'schemer', 'migrator')
 
@@ -49,6 +50,27 @@ class FooTest < Test::Unit::TestCase
     end
   end
   
+  context "with types" do
+    setup do
+      Foo.schema :foo, { :bar => :integer }, :baz
+      @foo = Foo.find(Foo.create!(:foo => '5', :bar => 5).id)
+    end
+    
+    should "create foo, bar and baz columns" do
+      assert @foo.respond_to?(:foo)
+      assert @foo.respond_to?(:bar)
+      assert @foo.respond_to?(:baz)
+    end
+    
+    should "create bar column using integer datatype" do
+      assert_equal 5, @foo.bar
+    end
+    
+    should "create foo column using string datatype" do
+      assert_equal '5', @foo.foo
+    end
+  end
+  
   context "building a Rails migration" do
     setup do
       Foo.schema :foo, :bar

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification 
 name: citrusbyte-schemer
 version: !ruby/object:Gem::Version 
-  version: 0.0.8
+  version: 0.0.9
 platform: ruby
 authors: 
 - Ben Alavi
@@ -9,7 +9,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2009-04-03 00:00:00 -07:00
+date: 2009-07-05 00:00:00 -07:00
 default_executable: 
 dependencies: []