Unify collection #to_comma extensions

.rubocop_todo.yml

@@ -108,13 +108,14 @@ Style/BlockDelimiters:
     - 'spec/comma/data_extractor_spec.rb'
     - 'spec/comma/header_extractor_spec.rb'
 
-# Offense count: 11
+# Offense count: 12
 # Configuration parameters: AllowedConstants.
 Style/Documentation:
   Exclude:
     - 'spec/**/*'
     - 'test/**/*'
     - 'lib/comma/array.rb'
+    - 'lib/comma/collection_export.rb'
     - 'lib/comma/data_extractor.rb'
     - 'lib/comma/data_mapper_collection.rb'
     - 'lib/comma/extractor.rb'

lib/comma.rb

@@ -11,10 +11,10 @@ module Comma
   }.freeze
 end
 
-require 'comma/data_mapper_collection' if defined? DataMapper
-
 require 'comma/options'
 require 'comma/generator'
+require 'comma/collection_export'
+require 'comma/data_mapper_collection' if defined? DataMapper
 require 'comma/array'
 require 'comma/object'
 

lib/comma/array.rb

@@ -1,7 +1,5 @@
 # frozen_string_literal: true
 
 class Array
-  def to_comma(style = :default)
-    Comma::Generator.new(self, style).run(:each)
-  end
+  include Comma::CollectionExport
 end

lib/comma/collection_export.rb

@@ -0,0 +1,9 @@
+# frozen_string_literal: true
+
+module Comma
+  module CollectionExport
+    def to_comma(style = :default)
+      Comma::Generator.new(self, style).run(:each)
+    end
+  end
+end

lib/comma/data_mapper_collection.rb

@@ -3,9 +3,7 @@
 if defined?(DataMapper)
   module DataMapper
     class Collection
-      def to_comma(style = :default)
-        Comma::Generator.new(self, style).run(:each)
-      end
+      include Comma::CollectionExport
     end
   end
 end

lib/comma/mongoid.rb

@@ -6,9 +6,7 @@
 
   module Mongoid
     class Criteria
-      def to_comma(style = :default)
-        Comma::Generator.new(self, style).run(:each)
-      end
+      include Comma::CollectionExport
     end
   end
 rescue LoadError => e