Merge pull request #3 from seuros/fix-issue-2-cli-delegated-mysql

fix: resolve GitHub issue #2 CLI and provider errors

Author: seuros

lib/rails_lens/analyzers/delegated_types.rb

@@ -17,7 +17,12 @@ def analyze
 
         lines << "Type Column: #{delegated_type_info[:type_column]}"
         lines << "ID Column: #{delegated_type_info[:id_column]}"
-        lines << "Types: #{delegated_type_info[:types].join(', ')}"
+        types_list = if delegated_type_info[:types].respond_to?(:keys)
+                       delegated_type_info[:types].keys
+                     else
+                       Array(delegated_type_info[:types])
+                     end
+        lines << "Types: #{types_list.join(', ')}"
 
         lines.join("\n")
       end

lib/rails_lens/cli.rb

@@ -15,6 +15,7 @@ class CLI < Thor
 
     desc 'annotate', 'Annotate Rails models with schema information'
     option :models, type: :array, desc: 'Specific models to annotate'
+    option :include_abstract, type: :boolean, desc: 'Include abstract classes'
     option :position, type: :string, enum: %w[before after top bottom], desc: 'Annotation position'
     option :routes, type: :boolean, desc: 'Annotate controller routes'
     option :mailers, type: :boolean, desc: 'Annotate mailer methods'

lib/rails_lens/schema/adapters/mysql.rb

@@ -117,7 +117,7 @@ def table_storage_engine
         rescue ActiveRecord::StatementInvalid => e
           Rails.logger.debug { "Failed to fetch storage engine for #{table_name}: #{e.message}" }
           nil
-        rescue Mysql2::Error => e
+        rescue => e
           Rails.logger.debug { "MySQL error fetching storage engine: #{e.message}" }
           nil
         end
@@ -137,7 +137,7 @@ def table_charset
         rescue ActiveRecord::StatementInvalid => e
           Rails.logger.debug { "Failed to fetch charset for #{table_name}: #{e.message}" }
           nil
-        rescue Mysql2::Error => e
+        rescue => e
           Rails.logger.debug { "MySQL error fetching charset: #{e.message}" }
           nil
         end
@@ -155,7 +155,7 @@ def table_collation
         rescue ActiveRecord::StatementInvalid => e
           Rails.logger.debug { "Failed to fetch collation for #{table_name}: #{e.message}" }
           nil
-        rescue Mysql2::Error => e
+        rescue => e
           Rails.logger.debug { "MySQL error fetching collation: #{e.message}" }
           nil
         end
@@ -202,7 +202,7 @@ def has_partitions?
           # Table doesn't exist or no permission to query information_schema
           Rails.logger.debug { "Failed to check partitions for #{table_name}: #{e.message}" }
           false
-        rescue Mysql2::Error => e
+        rescue => e
           # MySQL specific errors (connection issues, etc)
           Rails.logger.debug { "MySQL error checking partitions: #{e.message}" }
           false
@@ -231,7 +231,7 @@ def add_partitions(lines)
         rescue ActiveRecord::StatementInvalid => e
           # Permission denied or table doesn't exist
           Rails.logger.debug { "Failed to fetch partitions for #{table_name}: #{e.message}" }
-        rescue Mysql2::Error => e
+        rescue => e
           # MySQL specific errors
           Rails.logger.debug { "MySQL error fetching partitions: #{e.message}" }
         end
@@ -269,7 +269,7 @@ def add_partitions_toml(lines)
         rescue ActiveRecord::StatementInvalid => e
           # Permission denied or table doesn't exist
           Rails.logger.debug { "Failed to fetch partitions for #{table_name}: #{e.message}" }
-        rescue Mysql2::Error => e
+        rescue => e
           # MySQL specific errors
           Rails.logger.debug { "MySQL error fetching partitions: #{e.message}" }
         end