#
# Fluentd
#
#    Licensed 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 'fluent/plugin/output'

module Fluent::Plugin
  class StdoutOutput < Output
    Fluent::Plugin.register_output('stdout', self)

    helpers :inject, :formatter, :compat_parameters

    DEFAULT_LINE_FORMAT_TYPE = 'stdout'
    DEFAULT_FORMAT_TYPE = 'json'

    desc "If Fluentd logger outputs logs to a file (with -o option), this plugin outputs events to the file as well."
    config_param :use_logger, :bool, default: true

    config_section :buffer do
      config_set_default :chunk_keys, ['tag']
      config_set_default :flush_at_shutdown, true
      config_set_default :chunk_limit_size, 10 * 1024
    end

    config_section :format do
      config_set_default :@type, DEFAULT_LINE_FORMAT_TYPE
      config_set_default :output_type, DEFAULT_FORMAT_TYPE
    end

    def prefer_buffered_processing
      false
    end

    def multi_workers_ready?
      true
    end

    def dest_io
      @use_logger ? $log : $stdout
    end

    attr_accessor :formatter

    def configure(conf)
      compat_parameters_convert(conf, :inject, :formatter)

      super

      @formatter = formatter_create
    end

    def process(tag, es)
      es = inject_values_to_event_stream(tag, es)
      es.each {|time,record|
        dest_io.write(format(tag, time, record))
      }
      dest_io.flush
    end

    def format(tag, time, record)
      record = inject_values_to_record(tag, time, record)
      @formatter.format(tag, time, record).chomp + "\n"
    end

    def write(chunk)
      chunk.write_to(dest_io)
    end
  end
end
