add sti and polymorphysm

Author: hmdros

lib/rails_stats/console_formatter.rb

@@ -18,6 +18,8 @@ def to_s
       end
 
       print_code_test_stats
+      print_sti_stats
+      print_polymorphic_stats
       print_schema_stats
     end
 
@@ -55,6 +57,14 @@ def print_code_test_stats
         puts ""
       end
 
+      def print_sti_stats
+        puts "  STI models count: #{calculator.sti} STI classes"
+      end
+
+      def print_polymorphic_stats
+        puts "  Polymorphic models count: #{calculator.polymorphic} polymorphic associations"
+      end
+
       def print_schema_stats
         if File.exist?(calculator.schema_path)
           puts "  Schema Stats: #{calculator.schema} `create_table` calls in schema.rb"

lib/rails_stats/json_formatter.rb

@@ -18,6 +18,8 @@ def result
       @result << stat_hash("Total", @grand_total).merge(code_test_hash) if @grand_total
 
       @result << { "schema_stats" => schema_info }
+      @result << { "sti_stats" => print_sti_stat }
+      @result << { "polymorphic_stats" => print_polymorphic_stats }
 
       @result
     end
@@ -53,6 +55,22 @@ def stat_hash(name, statistics)
         }
       end
 
+      def print_sti_stat
+        if calculator.sti
+          {
+            "sti_models_count" => calculator.sti,
+          }
+        end
+      end
+
+      def print_polymorphic_stats
+        if calculator.polymorphic
+          {
+            "polymorphic_models_count" => calculator.polymorphic,
+          }
+        end
+      end
+
       def schema_info
         if File.exist?(calculator.schema_path)
           {

lib/rails_stats/stats_calculator.rb

@@ -17,10 +17,12 @@ def initialize(root_directory)
       @code_loc       = calculate_code
       @test_loc       = calculate_tests
       @schema         = calculate_create_table_calls
+      @sti            = calculate_sti
+      @polymorphic   = calculate_polymorphic
       @files_total, @code_total, @tests_total, @grand_total = calculate_totals
     end
 
-    attr_reader :code_loc, :code_total, :files_total, :grand_total, :statistics, :test_loc, :tests_total, :schema, :schema_path, :structure_path
+    attr_reader :code_loc, :code_total, :files_total, :grand_total, :statistics, :test_loc, :tests_total, :schema, :schema_path, :structure_path, :sti, :polymorphic
 
     private
 
@@ -49,7 +51,6 @@ def calculate_projects
         out
       end
 
-
       def app_projects
         @app_projects ||= calculate_app_projects
       end
@@ -82,7 +83,6 @@ def calculate_test_projects
         end
       end
 
-
       def calculate_root_projects
         [RootStatistics.new(@root_directory)]
       end
@@ -154,5 +154,29 @@ def count_create_table_calls(file_path, keyword)
         end
         create_table_count
       end
+
+      def calculate_sti
+        @sti = 0
+        Dir.glob(File.join(@root_directory, "app", "models", "*.rb")).each do |file|
+          File.foreach(file) do |line|
+            if line =~ /class\s+(\w+)\s*<\s*(\w+)/ && !(line =~ /class\s+(\w+)\s*<\s*(ActiveRecord::Base|ApplicationRecord)/)
+              @sti += 1
+            end
+          end
+        end
+        @sti
+      end
+
+      def calculate_polymorphic
+        polymorphic_count = 0
+        Dir.glob(File.join(@root_directory, "app", "models", "*.rb")).each do |file|
+          File.foreach(file) do |line|
+            if line =~ /belongs_to\s+:\w+,\s+polymorphic:\s+true/
+              polymorphic_count += 1
+            end
+          end
+        end
+        polymorphic_count
+      end
   end
 end