Module: LogStash::Config::Mixin::DSL

Defined in:
lib/logstash/config/mixin.rb

Overview

def replace_env_placeholders

Instance Attribute Summary collapse

Instance Method Summary collapse

Instance Attribute Details

#flagsObject

Returns the value of attribute flags.



187
188
189
# File 'lib/logstash/config/mixin.rb', line 187

def flags
  @flags
end

Instance Method Details

#config(name, opts = {}) ⇒ Object

Define a new configuration setting



211
212
213
214
215
216
217
218
219
220
221
222
# File 'lib/logstash/config/mixin.rb', line 211

def config(name, opts={})
  @config ||= Hash.new
  # TODO(sissel): verify 'name' is of type String, Symbol, or Regexp

  name = name.to_s if name.is_a?(Symbol)
  @config[name] = opts  # ok if this is empty

  if name.is_a?(String)
    define_method(name) { instance_variable_get("@#{name}") }
    define_method("#{name}=") { |v| instance_variable_set("@#{name}", v) }
  end
end

#config_name(name = nil) ⇒ Object

If name is given, set the name and return it. If no name given (nil), return the current name.



191
192
193
194
195
# File 'lib/logstash/config/mixin.rb', line 191

def config_name(name = nil)
  @config_name = name if !name.nil?
  LogStash::Config::Registry.registry[@config_name] = self
  return @config_name
end

#default(name, value) ⇒ Object

def config



224
225
226
227
# File 'lib/logstash/config/mixin.rb', line 224

def default(name, value)
  @defaults ||= {}
  @defaults[name.to_s] = value
end

#default?(name) ⇒ Boolean

Returns:

  • (Boolean)


237
238
239
# File 'lib/logstash/config/mixin.rb', line 237

def default?(name)
  return @defaults && @defaults.include?(name)
end

#get_configObject



229
230
231
# File 'lib/logstash/config/mixin.rb', line 229

def get_config
  return @config
end

#get_default(name) ⇒ Object

def get_config



233
234
235
# File 'lib/logstash/config/mixin.rb', line 233

def get_default(name)
  return @defaults && @defaults[name]
end

#hash_or_array(value) ⇒ Object



563
564
565
566
567
568
# File 'lib/logstash/config/mixin.rb', line 563

def hash_or_array(value)
  if !value.is_a?(Hash)
    value = [*value] # coerce scalar to array if necessary
  end
  return value
end

#inherited(subclass) ⇒ Object

This is called whenever someone subclasses a class that has this mixin.



253
254
255
256
257
258
259
260
261
262
263
264
265
# File 'lib/logstash/config/mixin.rb', line 253

def inherited(subclass)
  # Copy our parent's config to a subclass.
  # This method is invoked whenever someone subclasses us, like:
  # class Foo < Bar ...
  subconfig = Hash.new
  if !@config.nil?
    @config.each do |key, val|
      subconfig[key] = val
    end
  end
  subclass.instance_variable_set("@config", subconfig)
  @@version_notice_given = false
end

#milestone(m = nil) ⇒ Object

Deprecated: Declare the version of the plugin inside the gemspec.



205
206
207
208
# File 'lib/logstash/config/mixin.rb', line 205

def milestone(m = nil)
  @logger = Cabin::Channel.get(LogStash)
  @logger.debug(I18n.t('logstash.plugin.deprecated_milestone', :plugin => config_name))
end

#options(opts) ⇒ Object



241
242
243
244
245
246
247
248
249
250
# File 'lib/logstash/config/mixin.rb', line 241

def options(opts)
  # add any options from this class
  prefix = self.name.split("::").last.downcase
  @flags.each do |flag|
    flagpart = flag[:args].first.gsub(/^--/,"")
    # TODO(sissel): logger things here could help debugging.

    opts.on("--#{prefix}-#{flagpart}", *flag[:args][1..-1], &flag[:block])
  end
end

#plugin_status(status = nil) ⇒ Object

Deprecated: Declare the version of the plugin inside the gemspec.



199
200
201
# File 'lib/logstash/config/mixin.rb', line 199

def plugin_status(status = nil)
  milestone(status)
end

def validate



282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
# File 'lib/logstash/config/mixin.rb', line 282

def print_version_notice
  return if @@version_notice_given

  begin
    plugin_version = LogStash::Util::PluginVersion.find_plugin_version!(@plugin_type, @config_name)

    if plugin_version < PLUGIN_VERSION_1_0_0
      if plugin_version < PLUGIN_VERSION_0_9_0
        @logger.info(I18n.t("logstash.plugin.version.0-1-x", 
                            :type => @plugin_type,
                            :name => @config_name,
                            :LOGSTASH_VERSION => LOGSTASH_VERSION))
      else
        @logger.info(I18n.t("logstash.plugin.version.0-9-x", 
                            :type => @plugin_type,
                            :name => @config_name,
                            :LOGSTASH_VERSION => LOGSTASH_VERSION))
      end
    end
  rescue LogStash::PluginNoVersionError
    # If we cannot find a version in the currently installed gems we
    # will display this message. This could happen in the test, if you 
    # create an anonymous class to test a plugin.
    @logger.warn(I18n.t("logstash.plugin.no_version",
                            :type => @plugin_type,
                            :name => @config_name,
                            :LOGSTASH_VERSION => LOGSTASH_VERSION))
  ensure 
    @@version_notice_given = true
  end
