Contents

require 'csv_generator/version'

class CsvGenerator
  attr_reader :io

  class << self
    def generate(path, options = {}, &block)
      mode = options[:mode] || 'w'
      permission = options[:permission] || 0644

      File.open(path, mode, permission) do |io|
        with io, &block
      end
    end

    def with(io)
      generator = new(io)

      yield generator if block_given?
      generator
    end
  end

  def initialize(io)
    @io = io
  end

  def <<(row_values)
    io.write row_generator.generate(row_values)
  end

  private

  def row_generator
    @row_generator ||= RowGenerator.new
  end

  class RowGenerator
    def generate(row_values)
      row_values.map { |value| stringify value }.join(field_separator) + line_separator
    end

    private

    def line_separator
      "\r\n"
    end

    def field_separator
      ','
    end

    def quote_charactor
      '"'
    end

    def escaped_quote
      '""'
    end

    def stringify(value)
      case value
      when nil
        ''
      when String
        quote value
      else
        value
      end
    end

    def quote(str)
      escaped_string = str.gsub quote_charactor, escaped_quote
      %("#{escaped_string}")
    end
  end
end

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
csv_generator-0.5.0	lib/csv_generator.rb