openxml-xlsx 0.1.2 → 0.2.0

data/lib/openxml/xlsx/elements/row.rb

@@ -0,0 +1,37 @@
+module OpenXml
+  module Xlsx
+    module Elements
+      class Row
+        attr_reader :worksheet, :number, :height, :hidden, :cells
+
+        def initialize(worksheet, options={})
+          @worksheet = worksheet
+          @number = options.fetch(:number)
+          @height = options[:height]
+          @hidden = options[:hidden]
+          @cells = []
+
+          Array(options[:cells]).each do |attributes|
+            add_cell attributes
+          end
+        end
+
+        def add_cell(attributes)
+          cells.push Xlsx::Elements::Cell.new(self, attributes)
+        end
+
+        def to_xml(xml)
+          attributes = {"r" => number}
+          attributes.merge!("ht" => height, "customHeight" => 1) if height
+          attributes.merge!("hidden" => 1) if hidden
+          xml.row(attributes) do
+            cells.each do |cell|
+              cell.to_xml(xml)
+            end
+          end
+        end
+
+      end
+    end
+  end
+end

data/lib/openxml/xlsx/elements/style.rb

@@ -0,0 +1,29 @@
+module OpenXml
+  module Xlsx
+    module Elements
+      class Style < Struct.new(:format_id, :font_id, :fill_id, :border_id, :alignment)
+
+        def initialize(format_id=0, font_id=0, fill_id=0, border_id=0, alignment=nil)
+          super format_id || 0, font_id || 0, fill_id || 0, border_id || 0, alignment
+        end
+
+        def to_xml(xml)
+          attributes = {
+            numFmtId: format_id,
+            fontId: font_id,
+            fillId: fill_id,
+            borderId: border_id }
+          attributes.merge!(applyNumberFormat: 1) if format_id > 0
+          attributes.merge!(applyFont: 1) if font_id > 0
+          attributes.merge!(applyFill: 1) if fill_id > 0
+          attributes.merge!(applyBorder: 1) if border_id > 0
+          attributes.merge!(applyAlignment: 1) if alignment
+          xml.xf(attributes) do
+            alignment.to_xml(xml) if alignment
+          end
+        end
+
+      end
+    end
+  end
+end

data/lib/openxml/xlsx/elements/table_column.rb

@@ -0,0 +1,8 @@
+module OpenXml
+  module Xlsx
+    module Elements
+      class TableColumn < Struct.new(:name)
+      end
+    end
+  end
+end

data/lib/openxml/xlsx/elements/theme_color.rb

@@ -0,0 +1,13 @@
+module OpenXml
+  module Xlsx
+    module Elements
+      class ThemeColor < Struct.new(:theme, :tint)
+
+        def to_xml(name, xml)
+          xml.public_send(name, theme: theme, tint: tint)
+        end
+
+      end
+    end
+  end
+end

data/lib/openxml/xlsx/package.rb

@@ -0,0 +1,52 @@
+require "openxml/package"
+
+module OpenXml
+  module Xlsx
+    class Package < OpenXml::Package
+      attr_reader :xl_rels,
+                  :shared_strings,
+                  :stylesheet,
+                  :workbook
+
+
+      content_types do
+        override "/xl/workbook.xml", TYPE_WORKBOOK
+        override "/xl/worksheets/sheet1.xml", TYPE_WORKSHEET
+        override "/xl/sharedStrings.xml", TYPE_SHARED_STRINGS
+        override "/xl/styles.xml", TYPE_STYLES
+      end
+
+
+      def initialize
+        super
+        rels.add_relationship REL_DOCUMENT, "xl/workbook.xml"
+
+        @xl_rels = OpenXml::Parts::Rels.new([
+          { "Type" => REL_SHARED_STRINGS, "Target" => "sharedStrings.xml" },
+          { "Type" => REL_STYLES, "Target" => "styles.xml" }
+        ])
+        @shared_strings = Xlsx::Parts::SharedStrings.new
+        @stylesheet = Xlsx::Parts::Stylesheet.new
+        @workbook = Xlsx::Parts::Workbook.new(self)
+
+        # docProps/app.xml
+        # docProps/core.xml
+        add_part "xl/_rels/workbook.xml.rels", xl_rels
+        # xl/calcChain.xml
+        add_part "xl/sharedStrings.xml", shared_strings
+        add_part "xl/styles.xml", stylesheet
+        # xl/theme/theme1.xml
+        add_part "xl/workbook.xml", workbook
+      end
+
+      def string_ref(string)
+        shared_strings.reference_of(string)
+      end
+
+      def style_ref(style)
+        stylesheet.reference_of(style)
+      end
+
+    end
+  end
+end

data/lib/{xlsx → openxml/xlsx}/parts.rb

@@ -1,5 +1,7 @@
-module Xlsx
-  module Parts
+module OpenXml
+  module Xlsx
+    module Parts
+    end
   end
 end
 

data/lib/openxml/xlsx/parts/shared_strings.rb

