sonots/haikanko

class Fluent::IkachanOutput < Fluent::Output
  Fluent::Plugin.register_output('ikachan', self)

  config_param :host, :string
  config_param :port, :integer, :default => 4979
  config_param :channel, :string
  config_param :message, :string, :default => nil
  config_param :out_keys, :string, :default => ""
  config_param :privmsg_message, :string, :default => nil
  config_param :privmsg_out_keys, :string, :default => ""
  config_param :max_lines, :integer, :default => 10
  config_param :time_key, :string, :default => nil
  config_param :time_format, :string, :default => nil
  config_param :tag_key, :string, :default => 'tag'

  def initialize
    super
    require 'net/http'
    require 'uri'
  end

  def configure(conf)
    super

    @channel = '#' + @channel
    @join_uri = URI.parse "http://#{@host}:#{@port}/join"
    @notice_uri = URI.parse "http://#{@host}:#{@port}/notice"
    @privmsg_uri = URI.parse "http://#{@host}:#{@port}/privmsg"

    @out_keys = @out_keys.split(',')
    @privmsg_out_keys = @privmsg_out_keys.split(',')

    if @message.nil? and @privmsg_message.nil?
      raise Fluent::ConfigError, "Either 'message' or 'privmsg_message' must be specifed."
    end

    begin
      @message % (['1'] * @out_keys.length) if @message
    rescue ArgumentError
      raise Fluent::ConfigError, "string specifier '%s' and out_keys specification mismatch"
    end

    begin
      @privmsg_message % (['1'] * @privmsg_out_keys.length) if @privmsg_message
    rescue ArgumentError
      raise Fluent::ConfigError, "string specifier '%s' of privmsg_message and privmsg_out_keys specification mismatch"
    end

    if @time_key
      if @time_format
        f = @time_format
        tf = Fluent::TimeFormatter.new(f, true) # IRC notification is formmatted as localtime only...
        @time_format_proc = tf.method(:format)
        @time_parse_proc = Proc.new {|str| Time.strptime(str, f).to_i }
      else
        @time_format_proc = Proc.new {|time| time.to_s }
        @time_parse_proc = Proc.new {|str| str.to_i }
      end
    end
  end

  def start
    res = Net::HTTP.post_form(@join_uri, {'channel' => @channel})
    if res.code.to_i == 200
      # ok
    elsif res.code.to_i == 403 and res.body == "joinned channel: #{@channel}"
      # ok
    else
      raise Fluent::ConfigError, "failed to connect ikachan server #{@host}:#{@port}"
    end
  end

  def shutdown
  end

  def emit(tag, es, chain)
    messages = []
    privmsg_messages = []

    es.each {|time,record|
      messages << evaluate_message(@message, @out_keys, tag, time, record) if @message
      privmsg_messages << evaluate_message(@privmsg_message, @privmsg_out_keys, tag, time, record) if @privmsg_message
    }

    messages.each do |msg|
      begin
        lines = msg.split("\n")
        limited_lines = lines[0, @max_lines]
        limited_lines += [lines.last] if lines.size > @max_lines
        limited_lines += ["too many error occured. check email please"] if limited_lines.size != lines.size
        limited_lines.each do |m|
          res = Net::HTTP.post_form(@notice_uri, {'channel' => @channel, 'message' => m})
        end
      rescue
        $log.warn "out_ikachan: failed to send notice to #{@host}:#{@port}, #{@channel}, message: #{msg}"
      end
    end

    privmsg_messages.each do |msg|
      begin
        lines = msg.split("\n")
        limited_lines = lines[0, @max_lines]
        limited_lines += [lines.last] if lines.size > @max_lines
        limited_lines += ["too many error occured. check email please"] if limited_lines.size != lines.size
        limited_lines.each do |m|
          res = Net::HTTP.post_form(@privmsg_uri, {'channel' => @channel, 'message' => m})
        end
      rescue
        $log.warn "out_ikachan: failed to send privmsg to #{@host}:#{@port}, #{@channel}, message: #{msg}"
      end
    end

    chain.next
  end

  private

  def evaluate_message(message, out_keys, tag, time, record)
    values = []
    last = out_keys.length - 1

    values = out_keys.map do |key|
      case key
      when @time_key
        @time_format_proc.call(time)
      when @tag_key
        tag
      else
        record[key].to_s
      end
    end

    (message % values).gsub(/\\n/, "\n")
  end

end