# https://github.com/rails/rails/pull/29180
# storing global request info
require 'active_support'

module ContextualizedLogs
  class CurrentContext < ActiveSupport::CurrentAttributes
    # ⚠️ do not use this class to store any controller specific info..

    attribute :custom_attributes,
      :request_uuid, :request_user_agent, :request_origin, :request_referer, :request_xhr, # request
      :current_job_id, :enqueued_jobs_ids, :worker, :worker_args, # sidekiq
      :request_remote_ip, :request_ip, :request_remote_addr, :request_x_forwarded_for, # ips
      :errors,
      :contextualize_model_enabled, # enable model context values
      :context_values, :context_values_count, # context values
      :resource_name # controller_action to correlate APM metrics

    MAX_CONTEXT_VALUES = 100

    class << self
      def context
        # https://docs.hq.com/logs/processing/attributes_naming_convention/#source-code

        data = {}

        data[:resource_name] = resource_name unless resource_name.nil?

        #  normalized
        data[:http] = data[:http] || {}
        data[:http][:referer] = request_referer unless request_referer.nil?
        data[:http][:request_id] = request_uuid unless request_uuid.nil?
        data[:http][:useragent] = request_user_agent unless request_user_agent.nil?
        data[:http][:origin] = request_origin unless request_origin.nil?
        data.delete(:http) if data[:http].empty?

        #  normalized
        data[:network] = data[:network] || { client: {} }
        data[:network][:client][:ip] = request_ip unless request_ip.nil?
        data[:network][:client][:remote_addr] = request_remote_addr unless request_remote_addr.nil?
        data[:network][:client][:remote_ip] = request_remote_ip unless request_remote_ip.nil?
        data[:network][:client][:x_forwarded_for] = request_x_forwarded_for unless request_x_forwarded_for.nil?
        data.delete(:network) if data[:network][:client].empty?

        # eventual error response
        #  normalized
        data[:errors] = errors unless errors.nil?

        # context_values
        unless context_values.nil?
          if context_values.is_a?(Hash) && !context_values.empty?
            data[:context_values] = {}
            context_values.each { |k, v| data[:context_values][k.to_sym] = v }
          end
        end

        unless current_job_id.nil? && worker.nil?
          data[:job] = { id: current_job_id, worker: worker }
          data[:job][:args] = worker_args if worker_args
        end

        data.deep_merge(custom_attributes || {}) 
      end

      def to_json
        attributes.to_json
      end

      def add_context(key, value)
        self.context_values_count ||= 0
        self.context_values_count += 1
        if self.context_values_count >= MAX_CONTEXT_VALUES
          Rails.logger.warn('high number of context values') if self.context_values_count == MAX_CONTEXT_VALUES
          return

        end
        self.context_values ||= {}
        self.context_values[key] ||= []
        unless self.context_values[key].include?(value)
          self.context_values[key] << value
        end
      end

      def from_json(json)
        return unless json

        begin
          values = JSON.parse(json).deep_symbolize_keys
          values.each { |k, v| send("#{k}=", v) }
        rescue
        end
      end

      def add_error(error)
        return if error.nil?

        self.errors ||= []
        self.errors << error
      end
    end
  end
end