#include "common.hpp"
#include <llarp.hpp>
#include <router/router.cpp>
#include "llarp/handlers/pyhandler.hpp"
namespace llarp
{
  void
  Context_Init(py::module& mod)
  {
    using Context_ptr = std::shared_ptr<Context>;
    py::class_<Context, Context_ptr>(mod, "Context")
        .def("Setup", [](Context_ptr self) -> bool { return self->Setup() == 0; })
        .def("Run", [](Context_ptr self) -> int { return self->Run(llarp_main_runtime_opts{}); })
        .def("Stop", [](Context_ptr self) { self->CloseAsync(); })
        .def("IsUp", &Context::IsUp)
        .def("IsRelay", [](Context_ptr self) -> bool { return self->router->IsServiceNode(); })
        .def("LooksAlive", &Context::LooksAlive)
        .def("Configure", &Context::Configure)
        .def(
            "TrySendPacket",
            [](Context_ptr self, std::string from, service::Address to, std::string pkt) {
              auto ep = self->router->hiddenServiceContext().GetEndpointByName(from);
              std::vector<byte_t> buf;
              buf.resize(pkt.size());
              std::copy_n(pkt.c_str(), pkt.size(), buf.data());
              return ep and ep->SendToServiceOrQueue(to, std::move(buf), service::eProtocolControl);
            })
        .def(
            "AddEndpoint",
            [](Context_ptr self, handlers::PythonEndpoint_ptr ep) {
              self->router->hiddenServiceContext().InjectEndpoint(ep->OurName, ep);
            })
        .def("CallSafe", &Context::CallSafe);
  }
}  // namespace llarp