#
# Fluent
#
# Copyright (C) 2011 FURUHASHI Sadayuki
#
#    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.
#
module Fluent
  class FileOutput < TimeSlicedOutput
    Plugin.register_output('file', self)

    SUPPORTED_COMPRESS = {
      'gz' => :gz,
      'gzip' => :gz,
    }

    config_param :path, :string

    config_param :time_format, :string, :default => nil

    config_param :compress, :default => nil do |val|
      c = SUPPORTED_COMPRESS[val]
      unless c
        raise ConfigError, "Unsupported compression algorithm '#{val}'"
      end
      c
    end

    config_param :symlink_path, :string, :default => nil

    def initialize
      require 'zlib'
      require 'time'
      super
    end

    def configure(conf)
      if path = conf['path']
        @path = path
      end
      unless @path
        raise ConfigError, "'path' parameter is required on file output"
      end

      if pos = @path.index('*')
        @path_prefix = @path[0,pos]
        @path_suffix = @path[pos+1..-1]
        conf['buffer_path'] ||= "#{@path}"
      else
        @path_prefix = @path+"."
        @path_suffix = ".log"
        conf['buffer_path'] ||= "#{@path}.*"
      end

      super

      @timef = TimeFormatter.new(@time_format, @localtime)
    end

    def format(tag, time, record)
      time_str = @timef.format(time)
      "#{time_str}\t#{tag}\t#{Yajl.dump(record)}\n"
    end

    def write(chunk)
      case @compress
      when nil
        suffix = ''
      when :gz
        suffix = ".gz"
      end

      i = 0
      begin
        path = "#{@path_prefix}#{chunk.key}_#{i}#{@path_suffix}#{suffix}"
        i += 1
      end while File.exist?(path)
      FileUtils.mkdir_p File.dirname(path)

      case @compress
      when nil
        File.open(path, "a", DEFAULT_FILE_PERMISSION) {|f|
          chunk.write_to(f)
        }
      when :gz
        Zlib::GzipWriter.open(path) {|f|
          chunk.write_to(f)
        }
      end
      create_symlink(path, suffix) if @symlink_path

      return path  # for test
    end

    def secondary_init(primary)
      # don't warn even if primary.class is not FileOutput
    end

    private

    def create_symlink(path, suffix)
      FileUtils.ln_sf(path, "#{@symlink_path}#{suffix}")
    end
  end
end