arpry 0.2.0 → 0.3.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 2db01731aef25421eaef49a72b1a89d4e36476bcb44841cb957ad6cb9082d2a4
-  data.tar.gz: 4c070847760cb4a1b638b5dbcbf7304cb3e195d2cbbf873fd0b115a659407761
+  metadata.gz: cc115cecaf26c426cc0eb12660c865579941de482cd5c04b1b757eb62038ca6e
+  data.tar.gz: e29210c78f1a7d2685f14d1ac14f2ac2029e7b3ea25f298d64fed038eb12ec41
 SHA512:
-  metadata.gz: bcd25b8f5f2d61776052019695d0d17a0602806bca913e1f273980d8ded2ab69f859a4129f8018d2d4b39a746593e3ac7813c99d9461983cc2cbc8df8c993671
-  data.tar.gz: 322477d5332c84d396dcae64c980c72052d5c8efb5108019106a1741b1f54eebf0d534f64c94888aa4b6b1ea58285ce66f010c79f0057e06af19c28c65e6dbad
+  metadata.gz: 19b2a6032392028d8efa6bf37265b6fbc4fed08bdaa65a0b4486d4d7f877f15e016c79e3079aa4520e3865c385214d8ad16293be8080f891cfa13f333d18c388
+  data.tar.gz: 12794da7d0774a2c48fda0df77fd03bd5de3dc9bd68ac86a24ec97dc18948941e1496e2a1894a922e0131cff6e89b713ffb8a0af5e13293ce0700a4d4387c742

data/README.md

@@ -11,20 +11,13 @@ You can get the same experience as Rails Console to explore database with not-Ra
 
 ## Installation
 
-Add this line to your application's Gemfile:
+```bash
+$ gem install arpry
 
-```ruby
-gem 'arpry'
+# Install adapter gem(s) you need
+$ gem install sqlite3 # Or pg, mysql2, ...
 ```
 
-And then execute:
-
-    $ bundle
-
-Or install it yourself as:
-
-    $ gem install arpry
-
 ## Usage
 
 Explore database with `arpry` command.
@@ -35,6 +28,8 @@ $ arpry /path/to/databasefile.sqlite3
 
 # For postgresql
 $ arpry --adapter postgresql --host localhost --user YOUR_USER_NAME --password YOUR_PASSWORD --database YOUR_DB_NAME
+
+# Probably it works with other RDBMS such as MySQL.
 ```
 
 See `arpry --help` for more information of command line options.

data/lib/arpry/cli.rb

@@ -11,7 +11,8 @@ module Arpry
 
       ApplicationRecord.establish_connection(@params)
 
-      generate_classes
+      classes = generate_classes
+      define_foreign_keys(classes)
 
       binding.pry(Namespace)
 
@@ -37,12 +38,14 @@ module Arpry
     end
 
     def generate_classes
-      classes = ApplicationRecord.connection.tables.map do |table|
+      ApplicationRecord.connection.tables.map do |table|
         Namespace.const_set(table.classify, Class.new(ApplicationRecord) do
           self.table_name = table
         end)
       end
+    end
 
+    def define_foreign_keys(classes)
       relations = Array.new(classes.size) do
         Array.new(classes.size)
       end
@@ -56,6 +59,12 @@ module Arpry
 
           relations[idx][ref_klass_idx] = col.name
         end
+
+        klass.connection.foreign_keys(klass.table_name).each do |fk|
+          ref_klass_idx = classes.find_index {|c| c.table_name == fk.to_table }
+          next unless ref_klass_idx
+          relations[idx][ref_klass_idx] = fk.options[:column]
+        end
       end
 
       relations.each.with_index do |row, idx|

data/lib/arpry/version.rb

@@ -1,3 +1,3 @@
 module Arpry
-  VERSION = "0.2.0"
+  VERSION = "0.3.0"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: arpry
 version: !ruby/object:Gem::Version
-  version: 0.2.0
+  version: 0.3.0
 platform: ruby
 authors:
 - Masataka Pocke Kuwabara
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2018-12-31 00:00:00.000000000 Z
+date: 2019-01-04 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activerecord