atchu 0.0.1 → 0.0.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: e93c9db74950af711e707f4a011a5a8707fa63e4
-  data.tar.gz: 3b0a15146f78e151b7893b5b486f96fc2fd58e90
+  metadata.gz: d42132be4b71d56a6624717847bdd28d3951e252
+  data.tar.gz: 8845b8d2edbf2382b6c29d764d5a195694e62dbd
 SHA512:
-  metadata.gz: d083818d9262da3de18d9efe858bcd91dcc516e0e7015519b9b39ad46dc699da12c67db1fd7bbae9077b98f9090f7da542fc10edd4986c99d0ea240f69451f59
-  data.tar.gz: 8fc7b3715b5548c6a036aad6c29de16593a96168e66c69a8e34ab88bb106bf6db462d7d9e7e177afc158d6d24f8851c263d0680c84c46dc13d924437d5cbf5fc
+  metadata.gz: 931a67e2bb69818e36a48cd254f3553ab3470c7a1b1ef98ce6181c50ca07965fa5e817d8c4a7bcbd56d9452e1ed1cc1acf159321c5cd2989802478787f88c5d4
+  data.tar.gz: f115e21ee0fa428eaf6364bdc47e33280685f06e875475690cbcdd1117a1f11efffca0e9e8b87b9e6c062574e6585ec566a07dfac2e19d769f54a4c7ac7e4c81

data/README.md

@@ -2,3 +2,7 @@ atchu
 =====
 
 Make Active Records form DB with composite keys
