who_am_i 0.0.1 → 0.0.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 2253bede51a5e4fa72f405ab2ff5f688623e28e3
-  data.tar.gz: d75e93a1227f87b1be4cdff7b3a08854a62d4d90
+  metadata.gz: 55880585cfea925d0d39e6e92d909fbaa8a16344
+  data.tar.gz: 5fd295e65e97840d2f053402b778cad610543a22
 SHA512:
-  metadata.gz: ad5bae82858ef5f415ca30934b87807358ed8c3dd16d10d1c03b1809a1fd3922f9a0232549b3e26e7bdb6cca8f8dd5de44514df98910debb12b75ea9d1f07c75
-  data.tar.gz: f287bae2249961e3db22ceb871120a5f8bb475de5b20c84ebffe0c98094da4a3628270123413d58acce02b0c28e45cdc652f12e1afc4c84f10a98d52bb0b9d83
+  metadata.gz: 4d85064a47d41c6e03597ccc893bcfb5860f88a027a0c95051acb665b6a9a0fc41d6b5db037b666e8cfd1b3fce1746abe73eaa2cdca12e2cd8eda34643cd4035
+  data.tar.gz: 8c242e8b3ce144068ca7cbb9c80103fde3597731936702cca4ae3382a2898f73784baee06afd40d083910bbc57c52eacbfdb1ae12827ec535659c58aa6317331

data/gemfiles/rails32.gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: ..
   specs:
-    who_am_i (0.0.0)
+    who_am_i (0.0.2)
       activerecord (>= 3.2, < 5.2)
       parser
       proc_party

data/gemfiles/rails40.gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: ..
   specs:
-    who_am_i (0.0.0)
+    who_am_i (0.0.2)
       activerecord (>= 3.2, < 5.2)
       parser
       proc_party

data/gemfiles/rails41.gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: ..
   specs:
-    who_am_i (0.0.0)
+    who_am_i (0.0.2)
       activerecord (>= 3.2, < 5.2)
       parser
       proc_party

data/gemfiles/rails42.gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: ..
   specs:
-    who_am_i (0.0.0)
+    who_am_i (0.0.2)
       activerecord (>= 3.2, < 5.2)
       parser
       proc_party

data/gemfiles/rails50.gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: ..
   specs:
-    who_am_i (0.0.0)
+    who_am_i (0.0.2)
       activerecord (>= 3.2, < 5.2)
       parser
       proc_party

data/gemfiles/rails51.gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: ..
   specs:
-    who_am_i (0.0.0)
+    who_am_i (0.0.2)
       activerecord (>= 3.2, < 5.2)
       parser
       proc_party

data/lib/who_am_i/extracted_class.rb

@@ -23,6 +23,7 @@ module WhoAmI
       self.claimed_superclass = claimed_superclass
       self.table_name = table_name
       self.abstract_class = abstract_class
+      @skipped_because = []
     end
 
     def activerecord?
@@ -45,6 +46,14 @@ module WhoAmI
       end
     end
 
+    def skipped?
+      @skipped_because.size > 0
+    end
+
+    def skip(reason)
+      @skipped_because.push(reason)
+    end
+
     alias_method :to_s, :class_name
   end
 end

data/lib/who_am_i/function/annotate_models.rb

@@ -1,7 +1,6 @@
 module WhoAmI
   module Function
     class AnnotateModels
-      using Refinement::YieldSelf
       include ProcParty
 
       def initialize(config, tables)
@@ -14,10 +13,14 @@ module WhoAmI
           .flat_map(&Ls.new)
           .flat_map(&ParseModel.new)
           .tap(&ExtractModelData.new)
-          .yield_self(&ResolveTables.new)
+          .select(&:activerecord?)
           .reject(&:abstract_class?)
+          .each(&ResolveTable.new(@tables))
+          .reject(&:skipped?)
           .each(&ComputeComment.new)
+          .reject(&:skipped?)
           .each(&ComputeContent.new)
+          .reject(&:skipped?)
           .each(&WriteModel.new)
       end
 

data/lib/who_am_i/function/load_inflections.rb

@@ -0,0 +1,13 @@
+module WhoAmI
+  module Function
+    class LoadInflections
+      include ProcParty
+
+      def call
+        if File.exist?("config/initializers/inflections.rb")
+          load "config/initializers/inflections.rb"
+        end
+      end
+    end
+  end
+end

data/lib/who_am_i/function/main.rb

@@ -8,6 +8,7 @@ module WhoAmI
       def call
         config = LoadConfig.new(@root).call
         tables = GetTables.new.call