@@ -0,0 +1,28 @@
+module OpenXml
+  module Xlsx
+    module Parts
+      class SharedStrings < OpenXml::Part
+        attr_reader :strings
+
+        def initialize
+          @strings = Hash.new { |hash, key| hash[key] = hash.length }
+        end
+
+        def reference_of(string)
+          strings[string]
+        end
+
+        def to_xml
+          build_standalone_xml do |xml|
+            xml.sst(xmlns: "http://schemas.openxmlformats.org/spreadsheetml/2006/main", uniqueCount: strings.length) do
+              strings.each do |string, i|
+                xml.si { xml.t(string) }
+              end
+            end
+          end
+        end
+
+      end
+    end
+  end
+end

data/lib/openxml/xlsx/parts/stylesheet.rb

@@ -0,0 +1,80 @@
+module OpenXml
+  module Xlsx
+    module Parts
+      class Stylesheet < OpenXml::Part
+        include Xlsx::Elements
+
+        attr_reader :formats, :fonts, :fills, :borders, :styles
+
+        def initialize
+          @formats = []
+          @fonts = [Font.new("Calibri", 12)]
+          @fills = [PatternFill.new("none"), PatternFill.new("gray125")]
+          @borders = [Border.new]
+          @styles = [Style.new]
+        end
+
+        def reference_of(options={})
+          case format = options[:format]
+          when NumberFormat
+            options[:format_id] = Xlsx.index!(formats, format) + NUMBER_FORMAT_START_ID
+          when ImpliedNumberFormat
+            options[:format_id] = format.id
+          end
+          options[:font_id] = Xlsx.index!(fonts, options[:font]) if options.key? :font
+          options[:fill_id] = Xlsx.index!(fills, options[:fill]) if options.key? :fill
+          options[:border_id] = Xlsx.index!(borders, options[:border]) if options.key? :border
+
+          style = Style.new(
+            options.fetch(:format_id, 0),
+            options.fetch(:font_id, 0),
+            options.fetch(:fill_id, 0),
+            options.fetch(:border_id, 0),
+            options.fetch(:alignment, nil))
+
+          Xlsx.index!(styles, style)
+        end
+
+        def to_xml
+          build_standalone_xml do |xml|
+            xml.styleSheet(xmlns: "http://schemas.openxmlformats.org/spreadsheetml/2006/main", "xmlns:mc" => "http://schemas.openxmlformats.org/markup-compatibility/2006") do
+              xml.numFmts(count: formats.length) do
+                formats.each_with_index do |format, index|
+                  format.to_xml(index + NUMBER_FORMAT_START_ID, xml)
+                end
+              end
+
+              xml.fonts(count: fonts.length) do
+                fonts.each do |font|
+                  font.to_xml(xml)
+                end
+              end
+
+              xml.fills(count: fills.length) do
+                fills.each do |fill|
+                  fill.to_xml(xml)
+                end
+              end
+
+              xml.borders(count: borders.length) do
+                borders.each do |border|
+                  border.to_xml(xml)
+                end
+              end
+
+              xml.cellXfs(count: styles.length) do
+                styles.each do |style|
+                  style.to_xml(xml)
+                end
+              end
+            end
+          end
+        end
+
+      private
+        NUMBER_FORMAT_START_ID = 165.freeze
+
+      end
+    end
+  end
+end

data/lib/openxml/xlsx/parts/table.rb

@@ -0,0 +1,36 @@
+module OpenXml
+  module Xlsx
+    module Parts
+      class Table < OpenXml::Part
+        attr_reader :id, :name, :ref, :columns
+
+        def initialize(id, name, ref, columns)
+          @id = id
+          @name = name
+          @ref = ref
+          @columns = columns
+        end
+
+        def filename
+          "table#{id}.xml"
+        end
+
+        def to_xml
+          build_standalone_xml do |xml|
+            xml.table(xmlns: "http://schemas.openxmlformats.org/spreadsheetml/2006/main",
+              id: id, name: name, displayName: name, ref: ref, totalsRowShown: 0) do
+              xml.autoFilter ref: ref
+              xml.tableColumns(count: columns.length) do
+                columns.each_with_index do |column, index|
+                  xml.tableColumn(id: index + 1, name: column.name)
+                end
+              end
+              xml.tableStyleInfo(name: "TableStyleLight6", showFirstColumn: 0, showLastColumn: 0, showRowStripes: 1, showColumnStripes: 0)
+            end
+          end
+        end
+
+      end
+    end
+  end
+end

data/lib/openxml/xlsx/parts/workbook.rb

