#include "common.hpp"

#include "router/abstractrouter.hpp"
#include "tooling/hive_router.hpp"

namespace llarp
{
  void
  AbstractRouter_Init(py::module& mod)
  {
    py::class_<AbstractRouter>(mod, "AbstractRouter")
        .def("rc", &AbstractRouter::rc)
        .def("Stop", &AbstractRouter::Stop)
        .def("peerDb", &AbstractRouter::peerDb);
  }

}  // namespace llarp

namespace tooling
{
  void
  HiveRouter_Init(py::module& mod)
  {
    py::class_<HiveRouter, llarp::AbstractRouter>(mod, "HiveRouter")
        .def("disableGossiping", &HiveRouter::disableGossiping)
        .def("enableGossiping", &HiveRouter::enableGossiping);
  }
}  // namespace tooling