+        LoadInflections.new.call
 
         AnnotateModels.new(config[:enabled][:models], tables).call
       end

data/lib/who_am_i/function/resolve_class_relationships.rb

@@ -18,7 +18,8 @@ module WhoAmI
         object_space["::ActiveRecord::Base"] =
           ExtractedClass.new(
             "ActiveRecord::Base",
-            outerclass: object_space[""]
+            outerclass: object_space[""],
+            abstract_class: true
           )
 
         object_space.each do |class_name, extracted_class|

data/lib/who_am_i/function/resolve_table.rb

@@ -0,0 +1,66 @@
+module WhoAmI
+  module Function
+    class ResolveTable
+      include ProcParty
+
+      def initialize(known_tables)
+        @known_tables = known_tables
+      end
+
+      def call(extracted_class)
+        set_table_name_for(extracted_class)
+      end
+
+      private
+
+      def set_table_name_for(extracted_class)
+        if extracted_class.abstract_class?
+          return
+        end
+
+        if extracted_class.table_name
+          return
+        end
+
+        if extracted_class.resolved_superclass
+          if !extracted_class.resolved_superclass.abstract_class?
+            set_table_name_for(extracted_class.resolved_superclass)
+
+            extracted_class.table_name =
+              extracted_class.resolved_superclass.table_name
+
+            return
+          end
+        end
+
+        extracted_class.table_name =
+          guess_table_from(extracted_class.class_name)
+
+        if extracted_class.table_name.nil?
+          extracted_class.skip("no table could be determined")
+        end
+      end
+
+      def guess_table_from(class_name)
+        loop do
+          guess =
+            class_name
+              .underscore
+              .pluralize
+              .sub(%r{\A/}, "")
+              .sub(%r{/}, "_")
+
+          if @known_tables.include?(guess)
+            break guess
+          end
+
+          if class_name !~ /::/
+            break
+          end
+
+          class_name = class_name.split("::", 2).last
+        end
+      end
+    end
+  end
+end

data/lib/who_am_i/version.rb

@@ -1,3 +1,3 @@
 module WhoAmI
-  VERSION = "0.0.1".freeze
+  VERSION = "0.0.2".freeze
 end

data/lib/who_am_i.rb

@@ -17,13 +17,14 @@ require "who_am_i/function/remove_annotation"
 require "who_am_i/function/main"
 require "who_am_i/function/load_config"
 require "who_am_i/function/get_tables"
+require "who_am_i/function/load_inflections"
 require "who_am_i/function/annotate_models"
 require "who_am_i/function/ls"
 require "who_am_i/function/parse_model"
 require "who_am_i/function/extract_model_data"
 require "who_am_i/function/resolve_class_relationships"
 require "who_am_i/function/resolve_active_record"
-require "who_am_i/function/resolve_tables"
+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"

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: who_am_i
 version: !ruby/object:Gem::Version
-  version: 0.0.1
+  version: 0.0.2
 platform: ruby
 authors:
 - Zach Ahn
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2018-01-02 00:00:00.000000000 Z
+date: 2018-01-04 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler
@@ -242,13 +242,14 @@ files:
 - lib/who_am_i/function/extract_model_data.rb
 - lib/who_am_i/function/get_tables.rb
 - lib/who_am_i/function/load_config.rb
+- lib/who_am_i/function/load_inflections.rb
 - lib/who_am_i/function/ls.rb
 - lib/who_am_i/function/main.rb
 - lib/who_am_i/function/parse_model.rb
 - lib/who_am_i/function/remove_annotation.rb
 - lib/who_am_i/function/resolve_active_record.rb
 - lib/who_am_i/function/resolve_class_relationships.rb
-- lib/who_am_i/function/resolve_tables.rb
+- lib/who_am_i/function/resolve_table.rb
 - lib/who_am_i/function/write_model.rb
 - lib/who_am_i/rake.rb
 - lib/who_am_i/refinement/yield_self.rb

data/lib/who_am_i/function/resolve_tables.rb

@@ -1,25 +0,0 @@
-module WhoAmI
-  module Function
-    class ResolveTables
-      include ProcParty
-
-      def call(extracted_classes)
-        extracted_classes.each do |extracted_class|
-          if !extracted_class.table_name.nil?
-            next
-          end
-
-          extracted_class.table_name =
-            if Kernel.const_defined?(extracted_class.class_name)
-              Kernel.const_get(extracted_class.class_name).table_name
-            else
-              extracted_class.class_name
-                .underscore
-                .pluralize
-                .sub(%r{\A/}, "")
-            end
-        end
-      end
-    end
-  end
-end