hoss-agent 1.0.7

data/lib/hoss/context/request/url.rb

@@ -0,0 +1,59 @@
+# Licensed to Elasticsearch B.V. under one or more contributor
+# license agreements. See the NOTICE file distributed with
+# this work for additional information regarding copyright
+# ownership. Elasticsearch B.V. licenses this file to you under
+# the Apache License, Version 2.0 (the "License"); you may
+# not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#   http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing,
+# software distributed under the License is distributed on an
+# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+# KIND, either express or implied.  See the License for the
+# specific language governing permissions and limitations
+# under the License.
+
+# frozen_string_literal: true
+
+module Hoss
+  # @api private
+  class Context
+    # @api private
+    class Request
+      # @api private
+      class Url
+        SKIPPED_PORTS = {
+          'http' => 80,
+          'https' => 443
+        }.freeze
+
+        def initialize(req)
+          @protocol = req.scheme
+          @hostname = req.host
+          @port = req.port.to_s
+          @pathname = req.path
+          @search = req.query_string
+          @hash = nil
+          @full = build_full_url req
+        end
+
+        attr_reader :protocol, :hostname, :port, :pathname, :search, :hash,
+          :full
+
+        private
+
+        def build_full_url(req)
+          url = "#{req.scheme}://#{req.host}"
+
+          if req.port != SKIPPED_PORTS[req.scheme]
+            url += ":#{req.port}"
+          end
+
+          url + req.fullpath
+        end
+      end
+    end
+  end
+end

data/lib/hoss/context/response.rb

@@ -0,0 +1,47 @@
+# Licensed to Elasticsearch B.V. under one or more contributor
+# license agreements. See the NOTICE file distributed with
+# this work for additional information regarding copyright
+# ownership. Elasticsearch B.V. licenses this file to you under
+# the Apache License, Version 2.0 (the "License"); you may
+# not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#   http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing,
+# software distributed under the License is distributed on an
+# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+# KIND, either express or implied.  See the License for the
+# specific language governing permissions and limitations
+# under the License.
+
+# frozen_string_literal: true
+
+module Hoss
+  class Context
+    # @api private
+    class Response
+      def initialize(
+        status_code,
+        headers: {},
+        headers_sent: true,
+        finished: true
+      )
+        @status_code = status_code
+        @headers_sent = headers_sent
+        @finished = finished
+
+        self.headers = headers
+      end
+
+      attr_accessor :status_code, :headers_sent, :finished
+      attr_reader :headers
+
+      def headers=(headers)
+        @headers = headers&.each_with_object({}) do |(k, v), hsh|
+          hsh[k] = v.to_s
+        end
+      end
+    end
+  end
+end

data/lib/hoss/context/user.rb

@@ -0,0 +1,59 @@
+# Licensed to Elasticsearch B.V. under one or more contributor
+# license agreements. See the NOTICE file distributed with
+# this work for additional information regarding copyright
+# ownership. Elasticsearch B.V. licenses this file to you under
+# the Apache License, Version 2.0 (the "License"); you may
+# not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#   http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing,
+# software distributed under the License is distributed on an
+# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+# KIND, either express or implied.  See the License for the
+# specific language governing permissions and limitations
+# under the License.
+
+# frozen_string_literal: true
+
+module Hoss
+  class Context
+    # @api private
+    class User
+      def initialize(id: nil, email: nil, username: nil)
+        @id = id
+        @email = email
+        @username = username
+      end
+
+      def self.infer(config, record)
+        return unless record
+
+        new(
+          id: safe_get(record, config.current_user_id_method)&.to_s,
+          email: safe_get(record, config.current_user_email_method),
+          username: safe_get(record, config.current_user_username_method)
+        )
+      end
+
+      attr_accessor :id, :email, :username
+
+      def empty?
+        !id && !email && !username
+      end
+
+      def any?
+        !empty?
+      end
+
+      class << self
+        private
+
+        def safe_get(record, method_name)
+          record.respond_to?(method_name) ? record.send(method_name) : nil
+        end
+      end
+    end
+  end
+end

data/lib/hoss/context_builder.rb

