nostr/strfry
1
0
Fork 0
mirror of https://github.com/hoytech/strfry.git synced 2025-06-17 08:48:51 +00:00
Code Issues Packages Projects Releases Wiki Activity

EventStreamer

Browse Source
This commit is contained in:
Doug Hoyte
2023-08-16 23:58:44 -04:00
parent d66de14667
commit 7edc9edf1f
3 changed files with 118 additions and 83 deletions
Download Patch File Download Diff File Expand all files Collapse all files

85
src/EventStreamer.h Normal file
View File

@ -0,0 +1,85 @@
#pragma once
#include <hoytech/protected_queue.h>
#include "golpe.h"
#include "WSConnection.h"
struct EventStreamer {
std::string url;
WSConnection ws;
std::string dir;
tao::json::value filter;
std::function<void(tao::json::value &&, const WSConnection &ws)> onEvent;
private:
hoytech::protected_queue<std::shared_ptr<std::string>> inbox;
public:
EventStreamer(const std::string &url, const std::string &dir, tao::json::value filter = tao::json::empty_object) : url(url), ws(url), dir(dir), filter(filter) {
filter["limit"] = 0;
}
void sendEvent(std::shared_ptr<std::string> msg) {
inbox.push_move(std::move(msg));
}
void trigger() {
ws.trigger();
}
void run() {
ws.onConnect = [&]{
if (dir == "down" || dir == "both") {
auto encoded = tao::json::to_string(tao::json::value::array({ "REQ", "X", filter }));
ws.send(encoded);
}
};
ws.onMessage = [&](auto msg, uWS::OpCode, size_t){
auto origJson = tao::json::from_string(msg);
if (origJson.is_array()) {
if (origJson.get_array().size() < 2) throw herr("array too short");
auto &msgType = origJson.get_array().at(0);
if (msgType == "EOSE") {
return;
} else if (msgType == "NOTICE") {
LW << "NOTICE message from " << url << " : " << tao::json::to_string(origJson);
return;
} else if (msgType == "OK") {
if (!origJson.get_array().at(2).get_boolean()) {
LW << "Event not written by " << url << " : " << origJson;
}
} else if (msgType == "EVENT") {
if (dir == "down" || dir == "both") {
if (origJson.get_array().size() < 3) throw herr("array too short");
auto &evJson = origJson.at(2);
// FIXME: validate that the event actually matches provided filter?
if (onEvent) onEvent(std::move(evJson), ws);
} else {
LW << "Unexpected EVENT from " << url;
}
} else {
throw herr("unexpected first element");
}
} else {
throw herr("unexpected message");
}
};
ws.onTrigger = [&]{
auto msgs = inbox.pop_all();
for (auto &msg : msgs) ws.send(*msg);
};
ws.run();
}
};

31
src/apps/mesh/cmd_router.cpp Normal file
View File

@ -0,0 +1,31 @@
#include <docopt.h>
#include <tao/json.hpp>
#include <hoytech/file_change_monitor.h>
#include "golpe.h"
#include "EventStreamer.h"
#include "WriterPipeline.h"
#include "PluginWritePolicy.h"
#include "events.h"
static const char USAGE[] =
R"(
Usage:
router <routerConfigFile>
Options:
)";
void cmd_router(const std::vector<std::string> &subArgs) {
std::map<std::string, docopt::value> args = docopt::docopt(USAGE, subArgs, true, "");
std::string routerConfigFile = args["<routerConfigFile>"].asString();
tao::config::value configJson = loadRawTaoConfig(routerConfigFile);
std::cout << configJson << std::endl;
}

85
src/apps/mesh/cmd_stream.cpp
View File

@ -1,15 +1,13 @@
#include <docopt.h>
#include <tao/json.hpp>
#include <hoytech/protected_queue.h>
#include <hoytech/file_change_monitor.h>
#include "golpe.h"
#include "EventStreamer.h"
#include "WriterPipeline.h"
#include "Subscription.h"
#include "WSConnection.h"
#include "events.h"
#include "PluginWritePolicy.h"
#include "events.h"
static const char USAGE[] =
@ -23,85 +21,6 @@ R"(
struct EventStreamer {
std::string url;
WSConnection ws;
std::string dir;
tao::json::value filter;
std::function<void(tao::json::value &&, const WSConnection &ws)> onEvent;
private:
hoytech::protected_queue<std::shared_ptr<std::string>> inbox;
public:
EventStreamer(const std::string &url, const std::string &dir, tao::json::value filter = tao::json::empty_object) : url(url), ws(url), dir(dir), filter(filter) {
filter["limit"] = 0;
}
void sendEvent(std::shared_ptr<std::string> msg) {
inbox.push_move(std::move(msg));
}
void trigger() {
ws.trigger();
}
void run() {
ws.onConnect = [&]{
if (dir == "down" || dir == "both") {
auto encoded = tao::json::to_string(tao::json::value::array({ "REQ", "X", filter }));
ws.send(encoded);
}
};
ws.onMessage = [&](auto msg, uWS::OpCode, size_t){
auto origJson = tao::json::from_string(msg);
if (origJson.is_array()) {
if (origJson.get_array().size() < 2) throw herr("array too short");
auto &msgType = origJson.get_array().at(0);
if (msgType == "EOSE") {
return;
} else if (msgType == "NOTICE") {
LW << "NOTICE message from " << url << " : " << tao::json::to_string(origJson);
return;
} else if (msgType == "OK") {
if (!origJson.get_array().at(2).get_boolean()) {
LW << "Event not written by " << url << " : " << origJson;
}
} else if (msgType == "EVENT") {
if (dir == "down" || dir == "both") {
if (origJson.get_array().size() < 3) throw herr("array too short");
auto &evJson = origJson.at(2);
// FIXME: validate that the event actually matches provided filter?
if (onEvent) onEvent(std::move(evJson), ws);
} else {
LW << "Unexpected EVENT from " << url;
}
} else {
throw herr("unexpected first element");
}
} else {
throw herr("unexpected message");
}
};
ws.onTrigger = [&]{
auto msgs = inbox.pop_all();
for (auto &msg : msgs) ws.send(*msg);
};
ws.run();
}
};
void cmd_stream(const std::vector<std::string> &subArgs) {
std::map<std::string, docopt::value> args = docopt::docopt(USAGE, subArgs, true, "");