rare_map 0.9.5.beta

data/LICENSE.txt

@@ -0,0 +1,13 @@
+Copyright 2012 Wei-Ming Wu
+
+Licensed under the Apache License, Version 2.0 (the "License"); you
+may not use this file except in compliance with the License. You may
+obtain a copy of the License at
+
+http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
+implied. See the License for the specific language governing
+permissions and limitations under the License.

data/README.rdoc

@@ -0,0 +1,19 @@
+= rare_map
+
+Description goes here.
+
+== Contributing to rare_map
+ 
+* Check out the latest master to make sure the feature hasn't been implemented or the bug hasn't been fixed yet.
+* Check out the issue tracker to make sure someone already hasn't requested it and/or contributed it.
+* Fork the project.
+* Start a feature/bugfix branch.
+* Commit and push until you are happy with your contribution.
+* Make sure to add tests for it. This is important so I don't break it in a future version unintentionally.
+* Please try not to mess with the Rakefile, version, or history. If you want to have your own version, or is otherwise necessary, that is fine, but please isolate to its own commit so I can cherry-pick around it.
+
+== Copyright
+
+Copyright (c) 2012 Wei-Ming Wu. See LICENSE.txt for
+further details.
+

data/lib/rare_map/database_to_schema_builder.rb

@@ -0,0 +1,27 @@
+require 'rare_map/rails_locator'
+require 'rare_map/schema_to_hash_mapping'
+
+module DatabaseToSchemaBuilder
+  
+  def self.build
+    file = File.open(RailsLocator.locate + 'config/database.yml') { |f| f.read }
+    database = file.split("\n").select do |line|
+      line.match(/^[a-z]+/) && line != 'development:' && line != 'test:' && line != 'production:'
+    end
+    database.map! { |db| db.delete(':') }
+    
+    database_hash = Hash.new { |hash, key| hash[key] = [] }
+    database.each { |db| database_hash[db.split('_').first.to_sym] << { :dbname => db } }
+    
+    database_hash.each do |group, dbs|
+      dbs.each do |db|
+        puts "db:schema:dump RAILS_ENV=#{db[:dbname]}"
+        system "rake db:schema:dump RAILS_ENV=#{db[:dbname]}"
+        schema = File.open(RailsLocator.locate + 'db/schema.rb') { |f| f.read }
+        db[:schema] = SchemaToHashMapping.convert(schema.split("\n"))
+      end
+    end
+    database_hash
+  end
+  
+end

data/lib/rare_map/inheritor_util.rb

@@ -0,0 +1,50 @@
+require 'rare_map/rails_locator'
+
+module InheritorUtil
+  
+  def self.to_class_name(name)
+    name.to_s.split('_').map { |word| word.capitalize }.join
+  end
+  
+  def self.remove_underline_between_numbers(line)
+    line = line.to_s
+    md = line.match(/(\d+)(_)(\d+)/)
+    md.size.times { line.gsub!(/(\d+)_(\d+)/, '\1\2') } unless md.nil?
+    line.gsub!(/([a-zA-Z])(_)(\d)/, '\1\3')
+    line.to_sym
+  end
+  
+  def self.write_model_file(file_name, content, group_name)
+    write_file(file_name, content, group_name, 'models')
+  end
+  
+  def self.write_controller_file(file_name, content, group_name)
+    write_file(file_name, content, group_name, 'controllers')
+  end
+  
+  def self.complex_flatten(node, flatAry = [])
+    if node.kind_of?(Array)
+      node.map { |leaf| complex_flatten(leaf, flatAry) }
+    elsif node.kind_of?(Hash)
+      node.each do |key, val|
+        flatAry << key
+        complex_flatten(val, flatAry)
+      end
+    else
+      flatAry << node
+    end
+    flatAry
+  end
+  
+  
+  private
+  
+  
+  def self.write_file(file_name, content, group_name, mvc)
+    Dir.mkdir(RailsLocator.locate + "app/#{mvc}/#{group_name}") unless Dir.exist?(RailsLocator.locate + "app/#{mvc}/#{group_name}")
+    f = File.new(RailsLocator.locate + "app/#{mvc}/#{group_name}/#{group_name}_#{remove_underline_between_numbers(file_name)}.rb", 'w')
+    f.write(content)
+    f.close
+  end
+  
+end

data/lib/rare_map/rails_locator.rb

@@ -0,0 +1,22 @@
+module RailsLocator
+  
+ def self.locate(level = 5)
+    rails_dirs = ['app', 'config', 'db', 'doc', 'lib']
+    
+    level.times do |i|
+      found = true
+      path = ''
+      
+      i.times { path += '../' }
+      
+      rails_dirs.each do |dir|
+        found = false unless Dir.exist?(path + dir)
+      end
+      
+      return path if found
+    end
+    
+    nil
+  end
+  
+end

