You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
106 lines
3.1 KiB
106 lines
3.1 KiB
#include <memory>
|
|
#include <string>
|
|
#include <stdexcept>
|
|
|
|
#include <lib/libwebsockets.h>
|
|
#include "websockets.h"
|
|
|
|
#include "json.h"
|
|
#include "json_ext.h"
|
|
|
|
#include "basics.h"
|
|
#include "app.h"
|
|
|
|
|
|
std::unique_ptr<App> app;
|
|
|
|
using namespace websockets;
|
|
|
|
websockets::TestProtocol<User> default_protocol{
|
|
// connection established
|
|
[](User& user, basic_websocket_info){
|
|
app->login(user);
|
|
std::cout << "New user!\n";
|
|
},
|
|
// connection closed
|
|
[](User& user, basic_websocket_info){
|
|
app->logout(user);
|
|
std::cout << "User left: " << &user << ": " << user.name << std::endl;
|
|
},
|
|
// write (will always come after receive)
|
|
[](User& user, basic_websocket_info) -> std::string{
|
|
js::Object ret;
|
|
if(user.update_lines){
|
|
ret["lines"] = vector_to_json(app->lines, [](AbstractLine const & l){return cheap_line_type(l);});
|
|
user.update_lines = false;
|
|
} else {
|
|
ret["balls"] = vector_to_json(app->sim.balls, [](ball_type const & b){return cheap_ball_type(b);});
|
|
}
|
|
return write_json(ret);
|
|
},
|
|
// receive
|
|
[](User& user, std::string in, basic_websocket_info){
|
|
try{
|
|
auto object = parse_json(in).getObject();
|
|
auto command = object["command"];
|
|
if(command == "add line"){
|
|
app->add_line(from_json<AbstractLine>(object["data"]));
|
|
std::cout << "Line from: " << &user << ": " << user.name << std::endl;
|
|
} else if (command == "update user") {
|
|
user = from_json<IncomingPacket>(object["data"]);
|
|
std::cout << "Updated user: " << &user << ": " << user.name << std::endl;
|
|
}
|
|
} catch(std::exception& e) {
|
|
throw websockets::runtime_error(e.what());
|
|
}
|
|
}
|
|
};
|
|
|
|
websockets::TestProtocol<Empty> observer_protocol{
|
|
// connection established
|
|
[](Empty& user, basic_websocket_info){},
|
|
// connection closed
|
|
[](Empty& user, basic_websocket_info){},
|
|
// write (will always come after receive)
|
|
[](Empty& user, basic_websocket_info) -> std::string{
|
|
js::Object ret;
|
|
ret["people"] = vector_to_json(app->people_online, [](User* u){return *u;});
|
|
ret["balls"] = vector_to_json(app->sim.balls);
|
|
ret["lines"] = vector_to_json(app->sim.lines);
|
|
return write_json(ret);
|
|
},
|
|
// receive
|
|
[](Empty& user, std::string in, basic_websocket_info){}
|
|
};
|
|
|
|
websockets::TestProtocol<Empty> uberclient_protocol{
|
|
// connection established
|
|
[](Empty& user, basic_websocket_info binfo){
|
|
if(app->uberclient) throw websockets::runtime_error("Warning: another uberclient connection\n");
|
|
app->uberclient = binfo.wsi;
|
|
},
|
|
// connection closed
|
|
[](Empty& user, basic_websocket_info){
|
|
app->uberclient = nullptr;
|
|
},
|
|
// write (will always come after receive)
|
|
[](Empty& user, basic_websocket_info) -> std::string{
|
|
static int i = 0;
|
|
return std::to_string(i++);
|
|
},
|
|
// receive
|
|
[](Empty& user, std::string in, basic_websocket_info){}
|
|
};
|
|
|
|
int main(int argc, char **argv){
|
|
app.reset(new App);
|
|
|
|
libwebsocket_protocols protocols[] = {
|
|
WSstandard_protocol("default", default_protocol),
|
|
WSstandard_protocol("observer", observer_protocol),
|
|
WSstandard_protocol("uberclient", uberclient_protocol),
|
|
{ NULL, NULL, 0 } // end of list
|
|
};
|
|
|
|
return websockets::default_main(argc, argv, protocols, []{app->update();});
|
|
}
|
|
|