aderyabin/sniffer

# frozen_string_literal: true

require 'dry-initializer'
require 'json'
require_relative 'request_policy'

module Sniffer
  # Sniffer data item stores a request info
  class DataItem
    extend Dry::Initializer

    attr_writer :request, :response

    option :request, optional: true
    option :response, optional: true

    def to_h
      {
        request: request&.to_h,
        response: response&.to_h
      }
    end

    def to_log
      return {} unless Sniffer.config.logger

      request.to_log.merge(response.to_log)
    end

    def to_json(*_args)
      to_log.to_json
    end

    def allowed_to_sniff?
      return true unless request

      RequestPolicy.call(request)
    end

    # Basic object for request and response objects
    class HttpObject
      extend Dry::Initializer

      def log_message
        raise NotImplementedError
      end

      def log_settings
        Sniffer.config.log || {}
      end
    end

    # Stores http request data
    class Request < HttpObject
      option :host, optional: true
      option :port, optional: true
      option :query, optional: true
      option :method, optional: true
      option :headers, optional: true
      option :body, optional: true

      attr_writer :host, :port, :query, :method, :headers, :body

      def to_h
        {
          host: host,
          query: query,
          port: port,
          headers: headers,
          body: body&.to_s,
          method: method
        }
      end

      # rubocop:disable Metrics/AbcSize,Metrics/MethodLength
      def to_log
        {}.tap do |hash|
          if log_settings["request_url"]
            hash[:port] = port
            hash[:host] = host
            hash[:query] = query
          end

          if log_settings["request_headers"]
            headers.each do |(k, v)|
              hash[:"rq_#{k.to_s.tr("-", '_').downcase}"] = v
            end
          end

          hash[:method] = method if log_settings["request_method"]
          hash[:request_body] = body.to_s if log_settings["request_body"]
        end
      end
    end
    # rubocop:enable Metrics/AbcSize,Metrics/MethodLength

    # Stores http response data
    class Response < HttpObject
      attr_writer :status, :headers, :body, :timing

      option :status, optional: true
      option :headers, optional: true
      option :body, optional: true
      option :timing, optional: true

      def to_h
        {
          status: status,
          headers: headers,
          body: body&.to_s,
          timing: timing
        }
      end

      # rubocop:disable Metrics/AbcSize
      def to_log
        {}.tap do |hash|
          hash[:status] = status if log_settings["response_status"]

          if log_settings["response_headers"]
            headers.each do |(k, v)|
              hash[:"rs_#{k.to_s.tr("-", '_').downcase}"] = v
            end
          end

          hash[:timing] = timing if log_settings["timing"]
          hash[:response_body] = body.to_s if log_settings["response_body"]
        end
      end
      # rubocop:enable Metrics/AbcSize
    end
  end
end