hoss-agent 1.0.6

data/lib/hoss/config/bytes.rb

@@ -0,0 +1,42 @@
+# 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 Config
+    # @api private
+    class Bytes
+      MULTIPLIERS = {
+        'kb' => 1024,
+        'mb' => 1024 * 1_000,
+        'gb' => 1024 * 100_000
+      }.freeze
+      REGEX = /^(\d+)(b|kb|mb|gb)?$/i.freeze
+
+      def initialize(default_unit: 'kb')
+        @default_unit = default_unit
+      end
+
+      def call(value)
+        _, amount, unit = REGEX.match(String(value)).to_a
+        unit ||= @default_unit
+        MULTIPLIERS.fetch(unit.downcase, 1) * amount.to_i
+      end
+    end
+  end
+end

data/lib/hoss/config/duration.rb

@@ -0,0 +1,40 @@
+# 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 Config
+    # @api private
+    class Duration
+      MULTIPLIERS = { 'ms' => 0.001, 'm' => 60 }.freeze
+      REGEX = /^(-)?(\d+)(m|ms|s)?$/i.freeze
+
+      def initialize(default_unit: 's')
+        @default_unit = default_unit
+      end
+
+      def call(str)
+        _, negative, amount, unit = REGEX.match(String(str)).to_a
+        unit ||= @default_unit
+        seconds = MULTIPLIERS.fetch(unit.downcase, 1) * amount.to_i
+        seconds = 0 - seconds if negative
+        seconds
+      end
+    end
+  end
+end

data/lib/hoss/config/options.rb

@@ -0,0 +1,154 @@
+# 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 Config
+    # @api private
+    module Options
+      # @api private
+      class Option
+        def initialize(
+          key,
+          value: nil,
+          type: :string,
+          default: nil,
+          converter: nil
+        )
+          @key = key
+          @type = type
+          @default = default
+          @converter = converter
+
+          set(value || default)
+        end
+
+        attr_reader :key, :value, :default, :type
+
+        def set(value)
+          @value = normalize(value)
+        end
+
+        def env_key
+          "HOSS_#{key.upcase}"
+        end
+
+        private
+
+        # rubocop:disable Metrics/CyclomaticComplexity
+        def normalize(val)
+          return if val.nil?
+
+          if @converter
+            return @converter.call(val)
+          end
+
+          case type
+          when :string then val.to_s
+          when :int then val.to_i
+          when :float then val.to_f
+          when :bool then normalize_bool(val)
+          when :list then normalize_list(val)
+          when :dict then normalize_dict(val)
+          when :url then normalize_url(val)
+          else
+            # raise "Unknown options type '#{type.inspect}'"
+            val
+          end
+        end
+        # rubocop:enable Metrics/CyclomaticComplexity
+
+        def normalize_bool(val)
+          return val unless val.is_a?(String)
+          !%w[0 false].include?(val.strip.downcase)
+        end
+
+        def normalize_list(val)
+          return Array(val) unless val.is_a?(String)
+          val.split(/[ ,]/)
+        end
+
+        def normalize_dict(val)
+          return val unless val.is_a?(String)
+          Hash[val.split(/[&,]/).map { |kv| kv.split('=') }]
+        end
+
+        def normalize_url(val)
+          val = val.to_s
+          val.end_with?('/') ? val.chomp('/') : val
+        end
+      end
+
+      # @api private
+      module ClassMethods
+        def schema
+          @schema ||= {}
+        end
+
+        def option(key, **args)
+          schema[key] = args
+        end
+      end
+
+      # @api private
+      module InstanceMethods
+        def load_schema
+          Hash[self.class.schema.map do |key, args|
+            [key, Option.new(key, **args)]
+          end]
+        end
+
+        def method_missing(name, *value)
+          name_str = name.to_s
+
+          if name_str.end_with?('=')
+            key = name_str[0...-1].to_sym
+            set(key, value.first)
+
+          elsif name_str.end_with?('?')
+            key = name_str[0...-1].to_sym
+            options.key?(key) ? options[key].value : super
+
+          elsif options.key?(name)
+            options.fetch(name).value
+
+          else
+            super
+          end
+        end
+
+        def [](key)
+          options[key]
+        end
+        alias :get :[]
+
+        def set(key, value)
+          options.fetch(key.to_sym).set(value)
+        rescue KeyError
+          warn format("Unknown option '%s'", key)
+          nil
+        end
+      end
+
+      def self.extended(kls)
+        kls.instance_eval { extend ClassMethods }
+        kls.class_eval { include InstanceMethods }
+      end
+    end
+  end
+end

data/lib/hoss/config/regexp_list.rb

@@ -0,0 +1,30 @@
+# 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 Config
+    # @api private
+    class RegexpList
+      def call(value)
+        value = value.is_a?(String) ? value.split(',') : Array(value)
+        value.map(&Regexp.method(:new))
+      end
+    end
+  end
+end

data/lib/hoss/config/wildcard_pattern_list.rb

@@ -0,0 +1,54 @@
+# 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 Config
+    # @api private
+    class WildcardPatternList
+      # @api private
+      class WildcardPattern
+        def initialize(str)
+          @pattern = convert(str)
+        end
+
+        def match?(other)
+          !!@pattern.match(other)
+        end
+
+        alias :match :match?
+
+        private
+
+        def convert(str)
+          parts =
+            str.chars.each_with_object([]) do |char, arr|
+              arr << (char == '*' ? '.*' : Regexp.escape(char))
+            end
+
+          Regexp.new('\A' + parts.join + '\Z', Regexp::IGNORECASE)
+        end
+      end
+
+      def call(value)
+        value = value.is_a?(String) ? value.split(',') : Array(value)
+        value.map(&WildcardPattern.method(:new))
+      end
+    end
+  end
+end

data/lib/hoss/context.rb

@@ -0,0 +1,64 @@
+# 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/context/request'
+require 'hoss/context/request/socket'
+require 'hoss/context/request/url'
+require 'hoss/context/response'
+require 'hoss/context/user'
+
+module Hoss
+  # @api private
+  class Context
+    def initialize(custom: {}, labels: {}, user: nil, service: nil)
+      @custom = custom
+      @labels = labels
+      @user = user || User.new
+      @service = service
+    end
+
+    Service = Struct.new(:framework)
+    Framework = Struct.new(:name, :version)
+
+    attr_accessor :request
+    attr_accessor :response
+    attr_accessor :user
+    attr_reader :custom
+    attr_reader :labels
+    attr_reader :service
+
+    # rubocop:disable Metrics/CyclomaticComplexity
+    def empty?
+      return false if labels.any?
+      return false if custom.any?
+      return false if user.any?
+      return false if service
+      return false if request || response
+
+      true
+    end
+    # rubocop:enable Metrics/CyclomaticComplexity
+
+    def set_service(framework_name: nil, framework_version: nil)
+      @service = Service.new(
+        Framework.new(framework_name, framework_version)
+      )
+    end
+  end
+end

data/lib/hoss/context/request.rb

@@ -0,0 +1,28 @@
+# 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 Request
+      attr_accessor :body, :cookies, :env, :headers, :http_version, :method,
+        :socket, :url
+    end
+  end
+end

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

@@ -0,0 +1,36 @@
+# 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 Socket
+        def initialize(req)
+          @remote_addr = req.env['REMOTE_ADDR']
+          @encrypted = req.scheme == 'https'
+        end
+
+        attr_reader :remote_addr, :encrypted
+      end
+    end
+  end
+end