# frozen_string_literal: true

require 'rest-client'

class MetricsHelper
  @@obs_endpoint_exists = false

  def initialize(logger, metric_prefix, jpd_url, username, apikey, token, common_jpd, verify_ssl, request_timeout)
    @logger = logger
    @metric_prefix = metric_prefix
    @jpd_url = jpd_url
    @username = username
    @apikey = apikey
    @token = token
    @common_jpd = common_jpd
    @verify_ssl = verify_ssl
    @request_timeout = request_timeout
  end

  def get_metrics
    @logger.debug("Get metrics started")
    url = nil
    url = case @metric_prefix
          when 'jfrog.artifactory'
            "#{@jpd_url}/artifactory/api/v1/metrics"
          when 'jfrog.xray'
            "#{@jpd_url}/xray/api/v1/metrics"
          else
            "#{@jpd_url}/artifactory/api/v1/metrics"
          end
    
    @logger.info("Executing #{@metric_prefix} metrics collection from: #{url}")
    if !@token.nil? && @token != ''
      execute_rest_call(url, @username, nil, @token, true, @verify_ssl, @request_timeout)
    elsif !@apikey.nil? && @apikey != ''
      execute_rest_call(url, @username, @apikey, nil, false, @verify_ssl, @request_timeout)
    end
    @logger.debug("Get metrics finished")
  end

  def get_additional_metrics
    @logger.info("Aadditional metrics collection started")
    if (@metric_prefix == 'jfrog.artifactory' || @common_jpd == false) && !@token.nil? && @token != ''
      url = "#{@jpd_url}/observability/api/v1/metrics"
      @logger.info("Collecting additional metrics from: #{url}")
      check_endpoint(url, @token, @verify_ssl, @request_timeout) if @@obs_endpoint_exists == nil? || !@@obs_endpoint_exists
      execute_rest_call(url, @username, nil, @token, true, @verify_ssl, @request_timeout) if @@obs_endpoint_exists
    end
    @logger.info("Aadditional metrics collection finished")
  end

  def check_endpoint(url, token, verify_ssl, request_timeout)
    @logger.debug("Checking connectivity to endpoint: #{url} started")
    request = RestClient::Request.new(
      method: :get,
      url: url,
      headers: { Authorization: "Bearer #{token}"},
      verify_ssl: verify_ssl,
      timeout: request_timeout
    )

    request.execute do |response, request, result|
      if response.code == 200
        @@obs_endpoint_exists = true
        @logger.info("#{url} exists: #{@@obs_endpoint_exists}. Storing the result for next executions")
      else
        @@obs_endpoint_exists = false 
        @logger.info("Cannot verify endpoint. Skipping metrics collection from  #{url}. Received response code: #{response.code}, Response body:\n#{response.body}")
      end
    end
    @logger.debug("Checking connectivity to endpoint: #{url} finished")
  end

  def execute_rest_call(url, user, password, token, use_token, verify_ssl, request_timeout)
    @logger.debug("Rest call to fetch metrics started")
    request = if use_token == true
      @logger.debug("Using token for authentication")
      RestClient::Request.new(
        method: :get,
        url: url,
        headers: { Authorization: "Bearer #{token}" },
        verify_ssl: verify_ssl,
        timeout: request_timeout
      )
    else
      @logger.debug("Using apiKey for authentication")
      RestClient::Request.new(
        method: :get,
        url: url,
        user: user,
        password: password,
        verify_ssl: verify_ssl,
        timeout: request_timeout
      )
    end

    request.execute do |response, request, result|
      @logger.debug("Recieved response body: #{response.body} when fetching metrics")
      case response.code
      when 200
        @logger.info("#{@metric_prefix} metrics were successfully collected from url: #{url}")
        return response.body
      else
        @logger.info("Cannot fetch #{@metric_prefix} metrics from url: #{url}. Received response code: #{response.code}, Response body:\n#{response.body}")
        raise "Unexpected response code: #{response.code} when calling #{url}"
      end
    end
  end

end