msg_dispatcher.h

#ifndef MSGDISPATCHER_H
#define MSGDISPATCHER_H

#include <memory>
#include <unordered_map>

#include "AKIpUptaneMes.h"
#include "asn1/asn1_message.h"

#include "aktualizr_secondary_interface.h"

class MsgDispatcher {
 public:
  enum HandleStatusCode { kUnkownMsg = -1, kOk, kRebootRequired };

  using Handler = std::function<HandleStatusCode(Asn1Message&, Asn1Message&)>;

 public:
  MsgDispatcher() = default;
  virtual ~MsgDispatcher() = default;

  MsgDispatcher(const MsgDispatcher&) = delete;
  MsgDispatcher(const MsgDispatcher&&) = delete;
  MsgDispatcher& operator=(const MsgDispatcher&) = delete;
  MsgDispatcher& operator=(const MsgDispatcher&&) = delete;

 public:
  virtual void registerHandler(AKIpUptaneMes_PR msg_id, Handler handler);
  virtual HandleStatusCode handleMsg(const Asn1Message::Ptr& in_msg, Asn1Message::Ptr& out_msg);

 private:
  std::unordered_map<unsigned int, MsgDispatcher::Handler> handler_map_;
};

class AktualizrSecondaryMsgDispatcher : public MsgDispatcher {
 public:
  AktualizrSecondaryMsgDispatcher(IAktualizrSecondary& secondary);

 protected:
  HandleStatusCode getInfoHdlr(Asn1Message& in_msg, Asn1Message& out_msg);
  HandleStatusCode getManifestHdlr(Asn1Message& in_msg, Asn1Message& out_msg);
  HandleStatusCode putMetaHdlr(Asn1Message& in_msg, Asn1Message& out_msg);
  HandleStatusCode sendFirmwareHdlr(Asn1Message& in_msg, Asn1Message& out_msg);
  HandleStatusCode installHdlr(Asn1Message& in_msg, Asn1Message& out_msg);

 private:
  IAktualizrSecondary& secondary_;
};

#endif  // MSGDISPATCHER_H