Adjust stats, formatter and calculator to print schema/structure file information

Author: hmdros

lib/rails_stats/console_formatter.rb

@@ -18,6 +18,7 @@ def to_s
       end
 
       print_code_test_stats
+      print_schema_stats
     end
 
     private
@@ -53,5 +54,15 @@ def print_code_test_stats
         puts "  Code LOC: #{code}     Test LOC: #{tests}     Code to Test Ratio: 1:#{sprintf("%.1f", tests.to_f/code)}  Files: #{calculator.files_total}"
         puts ""
       end
+
+      def print_schema_stats
+        if File.exist?(calculator.schema_path)
+          puts "  Schema Stats: #{calculator.schema} `create_table` calls in schema.rb"
+        elsif File.exist?(calculator.structure_path)
+          puts "  Schema Stats: #{calculator.schema} `CREATE TABLE` calls in structure.sql"
+        else
+          puts "  Schema Stats: No schema.rb or structure.sql file found"
+        end
+      end
   end
 end

lib/rails_stats/json_formatter.rb

@@ -17,6 +17,8 @@ def result
       @result << stat_hash("Tests", @tests_total).merge(code_test_hash) if @tests_total
       @result << stat_hash("Total", @grand_total).merge(code_test_hash) if @grand_total
 
+      @result << { "schema_stats" => schema_info }
+
       @result
     end
 
@@ -50,5 +52,16 @@ def stat_hash(name, statistics)
           "loc_over_m" => loc_over_m.to_s
         }
       end
+
+      def schema_info
+        if File.exist?(calculator.schema_path)
+          {
+            "schema_stats" => calculator.schema,
+            "create_table_calls" => calculator.create_table_calls
+          }
+        else
+          { "schema_stats" => "No schema.rb file found" }
+        end
+      end
   end
 end

lib/rails_stats/stats_calculator.rb

@@ -9,15 +9,18 @@ class StatsCalculator
 
     def initialize(root_directory)
       @root_directory = root_directory
+      @schema_path    = File.join(@root_directory, "db", "schema.rb")
+      @structure_path = File.join(@root_directory, "db", "structure.sql")
       @key_concepts   = calculate_key_concepts
       @projects       = calculate_projects
       @statistics     = calculate_statistics
       @code_loc       = calculate_code
       @test_loc       = calculate_tests
+      @schema         = calculate_create_table_calls
       @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
+    attr_reader :code_loc, :code_total, :files_total, :grand_total, :statistics, :test_loc, :tests_total, :schema, :schema_path, :structure_path
 
     private
 
@@ -133,5 +136,23 @@ def calculate_tests
         @statistics.each { |k, v| @test_loc += v.code_lines if v.test }
         @test_loc
       end
+
+      def calculate_create_table_calls
+        if @schema_path && File.exist?(@schema_path)
+          count_create_table_calls(schema_path, "create_table")
+        elsif @structure_path && File.exist?(@structure_path)
+          count_create_table_calls(structure_path, "CREATE TABLE")
+        else
+          0
+        end
+      end
+
+      def count_create_table_calls(file_path, keyword)
+        create_table_count = 0
+        File.foreach(file_path) do |line|
+          create_table_count += 1 if line.strip.start_with?(keyword)
+        end
+        create_table_count
+      end
   end
 end