schema_generator 0.2.0 → 0.9.0

data/schema_generator.rb

@@ -1,4 +1,5 @@
 require 'pathname'
+require 'mysql'
 
 class SchemaGenerator < Rails::Generator::Base
   def initialize(*runtime_args)
@@ -43,7 +44,7 @@ class SchemaGenerator < Rails::Generator::Base
     return nil unless file =~/\.rb$/
 
     begin
-      file =~ %r{^(db/migrate/#{migration_number}_(.*)).rb$}
+      file =~ %r{^(db/migrate/0*#{migration_number}_(.*)).rb$}
       include_name = $1.to_s
       class_base_name = $2.camelize
 
@@ -59,6 +60,7 @@ class SchemaGenerator < Rails::Generator::Base
       m.template 'schema.postgresql.sql',   File.join('db', "schema.postgresql.sql")
       m.template 'schema.mysql.sql',   File.join('db', "schema.mysql.sql")
       m.template 'schema.sqlite.sql',   File.join('db', "schema.sqlite.sql")
+      m.template 'schema.rb',   File.join('db', "schema.rb")
     end
   end
 end
@@ -76,16 +78,26 @@ module ActiveRecord
 
     class << self
       alias_method :old_find, :find
+      alias_method :old_update_all, :update_all
       alias_method :old_create, :create
+      alias_method :old_transaction, :transaction
 
       def dbmigrate_find(*args)
         []
       end
 
+      def dbmigrate_update_all(*args)
+        true
+      end
+
       def dbmigrate_create(attributes = nil)
         DBMigrator::Database.new_data(self,attributes)
       end
       
+      def dbmigrate_transaction(*objects)
+        yield
+      end
+      
     end
   end
   
@@ -106,7 +118,7 @@ module ActiveRecord
 
     def self.add_column(tablename,name,type,options={})
       table=DBMigrator::Database.tables.find{|t| t.name == tablename}
-      if(table.fields.find{|f| f.name == name})
+      if(table.fields.find{|f| f.name.to_s == name.to_s})
         raise "Duplicate field definition for #{tablename}.#{name}"
       end
       table.column(name,type,options)
@@ -116,6 +128,27 @@ module ActiveRecord
       table = DBMigrator::Database.tables.find{|t| t.name == tablename}
       field = table.fields.delete_if {|f| f.name == name}
     end
+    
+    def self.change_column(tablename, name, type, options={})
+      table=DBMigrator::Database.tables.find{|t| t.name == tablename}
+      column = table.fields.find{|f| f.name.to_s == name.to_s}
+      if not column
+        raise "Missing field definition for #{tablename}.#{name} in change_column"
+      else
+        column.type = type
+        column.options = options
+      end
+    end
+    
+    def self.rename_column(tablename, oldname, newname)
+      table = DBMigrator::Database.tables.find{|t| t.name == tablename}
+      column = table.fields.find{|f| f.name.to_s == oldname.to_s}
+      if not column
+        raise "Missing field definition for #{tablename}.#{oldname} in rename_column (found #{table.fields.collect {|f| f.name}.join(', ')})"
+      else
+        column.name = newname
+      end
+    end
 
     def self.add_index(tablename,name,options={})
       if(DBMigrator::Database.indexes.find {|i| i.table == tablename and i.name == name})
@@ -199,6 +232,8 @@ module DBMigrator
       class << ActiveRecord::Base  
         alias find old_find
         alias create old_create
+        alias update_all old_update_all
+        alias transaction old_transaction
       end
 
       ActiveRecord::Base.class_eval 'alias save old_save'
@@ -208,6 +243,8 @@ module DBMigrator
       class << ActiveRecord::Base  
         alias find dbmigrate_find
         alias create dbmigrate_create
+        alias update_all dbmigrate_update_all
+        alias transaction dbmigrate_transaction
       end
       ActiveRecord::Base.class_eval 'alias save dbmigrate_save'
     end
@@ -217,11 +254,11 @@ module DBMigrator
         @@commandstring=''
         case databasename
         when 'postgresql'
-          @@dbdriver = ActiveRecord::ConnectionAdapters::PostgreSQLAdapter.new(nil)
+          @@dbdriver = ActiveRecord::ConnectionAdapters::PostgreSQLAdapter.new(nil,nil)
         when 'mysql'
-          @@dbdriver = ActiveRecord::ConnectionAdapters::MysqlAdapter.new(nil,nil)
+          @@dbdriver = ActiveRecord::ConnectionAdapters::MysqlAdapter.new(nil,nil,nil)
         when 'sqlite'
-          @@dbdriver = ActiveRecord::ConnectionAdapters::SQLiteAdapter.new(nil)
+          @@dbdriver = ActiveRecord::ConnectionAdapters::SQLiteAdapter.new(nil,nil)
         else
           raise 'Unknown driver'
         end

data/templates/schema.rb

@@ -0,0 +1,20 @@
+# This file is autogenerated. Instead of editing this file, please use the
+# migrations feature of ActiveRecord to incrementally modify your database, and
+# then regenerate this schema definition.
+
+ActiveRecord::Schema.define(:version => <%= DBMigrator::Database.version %>) do
+
+<% DBMigrator::Database.tables.sort_by {|table| table.name.to_s }.each do |table| -%>
+  create_table "<%= table.name %>"<% table.options.keys.each do |k| %>, :<%= k %> => <%= table.options[k].inspect %><% end %>, :force => true do |t|
+<% table.fields.each do |field| -%>
+    t.column "<%= field.name %>", :<%= field.type %><% field.options.keys.each do |k| %>, :<%= k %> => <%= field.options[k].inspect %><% end %>
+<% end -%>
+  end
+
+<% end -%>
+
+<% DBMigrator::Database.indexes.sort_by {|index| index.table.to_s}.each do |index| -%>
+  add_index "<%= index.table %>", ["<%= index.name %>"], :name => "<%= index.table %>_<%= index.name %>_index" 
+<% end -%>
+
+end

metadata

@@ -3,8 +3,8 @@ rubygems_version: 0.8.11
 specification_version: 1
 name: schema_generator
 version: !ruby/object:Gem::Version 
-  version: 0.2.0
-date: 2005-10-24 00:00:00 -07:00
+  version: 0.9.0
+date: 2005-12-10 00:00:00 -08:00
 summary: "The rails schema generator generates complete SQL schemas from a collection of
   rails migrations.  It currently produces one schema file each for MySQL,
   PostgreSQL, and SQLite."
@@ -33,6 +33,7 @@ authors:
 files: 
   - templates/schema.mysql.sql
   - templates/schema.postgresql.sql
+  - templates/schema.rb
   - templates/schema.sqlite.sql
   - schema_generator.rb
   - README