end

#secure_params!(params) ⇒ Object

def validate_value



555
556
557
558
559
560
561
# File 'lib/logstash/config/mixin.rb', line 555

def secure_params!(params)
  params.each do |key, value|
    if @config[key][:validate] == :password && !value.is_a?(::LogStash::Util::Password)
      params[key] = ::LogStash::Util::Password.new(value)
    end
  end
end

#validate(params) ⇒ Object

def inherited



267
268
269
270
271
272
273
274
275
276
277
278
279
280
# File 'lib/logstash/config/mixin.rb', line 267

def validate(params)
  @plugin_name = config_name
  @plugin_type = ancestors.find { |a| a.name =~ /::Base$/ }.config_name
  @logger = Cabin::Channel.get(LogStash)
  is_valid = true

  print_version_notice

  is_valid &&= validate_check_invalid_parameter_names(params)
  is_valid &&= validate_check_required_parameter_names(params)
  is_valid &&= validate_check_parameter_values(params)

  return is_valid
end

#validate_check_invalid_parameter_names(params) ⇒ Object



314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
# File 'lib/logstash/config/mixin.rb', line 314

def validate_check_invalid_parameter_names(params)
  invalid_params = params.keys
  # Filter out parameters that match regexp keys.
  # These are defined in plugins like this:
  #   config /foo.*/ => ...
  @config.each_key do |config_key|
    if config_key.is_a?(Regexp)
      invalid_params.reject! { |k| k =~ config_key }
    elsif config_key.is_a?(String)
      invalid_params.reject! { |k| k == config_key }
    end
  end

  if invalid_params.size > 0
    invalid_params.each do |name|
      @logger.error("Unknown setting '#{name}' for #{@plugin_name}")
    end
    return false
  end # if invalid_params.size > 0
  return true
end

#validate_check_parameter_values(params) ⇒ Object



356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
# File 'lib/logstash/config/mixin.rb', line 356

def validate_check_parameter_values(params)
  # Filter out parametrs that match regexp keys.
  # These are defined in plugins like this:
  #   config /foo.*/ => ... 
  is_valid = true

  params.each do |key, value|
    @config.keys.each do |config_key|
      next unless (config_key.is_a?(Regexp) && key =~ config_key) \
                  || (config_key.is_a?(String) && key == config_key)
      config_val = @config[config_key][:validate]
      #puts "  Key matches."
      success, result = validate_value(value, config_val)
      if success 
        # Accept coerced value if success
        # Used for converting values in the config to proper objects.
        params[key] = result if !result.nil?
      else
        @logger.error(I18n.t("logstash.agent.configuration.setting_invalid",
                             :plugin => @plugin_name, :type => @plugin_type,
                             :setting => key, :value => value.inspect,
                             :value_type => config_val,
                             :note => result))
      end
      #puts "Result: #{key} / #{result.inspect} / #{success}"
      is_valid &&= success

      break # done with this param key
    end # config.each
  end # params.each

  return is_valid
end

#validate_check_required_parameter_names(params) ⇒ Object

def validate_check_invalid_parameter_names



336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
# File 'lib/logstash/config/mixin.rb', line 336

def validate_check_required_parameter_names(params)
  is_valid = true

  @config.each do |config_key, config|
    next unless config[:required]

    if config_key.is_a?(Regexp)
      next if params.keys.select { |k| k =~ config_key }.length > 0
    elsif config_key.is_a?(String)
      next if params.keys.member?(config_key)
    end
    @logger.error(I18n.t("logstash.agent.configuration.setting_missing",
                         :setting => config_key, :plugin => @plugin_name,
                         :type => @plugin_type))
    is_valid = false
  end

  return is_valid
end

#validate_value(value, validator) ⇒ Object



400
401
402
403
404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
426
427
428
429
430
431
432
433
434
435
436
437
438
439
440
441
442
443
444
445
446
447
448
449
450
451
452
453
454
455
456
457
458
459
460
461
462
463
464
465
466
467
468
469
470
471
472
473
474
475
476
477
478
479
480
481
482
483
484
485
486
487
488
489
490
491
492
493
494
495
496
497
498
499
500
501
502
503
504
505
506
507
508
509
510
511
512
513
514
515
516
517
518
519
520
521
522
523
524
525
526
527
528
529
530
531
532
533
534
535
536
537
538
539
540
541
542
543
544
545
546
547
548
549
550
551
552
553
# File 'lib/logstash/config/mixin.rb', line 400

