require 'awesome_print'

module Hieracles
  module Formats
    # format accepting colors
    # for display in the terminal
    class Console < Hieracles::Format
      include Hieracles::Utils

      COLORS = [
        "\e[31m%s\e[0m",
        "\e[32m%s\e[0m",
        "\e[33m%s\e[0m",
        "\e[34m%s\e[0m",
        "\e[35m%s\e[0m",
        "\e[37m%s\e[0m",
        "\e[38m%s\e[0m",
        "\e[36m%s\e[0m",
        "\e[97m%s\e[0m",
        "\e[35;1m%s\e[0m"
      ]

      def initialize(node)
        @colors = {}
        super(node)
      end

      def info(filter)
        build_list(@node.info, @node.notifications, filter)
      end

      def facts(filter)
        build_list(@node.facts, @node.notifications, filter)
      end

      def build_list(hash, notifications, filter)
        back = ''
        if hash.class.name == 'Array'
          hash.each do |v|
            back << "#{v}\n"
          end
        else
          back << build_notifications(notifications) if notifications
          if filter[0]
            hash.select! { |k, v| Regexp.new(filter[0]).match(k.to_s) }
          end
          length = max_key_length(hash) + 2
          title = format(COLORS[8], "%-#{length}s")
          hash.each do |k, v|
            if v.class.name == 'Hash' || v.class.name == 'Array'
              v = v.ai({ indent: 10, raw: true}).strip
            end
            back << format("#{title} %s\n", k, v)
          end
        end
        back
      end

      def build_notifications(notifications)
        back = "\n"
        notifications.each do |v|
          back << format("#{COLORS[9]}\n", "*** #{v.source}: #{v.message} ***")
        end
        back << "\n"
        back
      end

      def files(_)
        @node.files.join("\n") + "\n"
      end

      def paths(_)
        @node.paths.join("\n") + "\n"
      end

      def show_params(without_common, args)
        filter = args[0]
        output = "[-] (merged)\n"
        @node.files(without_common).each_with_index do |f, i|
          output << format("#{COLORS[i]}\n", "[#{i}] #{f}")
          @colors[f] = i
        end
        output << "\n"
        @node.params(without_common).each do |key, v|
          if !filter || Regexp.new(filter).match(key)
            filecolor_index = @colors[v[:file]]
            if v[:overriden]
              output << format(
                "%s #{COLORS[7]} %s\n", "[-]",
                key,
                sanitize(v[:value])
                )
              v[:found_in].each do |val|
                filecolor_index = @colors[val[:file]]
                output << format(
                  "    #{COLORS[8]}\n",
                  "[#{filecolor_index}] #{key} #{val[:value]}"
                  )
              end
            else
              filecolor = COLORS[filecolor_index]
              output << format(
                "#{filecolor} #{COLORS[7]} %s\n", "[#{filecolor_index}]",
                key,
                sanitize(v[:value])
                )
            end
          end
        end
        output
      end

      def build_modules_line(key, value)
        length = max_key_length(@node.modules) + 3
        value_color = '%s'
        value_color = COLORS[0] if /not found/i.match value
        value_color = COLORS[2] if /\(duplicate\)/i.match value
        format("%-#{length}s #{value_color}\n", key, value)
      end

    end
  end
end