#ifndef LLARP_ROUTING_HANDLER_HPP
#define LLARP_ROUTING_HANDLER_HPP

namespace llarp
{
  struct AbstractRouter;

  namespace dht
  {
    struct IMessage;
  }

  namespace service
  {
    struct ProtocolFrame;
  }

  namespace routing
  {
    struct DataDiscardMessage;
    struct GrantExitMessage;
    struct ObtainExitMessage;
    struct RejectExitMessage;
    struct TransferTrafficMessage;
    struct UpdateExitMessage;
    struct UpdateExitVerifyMessage;
    struct CloseExitMessage;
    struct PathTransferMessage;
    struct PathConfirmMessage;
    struct PathLatencyMessage;

    // handles messages on the routing level
    struct IMessageHandler
    {
      virtual bool
      HandleObtainExitMessage(const ObtainExitMessage *msg,
                              AbstractRouter *r) = 0;

      virtual bool
      HandleGrantExitMessage(const GrantExitMessage *msg,
                             AbstractRouter *r) = 0;

      virtual bool
      HandleRejectExitMessage(const RejectExitMessage *msg,
                              AbstractRouter *r) = 0;

      virtual bool
      HandleTransferTrafficMessage(const TransferTrafficMessage *msg,
                                   AbstractRouter *r) = 0;

      virtual bool
      HandleUpdateExitMessage(const UpdateExitMessage *msg,
                              AbstractRouter *r) = 0;

      virtual bool
      HandleUpdateExitVerifyMessage(const UpdateExitVerifyMessage *msg,
                                    AbstractRouter *r) = 0;

      virtual bool
      HandleCloseExitMessage(const CloseExitMessage *msg,
                             AbstractRouter *r) = 0;

      virtual bool
      HandleDataDiscardMessage(const DataDiscardMessage *msg,
                               AbstractRouter *r) = 0;

      virtual bool
      HandlePathTransferMessage(const PathTransferMessage *msg,
                                AbstractRouter *r) = 0;

      virtual bool
      HandleHiddenServiceFrame(const service::ProtocolFrame *msg) = 0;

      virtual bool
      HandlePathConfirmMessage(const PathConfirmMessage *msg,
                               AbstractRouter *r) = 0;

      virtual bool
      HandlePathLatencyMessage(const PathLatencyMessage *msg,
                               AbstractRouter *r) = 0;
      virtual bool
      HandleDHTMessage(const dht::IMessage *msg, AbstractRouter *r) = 0;
    };
  }  // namespace routing
}  // namespace llarp

#endif