mirror of https://github.com/oxen-io/lokinet
58 lines
2.0 KiB
C++
58 lines
2.0 KiB
C++
#include <common.hpp>
|
|
#include <llarp.hpp>
|
|
#include <llarp/tooling/hive_context.hpp>
|
|
#include <llarp/router/router.hpp>
|
|
#include <llarp/handlers/pyhandler.hpp>
|
|
#include <llarp/service/protocol_type.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 isRouter) {
|
|
self->Setup({false, false, isRouter});
|
|
})
|
|
.def("Run", [](Context_ptr self) -> int { return self->Run(RuntimeOptions{}); })
|
|
.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->SendToOrQueue(to, std::move(buf), service::ProtocolType::Control);
|
|
})
|
|
.def(
|
|
"AddEndpoint",
|
|
[](Context_ptr self, handlers::PythonEndpoint_ptr ep) {
|
|
self->router->hiddenServiceContext().InjectEndpoint(ep->OurName, ep);
|
|
})
|
|
.def("CallSafe", &Context::CallSafe);
|
|
}
|
|
|
|
} // namespace llarp
|
|
|
|
namespace tooling
|
|
{
|
|
void
|
|
HiveContext_Init(py::module& mod)
|
|
{
|
|
using HiveContext_ptr = std::shared_ptr<HiveContext>;
|
|
py::class_<tooling::HiveContext, HiveContext_ptr, llarp::Context>(mod, "HiveContext")
|
|
.def(
|
|
"getRouterAsHiveRouter",
|
|
&tooling::HiveContext::getRouterAsHiveRouter,
|
|
py::return_value_policy::reference);
|
|
}
|
|
} // namespace tooling
|