# 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.

require "fileutils"
require "delegate"

require "logstash/util/substitution_variables"
require "logstash/util/time_value"
require "i18n"

module LogStash
  class Settings

    include LogStash::Util::SubstitutionVariables
    include LogStash::Util::Loggable

    # The `LOGGABLE_PROXY` is included into `LogStash::Setting` to make all
    # settings-related logs and deprecations come from the same logger.
    LOGGABLE_PROXY = Module.new do
      define_method(:logger) { Settings.logger }
      define_method(:deprecation_logger) { Settings.deprecation_logger }

      def self.included(base)
        base.extend(self)
      end
    end

    # there are settings that the pipeline uses and can be changed per pipeline instance
    PIPELINE_SETTINGS_WHITE_LIST = [
      "config.debug",
      "config.support_escapes",
      "config.string",
      "dead_letter_queue.enable",
      "dead_letter_queue.flush_interval",
      "dead_letter_queue.flush_check_interval",
      "dead_letter_queue.max_bytes",
      "dead_letter_queue.storage_policy",
      "dead_letter_queue.retain.age",
      "metric.collect",
      "pipeline.plugin_classloaders",
      "path.config",
      "path.dead_letter_queue",
      "path.queue",
      "pipeline.batch.delay",
      "pipeline.batch.output_chunking.growth_threshold_factor",
      "pipeline.batch.metrics.sampling_mode",
      "pipeline.batch.size",
      "pipeline.id",
      "pipeline.reloadable",
      "pipeline.recoverable",
      "pipeline.system",
      "pipeline.workers",
      "pipeline.ordered",
      "pipeline.ecs_compatibility",
      "queue.checkpoint.acks",
      "queue.checkpoint.interval", # remove it for #17155
      "queue.checkpoint.writes",
      "queue.checkpoint.retry",
      "queue.compression",
      "queue.drain",
      "queue.max_bytes",
      "queue.max_events",
      "queue.page_capacity",
      "queue.type",
    ]

    # These are deprecated as pipeline override settings, they still exist as process-level settings
    DEPRECATED_PIPELINE_OVERRIDE_SETTINGS = [
      "config.reload.automatic",
      "config.reload.interval",
    ]

    def initialize
      @settings = {}
      # Theses settings were loaded from the yaml file
      # but we didn't find any settings to validate them,
      # lets keep them around until we do `validate_all` at that
      # time universal plugins could have added new settings.
      @transient_settings = {}
    end

    def register(setting)
      # Method #with_deprecated_alias returns collection containing couple of other settings.
      # In case the method is implemented in Ruby returns an Array while for the Java implementation
      # return a List, so the following type checking before going deep by one layer.
      return setting.map { |s| register(s) } if setting.kind_of?(Array) || setting.kind_of?(java.util.List)

      if @settings.key?(setting.name)
        raise ArgumentError.new("Setting \"#{setting.name}\" has already been registered as #{setting.inspect}")
      else
        @settings[setting.name] = setting
      end
    end

    def registered?(setting_name)
       @settings.key?(setting_name)
    end

    def get_setting(setting_name)
      setting = @settings[setting_name]
      raise ArgumentError.new("Setting \"#{setting_name}\" doesn't exist. Please check if you haven't made a typo.") if setting.nil?
      setting
    end

    def get_subset(setting_regexp)
      regexp = setting_regexp.is_a?(Regexp) ? setting_regexp : Regexp.new(setting_regexp)
      settings = self.class.new
      @settings.each do |setting_name, setting|
        next unless setting_name.match(regexp)
        settings.register(setting.clone)
      end
      settings
    end

    def names
      @settings.keys
    end

    def set?(setting_name)
      get_setting(setting_name).set?
    end

    def clone(*args)
      get_subset(".*")
    end
    alias_method :dup, :clone

    def get_default(setting_name)
      get_setting(setting_name).default
    end

    def get_value(setting_name)
      get_setting(setting_name).value
    end
    alias_method :get, :get_value

    def set_value(setting_name, value, graceful = false)
      get_setting(setting_name).set(value)
    rescue ArgumentError => e
      if graceful
        @transient_settings[setting_name] = value
      else
        raise e
      end
    end
    alias_method :set, :set_value

    def to_hash
      @settings.each_with_object({}) do |(name, setting), hash|
        next if (setting.kind_of? Setting::DeprecatedAlias) || (setting.kind_of? Java::org.logstash.settings.DeprecatedAlias)
        hash[name] = setting.value
      end
    end

    def merge(hash, graceful = false)
      hash.each {|key, value| set_value(key, value, graceful) }
      self
    end

    def merge_pipeline_settings(hash, graceful = false)
      hash.each do |key, _|
        if DEPRECATED_PIPELINE_OVERRIDE_SETTINGS.include?(key)
          deprecation_logger.deprecated("Config option \"#{key}\", set for pipeline \"#{hash['pipeline.id']}\", is " +
                                          "deprecated as a pipeline override setting. Please only set it at " +
                                          "the process level.")
          hash.delete(key)
        elsif !PIPELINE_SETTINGS_WHITE_LIST.include?(key)
          raise ArgumentError.new("Only pipeline related settings are expected. Received \"#{key}\". Allowed settings: #{PIPELINE_SETTINGS_WHITE_LIST}")
        end
      end
      merge(hash, graceful)
    end

    def format_settings
      output = []
      output << "-------- Logstash Settings (* means modified) ---------"
      @settings.each do |setting_name, setting|
        setting.format(output)
      end
      output << "--------------- Logstash Settings -------------------"
      output
    end

    def reset
      @settings.values.each(&:reset)
    end

    def from_yaml(yaml_path, file_name = "logstash.yml")
      settings = read_yaml(::File.join(yaml_path, file_name))
      self.merge(deep_replace(flatten_hash(settings), true), true)
      self
    end

    def post_process
      if @post_process_callbacks
        @post_process_callbacks.each do |callback|
          callback.call(self)
        end
      end

      # because we cannot rely on deprecation logger being wired up when the setters
      # are initially used, we re-emit setter-related deprecations after all post-processing
      # hooks have been activated (and therefore after logging has been configured)
      @settings.each_value do |setting|
        setting.observe_post_process if setting.respond_to?(:observe_post_process)
      end
    end

    def on_post_process(&block)
      @post_process_callbacks ||= []
      @post_process_callbacks << block
    end

    def validate_all
      # lets merge the transient_settings again to see if new setting were added.
      self.merge(@transient_settings)

      @settings.each do |name, setting|
        setting.validate_value
      end
    end

    def ==(other)
      return false unless other.kind_of?(::LogStash::Settings)
      self.to_hash == other.to_hash
    end

    private
    def read_yaml(path)
      YAML.safe_load(IO.read(path)) || {}
    end

    def flatten_hash(h, f = "", g = {})
      return g.update({ f => h }) unless h.is_a? Hash
      if f.empty?
        h.each { |k, r| flatten_hash(r, k, g) }
      else
        h.each { |k, r| flatten_hash(r, "#{f}.#{k}", g) }
      end
      g
    end
  end

  class Setting
    include LogStash::Settings::LOGGABLE_PROXY

    attr_reader :wrapped_setting

    def initialize(name, klass, default = nil, strict = true, &validator_proc)
      unless klass.is_a?(Class)
        raise ArgumentError.new("Setting \"#{name}\" must be initialized with a class (received #{klass})")
      end
      setting_builder = Java::org.logstash.settings.BaseSetting.create(name)
                            .defaultValue(default)
                            .strict(strict)
      if validator_proc
        setting_builder = setting_builder.validator(validator_proc)
      end

      @wrapped_setting = setting_builder.build()

      @klass = klass
      @validator_proc = validator_proc

      validate(default) if strict?
    end

    def default
      @wrapped_setting.default
    end

    def name
      @wrapped_setting.name
    end

    def initialize_copy(original)
      @wrapped_setting = original.wrapped_setting.clone
    end

    # To be used only internally
    def update_wrapper(wrapped_setting)
      @wrapped_setting = wrapped_setting
    end

    def value
      @wrapped_setting.value()
    end

    def set?
      @wrapped_setting.set?
    end

    def strict?
      @wrapped_setting.strict?
    end

    def set(value)
      validate(value) if strict?
      @wrapped_setting.set(value)
      @wrapped_setting.value
    end

    def reset
      @wrapped_setting.reset
    end

    def to_hash
      {
        "name" => @wrapped_setting.name,
        "klass" => @klass,
        "value" => @wrapped_setting.value,
        "value_is_set" => @wrapped_setting.set?,
        "default" => @wrapped_setting.default,
        # Proc#== will only return true if it's the same obj
        # so no there's no point in comparing it
        # also thereś no use case atm to return the proc
        # so let's not expose it
        #"validator_proc" => @validator_proc
      }
    end

    def inspect
      "<#{self.class.name}(#{name}): #{value.inspect}" + (@wrapped_setting.set? ? '' : ' (DEFAULT)') + ">"
    end

    def ==(other)
      self.to_hash == other.to_hash
    end

    def validate_value
      validate(value)
    end

    def with_deprecated_alias(deprecated_alias_name, obsoleted_version=nil)
      SettingWithDeprecatedAlias.wrap(self, deprecated_alias_name, obsoleted_version)
    end

    ##
    # Returns a Nullable-wrapped self, effectively making the Setting optional.
    def nullable
      Nullable.new(self)
    end

    def format(output)
      @wrapped_setting.format(output)
    end

    def clone(*args)
      copy = self.dup
      copy.update_wrapper(@wrapped_setting.clone())
      copy
    end

    protected
    def validate(input)
      if !input.is_a?(@klass)
        raise ArgumentError.new("Setting \"#{@wrapped_setting.name}\" must be a #{@klass}. Received: #{input} (#{input.class})")
      end

      if @validator_proc && !@validator_proc.call(input)
        raise ArgumentError.new("Failed to validate setting \"#{@wrapped_setting.name}\" with value: #{input}")
      end
    end

    class Coercible < Setting
      def initialize(name, klass, default = nil, strict = true, &validator_proc)
        unless klass.is_a?(Class)
          raise ArgumentError.new("Setting \"#{name}\" must be initialized with a class (received #{klass})")
        end

        @klass = klass
        @validator_proc = validator_proc

        # needed to have the name method accessible when invoking validate
        @wrapped_setting = Java::org.logstash.settings.BaseSetting.create(name)
                                      .defaultValue(default)
                                      .strict(strict)
                                      .build()

        if strict
          coerced_default = coerce(default)
          validate(coerced_default)
          updated_default = coerced_default
        else
          updated_default = default
        end

        # default value must be coerced to the right type before being set
        setting_builder = Java::org.logstash.settings.BaseSetting.create(name)
                              .defaultValue(updated_default)
                              .strict(strict)
        if validator_proc
          setting_builder = setting_builder.validator(validator_proc)
        end

        @wrapped_setting = setting_builder.build()
      end

      def set(value)
        coerced_value = coerce(value)
        validate(coerced_value)
        @wrapped_setting.set(coerced_value)
        coerced_value
      end

      def coerce(value)
        raise NotImplementedError.new("Please implement #coerce for #{self.class}")
      end
    end
    ### Specific settings #####

    java_import org.logstash.settings.BooleanSetting
    java_import org.logstash.settings.NumericSetting

    java_import org.logstash.settings.IntegerSetting
    java_import org.logstash.settings.PositiveIntegerSetting

    java_import org.logstash.settings.PortSetting # seems unused

    java_import org.logstash.settings.PortRangeSetting

    class Validator < Setting
      def initialize(name, default = nil, strict = true, validator_class = nil)
        @validator_class = validator_class
        super(name, ::Object, default, strict)
      end

      def validate(value)
        @validator_class.validate(value)
      end
    end
    
    java_import org.logstash.settings.StringSetting

    java_import org.logstash.settings.NullableStringSetting
    java_import org.logstash.settings.PasswordSetting
    java_import org.logstash.settings.ValidatedPasswordSetting

    java_import org.logstash.settings.CoercibleStringSetting

    java_import org.logstash.settings.ExistingFilePathSetting

    java_import org.logstash.settings.WritableDirectorySetting

    java_import org.logstash.settings.BytesSetting

    java_import org.logstash.settings.TimeValueSetting

    class ArrayCoercible < Coercible
      def initialize(name, klass, default, strict = true, &validator_proc)
        @element_class = klass
        super(name, ::Array, default, strict, &validator_proc)
      end

      def coerce(value)
        Array(value)
      end

      protected
      def validate(input)
        if !input.is_a?(@klass)
          raise ArgumentError.new("Setting \"#{@wrapped_setting.name}\" must be a #{@klass}. Received: #{input} (#{input.class})")
        end

        unless input.all? {|el| el.kind_of?(@element_class) }
          raise ArgumentError.new("Values of setting \"#{@wrapped_setting.name}\" must be #{@element_class}. Received: #{input.map(&:class)}")
        end

        if @validator_proc && !@validator_proc.call(input)
          raise ArgumentError.new("Failed to validate setting \"#{@wrapped_setting.name}\" with value: #{input}")
        end
      end
    end

    class StringArray < ArrayCoercible
      def initialize(name, default, strict = true, possible_strings = [], &validator_proc)
        @possible_strings = possible_strings
        super(name, ::String, default, strict, &validator_proc)
      end

      protected

      def validate(value)
        super(value)
        return unless @possible_strings&.any?

        invalid_value = coerce(value).reject { |val| @possible_strings.include?(val) }
        return unless invalid_value.any?

        raise ArgumentError,
          "Failed to validate the setting \"#{@wrapped_setting.name}\" value(s): #{invalid_value.inspect}. Valid options are: #{@possible_strings.inspect}"
      end
    end

    java_import org.logstash.settings.NullableSetting

    # @see Setting#nullable
    # @api internal
    class Nullable < SimpleDelegator
      def validate(value)
        return true if value.nil?

        __getobj__.send(:validate, value)
      end

      # prevent delegate from intercepting
      def validate_value
        validate(value)
      end
    end

    ##
    # @api private
    #
    # A DeprecatedAlias provides a deprecated alias for a setting, and is meant
    # to be used exclusively through `SettingWithDeprecatedAlias#wrap`
    class DeprecatedAlias < SimpleDelegator
      # include LogStash::Util::Loggable
      alias_method :wrapped, :__getobj__
      attr_reader :canonical_proxy, :obsoleted_version

      def initialize(canonical_proxy, alias_name, obsoleted_version)
        @canonical_proxy = canonical_proxy
        @obsoleted_version = obsoleted_version

        clone = @canonical_proxy.canonical_setting.clone
        clone.update_wrapper(clone.wrapped_setting.deprecate(alias_name))

        super(clone)
      end

      def set(value)
        do_log_setter_deprecation
        super
      end

      def value
        logger.warn(I18n.t("logstash.settings.deprecation.queried",
                           :deprecated_alias => name,
                           :canonical_name => canonical_proxy.name))
        @canonical_proxy.value
      end

      def validate_value
        # bypass deprecation warning
        wrapped.validate_value if set?
      end

      def observe_post_process
        do_log_setter_deprecation if set?
      end

      private

      def do_log_setter_deprecation
        deprecation_logger.deprecated(
          I18n.t("logstash.settings.deprecation.set",
                 :deprecated_alias => name,
                 :canonical_name => canonical_proxy.name,
                 :obsoleted_sentences =>
                   @obsoleted_version.nil? ?
                     I18n.t("logstash.settings.deprecation.obsoleted_future") :
                     I18n.t("logstash.settings.deprecation.obsoleted_version", :obsoleted_version => @obsoleted_version))
        )
      end
    end

    ##
    # A SettingWithDeprecatedAlias wraps any Setting to provide a deprecated
    # alias, and hooks `Setting#validate_value` to ensure that a deprecation
    # warning is fired when the setting is provided by its deprecated alias,
    # or to produce an error when both the canonical name and deprecated
    # alias are used together.
    class SettingWithDeprecatedAlias < SimpleDelegator

      ##
      # Wraps the provided setting, returning a pair of connected settings
      # including the canonical setting and a deprecated alias.
      # @param canonical_setting [Setting]: the setting to wrap
      # @param deprecated_alias_name [String]: the name for the deprecated alias
      # @param obsoleted_version [String]: the version of Logstash that deprecated alias will be removed
      #
      # @return [SettingWithDeprecatedAlias,DeprecatedSetting]
      def self.wrap(canonical_setting, deprecated_alias_name, obsoleted_version=nil)
        setting_proxy = new(canonical_setting, deprecated_alias_name, obsoleted_version)

        [setting_proxy, setting_proxy.deprecated_alias]
      end

      attr_reader :deprecated_alias
      alias_method :canonical_setting, :__getobj__

      def initialize(canonical_setting, deprecated_alias_name, obsoleted_version)
        super(canonical_setting)

        @deprecated_alias = DeprecatedAlias.new(self, deprecated_alias_name, obsoleted_version)
      end

      def set(value)
        canonical_setting.set(value)
      end

      def value
        return super if canonical_setting.set?

        # bypass warning by querying the wrapped setting's value
        return deprecated_alias.wrapped.value if deprecated_alias.set?

        default
      end

      def set?
        canonical_setting.set? || deprecated_alias.set?
      end

      def format(output)
        return super unless deprecated_alias.set? && !canonical_setting.set?

        output << "*#{self.name}: #{value.inspect} (via deprecated `#{deprecated_alias.name}`; default: #{default.inspect})"
      end

      def validate_value
        if deprecated_alias.set? && canonical_setting.set?
          fail(ArgumentError, I18n.t("logstash.settings.deprecation.ambiguous",
                                     :canonical_name => canonical_setting.name,
                                     :deprecated_alias => deprecated_alias.name))
        end

        super
      end
    end
  end

  SETTINGS = Settings.new
end