+
+
+##Credits:
+Inspired by [RMRE](https://github.com/bosko/rmre)

data/lib/atchu.rb

@@ -1 +1,21 @@
+require "atchu/connection"
+require "atchu/model_file_maker"
+require "atchu/model"
 
+binding.pry
+
+module Atchu
+  extend self
+
+  def load db_config
+    Connection.connect db_config
+    generate_models
+  end
+
+  private
+  def generate_models
+     tables = Connection.get.tables
+     models = tables.map{|table| Model.for_table(table)}
+     models.each{|model|ModelFileMaker.new(model).write_to "tmp"}
+  end
+end

data/lib/atchu/connection.rb

@@ -0,0 +1,20 @@
+require "yaml"
+
+module Atchu
+  module Connection
+    extend self
+
+    def connect yml_file
+      return if yml_file.empty?
+      if @connection.blank?
+        ActiveRecord::Base.establish_connection(YAML.load_file(yml_file)[:db])
+        @connection = ActiveRecord::Base.connection
+      end
+      @connection
+    end
+
+    def get
+      @connection
+    end
+  end
+end

data/lib/atchu/model.rb

@@ -0,0 +1,28 @@
+require "atchu/queries"
+require "active_support/inflector"
+
+class Model 
+  attr_accessor :table_name
+
+  def self.for_table table_name
+    model = Model.new
+    model.table_name = table_name
+    model
+  end
+
+  def class_name
+    table_name.classify
+  end
+
+  def file_name
+    table_name.classify.underscore
+  end
+
+  def primary_key
+    query = Atchu::Queries.primary_key table_name
+    keys = Atchu::Connection.get.select_all(query).rows
+    return "" if keys.eql? []
+    keys.reduce(:+).map{|x|x.intern}.to_s.gsub("[","").gsub("]","")
+  end
+
+end

data/lib/atchu/model_file_maker.rb

@@ -0,0 +1,30 @@
+require "erubis"
+
+module Atchu 
+  class ModelFileMaker
+
+    def initialize input_model
+      @model = input_model
+    end
+
+    def write_to output_folder
+      file_content = get_file_content
+      write_to_file output_folder,file_content
+    end
+
+    private
+    attr_accessor :model
+    def get_file_content
+      eruby = Erubis::Eruby.new(File.read(File.join(File.expand_path("./../template/", __FILE__), 'model.eruby')))
+      eruby.result({:class_name => model.class_name,:table_name=>model.table_name,:primary_key=>model.primary_key})
+    end
+
+    def write_to_file output_path,file_content
+      FileUtils.mkdir_p(output_path)
+      File.open(File.join(Dir.pwd,output_path, "#{model.file_name}.rb"), "w") { |file| file.write file_content }
+    end
+
+  end
+end
+
+

data/lib/atchu/queries.rb

@@ -0,0 +1,21 @@
+module Atchu 
+  module Queries 
+    extend self
+
+    def primary_key table_name
+      sql = <<-SQL
+SELECT               
+  pg_attribute.attname 
+FROM pg_index, pg_class, pg_attribute 
+WHERE 
+  pg_class.oid = '#{table_name}'::regclass AND
+  indrelid = pg_class.oid AND
+  pg_attribute.attrelid = pg_class.oid AND 
+  pg_attribute.attnum = any(pg_index.indkey)
+  AND indisprimary
+      SQL
+    end
+  end
+end
+
+

data/lib/atchu/template/model.eruby

@@ -0,0 +1,10 @@
+class <%= class_name %> < ActiveRecord::Base
+  <% unless class_name == table_name %>
+    <%= "self.table_name = '#{table_name}'" -%>
+  <% end %>
+
+  <% unless "id" == primary_key || primary_key.nil? %>
+    <%= "self.primary_key = #{primary_key}" -%>
+  <% end %>
+
+end

data/spec/spec_helper.rb

@@ -0,0 +1,2 @@
+require "atchu"
+require "pry"

data/spec/unit/model_file_maker_spec.rb

@@ -0,0 +1,11 @@
+describe Atchu::ModelFileMaker,"ModelFileMaker" do
+  before(:each) do
+    @model = instance_double("Model", :table_name => "table", :class_name => "klass", :file_name => "klass_file", :primary_key => [:id,:name])
+    @model_file = Atchu::ModelFileMaker.new(@model)
+  end
+
+  it "should create file for a model" do
+    @model_file.write_to "tmp"
+    expect(File).to exist("tmp/#{@model.file_name}.rb")
+  end
+end

data/spec/unit/model_spec.rb

@@ -0,0 +1,15 @@
+describe Model,"model with table name, 'car_parts'" do
+
+  before(:each) do
+    @model = Model.for_table("car_parts")
+  end
+
+  it "model class name is CarPart" do
+    expect(@model.class_name).to eq("CarPart")
+  end
+
+  it "model file name is car_part" do
+    expect(@model.file_name).to eq("car_part")
+  end
+end
+

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: atchu
 version: !ruby/object:Gem::Version
-  version: 0.0.1
+  version: 0.0.3
 platform: ruby
 authors:
 - Deepak
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2014-12-23 00:00:00.000000000 Z
+date: 2014-12-27 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rspec
@@ -108,6 +108,20 @@ dependencies:
     - - ">="
       - !ruby/object:Gem::Version
         version: '0'
+- !ruby/object:Gem::Dependency
+  name: erubis
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
 description: Reverse engineer DB with composite keys to Active Records
 email: 
 executables: []
@@ -116,7 +130,14 @@ extra_rdoc_files: []
 files:
 - README.md
 - lib/atchu.rb
+- lib/atchu/connection.rb
+- lib/atchu/model.rb
+- lib/atchu/model_file_maker.rb
+- lib/atchu/queries.rb
+- lib/atchu/template/model.eruby
 - spec/spec_helper.rb
+- spec/unit/model_file_maker_spec.rb
+- spec/unit/model_spec.rb
 homepage: https://github.com/paramadeep/atchu
 licenses:
 - MIT
@@ -142,4 +163,6 @@ signing_key:
 specification_version: 4
 summary: Reverse engineer DB with composite keys to Active Records
 test_files:
+- spec/unit/model_file_maker_spec.rb
+- spec/unit/model_spec.rb
 - spec/spec_helper.rb