#==============================================================================
# Copyright 2020 William McCumstie
#
# Redistribution and use in source and binary forms, with or without
# modification, are permitted provided that the following conditions are met:
#
# 1. Redistributions of source code must retain the above copyright notice,
# this list of conditions and the following disclaimer.
#
# 2. Redistributions in binary form must reproduce the above copyright notice,
# this list of conditions and the following disclaimer in the documentation
# and/or other materials provided with the distribution.
#
# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
# AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
# IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
# ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE
# LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
# CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
# SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
# INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
# CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
# ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
# POSSIBILITY OF SUCH DAMAGE.
#==============================================================================

require 'tty-table'

module OutputMode
  module Outputs
    class Tabulated < Output
      # @!attribute [r] renderer
      #   @return [Symbol] the renderer type, see: https://github.com/piotrmurach/tty-table#32-renderer
      # @!attribute [r] block
      #   @return [#call] an optional block of code that configures the renderer
      # @!attribute [r] colorize
      #   @return [Boolean] enable or disabled the colorization
      # @!attribute [r] header_color
      #   @return An optional header color or array of colors
      # @!attribute [r] row_color
      #   @return An optional data color or array of colors
      attr_reader :renderer, :default, :block, :yes, :no,
                  :header_color, :row_color, :colorize

      # @return [Hash] additional options to +TTY::Table+ renderer
      # @see https://github.com/piotrmurach/tty-table#33-options
      def config; super; end

      # @overload initialize(*procs, renderer: nil, **config)
      #   @param [Array] *procs see {OutputMode::Outputs::Base#initialize}
      #   @param [Symbol] :renderer override the default renderer
      #   @param [Hash] **config additional options to the renderer
      #   @yieldparam tty_table_renderer [TTY::Table::Renderer::Base] optional access the underlining TTY::Table renderer
      def initialize(*procs,
                     renderer: :unicode,
                     colorize: false,
                     header_color: nil,
                     row_color: nil,
                     **config,
                     &block)
        @renderer =  renderer
        @block = block
        @header_color = header_color
        @row_color = row_color
        @colorize = colorize
        super(*procs, **config)
      end

      # Implements the render method using +TTY::Table+
      # @see OutputMode::Outputs::Base#render
      # @see https://github.com/piotrmurach/tty-table
      def render(*data)
        table = TTY::Table.new header: processed_header
        data.each { |d| table << process_row(d) }
        table.render(renderer, **config, &block) || ''
      end

      private

      def has_header?
        callables.any? { |c| c.config.key?(:header) }
      end

      # Colorizes the header when requested
      def processed_header
        return nil unless has_header?
        callables.map do |callable|
          header = callable.config.fetch(:header, '')
          color = callable.config.fetch(:header_color, nil) || header_color
          case color
          when nil
            header.to_s
          when Array
            pastel.decorate(header.to_s, *color)
          else
            pastel.decorate(header.to_s, color)
          end
        end
      end

      # Colorizes the row when requested
      def process_row(model)
        callables.map do |callable|
          d = callable.generator(self).call(model)
          color = callable.config[:row_color] || row_color
          case color
          when NilClass
            d.to_s
          when Array
            pastel.decorate(d.to_s, *color)
          else
            pastel.decorate(d.to_s, color)
          end
        end
      end

      def pastel
        @pastel ||= Pastel::Color.new(enabled: colorize)
      end
    end
  end
end