# encoding: utf-8
require "logstash/outputs/base"
require "logstash/namespace"


class LogStash::Outputs::Mongodb < LogStash::Outputs::Base

  config_name "mongodb"

  # a MongoDB URI to connect to
  # See http://docs.mongodb.org/manual/reference/connection-string/
  config :endpoints, :validate => :array, :required => true
  
  # The database to use
  config :database, :validate => :string, :required => true

  # The user to authenticate
  config :user, :validate => :string, :required => false

  # The password of the user to authenticate
  config :password, :validate => :string, :required => false
   
  # The collection to use. This value can use `%{foo}` values to dynamically
  # select a collection based on data in the event.
  config :collection, :validate => :string, :required => true

  # If true, store the @timestamp field in mongodb as an ISODate type instead
  # of an ISO8601 string.  For more information about this, see
  # http://www.mongodb.org/display/DOCS/Dates
  config :isodate, :validate => :boolean, :default => false

  # Number of seconds to wait after failure before retrying
  config :retry_delay, :validate => :number, :default => 3, :required => false

  # If true, an "_id" field will be added to the document before insertion.
  # The "_id" field will use the timestamp of the event and overwrite an existing
  # "_id" field in the event.
  config :generateId, :validate => :boolean, :default => false

  public
  def register
    require "mongo"
    options = { database: @database }
    options[:user] = @user unless @user.nil? || @user.empty?
    options[:password] = @password unless @password.nil? || @password.empty?
    @client = Mongo::Client.new(@endpoints, options)
  end # def register

  public
  def receive(event)
    return unless output?(event)

    begin
      if @isodate
        # the mongodb driver wants time values as a ruby Time object.
        # set the @timestamp value of the document to a ruby Time object, then.
        document = event.to_hash
      else
        document = event.to_hash.merge("@timestamp" => event["@timestamp"].to_json)
      end
      if @generateId
        document['_id'] = BSON::ObjectId.new(nil, event["@timestamp"])
      end
      @client[event.sprintf(@collection)].insert_one(document)
    rescue Mongo::Error => e
      @logger.warn("Failed to send event to MongoDB", :event => event, :exception => e,
                   :backtrace => e.backtrace)
      sleep @retry_delay
      retry
    end
  end # def receive
end # class LogStash::Outputs::Mongodb