@@ -0,0 +1,112 @@
+# Licensed to Elasticsearch B.V. under one or more contributor
+# license agreements. See the NOTICE file distributed with
+# this work for additional information regarding copyright
+# ownership. Elasticsearch B.V. licenses this file to you under
+# the Apache License, Version 2.0 (the "License"); you may
+# not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#   http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing,
+# software distributed under the License is distributed on an
+# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+# KIND, either express or implied.  See the License for the
+# specific language governing permissions and limitations
+# under the License.
+
+# frozen_string_literal: true
+
+module Hoss
+  # @api private
+  class ContextBuilder
+    MAX_BODY_LENGTH = 2048
+    SKIPPED = '[SKIPPED]'
+
+    def initialize(config)
+      @config = config
+    end
+
+    attr_reader :config
+
+    def build(rack_env:, for_type:)
+      Context.new.tap do |context|
+        apply_to_request(context, rack_env: rack_env, for_type: for_type)
+      end
+    end
+
+    private
+
+    def apply_to_request(context, rack_env:, for_type:)
+      req = rails_req?(rack_env) ? rack_env : Rack::Request.new(rack_env)
+
+      context.request = Context::Request.new unless context.request
+      request = context.request
+
+      request.socket = Context::Request::Socket.new(req)
+      request.http_version = build_http_version rack_env
+      request.method = req.request_method
+      request.url = Context::Request::Url.new(req)
+
+      request.body = should_capture_body?(for_type) ? get_body(req) : SKIPPED
+
+      headers, env = get_headers_and_env(rack_env)
+      request.headers = headers if config.capture_headers?
+      request.env = env if config.capture_env?
+
+      request.cookies = req.cookies.dup
+
+      context
+    end
+
+    def should_capture_body?(for_type)
+      option = config.capture_body
+
+      return true if option == 'all'
+      return true if option == 'transactions' && for_type == :transaction
+      return true if option == 'errors' && for_type == :error
+
+      false
+    end
+
+    def get_body(req)
+      case req.media_type
+      when 'application/x-www-form-urlencoded', 'multipart/form-data'
+        req.POST.dup
+      else
+        body = req.body.read
+        req.body.rewind
+        body.byteslice(0, MAX_BODY_LENGTH).force_encoding('utf-8')
+      end
+    end
+
+    def rails_req?(env)
+      defined?(ActionDispatch::Request) && env.is_a?(ActionDispatch::Request)
+    end
+
+    def get_headers_and_env(rack_env)
+      # In Rails < 5 ActionDispatch::Request inherits from Hash
+      headers =
+        rack_env.respond_to?(:headers) ? rack_env.headers : rack_env
+
+      headers.each_with_object([{}, {}]) do |(key, value), (http, env)|
+        next unless key == key.upcase
+
+        if key.start_with?('HTTP_')
+          http[camel_key(key)] = value
+        else
+          env[key] = value
+        end
+      end
+    end
+
+    def camel_key(key)
+      key.gsub(/^HTTP_/, '').split('_').map(&:capitalize).join('-')
+    end
+
+    def build_http_version(rack_env)
+      return unless (http_version = rack_env['HTTP_VERSION'])
+      http_version.gsub(%r{HTTP/}, '')
+    end
+  end
+end

data/lib/hoss/deprecations.rb

