#!/usr/bin/env python
"""Message handlers."""

from grr_response_core.lib import rdfvalue

# TODO(amoser): Investigate if we can have a single place (handler_registry.py)
# where we define all the information needed for running message handlers.

session_id_map = {
    str(
        rdfvalue.SessionID(queue=rdfvalue.RDFURN("S"), flow_name="Stats")
    ): "StatsHandler",
    str(rdfvalue.SessionID(flow_name="ClientAlert")): "ClientAlertHandler",
    str(rdfvalue.SessionID(flow_name="Foreman")): "ForemanHandler",
    str(rdfvalue.SessionID(flow_name="Startup")): "ClientStartupHandler",
    str(rdfvalue.SessionID(flow_name="TransferStore")): "BlobHandler",
}


class MessageHandler(object):
  """The base class for all message handlers."""

  handler_name = ""

  def ProcessMessages(self, msgs):
    """This is where messages get processed.

    Override in derived classes.

    Args:
      msgs: The GrrMessages sent by the client.
    """