# encoding: utf-8
require "logstash/codecs/base"
require "logstash/namespace"
require "logstash/codecs/json"
require 'securerandom'

class LogStash::Codecs::GoogleAppengine < LogStash::Codecs::Base
  config_name "google_appengine"

  public

  def register
    @json = LogStash::Codecs::JSON.new
  end

  def decode(data)
    begin
      @json.decode(data) do |json|
        if is_parse_failure(json)
          return @logger.error("Failed to process data", :data => json)
        end

        flatten(json).each { |flattenedJson|
          yield LogStash::Event.new(flattenedJson)
        }
      end
    rescue => e
      @logger.error("Failed to process data", :error => e, :data => data)
    end
  end
end

private

def is_parse_failure(event)
  event["tags"] && event["tags"].include?("_jsonparsefailure")
end

def flatten(event)
  payload = event['protoPayload']
  lines = payload['line']
  payload.delete('@type')
  if lines
    payload.delete('line')
    lines.map.with_index { |line, i|
      merged = payload.merge(line)
      merged['lineId'] = merged['requestId'] + i.to_s
      merged
    }
  else
    payload['time'] = payload['endTime']
    payload['lineId'] = payload['requestId']
    [payload]
  end
end