@@ -0,0 +1,39 @@
+# Licensed to Elasticsearch B.V. under one or more contributor
+# license agreements. See the NOTICE file distributed with
+# this work for additional information regarding copyright
+# ownership. Elasticsearch B.V. licenses this file to you under
+# the Apache License, Version 2.0 (the "License"); you may
+# not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#   http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing,
+# software distributed under the License is distributed on an
+# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+# KIND, either express or implied.  See the License for the
+# specific language governing permissions and limitations
+# under the License.
+
+# frozen_string_literal: true
+
+module Hoss
+  # @api private
+  module Deprecations
+    def deprecate(name, replacement = nil)
+      alias_name = "#{name.to_s.chomp('=')}__deprecated_"
+      alias_name += '=' if name.to_s.end_with?('=')
+
+      class_eval <<-RUBY, __FILE__, __LINE__ + 1
+        alias :"#{alias_name}" :"#{name}"
+
+        def #{name}(*args, &block)
+          warn "[HOSS] [DEPRECATED] `#{name}' is being removed. " \
+               "#{replacement && "See `#{replacement}'."}" \
+               "\nCalled from \#{caller.first}"
+          send("#{alias_name}", *args, &block)
+        end
+      RUBY
+    end
+  end
+end

data/lib/hoss/error.rb

@@ -0,0 +1,49 @@
+# Licensed to Elasticsearch B.V. under one or more contributor
+# license agreements. See the NOTICE file distributed with
+# this work for additional information regarding copyright
+# ownership. Elasticsearch B.V. licenses this file to you under
+# the Apache License, Version 2.0 (the "License"); you may
+# not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#   http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing,
+# software distributed under the License is distributed on an
+# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+# KIND, either express or implied.  See the License for the
+# specific language governing permissions and limitations
+# under the License.
+
+# frozen_string_literal: true
+
+require 'hoss/stacktrace'
+require 'hoss/context'
+require 'hoss/error/exception'
+require 'hoss/error/log'
+
+module Hoss
+  # @api private
+  class Error
+    def initialize(culprit: nil, context: nil)
+      @id = SecureRandom.hex(16)
+      @culprit = culprit
+      @timestamp = Util.micros
+      @context = context
+    end
+
+    attr_accessor :id, :culprit, :exception, :log, :transaction_id,
+      :transaction, :context, :parent_id, :trace_id
+    attr_reader :timestamp
+
+    def inspect
+      "<Hoss::Error id:#{id}" \
+        " culprit:#{culprit}" \
+        " timestamp:#{timestamp}" \
+        " transaction_id:#{transaction_id}" \
+        " trace_id:#{trace_id}" \
+        " exception:#{exception.inspect}" \
+        '>'
+    end
+  end
+end

data/lib/hoss/error/exception.rb

@@ -0,0 +1,70 @@
+# Licensed to Elasticsearch B.V. under one or more contributor
+# license agreements. See the NOTICE file distributed with
+# this work for additional information regarding copyright
+# ownership. Elasticsearch B.V. licenses this file to you under
+# the Apache License, Version 2.0 (the "License"); you may
+# not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#   http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing,
+# software distributed under the License is distributed on an
+# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+# KIND, either express or implied.  See the License for the
+# specific language governing permissions and limitations
+# under the License.
+
+# frozen_string_literal: true
+
+module Hoss
+  class Error
+    # @api private
+    class Exception
+      MOD_SPLIT = '::'
+
+      def initialize(attrs = nil)
+        return unless attrs
+
+        attrs.each do |key, val|
+          send(:"#{key}=", val)
+        end
+      end
+
+      def self.from_exception(exception, **attrs)
+        new({
+          message: exception.message.to_s,
+          type: exception.class.to_s,
+          module: format_module(exception),
+          cause: exception.cause && Exception.from_exception(exception.cause)
+        }.merge(attrs))
+      end
+
+      attr_accessor(
+        :attributes,
+        :code,
+        :handled,
+        :message,
+        :module,
+        :stacktrace,
+        :type,
+        :cause
+      )
+
+      def inspect
+        '<Hoss::Error::Exception' \
+          " type:#{type}" \
+          " message:#{message}" \
+          '>'
+      end
+
+      class << self
+        private
+
+        def format_module(exception)
+          exception.class.to_s.split(MOD_SPLIT)[0...-1].join(MOD_SPLIT)
+        end
+      end
+    end
+  end
+end

data/lib/hoss/error/log.rb

@@ -0,0 +1,41 @@
+# Licensed to Elasticsearch B.V. under one or more contributor
+# license agreements. See the NOTICE file distributed with
+# this work for additional information regarding copyright
+# ownership. Elasticsearch B.V. licenses this file to you under
+# the Apache License, Version 2.0 (the "License"); you may
+# not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#   http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing,
+# software distributed under the License is distributed on an
+# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+# KIND, either express or implied.  See the License for the
+# specific language governing permissions and limitations
+# under the License.
+
+# frozen_string_literal: true
+
+module Hoss
+  class Error
+    # @api private
+    class Log
+      def initialize(message, attrs = {})
+        @message = message
+
+        attrs.each do |key, val|
+          send(:"#{key}=", val)
+        end
+      end
+
+      attr_accessor(
+        :level,
+        :logger_name,
+        :message,
+        :param_message,
+        :stacktrace
+      )
+    end
+  end
+end