Move CSV row encoding calculation to its own method

Author: wishdev

lib/csv.rb

@@ -1506,17 +1506,7 @@ def generate_line(row, **options)
       if options[:encoding]
         str.force_encoding(options[:encoding])
       else
-        fallback_encoding = nil
-        output_encoding = nil
-        row.each do |field|
-          next unless field.is_a?(String)
-          fallback_encoding ||= field.encoding
-          next if field.ascii_only?
-          output_encoding = field.encoding
-          break
-        end
-        output_encoding ||= fallback_encoding
-        if output_encoding
+        if output_encoding = row_encoding(row)
           str.force_encoding(output_encoding)
         end
       end
@@ -1960,6 +1950,20 @@ def table(path, **options)
     private_constant :ON_WINDOWS
 
     private
+
+    def row_encoding(row)
+      fallback_encoding = nil
+      output_encoding = nil
+      row.each do |field|
+        next unless field.is_a?(String)
+        fallback_encoding ||= field.encoding
+        next if field.ascii_only?
+        output_encoding = field.encoding
+        break
+      end
+      output_encoding || fallback_encoding
+    end
+
     def may_enable_bom_detection_automatically(filename_or_io,
                                                mode,
                                                options,