module Appsignal
  class EventFormatter
    module Moped
      class QueryFormatter < Appsignal::EventFormatter
        register 'query.moped'

        def format(payload)
          if payload[:ops] && payload[:ops].length > 0
            op = payload[:ops].first
            case op.class.to_s
            when 'Moped::Protocol::Command'
              return ['Command', {
                :database => op.full_collection_name,
                :selector => Appsignal::Utils.sanitize(op.selector)
              }.inspect]
            when 'Moped::Protocol::Query'
              return ['Query', {
                :database => op.full_collection_name,
                :selector => Appsignal::Utils.sanitize(op.selector),
                :flags    => op.flags,
                :limit    => op.limit,
                :skip     => op.skip,
                :fields   => op.fields
              }.inspect]
            when 'Moped::Protocol::Delete'
              return ['Delete', {
                :database => op.full_collection_name,
                :selector => Appsignal::Utils.sanitize(op.selector),
                :flags    => op.flags,
              }.inspect]
            when 'Moped::Protocol::Insert'
              return ['Insert', {
                :database   => op.full_collection_name,
                :documents  => Appsignal::Utils.sanitize(op.documents, true),
                :count      => op.documents.count,
                :flags      => op.flags,
              }.inspect]
            when 'Moped::Protocol::Update'
              return ['Update', {
                :database => op.full_collection_name,
                :selector => Appsignal::Utils.sanitize(op.selector),
                :update   => Appsignal::Utils.sanitize(op.update, true),
                :flags    => op.flags,
              }.inspect]
            when 'Moped::Protocol::KillCursors'
              return ['KillCursors', {
                :number_of_cursor_ids => op.number_of_cursor_ids
              }.inspect]
            else
              return [op.class.to_s.sub('Moped::Protocol::', ''), {
                :database => op.full_collection_name
              }.inspect]
            end
          end
        end

      end
    end
  end
end