module Columnist

# File lib/columnist.rb, line 96
def aligned(text, options = {})
    validate_options(options, :align, :width, :color, :bold)

    align = options[:align] || DEFAULTS[:align]
    width = options[:width] || DEFAULTS[:width]
    color = options[:color]
    bold  = options[:bold] || false

    line = case align
               when 'left'
                   text
               when 'right'
                   text.rjust(width)
               when 'center'
                   text.rjust((width - text.size)/2 + text.size)
               else
                   raise ArgumentError
           end

    line = line.send(color) if color
    line = line.send('bold') if bold

    puts line
end

# File lib/columnist.rb, line 22
def capture_output
    $stdout.rewind
    $stdout.read
ensure
    $stdout = STDOUT
end

# File lib/columnist.rb, line 134
def column(text, options = {})
    col = Columnist::Column.new(text, options)
    @row.add(col)
end

# File lib/columnist.rb, line 82
def datetime(options = {})
    validate_options(options, :align, :width, :format, :color, :bold)

    format = options[:format] || '%Y-%m-%d - %l:%M:%S%p'
    align  = options[:align] || DEFAULTS[:align]
    width  = options[:width] || DEFAULTS[:width]

    text = Time.now.strftime(format)

    raise Exception if text.size > width

    aligned(text, :align => align, :width => width, :color => options[:color], :bold => options[:bold])
end

# File lib/columnist.rb, line 60
def footer(options = {})
    section(:footer, options)
end

# File lib/columnist.rb, line 37
def formatter=(type = 'nested')
    name  = type.capitalize + 'Formatter'
    klass = %W{Columnist #{name}}.inject(Kernel) { |s, c| s.const_get(c) }

    # Each formatter is a singleton that responds to #instance
    @formatter = klass.instance
rescue
    raise ArgumentError, 'Invalid formatter specified'
end

# File lib/columnist.rb, line 47
def header(options = {})
    section(:header, options)
end

# File lib/columnist.rb, line 64
def horizontal_rule(options = {})
    validate_options(options, :char, :width, :color, :bold)

    # Got unicode?
    use_char = "\u2501" == 'u2501' ? '-' : "\u2501"

    char  = options[:char].is_a?(String) ? options[:char] : use_char
    width = options[:width] || DEFAULTS[:width]

    aligned(char * width, :width => width, :color => options[:color], :bold => options[:bold])
end

# File lib/columnist.rb, line 56
def progress(override = nil)
    self.formatter.progress(override)
end

# File lib/columnist.rb, line 51
def report(options = {}, &block)
    self.formatter ||= DEFAULTS[:formatter]
    self.formatter.format(options, block)
end

# File lib/columnist.rb, line 33
def restore_output
    $stdout = STDOUT
end

# File lib/columnist.rb, line 127
def row(options = {})
    options[:encoding] ||= @table.encoding
    @row               = Columnist::Row.new(options)
    yield
    @table.add(@row)
end

# File lib/columnist.rb, line 29
def suppress_output
    $stdout = StringIO.new
end

# File lib/columnist.rb, line 121
def table(options = {})
    @table = Columnist::Table.new(options)
    yield
    @table.output
end

# File lib/columnist.rb, line 76
def vertical_spacing(lines = 1)
    puts "\n" * lines
rescue
    raise ArgumentError
end

# File lib/columnist.rb, line 161
def assign_section_properties options
    validate_options(options, :title, :width, :align, :spacing, :timestamp, :rule, :color, :bold)

    title = options[:title] || 'Report'
    width = options[:width] || DEFAULTS[:width]
    align = options[:align] || DEFAULTS[:align]
    lines = options[:spacing] || 1
    color = options[:color]
    bold  = options[:bold] || false

    return [title, width, align, lines, color, bold]
end

# File lib/columnist.rb, line 141
def section(type, options)
    title, width, align, lines, color, bold = assign_section_properties(options)

    # This also ensures that width is a Fixnum
    raise ArgumentError if title.size > width

    if type == :footer
        vertical_spacing(lines)
        horizontal_rule(:char => options[:rule], :width => width, :color => color, :bold => bold) if options[:rule]
    end

    aligned(title, :align => align, :width => width, :color => color, :bold => bold)
    datetime(:align => align, :width => width, :color => color, :bold => bold) if options[:timestamp]

    if type == :header
        horizontal_rule(:char => options[:rule], :width => width, :color => color, :bold => bold) if options[:rule]
        vertical_spacing(lines)
    end
end