def validate_value(value, validator)
  # Validator comes from the 'config' pieces of plugins.
  # They look like this
  #   config :mykey => lambda do |value| ... end
  # (see LogStash::Inputs::File for example)
  result = nil

  if validator.nil?
    return true
  elsif validator.is_a?(Array)
    value = [*value]
    if value.size > 1
      return false, "Expected one of #{validator.inspect}, got #{value.inspect}"
    end

    if !validator.include?(value.first)
      return false, "Expected one of #{validator.inspect}, got #{value.inspect}"
    end
    result = value.first
  elsif validator.is_a?(Symbol)
    # TODO(sissel): Factor this out into a coersion method?
    # TODO(sissel): Document this stuff.
    value = hash_or_array(value)

    case validator
      when :codec
        if value.first.is_a?(String)
          value = LogStash::Plugin.lookup("codec", value.first).new
          return true, value
        else
          value = value.first
          return true, value
        end
      when :hash
        if value.is_a?(Hash)
          return true, value
        end

        if value.size % 2 == 1
          return false, "This field must contain an even number of items, got #{value.size}"
        end

        # Convert the array the config parser produces into a hash.
        result = {}
        value.each_slice(2) do |key, value|
          entry = result[key]
          if entry.nil?
            result[key] = value
          else
            if entry.is_a?(Array)
              entry << value
            else
              result[key] = [entry, value]
            end
          end
        end
      when :array
        result = value
      when :string
        if value.size > 1 # only one value wanted
          return false, "Expected string, got #{value.inspect}"
        end
        result = value.first
      when :number
        if value.size > 1 # only one value wanted
          return false, "Expected number, got #{value.inspect} (type #{value.class})"
        end

        v = value.first
        case v
          when Numeric
            result = v
          when String
            if v.to_s.to_f.to_s != v.to_s \
               && v.to_s.to_i.to_s != v.to_s
              return false, "Expected number, got #{v.inspect} (type #{v})"
            end
            if v.include?(".")
              # decimal value, use float.
              result = v.to_f
            else
              result = v.to_i
            end
        end # case v
      when :boolean
        if value.size > 1 # only one value wanted
          return false, "Expected boolean, got #{value.inspect}"
        end

        bool_value = value.first
        if !!bool_value == bool_value
          # is_a does not work for booleans
          # we have Boolean and not a string
          result = bool_value
        else
          if bool_value !~ /^(true|false)$/
            return false, "Expected boolean 'true' or 'false', got #{bool_value.inspect}"
          end

          result = (bool_value == "true")
        end
      when :ipaddr
        if value.size > 1 # only one value wanted
          return false, "Expected IPaddr, got #{value.inspect}"
        end

        octets = value.split(".")
        if octets.length != 4
          return false, "Expected IPaddr, got #{value.inspect}"
        end
        octets.each do |o|
          if o.to_i < 0 or o.to_i > 255
            return false, "Expected IPaddr, got #{value.inspect}"
          end
        end
        result = value.first
      when :password
        if value.size > 1
          return false, "Expected password (one value), got #{value.size} values?"
        end

        result = value.first.is_a?(::LogStash::Util::Password) ? value.first : ::LogStash::Util::Password.new(value.first)
      when :path
        if value.size > 1 # Only 1 value wanted
          return false, "Expected path (one value), got #{value.size} values?"
        end

        # Paths must be absolute
        #if !Pathname.new(value.first).absolute?
          #return false, "Require absolute path, got relative path #{value.first}?"
        #end

        if !File.exists?(value.first) # Check if the file exists
          return false, "File does not exist or cannot be opened #{value.first}"
        end

        result = value.first
      when :bytes
        begin
          bytes = Integer(value.first) rescue nil
          result = bytes || Filesize.from(value.first).to_i
        rescue ArgumentError
          return false, "Unparseable filesize: #{value.first}. possible units (KiB, MiB, ...) e.g. '10 KiB'. doc reference: http://www.elastic.co/guide/en/logstash/current/configuration.html#bytes"
        end
      else
        return false, "Unknown validator symbol #{validator}"
    end # case validator
  else
    return false, "Unknown validator #{validator.class}"
  end

  # Return the validator for later use, like with type coercion.
  return true, result
end

#validator_find(key) ⇒ Object

def validate_check_parameter_values



390
391
392
393
394
395
396
397
398
# File 'lib/logstash/config/mixin.rb', line 390

def validator_find(key)
  @config.each do |config_key, config_val|
    if (config_key.is_a?(Regexp) && key =~ config_key) \
       || (config_key.is_a?(String) && key == config_key)
      return config_val
    end
  end # @config.each
  return nil
end