@@ -0,0 +1,71 @@
+module OpenXml
+  module Xlsx
+    module Parts
+      class Workbook < OpenXml::Part
+        attr_reader :package, :worksheets, :tables, :defined_names
+
+        def initialize(package)
+          @package = package
+          @worksheets = []
+          @tables = []
+          @defined_names =[]
+          add_worksheet
+        end
+
+        def add_worksheet
+          worksheet = Worksheet.new(self, worksheets.length + 1)
+          package.xl_rels.add_relationship(
+            REL_WORKSHEET,
+            "worksheets/sheet#{worksheet.index}.xml",
+            "rId#{worksheet.index}")
+          package.add_part "xl/worksheets/_rels/sheet#{worksheet.index}.xml.rels", worksheet.rels
+          package.add_part "xl/worksheets/sheet#{worksheet.index}.xml", worksheet
+          worksheets.push worksheet
+        end
+
+        def add_table(table)
+          package.content_types.add_override "/xl/tables/#{table.filename}", TYPE_TABLE
+          package.add_part "xl/tables/#{table.filename}", table
+          tables.push table
+        end
+
+        def add_defined_names(*defined_names)
+          defined_names.flatten.each do |attributes|
+            add_defined_name attributes
+          end
+        end
+
+        def add_defined_name(attributes)
+          defined_names.push Xlsx::Elements::DefinedName.new(attributes[:name], attributes[:formula])
+        end
+
+        def to_xml
+          build_standalone_xml do |xml|
+            xml.workbook(root_namespaces) {
+              xml.bookViews {
+                xml.workbookView
+              }
+              xml.sheets { worksheets.each { |worksheet|
+                xml.sheet(
+                  "name" => worksheet.name,
+                  "sheetId" => worksheet.index,
+                  "r:id" => "rId#{worksheet.index}")
+              } }
+              xml.definedNames do
+                defined_names.each { |defined_name| defined_name.to_xml(xml) }
+              end if defined_names.any?
+            }
+          end
+        end
+
+      private
+
+        def root_namespaces
+          { "xmlns" => "http://schemas.openxmlformats.org/spreadsheetml/2006/main",
+            "xmlns:r" => 'http://schemas.openxmlformats.org/officeDocument/2006/relationships' }
+        end
+
+      end
+    end
+  end
+end

data/lib/openxml/xlsx/parts/worksheet.rb

@@ -0,0 +1,90 @@
+module OpenXml
+  module Xlsx
+    module Parts
+      class Worksheet < OpenXml::Part
+        attr_reader :workbook, :index, :rows, :tables, :rels, :cell_ranges
+
+        def initialize(workbook, index)
+          @workbook = workbook
+          @index = index
+          @rows = []
+          @tables = []
+          @cell_ranges = []
+          @rels = OpenXml::Parts::Rels.new
+          @column_widths = {}
+        end
+
+        def column_widths(*args)
+          return @column_widths if args.none?
+          @column_widths = args.first
+        end
+
+        def add_rows(*rows)
+          rows.flatten.each do |attributes|
+            add_row attributes
+          end
+        end
+
+        def add_row(attributes)
+          rows.push Xlsx::Elements::Row.new(self, attributes)
+        end
+
+        def merge_cells(*ranges)
+          ranges.each { |range| cell_ranges.push range }
+        end
+
+        def add_table(id, name, ref, columns)
+          table = Xlsx::Parts::Table.new(id, name, ref, columns)
+          rels.add_relationship(REL_TABLE, "../tables/#{table.filename}")
+          workbook.add_table table
+        end
+
+        def to_xml
+          build_standalone_xml do |xml|
+            xml.worksheet(root_namespaces) do
+              xml.sheetViews do
+                xml.sheetView(showGridLines: 0, tabSelected: 1, workbookViewId: 0)
+              end
+              xml.sheetFormatPr(baseColWidth: 10, defaultColWidth: 13.33203125, defaultRowHeight: 20, customHeight: 1)
+              xml.cols do
+                column_widths.each do |column, width|
+                  xml.col(min: column, max: column, width: width, customWidth: 1)
+                end
+              end if column_widths.any?
+              xml.sheetData do
+                rows.each { |row| row.to_xml(xml) }
+              end
+              xml.mergeCells(count: merge_cells.size) do
+                cell_ranges.each { |range| xml.mergeCell ref: range }
+              end if cell_ranges.any?
+              xml.pageMargins(left: 0.75, right: 0.75, top: 1, bottom: 1, header: 0.5, footer: 0.5)
+              xml.pageSetup(orientation: "portrait", horizontalDpi: 4294967292, verticalDpi: 4294967292)
+              xml.tableParts(count: tables.count) do
+                tables.each do |rel|
+                  xml.tablePart("r:id" => rel.id)
+                end
+              end if tables.any?
+            end
+          end
+        end
+
+        def name
+          "Sheet#{index}"
+        end
+
+      private
+
+        def root_namespaces
+          { "xmlns" => "http://schemas.openxmlformats.org/spreadsheetml/2006/main",
+            "xmlns:r" => "http://schemas.openxmlformats.org/officeDocument/2006/relationships",
+            "xmlns:mc" => "http://schemas.openxmlformats.org/markup-compatibility/2006" }
+        end
+
+        def tables
+          rels.select { |rel| rel.type == REL_TABLE }
+        end
+
+      end
+    end
+  end
+end