Merge Cells

lib/xlsxtream/worksheet.rb

@@ -9,6 +9,7 @@ def initialize(io, options = {})
       @rownum = 1
       @closed = false
       @options = options
+      @xml_merge_cells = nil
 
       write_header
     end
@@ -18,6 +19,18 @@ def <<(row)
       @rownum += 1
     end
     alias_method :add_row, :<<
+
+    # refs: ['A1:B1', 'C1:G1', 'A2:G2', ...]
+    def merge_cells(refs)
+      xml = String.new(%Q{<mergeCells r="#{refs.length}">})
+
+      refs.each do |ref|
+        xml << %Q{<mergeCell ref="#{ref}"/>}
+      end
+
+      xml << '</mergeCells>'
+      @xml_merge_cells = xml
+    end
 
     def close
       write_footer
@@ -57,6 +70,7 @@ def write_header
     def write_footer
       @io << XML.strip(<<-XML)
           </sheetData>
+          #{@xml_merge_cells}
         </worksheet>
       XML
     end

test/xlsxtream/worksheet_test.rb

@@ -101,5 +101,27 @@ def test_respond_to_name
       ws = Worksheet.new(StringIO.new, name: 'test')
       assert_equal 'test', ws.name
     end
+
+    def test_with_merge_cells
+      io = StringIO.new
+      ws = Worksheet.new(io)
+      ws << ['foo']
+      ws.add_row ['bar']
+      ws.add_row ['foobar']
+      ws.merge_cells(['A1:B1', 'A3:B3'])
+      ws.close
+      expected = \
+        '<?xml version="1.0" encoding="UTF-8" standalone="yes"?>'"\r\n" \
+        '<worksheet xmlns="http://schemas.openxmlformats.org/spreadsheetml/2006/main"><sheetData>' \
+          '<row r="1"><c r="A1" t="inlineStr"><is><t>foo</t></is></c></row>' \
+          '<row r="2"><c r="A2" t="inlineStr"><is><t>bar</t></is></c></row>' \
+          '<row r="3"><c r="A3" t="inlineStr"><is><t>foobar</t></is></c></row>' \
+        '</sheetData>' \
+        '<mergeCells r="2">' \
+          '<mergeCell ref="A1:B1"/>' \
+          '<mergeCell ref="A3:B3"/>' \
+        '</mergeCells></worksheet>'
+      assert_equal expected, io.string
+    end
   end
 end