95 lines
2.2 KiB
C++
95 lines
2.2 KiB
C++
#include "../common/common.h"
|
|
#include "../http/http_connector.h"
|
|
#include "../http/users.h"
|
|
#include "../ws/http_handshake.h"
|
|
#include "../ws/ssl_ws.h"
|
|
#include "../sql/db_init.h"
|
|
#include "../sql/db_handle.h"
|
|
#include "../parser/parser.h"
|
|
|
|
int connect_ws() {
|
|
|
|
bool done = false;
|
|
std::string input;
|
|
ws::websocket_endpoint endpoint;
|
|
|
|
/* Get session id */
|
|
std::string sid = ws::http_handshake::get_sid();
|
|
|
|
/* Start connection */
|
|
std::string uri = "wss://nasfaq.biz/socket/?user=314d0bda-d7f0-4636-aed7-5ea02743604b&EIO=4&transport=websocket&sid=" + sid;
|
|
int id = endpoint.connect(uri);
|
|
if(id != -1) {
|
|
std::cout << ">Created nasfaq websocket connection" << std::endl;
|
|
} else {
|
|
return -1;
|
|
}
|
|
sleep(1);
|
|
|
|
/* Display connection metadata */
|
|
ws::connection_metadata::ptr metadata = endpoint.get_metadata(id);
|
|
if (metadata) {
|
|
std::cout << *metadata << std::endl;
|
|
} else {
|
|
std::cout << ">Unknown connection id " << id << std::endl;
|
|
}
|
|
|
|
/* Populate database if needed. */
|
|
db::create::database::all();
|
|
|
|
/* Open up database connection. */
|
|
pqxx::connection C("dbname = nasfaq user = steaky \
|
|
hostaddr = 127.0.0.1 port = 5432");
|
|
|
|
if (C.is_open()) {
|
|
std::cout << "Opened database successfully: " << C.dbname() << std::endl;
|
|
} else {
|
|
std::cout << "Can't open database" << std::endl;
|
|
return 1;
|
|
}
|
|
|
|
/* Update users. */
|
|
db::push_users(&C, users::get_users());
|
|
|
|
/* Set up parser */
|
|
parser::parser p(metadata, &C);
|
|
|
|
/* Loop and print data through parser*/
|
|
while(!done) {
|
|
input = std::cin.get();
|
|
|
|
if(input == "q" || input == "quit") {
|
|
done = true;
|
|
} else if (input == "p") {
|
|
std::cout << "PROCESSING QUEUE " << std::endl;
|
|
p.process_queue_start();
|
|
} else if (input == "s") {
|
|
std::cout << "STOPPING QUEUE PROCESSING" << std::endl;
|
|
p.process_queue_stop();
|
|
} else if (input == "t") {
|
|
std::cout << "TEST" << std::endl;
|
|
auto ret = db::pull_last_cycle(&C);
|
|
std::cout << ret << std::endl;
|
|
}
|
|
}
|
|
|
|
/* Close websocket */
|
|
std::stringstream ss(input);
|
|
|
|
std::string cmd;
|
|
int close_code = websocketpp::close::status::normal;
|
|
std::string reason;
|
|
|
|
ss >> cmd >> id >> close_code;
|
|
std::getline(ss, reason);
|
|
|
|
endpoint.close(id, close_code, reason);
|
|
|
|
return 0;
|
|
}
|
|
|
|
int main(void) {
|
|
connect_ws();
|
|
return 1;
|
|
}
|