Merge pull request #18 from kzkn/generate-line-options

Add `csv_options` as an option to pass to `CSV.generate_line`

lib/csb/builder.rb

@@ -5,21 +5,22 @@ module Csb
   class Builder
     UTF8_BOM = "\xEF\xBB\xBF".freeze
 
-    attr_reader :output, :utf8_bom, :items, :cols
+    attr_reader :output, :utf8_bom, :items, :cols, :csv_options
     attr_accessor :items
 
-    def initialize(output = '', items: [], utf8_bom: false)
+    def initialize(output = '', items: [], utf8_bom: false, csv_options: {})
       @output = output
       @utf8_bom = utf8_bom
       @cols = Cols.new
       @items = items
+      @csv_options = csv_options
     end
 
     def build
       output << UTF8_BOM if utf8_bom
-      output << CSV.generate_line(cols.headers)
+      output << CSV.generate_line(cols.headers, **csv_options)
       items.each do |item|
-        output << CSV.generate_line(cols.values_by_item(item))
+        output << CSV.generate_line(cols.values_by_item(item), **csv_options)
       rescue => error
         break if Csb.configuration.ignore_class_names.include?(error.class.name)
 

lib/csb/template.rb

@@ -1,12 +1,13 @@
 module Csb
   class Template
-    attr_accessor :utf8_bom, :filename, :streaming, :items, :cols
+    attr_accessor :utf8_bom, :filename, :streaming, :items, :cols, :csv_options
 
     def initialize(utf8_bom:, streaming:)
       @utf8_bom = utf8_bom
       @streaming = streaming
       @cols = Cols.new
       @items = []
+      @csv_options = {}
     end
 
     def build
@@ -20,15 +21,15 @@ def streaming?
     private
 
     def build_string
-      builder = Builder.new(utf8_bom: utf8_bom, items: items)
+      builder = Builder.new(utf8_bom: utf8_bom, items: items, csv_options: csv_options)
       builder.cols.copy!(cols)
       builder.build
     end
 
     def build_enumerator
       Enumerator.new do |y|
         begin
-          builder = Builder.new(y, utf8_bom: utf8_bom, items: items)
+          builder = Builder.new(y, utf8_bom: utf8_bom, items: items, csv_options: csv_options)
           builder.cols.copy!(cols)
           builder.build
         rescue => error

spec/lib/csb/builder_spec.rb

@@ -28,5 +28,11 @@
 
       it { is_expected.to eq "\xEF\xBB\xBFName,Email,Dummy\ntester1,[email protected],\ntester2,[email protected],\n" }
     end
+
+    context 'with csv_options' do
+      let(:builder) { Csb::Builder.new(items: items, csv_options: { col_sep: "\t" }) }
+
+      it { is_expected.to eq "Name\tEmail\tDummy\ntester1\t[email protected]\t\ntester2\t[email protected]\t\n" }
+    end
   end
 end

spec/lib/csb/template_spec.rb

@@ -33,5 +33,17 @@
 
       it { is_expected.to eq "Name,Email,Dummy\ntester1,[email protected],\ntester2,[email protected],\n" }
     end
+
+    context 'with csv_options' do
+      subject { template.build }
+
+      let(:template) { Csb::Template.new(streaming: false, utf8_bom: false) }
+
+      before do
+        template.csv_options = { row_sep: "\r\n" }
+      end
+
+      it { is_expected.to eq "Name,Email,Dummy\r\ntester1,[email protected],\r\ntester2,[email protected],\r\n" }
+    end
   end
 end