fix issue 294 with writing headers with quotes (#295)

CHANGELOG.md

@@ -1,6 +1,9 @@
 
 # SmarterCSV 1.x Change Log
 
+## 1.13.1 (2024-12-12)
+  * fix bug with SmarterCSV.generate with `force_quotes: true` ([issue 294](https://github.com/tilo/smarter_csv/issues/294))
+
 ## 1.13.0 (2024-11-06) ⚡ POTENTIALLY BREAKING ⚡
 
   CHANGED DEFAULT BEHAVIOR

lib/smarter_csv/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module SmarterCSV
-  VERSION = "1.13.0"
+  VERSION = "1.13.1"
 end

lib/smarter_csv/writer.rb

@@ -81,6 +81,7 @@ def <<(data)
     def finalize
       # Map headers if :map_headers option is provided
       mapped_headers = @headers.map { |header| @map_headers[header] || header }
+      mapped_headers = mapped_headers.map{|x| escape_csv_field(x)} if @force_quotes
 
       @temp_file.rewind
       @output_file.write(mapped_headers.join(@col_sep) + @row_sep)

spec/smarter_csv/writer_spec.rb

@@ -288,7 +288,24 @@
       writer << [{ a: 'hello', b: 'world' }]
       writer.finalize
 
-      expect(File.read(file_path)).to eq("a,b#{row_sep}\"hello\",\"world\"#{row_sep}")
+      expect(File.read(file_path)).to eq("\"a\",\"b\"#{row_sep}\"hello\",\"world\"#{row_sep}")
+    end
+
+    context 'force_quotes also applies to headers' do
+      let(:options) { {force_quotes: true} }
+      let(:data) do
+        { name: 'John', age: 30, city: 'New York' }
+      end
+
+      it 'writes the given headers and data correctly' do
+        writer = SmarterCSV::Writer.new(file_path, options)
+        writer << data
+        writer.finalize
+        output = File.read(file_path)
+
+        expect(output).to include("\"name\",\"age\",\"city\"#{row_sep}")
+        expect(output).to include("\"John\",\"30\",\"New York\"#{row_sep}")
+      end
     end
   end