gem_activity_tracker 1.0.1 → 1.0.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: dbf3e335476faecbfe09e7b427ec0706ad1cbfc397ceadd74ff8bd2dd68d56ae
-  data.tar.gz: 83d525f61caafca1a605244dca22be4859045739a04ea984a7edd9267361af51
+  metadata.gz: 41af2cc25b92505945ab8ca305644b17b7df8e13b94e0b17ef88caaad2b14a4f
+  data.tar.gz: aac8db37f891708a2f7c21768657c87b8ed98e1b09be8839afa2c825578ae883
 SHA512:
-  metadata.gz: 55066b038228ba20827e13f17fd962d0e4da8c9056ca4d5c67d4b3b414fe01ac4e6f067c4652169089cff9d13407819e997203e28dfb7ec2c5eb5e10ace8960a
-  data.tar.gz: 354d740e41817c937009444eec07ffd896e0f41918896c1eb4725025b0592447459bd95d809e4673eff13c688328904b33dbb4af1bc7af2c7fe36bb7074e609e
+  metadata.gz: 13b648d787fc0ad59444df78e924c290b226f4bbf4325400e6665e90554cb26be7cb594047ddc3fbb38c42765562449b648834a6eafa97792c5926fec704eec5
+  data.tar.gz: 12eb408462c35e041cabba7ca4629f5a541d168f96529dfd17eb23c1321239519be0619046821245789b698cca486477632bfe13d3b16159bdc6adb358983fe1

data/lib/gem_activity_tracker/railtie.rb

@@ -6,7 +6,7 @@ module GemActivityTracker
       Rails.application.config.after_initialize do
         # Delay the execution slightly to avoid blocking boot
         Thread.new do
-          sleep(2) # Give Rails a moment to fully start
+          sleep(3) # Give Rails a moment to fully start
 
           begin
             path = Rails.root.to_s

data/lib/gem_activity_tracker/tracker.rb

@@ -3,6 +3,7 @@ require 'json'
 require 'fileutils'
 require 'csv'
 require 'digest'
+require 'erb'
 require 'active_support/all'
 require 'active_record'
 require 'active_support/dependencies'
@@ -13,7 +14,6 @@ module GemActivityTracker
       puts "🔍 Tracking project at: #{path}"
 
       if !defined?(Rails) || Rails.root.to_s != path
-        # Only require environment if we're not already inside a Rails app
         $LOAD_PATH.unshift(path)
         ENV['RAILS_ENV'] ||= 'development'
         require File.join(path, 'config', 'environment')
@@ -21,16 +21,30 @@ module GemActivityTracker
 
       data = collect_data(path)
       FileUtils.mkdir_p("#{path}/activity_tracker")
-      File.write("#{path}/activity_tracker/report.yml", data.to_yaml)
-
+      File.write("#{path}/activity_tracker/report.yml", data.to_yaml(line_width: -1))
       puts "✅ Report generated at: #{path}/activity_tracker/report.yml"
     end
-    
+
+    def self.safe_yaml_load(file_path)
+      content = ERB.new(File.read(file_path)).result
+      if YAML.respond_to?(:safe_load)
+        begin
+          YAML.safe_load(content, permitted_classes: [Symbol], permitted_symbols: [], aliases: true)
+        rescue ArgumentError
+          YAML.load(content)
+        end
+      else
+        YAML.load(content)
+      end
+    rescue => e
+      puts "❌ YAML Load Error: #{e.message}"
+      {}
+    end
+
     def self.collect_data(path)
       {
         project_name: File.basename(path),
         ruby_version: `ruby -v`.strip,
-        # rails_version: get_rails_version(path),
         rails_version: Rails.version,
         database: detect_database(path),
         models: analyze_models(path),
@@ -40,39 +54,36 @@ module GemActivityTracker
         services: list_and_count_files(path, "app/services"),
         migrations: migration_changes(path),
         schema_hash: schema_hash(path),
-        routes: get_routes(path),
+        # routes: get_routes(path),
         git_log: get_git_log(path)
       }
     end
 
     def self.report
       file = "activity_tracker/report.yml"
-      puts File.exist?(file) ? YAML.load_file(file).to_yaml : "❌ No report found."
+      puts File.exist?(file) ? safe_yaml_load(file).to_yaml(line_width: -1) : "❌ No report found."
     end
 
     def self.list_and_count_files(base, dir)
       path = File.join(base, dir)