data/lib/rare_map/schema_to_hash_mapping.rb

@@ -0,0 +1,53 @@
+module SchemaToHashMapping
+  
+  def self.convert(schema)
+    schema_hash = {}
+    table_name_sym = :Blank
+    
+    schema.each do |line|
+      case line.strip!
+      when /^create_table/
+        table_name_sym = filter_table_name_sym(line)
+        schema_hash[table_name_sym] = {}
+        if line.match(/:primary_key\s*=>\s*"[^"]+",/)
+          schema_hash[table_name_sym][:primary_key] = line.match(/:primary_key\s*=>\s*"[^"]+",/).to_s.delete('",').split(/\s+/)[2].to_sym
+        end
+      when /^t\./
+        column_name_sym = filter_column_name_sym(line)
+        column_type_sym = filter_column_type_sym(line)
+        schema_hash[table_name_sym][column_name_sym] ||= {}
+        schema_hash[table_name_sym][column_name_sym][:type] = table_name_sym
+        schema_hash[table_name_sym][column_name_sym][:unique] = false
+      when /^add_index.*:unique\s*=>\s*true/
+        uniqueColumns = filter_unique_columns(line)
+        uniqueColumns.each do |column|
+          schema_hash[table_name_sym][column][:unique] = true
+        end
+      end
+    end
+    
+    return schema_hash
+  end
+  
+  
+  private
+  
+  
+  def self.filter_unique_columns(line)
+    cols = line.match(/\[[^\]]*\]/).to_s.delete('"[],').split(/\s+/).map { |col| col.to_sym }
+    cols.size == 1 ? cols : []
+  end
+  
+  def self.filter_table_name_sym(line)
+    line.delete('",').split(/\s+/)[1].to_sym
+  end
+  
+  def self.filter_column_name_sym(line)
+    line.delete('",').split(/\s+/)[1].to_sym
+  end
+  
+  def self.filter_column_type_sym(line)
+    line.delete('",').split(/\s+/)[0][2 .. -1].to_sym
+  end
+  
+end

data/lib/rare_map/schema_to_model_builder.rb

@@ -0,0 +1,74 @@
+require 'rare_map/inheritor_util'
+
+module SchemaToModelBuilder
+  
+  def self.build(schema_hash, rel_hash, group_name, database_hash)
+    schema_hash.each do |table_name, table_content|
+      model_content = generate_header(table_name, schema_hash, group_name, database_hash)
+      model_content << generate_attributes_permission(table_content, 'attr_accessible') unless table_content.size == 0
+      model_content << generate_relationships(table_name, rel_hash, group_name)
+      model_content << generate_footer
+      InheritorUtil.write_model_file(table_name, model_content, group_name)
+    end
+  end
+  
+  
+  private
+  
+  
+  def self.generate_relationships(table_name, rel_hash, group_name)
+    return '' if rel_hash.count { |_, hash| hash[table_name] != nil } == 0
+    
+    has_one_ary = []
+    rel_hash[:has_one][table_name].each do |table, foreign_key|
+      has_one_ary << "  has_one :#{table}, :foreign_key => '#{foreign_key}', :class_name => '#{group_name.to_s.capitalize}#{InheritorUtil.to_class_name(table)}'"
+    end if rel_hash[:has_one][table_name] != nil
+    
+    has_many_ary = []
+    rel_hash[:has_many][table_name].each do |table, foreign_key|
+      has_many_ary << "  has_many :#{table}, :foreign_key => '#{foreign_key}', :class_name => '#{group_name.to_s.capitalize}#{InheritorUtil.to_class_name(table)}'"
+    end if rel_hash[:has_many][table_name] != nil
+    
+    belongs_to_ary = []
+    rel_hash[:belongs_to][table_name].each do |table, foreign_key|
+      belongs_to_ary << "  belongs_to :#{table}, :foreign_key => '#{foreign_key}', :class_name => '#{group_name.to_s.capitalize}#{InheritorUtil.to_class_name(table)}'"
+    end if rel_hash[:belongs_to][table_name] != nil
+    
+    has_many_through_ary = []
+    rel_hash[:has_many_through][table_name].each do |table, foreign_key, through|
+      has_many_through_ary << "  has_many :#{table}, :foreign_key => '#{foreign_key}', :class_name => '#{group_name.to_s.capitalize}#{InheritorUtil.to_class_name(table)}', :through => :#{through}"
+    end if  rel_hash[:has_many_through][table_name] != nil
+    
+    has_one_ary.join("\n") << "\n" <<
+    has_many_ary.join("\n") << "\n" <<
+    belongs_to_ary.join("\n") + "\n" <<
+    has_many_through_ary.join("\n") << "\n"
+  end
+  
+  def self.generate_attributes_permission(table_content, permission)
+    table_content.keys.inject('  ' + permission) { |o, i| o = o + " :#{i}," }[0 .. -2] + "\n"
+  end
+  
+  def self.generate_header(table_name, schema_hash, group_name, database_hash)
+    "class #{group_name.to_s.capitalize}#{InheritorUtil.to_class_name(table_name)} < ActiveRecord::Base\n" <<
+    "  establish_connection :#{get_connection_name(table_name, group_name, database_hash)}\n" <<
+    "  self.table_name = '#{table_name}'\n" <<
+    "  self.inheritance_column = 'ruby_type'\n" <<
+    "  self.primary_key = '#{check_primary_key(table_name, schema_hash)}'\n"
+  end
+  
+  def self.get_connection_name(table_name, group_name, database_hash)
+    database_hash[group_name].each do |db|
+      return db[:dbname] unless db[:schema][table_name].nil?
+    end
+  end
+  
+  def self.check_primary_key(table_name, schema_hash)
+    schema_hash[table_name][:primary_key].nil? ? 'id' : schema_hash[table_name][:primary_key]
+  end
+  
+  def self.generate_footer
+    "end\n"
+  end
+  
+end

