dbd4 1.0.2

data/lib/dbd4/ordered_hash.rb

@@ -0,0 +1,182 @@
+module DBD4
+  class OrderedHash < Hash
+  
+  	attr_accessor :order
+  
+  	class << self
+  
+  		def [] *args
+  			hsh = OrderedHash.new
+  			if Hash === args[0]
+  				hsh.replace args[0]
+  			elsif (args.size % 2) != 0
+  				raise ArgumentError, "odd number of elements for Hash"
+  			else
+  				hsh[args.shift] = args.shift while args.size > 0
+  			end
+  			hsh
+  		end
+  
+  	end
+  
+  	def initialize(*a, &b)
+  		super
+  		@order = []
+  	end
+  
+  	def store_only a,b
+  		store a,b
+  	end
+  
+  	alias orig_store store
+  
+  	def store a,b
+  		@order.push a unless has_key? a
+  		super a,b
+  	end
+  
+  	alias []= store
+  
+  	def == hsh2
+  		return hsh2==self if !hsh2.is_a?(OrderedHash)
+  		return false if @order != hsh2.order
+  		super hsh2
+  	end
+  
+  	def clear
+  		@order = []
+  		super
+  	end
+  
+  	def delete key
+  		@order.delete key
+  		super
+  	end
+  
+  	def each_key
+  		@order.each { |k| yield k }
+  		self
+  	end
+  
+  	def each_value
+  		@order.each { |k| yield self[k] }
+  		self
+  	end
+  
+  	def each
+  		@order.each { |k| yield k,self[k] }
+  		self
+  	end
+  
+  	alias each_pair each
+  
+  	def delete_if
+  		@order.clone.each { |k|
+  			delete k if yield
+  		}
+  		self
+  	end
+  
+  	def values
+  		ary = []
+  		@order.each { |k| ary.push self[k] }
+  		ary
+  	end
+  
+  	def keys
+  		@order
+  	end
+  
+  	def invert
+  		hsh2 = Hash.new
+  		@order.each { |k| hsh2[self[k]] = k }
+  		hsh2
+  	end
+  
+  	def reject &block
+  		self.dup.delete_if( &block )
+  	end
+  
+  	def reject! &block
+  		hsh2 = reject( &block )
+  		self == hsh2 ? nil : hsh2
+  	end
+  
+  	def replace hsh2
+  		@order = hsh2.keys
+  		super hsh2
+  	end
+  
+  	def shift
+  		key = @order.first
+  		key ? [key,delete(key)] : super
+  	end
+  
+  	def unshift k,v
+  		unless self.include? k
+  			@order.unshift k
+  			orig_store(k,v)
+  			true
+  		else
+  			false
+  		end
+  	end
+  
+  	def push k,v
+  		unless self.include? k
+  			@order.push k
+  			orig_store(k,v)
+  			true
+  		else
+  			false
+  		end
+  	end
+  
+  	def pop
+  		key = @order.last
+  		key ? [key,delete(key)] : nil
+  	end
+  
+  	def first
+  		self[@order.first]
+  	end
+  
+  	def last
+  		self[@order.last]
+  	end
+  
+  	def to_a
+  		ary = []
+  		each { |k,v| ary << [k,v] }
+  		ary
+  	end
+  
+  	def to_s
+  		self.to_a.to_s
+  	end
+  
+  	def inspect
+  		ary = []
+  		each {|k,v| ary << k.inspect + "=>" + v.inspect}
+  		'{' + ary.join(", ") + '}'
+  	end
+  
+  	def update hsh2
+  		hsh2.each { |k,v| self[k] = v }
+  		self
+  	end
+  
+  	alias :merge! update
+  
+  	def merge hsh2
+  		self.dup update(hsh2)
+  	end
+  
+  	def select
+  		ary = []
+  		each { |k,v| ary << [k,v] if yield k,v }
+  		ary
+  	end
+  
+  end
+end

