# frozen_string_literal: true

require File.join(File.dirname(__FILE__), 'r7_insight', 'host')

require 'logger'

# Rapid7 Insight Platform Ruby Logging functionality
module R7Insight
  def self.new(token, region, options = {})
    opt_local     = options[:local]                     || false
    opt_debug     = options[:debug]                     || false
    opt_ssl       = !options.include?(:ssl) ? true : options[:ssl]
    opt_tag       = options[:tag]                       || false
    opt_log_level = options[:log_level]                 || Logger::DEBUG

    opt_datahub_enabled = options[:datahub_enabled]     || false
    opt_datahub_endpoint = options[:datahub_endpoint]   || ['', 10_000]
    opt_host_id = options[:host_id] || ''
    opt_custom_host = options[:custom_host]             || [false, '']

    opt_udp_port = options[:udp_port]                   || nil
    opt_use_data_endpoint = options[:data_endpoint]     || false

    check_params(token, region, opt_datahub_enabled, opt_udp_port)

    host = R7Insight::Host.new(token, region, opt_local, opt_debug, opt_ssl,
                               opt_datahub_endpoint, opt_host_id, opt_custom_host,
                               opt_udp_port, opt_use_data_endpoint)

    if defined?(ActiveSupport::TaggedLogging) &&  opt_tag
      logger = ActiveSupport::TaggedLogging.new(Logger.new(host))
    elsif defined?(ActiveSupport::Logger)
      logger = ActiveSupport::Logger.new(host)
      logger.formatter = host.formatter if host.respond_to?(:formatter)
    else
      logger = Logger.new(host)
      logger.formatter = host.formatter if host.respond_to?(:formatter)
    end

    logger.level = opt_log_level

    logger
  end

  def self.check_params(token, region, opt_datahub_enabled, opt_udp_port)
    # test Token only when DataHub and UDP are not enabled
    return unless !opt_datahub_enabled && !opt_udp_port

    # Check if the key is valid UUID format
    if (token =~ /\A(urn:uuid:)?[\da-f]{8}-([\da-f]{4}-){3}[\da-f]{12}\z/i).nil?
      puts "\nLE: It appears the R7INSIGHT_TOKEN you entered is invalid!\n"
    end

    puts "\nLE: You need to specify region, such as 'eu'" unless region
  end
end