data/lib/rare_map/schema_to_relation_builder.rb

@@ -0,0 +1,104 @@
+module BelongsTo2HasManyThroughtRelationBuilder
+  
+  def self.build(belongs_to_hash, has_one_hash, has_many_hash)
+    belongs_to_hash = belongs_to_hash.clone
+    has_many_through_hash = {}
+  
+    belongs_to_hash = belongs_to_hash.delete_if { |table, rels| rels.size < 2 }
+    belongs_to_hash.each do |table, rels|
+      rels.each do |from_table, from_table_key|
+        rels.each do |to_table, to_table_key|
+          if from_table != to_table
+            has_many_through_hash[from_table] ||= []
+            rel = [to_table, to_table_key, table]
+            has_many_through_hash[from_table] << rel 
+          end
+        end
+      end
+    end
+    
+    final_has_many_through_hash = {}
+    has_many_through_hash.each do |table, rels|
+      rels.each do |rel|
+        if has_many_hash[table] != nil && has_many_hash[table].count { |i| i.first == rel.first } == 0
+          final_has_many_through_hash[table] ||= []
+          final_has_many_through_hash[table] << rel
+        end
+      end
+    end
+    has_many_through_hash = final_has_many_through_hash
+    
+    final_has_many_through_hash = {}
+    has_many_through_hash.each do |table, rels|
+      rels.each do |rel|
+        if has_one_hash[table] != nil && has_one_hash[table].count { |i| i.first == rel.first } == 0
+          final_has_many_through_hash[table] ||= []
+          final_has_many_through_hash[table] << rel
+        end
+      end
+    end
+    has_many_through_hash = final_has_many_through_hash
+    
+    final_has_many_through_hash = {}
+    has_many_through_hash.each do |table, rels|
+      rels.each do |rel|
+        if belongs_to_hash[table] != nil && belongs_to_hash[table].count { |i| i.first == rel.first } == 0
+          final_has_many_through_hash[table] ||= []
+          final_has_many_through_hash[table] << rel
+        end
+      end
+    end
+    has_many_through_hash = final_has_many_through_hash
+    
+    has_many_through_hash
+  end
+  
+end
+
+module SchemaToRelationBuilder
+  
+  def self.build(schema_hash, relation_mark, alias_hash = {})
+    relation_mark ||= '_id'
+    has_one_hash = {}; has_many_hash = {}; belongs_to_hash = {}
+    
+    schema_hash.each do |table_name, table_content|
+      table_content.each do |column, type|
+        if alias_hash[column] != nil
+          belongs_to = alias_hash[column]
+          if schema_hash[belongs_to] != nil
+            if schema_hash[table_name][column][:unique] == true
+              has_one_hash[belongs_to] ||= []
+              has_one_hash[belongs_to] << [table_name, column]
+            else
+              has_many_hash[belongs_to] ||= []
+              has_many_hash[belongs_to] << [table_name, column]
+            end
+            belongs_to_hash[table_name] ||= []
+            belongs_to_hash[table_name] << [belongs_to, column]
+          end
+        elsif column.to_s.match(eval("/#{relation_mark}$/")) && column.to_s[0 .. -(relation_mark.length + 1)].to_sym != table_name
+          belongs_to = column.to_s[0 .. -(relation_mark.length + 1)].to_sym
+          if schema_hash[belongs_to] != nil
+            if schema_hash[table_name][column][:unique] == true
+              has_one_hash[belongs_to] ||= []
+              has_one_hash[belongs_to] << [table_name, column]
+            else
+              has_many_hash[belongs_to] ||= []
+              has_many_hash[belongs_to] << [table_name, column]
+            end
+            belongs_to_hash[table_name] ||= []
+            belongs_to_hash[table_name] << [belongs_to, column]
+          end
+        end
+      end
+    end
+    
+    has_many_through_hash = BelongsTo2HasManyThroughtRelationBuilder.build(belongs_to_hash, has_one_hash, has_many_hash)
+    
+    return { :has_one => has_one_hash,
+             :has_many => has_many_hash,
+             :belongs_to => belongs_to_hash,
+             :has_many_through => has_many_through_hash }
+  end
+  
+end