data/lib/dbd4/rails_migration_file.rb

@@ -0,0 +1,122 @@
+#--
+# Copyright (c) 2006 Daniel Shane
+#
+# Permission is hereby granted, free of charge, to any person obtaining
+# a copy of this software and associated documentation files (the
+# "Software"), to deal in the Software without restriction, including
+# without limitation the rights to use, copy, modify, merge, publish,
+# distribute, sublicense, and/or sell copies of the Software, and to
+# permit persons to whom the Software is furnished to do so, subject to
+# the following conditions:
+#
+# The above copyright notice and this permission notice shall be
+# included in all copies or substantial portions of the Software.
+#
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+# EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+# MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+# NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+# LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+# OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+# WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
+#--
+
+require 'slurp'
+
+module DBD4
+  class RailsMigrationFile
+    attr_writer :tablename
+    attr_reader :warnings, :tablename
+          
+    def initialize(options = {})
+      @warnings = Array.new
+      @infos = Array.new      
+      @tablename = options[:tablename]
+      
+      if options[:tablename]
+        if Inflector.singularize(@tablename) == @tablename
+          raise RailsModelFileError, "MigrationModelFile : tablename #{@tablename} is not valid, it must be plural, not singular"
+        end
+        partial_name = "_create_" + Inflector.underscore(Inflector.camelize(@tablename)) + ".rb"
+        fileList = Dir.new(File.join('db', 'migrate')).entries.delete_if { |e| e !~ /#{partial_name}/ }
+        if fileList.size == 0
+          raise "Could not find migration file starting with #{partial_name} in dir db/migrate"
+        end
+        if fileList.size > 1
+          raise "Found more than 1 migration file starting with #{partial_name} in dir db/migrate"
+        end
+        @migrationfile = File.join('db', 'migrate', fileList[0])
+      end  
+    end
+    
+    def columnsToString(table)
+      lines = Array.new
+      table.columns.each_value do |c|
+        next if c.primary_key?
+        d = c.datatype
+        case d.type
+          when 'BOOL'
+            lines << "      t.column :#{c.name}, :boolean"         
+          when 'TEXT'
+            lines << "      t.column :#{c.name}, :text"     
+          when 'VARCHAR'
+            if d.params['length']
+              lines << "      t.column :#{c.name}, :string, :length => #{d.params['length']}"
+            else
+              lines << "      t.column :#{c.name}, :string"
+            end
+          when 'FLOAT'
+            lines << "      t.column :#{c.name}, :float"
+          when 'DATETIME'
+            lines << "      t.column :#{c.name}, :datetime"
+          when 'DATE'
+            lines << "      t.column :#{c.name}, :date"
+          when 'INTEGER'
+            if d.params['length']
+              lines << "      t.column :#{c.name}, :integer, :length => #{d.params['length']}"
+            else
+              lines << "      t.column :#{c.name}, :integer"
+            end
+          when 'DECIMAL'
+            lines << "      t.column :#{c.name}, :decimal, :precision => #{d.params['length']}, :scale => #{d.params['decimals']}"
+          when 'BINARY'
+            lines << "      t.column :#{c.name}, :binary"
+          when 'BLOB'
+            lines << "      t.column :#{c.name}, :binary"
+          else
+            raise "Warning : table #{column.table.name} column #{column.name}, #{d.type} is not a valid type"
+        end 
+      end
+      if lines.size == 0
+        "      #t.column :name, :type"
+      else
+        lines.join("\n")
+      end
+    end
+    
+    def update(table)
+      raise "MigrationFile : migration file name not given" if ! @migrationfile 
+  
+      puts "Updating migration file for table #{@tablename} (file : #{@migrationfile})..."
+      newFile = File.open(@migrationfile, "w")
+      c = Inflector.camelize(@tablename)
+      newFile.puts <<"MIGRATIONFILE"
+class Create#{c} < ActiveRecord::Migration
+  def self.up
+    create_table :#{table.name} do |t|
+#{columnsToString(table)}
+    end
+  end
+  
+  def self.down
+    drop_table :#{@tablename}
+  end
+end 
+MIGRATIONFILE
+    end
+      
+    def to_str
+      @lines.join("")
+    end
+  end
+end

data/lib/dbd4/rails_model_file.rb

@@ -0,0 +1,279 @@
+#--
+# Copyright (c) 2006 Daniel Shane
+#
+# Permission is hereby granted, free of charge, to any person obtaining
+# a copy of this software and associated documentation files (the
+# "Software"), to deal in the Software without restriction, including
+# without limitation the rights to use, copy, modify, merge, publish,
+# distribute, sublicense, and/or sell copies of the Software, and to
+# permit persons to whom the Software is furnished to do so, subject to
+# the following conditions:
+#
+# The above copyright notice and this permission notice shall be
+# included in all copies or substantial portions of the Software.
+#
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+# EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+# MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+# NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+# LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+# OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+# WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
+#--
+
+require 'slurp'
+
+module DBD4
+  class RailsModelFile
+    attr_writer :modelfile, :modelname
+    attr_reader :warnings
+    
+    def init_variables
+      @content = nil
+      @lines = Array.new
+      @warnings = Array.new
+      @infos = Array.new
+      @elements = {
+        'has_one' => {},
+        'has_many' => {},
+        'belongs_to' => {},
+        'habtm' => {},
+        'has_and_belongs_to_many' => {},
+        'class' => {},
+        'primary_key' => {},
+        'table_name' => {},
+        'as' => {}
+      }
+    end
+    
+    def setContent(string)
+      init_variables
+      @content = string
+      @lines = string.readlines
+      load
+    end
+      
+    def initialize(options = {})
+      init_variables
+      
+      @modelname = options[:modelname]
+      
+      if options[:modelname]
+        if Inflector.singularize(@modelname) != @modelname
+          raise RailsModelFileError, "ModelFile : modelname #{@modelname} is not valid, it must not be plural"
+        end
+        @modelfile = File.join('app', 'models', Inflector.underscore(Inflector.camelize(@modelname)) + ".rb") 
+        if ! FileTest.exist?(@modelfile)
+          puts "Generating model #{@modelname}..."
+          generate unless $dryrun
+        end
+        load
+      end  
+    end
+    
+    def load
+      raise "ModelFile : modelname not given" if ! @modelname
+      raise "ModelFile : model file name not given" if ! @modelfile 
+  
+      @content = IO.read(@modelfile) if ! @content
+      @lines = IO.readlines(@modelfile) if @lines.empty?
+          
+      raise "ModelFile : Cannot load, no content was given" if ! @content
+      
+      for i in (0..@lines.size-1)
+        l = @lines[i]
+        comment = false
+        
+        if l.match(/^\s*#/)
+          #do nothing
+        elsif l.match(/^\s*class\s+([A-Za-z_0-9]+)/) and @elements['class'].empty?
+          @elements['class'] = { :start => i, :name => $~[1] }
+          if @elements['class'][:name] != Inflector.camelize(@modelname)
+            @warnings << "Warning in file #{@modelfile} at line #{i + 1}:"
+            @warnings << "  Class \"#{@elements['class'][:name]}\" does not match expected name  : \"" + Inflector.camelize(@modelname) + "\""
+            @warnings << "  Model name is \"#{@modelname}\" class name should be \"" + Inflector.camelize(@modelname) + "\""
+          end
+        elsif l.match(/^\s*set_primary_key(?:[ \t\("'])+([a-zA-Z_0-9]+)/)
+          if ! @elements['primary_key'].empty?
+            @warnings << "Warning in file #{@modelfile} at line #{i + 1}:"
+            @warnings << "  Duplicate set_primary_key declaration, it is already defined at line #{@elements['primary_key'][:start] + 1}"
+            @lines[i] = "#" + @lines[i]
+          else
+            @elements['primary_key'] = { :start => i, :name => $~[1] }
+          end
+        elsif l.match(/^\s*set_table_name(?:[ \t\("'])+([a-zA-Z_0-9]+)/)
+          if ! @elements['table_name'].empty?
+            @warnings << "Warning in file #{@modelfile} at line #{i + 1}:"
+            @warnings << "  Duplicate set_table_name declaration, it is already defined at line #{@elements['table_name'][:start] + 1}"
+            @lines[i] = "#" + @lines[i]
+          else
+            @elements['table_name'] = { :start => i, :name => $~[1] }
+          end
+        elsif l.match(/^\s*(has_one|has_many|has_and_belongs_to_many|habtm|belongs_to)\s+:([a-zA-Z_0-9]+)/)
+          r_type = $~[1]
+          t_name = $~[2]
+          
+          if r_type == 'has_and_belongs_to_many'
+            r_type = 'habtm'
+          end        
+          
+          if @elements[r_type][t_name]
+            @warnings << "Warning in file #{@modelfile} at line #{i + 1}:"
+            @warnings << "  Duplicate declaration #{r_type}, it is already defined at line #{@elements[r_type][t_name][:start] + 1}"
+            @lines[i] = "#" + @lines[i]
+            i += 1
+            while i < @lines.size
+              break if @lines[i].match(/^\s*(?:has_one|has_many|has_and_belongs_to_many|habtm|belongs_to|def|end)/)
+              @lines[i] = "#" + @lines[i] if @lines[i] !~ /^\s*#/
+              i += 1
+            end
+          else
+            r = @elements[r_type][t_name] = { :start => i, :end => i }
+            r[:foreign_key] = $~ if l.match(/:foreign_key\s*=>\s*["']([a-zA-Z_0-9]+)/)
+            r[:polymorphic] = $~ if l.match(/:polymorphic\s*=>\s*true/)
+            r[:join_table] = $~ if l.match(/:join_table\s*=>\s*["']([a-zA-Z_0-9]+)/)
+            r[:as] = $~ if l.match(/:as\s*=>\s*([a-zA-Z_0-9]+)/)
+            r[:association_foreign_key] = $~ if l.match(/:association_foreign_key\s*=>\s*["']([a-zA-Z_0-9]+)/)
+  
+            i += 1
+            while i < @lines.size
+              l2 = @lines[i]
+              break if @lines[i].match(/^\s*(?:has_one|has_many|has_and_belongs_to_many|habtm|belongs_to|def|end)/)
+              r[:foreign_key] = $~ if l2.match(/:foreign_key\s*=>\s*["']([a-zA-Z_0-9]+)/)
+              r[:polymorphic] = $~ if l2.match(/:polymorphic\s*=>\s*true/)
+              r[:as] = $~ if l2.match(/:as\s*=>\s*([a-zA-Z_0-9]+)/)
+              r[:join_table] = $~ if l2.match(/:join_table\s*=>\s*["']([a-zA-Z_0-9]+)/)
+              r[:association_foreign_key] = $~ if l2.match(/:association_foreign_key\s*=>\s*["']([a-zA-Z_0-9]+)/)
+              r[:end] = i
+              
+              i += 1
+            end
+          end
+        end
+      end
+      
+      if @elements['class'].empty?
+        @warnings << "Error : Could not find the class declaration in model file #{@modelfile}"
+        raise "Error : Could not find the class declaration in model file #{@modelfile}"
+      end
+    end
+    
+    def generate
+      if ! FileTest.exist?(@modelfile)
+        if @comments =~ /SCAFFOLD/
+          cmdline = ['scaffold', @modelname]
+        else 
+          cmdline = ['model', @modelname]
+        end
+        
+        @infos << "Calling generator: #{cmdline.join(' ')}"
+        Rails::Generator::Scripts::Generate.new.run(['model', @modelname]) if not $dryrun
+      end
+    end 
+  
+    def update(table)
+      raise "ModelFile : modelname not given" if ! @modelname
+      raise "ModelFile : model file name not given" if ! @modelfile 
+      raise "ModelFile : no content given" if @content == nil 
+       
+      table.relation_starts.each_value do |r|
+        dname = ""
+        if r.type == :one2one
+          dname = r.destination_table.modelname
+        else
+          dname = r.destination_table.name
+        end
+        
+        etype = 'has_one' if r.type == :one2one
+        etype = 'has_many' if r.type == :one2many
+        etype = 'habtm' if r.type == :many2many
+        element = @elements[etype][dname]
+          
+        #If the line exists
+        if element
+          if element[:as]
+            if r.destination_column.polymorphic
+              for i in (element[:start] .. element[:end])
+                @lines[i].sub!(/(:as\s*=>\s*["']?)[a-zA-Z_0-9]+/, "\\1#{r.destination_column.polymorphic}")
+              end
+            else
+              for i in (element[:start] .. element[:end])
+                @lines[i].sub!(/,\s*:as\s*=>\s*["']?[a-zA-Z_0-9]+["']?/, "")
+              end
+            end
+          else
+            @lines[element[:start]].sub!(/(#{etype}[^,]+)/, "\\1, :as => '#{r.destination_column.polymorphic}'")
+          end
+          @elements[etype].delete(dname)  
+        else
+          @lines[@elements['class'][:start]] += "  #{etype} :#{dname}"
+          if r.destination_column.polymorphic
+            @lines[@elements['class'][:start]] += ", :as => #{r.destination_column.polymorphic}"
+          end
+          @lines[@elements['class'][:start]] += "\n"
+        end
+      end
+      
+      table.relation_ends.each_value do |r|
+        dname = ""
+        if r.type == :many2many
+          dname = r.source_table.name
+        else
+          if r.destination_column.polymorphic
+            dname = r.destination_column.polymorphic
+          else
+            dname = r.source_table.modelname
+          end
+        end
+        
+        etype = 'belongs_to' if r.type == :one2one or r.type == :one2many
+        etype = 'habtm' if r.type == :many2many
+        element = @elements[etype][dname]
+          
+        #If the line exists
+        if element
+          if element[:polymorphic]
+            if ! r.destination_column.polymorphic
+              for i in (element[:start] .. element[:end])
+                @lines[i].sub!(/,\s*:polymorphic\s*=>\s*["']?[a-zA-Z_0-9]+["']?/, "")
+              end
+            end
+          else
+            if r.destination_column.polymorphic
+              @lines[element[:start]].sub!(/(#{etype}[^,\n]+)/, "\\1, :polymorphic => true")
+            else
+              # do nothing
+            end
+          end
+          @elements[etype].delete(dname)  
+        else
+          @lines[@elements['class'][:start]] += "  #{etype} :#{dname}"
+          if r.destination_column.polymorphic
+            @lines[@elements['class'][:start]] += ", :polymorphic => true"
+          end
+          @lines[@elements['class'][:start]] += "\n"
+        end
+      end
+      
+      @elements.each do |k1, v1|
+        next if k1 == 'class' or v1.empty?
+        v1.each do |k2, v2|
+          @warnings << "Warning in file #{@modelfile} at line #{v2[:start]}"
+          @warnings << "  #{k1} :#{k2} declaration is useless, will comment it"
+      
+          for i in (v2[:start] .. v2[:end])
+            @lines[i] = "#" + @lines[i]
+          end
+        end
+      end
+      
+      puts "Updating model #{@modelname} (file : #{@modelfile})..."
+      newFile = File.open(@modelfile, "w")
+      newFile.puts @lines
+    end
+      
+    def to_str
+      @lines.join("")
+    end
+  end
+end