+      return { count: 0, files: [] } unless Dir.exist?(path)
+
       files = Dir.glob("#{path}/**/*.rb").map { |f| f.gsub(base + '/', '') }
       { count: files.count, files: files }
     end
 
-    def self.get_rails_version(path)
-      gemfile = File.join(path, 'Gemfile.lock')
-      File.exist?(gemfile) ? File.read(gemfile)[/rails \((.*?)\)/, 1] : 'Not a Rails project'
-    end
-
     def self.detect_database(path)
       db_file = File.join(path, "config/database.yml")
       return 'Unknown' unless File.exist?(db_file)
 
-      config = YAML.load_file(db_file, aliases: true)
+      config = safe_yaml_load(db_file)
       config["development"]["adapter"] rescue "Unknown"
     end
 
-    def self.get_routes(path)
-      output = `cd #{path} && RAILS_ENV=development bundle exec rails routes 2>/dev/null`
-      output.empty? ? "No routes found or Rails not installed" : output
-    end
+    # def self.get_routes(path)
+    #   output = `cd #{path} && RAILS_ENV=development bundle exec rails routes 2>/dev/null`
+    #   output.empty? ? "No routes found or Rails not installed" : output
+    # end
 
     def self.migration_changes(path)
       files = Dir.glob("#{path}/db/migrate/*.rb")
@@ -89,87 +100,102 @@ module GemActivityTracker
     end
 
     def self.get_git_log(path)
+      return [] unless File.exist?(File.join(path, ".git"))
       Dir.chdir(path) do
-        log = `git log --pretty=format:'%h - %an (%ad): %s' --date=short -n 20`
+        log = `git log --pretty=format:'%h - %an (%ad): %s' --date=short -n 20 2>/dev/null`
         log.split("\n")
       end
+    rescue => e
+      ["Error reading git log: #{e.message}"]
     end
 
-    def self.analyze_models(path)
-      result = { count: 0, files: [], detailed: {} }
+def self.analyze_models(path)
+  result = { count: 0, files: [], detailed: {} }
 
-      ActiveSupport::Dependencies.autoload_paths += Dir["#{path}/app/models/**/"]
+  ActiveSupport::Dependencies.autoload_paths += Dir["#{path}/app/models/**/"]
 
-      Dir.glob("#{path}/app/models/**/*.rb").each do |file|
-        relative_path = file.gsub("#{path}/", '')
-        model_name = relative_path.sub('app/models/', '').sub('.rb', '').camelize
+  Dir.glob("#{path}/app/models/**/*.rb").each do |file|
+    next unless File.exist?(file)
 
-        begin
-          require_dependency file
-          model_class = model_name.constantize
-          next unless model_class < ActiveRecord::Base
-
-          result[:count] += 1
-          result[:files] << relative_path
-
-          # Validations
-          validations = Hash.new { |hash, key| hash[key] = [] }
-          model_class.validators.each do |validator|
-            validator.attributes.each do |attr|
-              validations[attr] << validator.class.name.demodulize.underscore
-            end
-          end
+    relative_path = file.gsub("#{path}/", '')
+    model_name = relative_path.sub('app/models/', '').sub('.rb', '').camelize
 
-          # Callbacks
-          callbacks = {}
-          ActiveRecord::Callbacks::CALLBACKS.each do |cb|
-            if model_class.respond_to?("_#{cb}_callbacks")
-              methods = model_class.send("_#{cb}_callbacks").map(&:filter).uniq
-              callbacks[cb] = methods.map(&:to_s) if methods.any?
-            end
-          end
+    begin
+      require_dependency file
+
+      begin
+        model_class = model_name.safe_constantize
+      rescue => e
+        result[:detailed][model_name] = { error: "safe_constantize failed: #{e.class} - #{e.message}" }
+        next
+      end
 
