# -*- coding: utf-8 -*-
module GrooveDl
  # Downloader Class
  class Displayer
    attr_reader :result, :type

    ##
    # Initialize Displayer
    #
    # @param [Array] result The result from the search
    # @param [String] type The search type
    #
    # @return [Nil]
    #
    def initialize(result, type)
      @result = result
      @type = type
    end

    ##
    # Display prompt to choose songs or playlists.
    #
    def render
      table = Terminal::Table.new(headings: headers, title: @type)
      @result.each do |data|
        add_row(table, data)
      end

      puts table.to_s
    end

    def headers
      return %w(Id Album Artist Song) if @type == 'Songs'
      return %w(Id Name Author NumSongs) if @type == 'Playlists'
    end

    ##
    # Add row into table
    #
    # @param [Terminal::Table] table Table in which row will be added
    # @param [Array] result The result from the search
    #
    # @return [Nil]
    #
    def add_row(table, data)
      table.add_row([data.id,
                     data.name,
                     data.username,
                     data.num_songs]) if @type == 'Playlists'
      table.add_row([data.id,
                     data.album,
                     data.artist,
                     data.name]) if @type == 'Songs'
    end
  end
end