Sutto-reversible_data 0.0.3 → 0.0.4

data/lib/reversible_data/table.rb

@@ -6,17 +6,19 @@ module ReversibleData
     
     attr_accessor :table_name, :model_name, :options
     
-    def initialize(table_name, model_name = nil, opts = {}, &blk)
+    def initialize(table_name, opts = {}, &blk)
       @table_name       = table_name.to_s.tableize.to_sym
-      @model_name       = (model_name || @table_name).to_s.classify
+      @model_name       = (opts[:class_name] || @table_name).to_s.classify
       @migrator         = blk
       @model_definition = nil
-      @options          = opts
+      default_options   = {:skip_model => Object.const_defined?(@model_name), :skip_table => connection.table_exists?(@table_name)}
+      @options          = opts.reverse_merge(default_options)
+      @blueprint        = nil
       self.known_models[table_name.to_sym] = self
     end
     
     def drop_table
-      return if @options[:skip_model]
+      return if @options[:skip_table]
       connection.drop_table(@table_name) if connection.table_exists?(@table_name)
     end
     
@@ -40,6 +42,7 @@ module ReversibleData
       return if Object.const_defined?(@model_name)
       @model = Class.new(ActiveRecord::Base)
       @model.class_eval(&@model_definition) unless @model_definition.nil?
+      @model.blueprint(&@blueprint) unless @blueprint.nil?
       Object.const_set(@model_name, @model)
     end
     
@@ -47,6 +50,10 @@ module ReversibleData
       @model_definition = blk unless blk.nil?
     end
     
+    def blueprint(&blk)
+      @blueprint = blk unless blk.nil?
+    end
+    
     def clear_model_definition!
       @model_definition = nil
     end

data/lib/reversible_data.rb

@@ -9,8 +9,12 @@ module ReversibleData
   autoload :TableManager,  'reversible_data/table_manager'
   autoload :ShouldaMacros, 'reversible_data/shoulda_macros'
   
-  def self.add(name, model_name = nil, opts = {}, &blk)
-    Table.new(name, model_name, opts, &blk)
+  def self.add(name, opts = {}, &blk)
+    Table.new(name, opts, &blk)
+  end
+  
+  def self.[](name)
+    Table.known_models[name]
   end
   
   def self.manager_for(*tables)

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification 
 name: Sutto-reversible_data
 version: !ruby/object:Gem::Version 
-  version: 0.0.3
+  version: 0.0.4
 platform: ruby
 authors: 
 - Darcy Laycock