data/lib/rare_map/version.rb

@@ -0,0 +1,10 @@
+class RareMap
+  module Version
+    MAJOR = 0
+    MINOR = 9
+    PATCH = 5
+    BUILD = 'beta'
+
+    STRING = [MAJOR, MINOR, PATCH, BUILD].compact.join('.')
+  end
+end

data/lib/rare_map.rb

@@ -0,0 +1,41 @@
+require 'rare_map/database_to_schema_builder'
+require 'rare_map/schema_to_relation_builder'
+require 'rare_map/schema_to_model_builder'
+require 'rare_map/rails_locator'
+
+class RareMap
+  
+  def self.mapping(opt = {})
+    unless RailsLocator.locate
+      $stderr.puts 'Run RareMap under Rails root directory'
+      return
+    end
+    
+    info = {}
+    
+    databases = DatabaseToSchemaBuilder.build
+    databases.each do |group_name, tables|  
+      foreign_keys = opt[:foreign_key][group_name]
+      aliases = opt[:alias][group_name]
+      primary_keys = opt[:primary_key][group_name]
+      
+      schemata = tables.inject({}) { |o, i| o = o.merge(i[:schema]) }
+      primary_keys.each do |table, primary_key|
+        schemata[table][:primary_key] = primary_key if schemata[table]
+      end
+      relations = SchemaToRelationBuilder.build(schemata, foreign_keys, aliases)
+      SchemaToModelBuilder.build(schemata, relations, group_name, databases)
+      
+      info[group_name] ||= {}
+      info[group_name][:schemata] = schemata
+      info[group_name][:relations] = relations
+    end
+    
+    $stdout.puts %{Add following line to your config/application.rb}
+    $stdout.puts %{config.autoload_paths += Dir[Rails.root.join('app', 'models', '{**}')]}
+    
+    info[:databases] = databases
+    info
+  end
+  
+end

metadata

@@ -0,0 +1,156 @@
+--- !ruby/object:Gem::Specification
+name: rare_map
+version: !ruby/object:Gem::Version
+  version: 0.9.5.beta
+  prerelease: 6
+platform: ruby
+authors:
+- Wei-Ming Wu
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2012-10-07 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: rails
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: shoulda
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: rdoc
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '3.12'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '3.12'
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: 1.0.0
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: 1.0.0
+- !ruby/object:Gem::Dependency
+  name: jeweler
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 1.8.4
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 1.8.4
+- !ruby/object:Gem::Dependency
+  name: rcov
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 0.9.11
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 0.9.11
+description: Translate legacy db to ActiveRecord models
+email: wnameless@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files:
+- LICENSE.txt
+- README.rdoc
+files:
+- lib/rare_map.rb
+- lib/rare_map/database_to_schema_builder.rb
+- lib/rare_map/inheritor_util.rb
+- lib/rare_map/rails_locator.rb
+- lib/rare_map/schema_to_hash_mapping.rb
+- lib/rare_map/schema_to_model_builder.rb
+- lib/rare_map/schema_to_relation_builder.rb
+- lib/rare_map/version.rb
+- LICENSE.txt
+- README.rdoc
+homepage: http://github.com/wnameless/rare_map
+licenses:
+- Apache License, Version 2.0
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+      segments:
+      - 0
+      hash: -2253654636828043832
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>'
+    - !ruby/object:Gem::Version
+      version: 1.3.1
+requirements: []
+rubyforge_project: 
+rubygems_version: 1.8.24
+signing_key: 
+specification_version: 3
+summary: RelationDB-ActiveRecord Mapper
+test_files: []