# frozen_string_literal: true

module Zypper
  #
  # Collection of util classes.
  #
  module Onlinesearch
    #
    # String class patch.
    #
    class ::String
      def black;          "\033[30m#{self}\033[0m" end
      def red;            "\033[31m#{self}\033[0m" end
      def green;          "\033[32m#{self}\033[0m" end
      def yellow;         "\033[33m#{self}\033[0m" end
      def blue;           "\033[34m#{self}\033[0m" end
      def magenta;        "\033[35m#{self}\033[0m" end
      def cyan;           "\033[36m#{self}\033[0m" end
      def gray;           "\033[37m#{self}\033[0m" end
      def bg_black;       "\033[40m#{self}\0330m"  end
      def bg_red;         "\033[41m#{self}\033[0m" end
      def bg_green;       "\033[42m#{self}\033[0m" end
      def bg_brown;       "\033[43m#{self}\033[0m" end
      def bg_blue;        "\033[44m#{self}\033[0m" end
      def bg_magenta;     "\033[45m#{self}\033[0m" end
      def bg_cyan;        "\033[46m#{self}\033[0m" end
      def bg_gray;        "\033[47m#{self}\033[0m" end
      def bold;           "\033[1m#{self}\033[22m" end
      def reverse_color;  "\033[7m#{self}\033[27m" end
      def cr;             "\r#{self}" end
      def clean;          "\e[K#{self}" end
      def new_line;       "\n#{self}" end
      def none;           self end
    end

    #
    # Float class patch.
    #
    class ::Float
      def to_human
        conv = {
          1024 => "B",
          1024 * 1024 => "KB",
          1024 * 1024 * 1024 => "MB",
          1024 * 1024 * 1024 * 1024 => "GB",
          1024 * 1024 * 1024 * 1024 * 1024 => "TB",
          1024 * 1024 * 1024 * 1024 * 1024 * 1024 => "PB",
          1024 * 1024 * 1024 * 1024 * 1024 * 1024 * 1024 => "EB"
        }
        conv.keys.sort.each do |mult|
          next if self >= mult

          suffix = conv[mult]
          return format("%<fnum>.2f %<suffix>s", fnum: self / (mult / 1024), suffix: suffix)
        end
      end
    end

    #
    # Array class patch.
    #
    class ::Array
      def max_column(field)
        max_by { |x| x[field].to_s.length }[field].to_s.length
      end
    end

    #
    # Default error code.
    #
    class ::StandardError
      def error_code
        1
      end
    end

    #
    # Color the error message.
    #
    class Messages
      def self.error(err)
        if err.instance_of? String
          puts " #{"[E]".bold.red} #{err}"
        else
          warn "Error! ".bold.red + err.message
        end
      end
    end

    #
    # Too short query string.
    #
    class QueryStringTooShort < StandardError
      def initialize(query)
        super "The query string '#{query}' is too short, be sure to use more than 3 characters."
      end

      def error_code
        2
      end
    end

    #
    # Too many redirections.
    #
    class TooManyRedirections < StandardError
      def initialize(url)
        super "#{url} generates too many redirections!"
      end
    end

    #
    # Invalid engine request.
    #
    class InvalidEngine < StandardError
      def initialize(engine)
        super "#{engine} is not a valid engine!"
      end

      def error_code
        4
      end
    end

    #
    # No item number provided.
    #
    class MissingItemNumber < StandardError
      def initialize
        super "No item number has been provided!"
      end
    end

    #
    # Empty cache folder.
    #
    class EmptyCache < StandardError
      def initialize
        super "The cache folder is already empty!"
      end
    end

    #
    # No internet connection.
    #
    class NoConnection < StandardError
      def initialize
        super "Internet connection has some trouble"
      end

      def error_code
        6
      end
    end

    #
    # Ctrl + C message error.
    #
    class Interruption < StandardError
      def initialize
        super "Ok ok... Exiting!"
      end
    end

    Signal.trap("INT") { raise Interruption }
    Signal.trap("TERM") { raise Interruption }
  end
end