-          # Enums
-          # debugger
-          enums = model_class.defined_enums.transform_values { |v| v.keys }
-
-          # Associations
-          associations = model_class.reflect_on_all_associations.group_by(&:macro).transform_values { |a| a.map(&:name) }
-
-          # Scopes
-          scopes = model_class.methods(false).select do |method|
-            model_class.method(method).source_location&.first&.include?('/models/')
-          end.select { |m| model_class.respond_to?(m) && model_class.send(m).is_a?(ActiveRecord::Relation) rescue false }
-
-          # Class methods
-          class_methods = (model_class.methods(false) - ActiveRecord::Base.methods)
-          instance_methods = (model_class.instance_methods(false) - ActiveRecord::Base.instance_methods).map(&:to_s)
-
-          # # Comments
-          # comment = File.foreach(file).first if File.read(file).lines.first.strip.start_with?('#')
-
-          result[:detailed][model_class.name] = {
-            table_name: model_class.table_name,
-            file: relative_path,
-            # comment: comment&.strip,
-            attributes: model_class.columns.map(&:name),
-            associations: associations,
-            validations: validations,
-            enums: enums,
-            callbacks: callbacks,
-            scopes: scopes.map(&:to_s),
-            class_methods: class_methods.map(&:to_s),
-            instance_methods: instance_methods,
-            methods_count: model_class.instance_methods(false).count
-          }
-
-        rescue => e
-          result[:detailed][model_name] = { error: "Failed to load: #{e.message}" }
+      unless model_class && model_class < ActiveRecord::Base
+        result[:detailed][model_name] = { error: "Not an ActiveRecord model or constant not found." }
+        next
+      end
+
+      result[:count] += 1
+      result[:files] << relative_path
+
+      # Validations
+      validations = Hash.new { |hash, key| hash[key] = [] }
+      model_class.validators.each do |validator|
+        validator.attributes.each do |attr|
+          validations[attr] << validator.class.name.demodulize.underscore
         end
       end
 
-      result
+      # Callbacks
+      callbacks = {}
+      ActiveRecord::Callbacks::CALLBACKS.each do |cb|
+        if model_class.respond_to?("_#{cb}_callbacks")
+          methods = model_class.send("_#{cb}_callbacks").map(&:filter).uniq
+          callbacks[cb] = methods.map(&:to_s) if methods.any?
+        end
+      end
+
+      # Enums
+      enums = model_class.defined_enums.transform_values { |v| v.keys }
+
+      # Associations with polymorphic check
+      raw_associations = model_class.reflect_on_all_associations
+      associations = raw_associations.group_by(&:macro).transform_values { |a| a.map(&:name) }
+      polymorphic_associations = raw_associations.select { |a| a.options[:polymorphic] }.map(&:name)
+
+      # Scopes
+      scopes = model_class.methods(false).select do |method|
+        model_class.method(method).source_location&.first&.include?('/models/')
+      end.select { |m| model_class.respond_to?(m) && model_class.send(m).is_a?(ActiveRecord::Relation) rescue false }
+
+      # Class & Instance methods
+      class_methods = (model_class.methods(false) - ActiveRecord::Base.methods)
+      instance_methods = (model_class.instance_methods(false) - ActiveRecord::Base.instance_methods).map(&:to_s)
+
+      result[:detailed][model_class.name] = {
+        table_name: model_class.table_name,
+        file: relative_path,
+        attributes: model_class.columns.map(&:name),
+        attribute_defaults: model_class.columns.each_with_object({}) { |col, hash| hash[col.name] = col.default },
+        sti: model_class.columns.any? { |col| col.name == model_class.inheritance_column },
+        polymorphic_associations: polymorphic_associations,
+        associations: associations,
+        validations: validations,
+        enums: enums,
+        callbacks: callbacks,
+        scopes: scopes.map(&:to_s),
+        class_methods: class_methods.map(&:to_s),
+        instance_methods: instance_methods,
+        methods_count: model_class.instance_methods(false).count
+      }
+
+    rescue => e
+      result[:detailed][model_name] = { error: "Failed to load model: #{e.class} - #{e.message}" }
     end
+  end
+
+  result
+end
 
 
 
@@ -177,7 +203,7 @@ module GemActivityTracker
       file = "#{path}/activity_tracker/report.yml"
       return puts "❌ Report not found." unless File.exist?(file)
 
-      data = YAML.load_file(file)
+      data = safe_yaml_load(file)
 
       case format
       when :json

data/lib/gem_activity_tracker/version.rb

@@ -1,3 +1,3 @@
 module GemActivityTracker
-  VERSION = "1.0.1"
+  VERSION = "1.0.2"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: gem_activity_tracker
 version: !ruby/object:Gem::Version
-  version: 1.0.1
+  version: 1.0.2
 platform: ruby
 authors:
 - Atul Yadav
@@ -25,7 +25,7 @@ dependencies:
       - !ruby/object:Gem::Version
         version: '3.0'
 description: 'A CLI gem to analyze Ruby/Rails projects and track activity: models,
-  migrations, routes, Git, schema, and more.'
+  migrations, Git, schema,STI, assocation,methods, acope, validation and more.'
 email:
 - atulyadav9039@gmail.com
 executables: