Contents

require "metacrunch/file"
require "axlsx"

module Metacrunch
  class File::XLSXDestination

    DEFAULT_OPTIONS = {
      worksheet_title: "My data"
    }

    def initialize(filename, columns, options = {})
      @filename = filename
      @columns = columns
      @options = DEFAULT_OPTIONS.deep_merge(options)

      @package = Axlsx::Package.new
      @workbook = @package.workbook
      @sheet = @workbook.add_worksheet(name: @options[:worksheet_title])

      @sheet.add_row(columns, types: :string)
    end

    def write(data)
      return if data.blank?
      raise ArgumentError, "Data must be an Array" unless data.is_a?(Array)

      if data.first.is_a?(Array)
        data.each do |d|
          @sheet.add_row(d, types: :string)
        end
      else
        @sheet.add_row(data, types: :string)
      end
    end

    def close
      # Make the first row a header
      @sheet.sheet_view.pane do |pane|
        pane.top_left_cell = "A2"
        pane.state = :frozen_split
        pane.y_split = 1
        pane.x_split = 0
        pane.active_pane = :bottom_right
      end

      # Add a filter
      @sheet.auto_filter = @sheet.dimension.sqref

      # Generate file
      @package.serialize(@filename)
    end

  end
end

Version data entries

3 entries across 3 versions & 1 rubygems

Version	Path
metacrunch-file-1.5.0	lib/metacrunch/file/xlsx_destination.rb
metacrunch-file-1.4.0	lib/metacrunch/file/xlsx_destination.rb
metacrunch-file-1.3.1	lib/metacrunch/file/xlsx_destination.rb