lokinet/pybind/llarp/context.cpp

#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