who_am_i 0.0.3 → 0.0.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: cd78ba2e07002c41c67d3d85128e49b1bcd40b1b
-  data.tar.gz: 4998e54c08a6a47deed886df900ca7f6d84b3a36
+  metadata.gz: 288863730bbd33ec0a782b65eb247bf38d8c7d38
+  data.tar.gz: 27ac823f61dfecff9099f5da67aa11449d521ca0
 SHA512:
-  metadata.gz: df44e63b4e02bf2196a61304eb0a4e93dce311df8ce4e5784582cfe17309a10bdbf0e1a07995b29edd94c90a6af5175e4a28cca97716a88f2d23f83780857a66
-  data.tar.gz: 1dee087a0dfb42f29f514e9d9c10c407ce13439b7396bff1a8f9d828d0112b7c38e06c825663b0912c9139fc8bbda4825fa27d170298c041b72df6ce6609d7e6
+  metadata.gz: dc4b1779e3649746e7d4de71bbfa5235166bdfea8c1af754fd6c446f6546d75dbffa80d1ec8f979bb61ec15b10dcedfe042dcf2a5ace1942cdb346a6fef491db
+  data.tar.gz: 5699729dd05db062fa99a7b4bc40dbab2cd8215dd9e9e057e3ea6ced5d52a209e6ab85c602a14dec5a690b198f539b011c394d8523772a980da37ae7e8052f90

data/README.md

@@ -23,7 +23,7 @@ Add this line to your application's Gemfile:
 
 ```ruby
 group :development do
-  gem "who_am_i", require: false
+  gem "who_am_i"
 end
 ```
 
@@ -31,21 +31,13 @@ And then execute:
 
     $ bundle
 
-In your `Rakefile`:
-
-```ruby
-require "who_am_i/rake"
-
-# ...
-# after `Rails.application.load_tasks`
-WhoAmI.load_rake_tasks
-```
-
 
 ## Usage
 
 Run `rake who_am_i` to update your model files.
 
+By default, `who_am_i` also runs after migrating your database.
+
 
 ## Development
 

data/lib/who_am_i/railtie.rb

@@ -0,0 +1,9 @@
+module WhoAmI
+  class Railtie < Rails::Railtie
+    railtie_name :who_am_i
+
+    rake_tasks do
+      load "who_am_i/tasks.rake"
+    end
+  end
+end

data/lib/who_am_i/table_column_info.rb

@@ -1,8 +1,8 @@
 module WhoAmI
   class TableColumnInfo
-    def initialize(basic_model_class:, column:)
-      @basic_model_class = basic_model_class
+    def initialize(table_name:, column:)
       @column = column
+      @primary_key = ActiveRecord::Base.get_primary_key(@table_name)
     end
 
     def name
@@ -27,11 +27,11 @@ module WhoAmI
     end
 
     def primary_key?
-      case @basic_model_class.primary_key
+      case @primary_key
       when Array
-        @basic_model_class.primary_key.include?(name)
+        @primary_key.include?(name)
       when String
-        @basic_model_class.primary_key == name
+        @primary_key == name
       else
         false
       end

data/lib/who_am_i/table_info.rb

@@ -8,24 +8,18 @@ module WhoAmI
 
     def columns
       @columns ||=
-        basic_model_class.columns.map do |column|
-          TableColumnInfo.new(basic_model_class: basic_model_class, column: column)
+        begin
+          columns_hash =
+            ActiveRecord::Base.connection.schema_cache.columns_hash(@table_name)
+
+          columns_hash.values.map do |column|
+            TableColumnInfo.new(table_name: @table_name, column: column)
+          end
         end
     end
 
     def indices
-      @indices ||=
-        basic_model_class.connection.indexes(@table_name)
-    end
-
-    private
-
-    def basic_model_class
-      @basic_model_class ||=
-        begin
-          table = @table_name
-          Class.new(ActiveRecord::Base) { self.table_name = table }
-        end
+      @indices ||= ActiveRecord::Base.connection.indexes(@table_name)
     end
   end
 end

data/lib/who_am_i/version.rb

@@ -1,3 +1,3 @@
 module WhoAmI
-  VERSION = "0.0.3".freeze
+  VERSION = "0.0.4".freeze
 end

data/lib/who_am_i.rb

@@ -32,3 +32,7 @@ require "who_am_i/function/resolve_table"
 require "who_am_i/function/compute_comment"
 require "who_am_i/function/compute_content"
 require "who_am_i/function/write_model"
+
+if defined?(Rails)
+  require "who_am_i/railtie"
+end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: who_am_i
 version: !ruby/object:Gem::Version
-  version: 0.0.3
+  version: 0.0.4
 platform: ruby
 authors:
 - Zach Ahn
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2018-01-08 00:00:00.000000000 Z
+date: 2018-01-11 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler
@@ -248,7 +248,7 @@ files:
 - lib/who_am_i/function/resolve_table.rb
 - lib/who_am_i/function/setup_environment.rb
 - lib/who_am_i/function/write_model.rb
-- lib/who_am_i/rake.rb
+- lib/who_am_i/railtie.rb
 - lib/who_am_i/refinement/hash_dig.rb
 - lib/who_am_i/refinement/yield_self.rb
 - lib/who_am_i/table_column_info.rb

data/lib/who_am_i/rake.rb

@@ -1,6 +0,0 @@
-module WhoAmI
-  def self.load_rake_tasks
-    require "who_am_i"
-    load(File.expand_path("tasks.rake", __dir__))
-  end
-end