Implement channels.

Author: NikoGrano

src/main.cpp

@@ -1,65 +1,154 @@
+#include <set>
 #include <websocketpp/config/asio_no_tls.hpp>
 #include <websocketpp/server.hpp>
-
 #include <iostream>
 
-typedef websocketpp::server<websocketpp::config::asio> server;
 
+struct connection_data {
+    int session;
+    std::string room;
+};
+
+struct custom_config : public websocketpp::config::asio {
+    // Default settings from our core config
+    typedef websocketpp::config::asio core;
+    
+    typedef core::concurrency_type concurrency_type;
+    typedef core::request_type request_type;
+    typedef core::response_type response_type;
+    typedef core::message_type message_type;
+    typedef core::con_msg_manager_type con_msg_manager_type;
+    typedef core::endpoint_msg_manager_type endpoint_msg_manager_type;
+    typedef core::alog_type alog_type;
+    typedef core::elog_type elog_type;
+    typedef core::rng_type rng_type;
+    typedef core::transport_type transport_type;
+    typedef core::endpoint_base endpoint_base;
+    
+    // Set a custom connection_base class to handle indetification.
+    typedef connection_data connection_base;
+};
+
+typedef websocketpp::server<custom_config> server;
+typedef server::connection_ptr connection_ptr;
+typedef server::message_ptr message_ptr;
+typedef server::connection_ptr connection_ptr;
+
+using websocketpp::connection_hdl;
 using websocketpp::lib::placeholders::_1;
 using websocketpp::lib::placeholders::_2;
 using websocketpp::lib::bind;
 
-// pull out the type of messages sent by our config
-typedef server::message_ptr message_ptr;
 
-// Define a callback to handle incoming messages
-void on_message(server* s, websocketpp::connection_hdl hdl, message_ptr msg) {
-    try {
-        s->send(hdl, msg->get_payload(), msg->get_opcode());
-    } catch (websocketpp::exception const & e) {
-        std::cout << "Error: "
-                  << "(" << e.what() << ")" << std::endl;
+// This class will be heart of our app.
+class EchoServer
+{
+public:
+    EchoServer()
+    {
+        // Try to not fail. Please. I have faith on you.
+        try {
+            // Set logging settings
+            _server.set_access_channels(websocketpp::log::alevel::none);
+            _server.clear_access_channels(websocketpp::log::alevel::none);
+            _server.init_asio();
+
+            // Message handler
+            _server.set_open_handler(bind(&EchoServer::onOpen,this,::_1));
+            _server.set_close_handler(bind(&EchoServer::onClose,this,::_1));
+            _server.set_message_handler(bind(&EchoServer::onMessage,this,::_1,::_2));
+            _server.listen(80);
+
+            // Tell something :)
+            std::cout << "Starting GameTactic CPP Echo server...\n";
+            std::cout << "--------------------------------------------------------\n";
+            std::cout << "|     Copyright 2019 Niko Granö <niko@ironlions.fi>    |\n";
+            std::cout << "|                 Licensed under GPLv3.                |\n";
+            std::cout << "|                 https://gametactic.eu                |\n";
+            std::cout << "--------------------------------------------------------\n\n";
+
+            // Start the server.
+            _server.start_accept();
+            _server.run();
+        } catch (websocketpp::exception const & e) {
+            std::cout << e.what() << std::endl;
+        } catch (...) {
+            std::cout << "Error: " << std::endl;
+        }
     }
-}
 
+    // Callback to handle incoming messages
+    void onMessage(connection_hdl conn, message_ptr msg) {
+        // Get connection
+        connection_ptr ptr = _server.get_con_from_hdl(conn);
+
+        // Check if client is requesting to join.
+        if (msg->get_payload().substr(0,5) == "join:") {
+            std::string room = msg->get_payload().substr(5);
+            ptr->room = room;
+            std::stringstream ss;
+            ss << "{\"success\":\"Room " << room << " selected.\"}";
+            ptr->send(ss.str());
+
+            return;
+        }
+
+        if (ptr->room.empty()) {
+            ptr->send("{\"error\":\"No room selected.\"}");
+
+            return;
+        }
+
+        try {
+            for (auto c : _connections) {
+                connection_ptr _ptr = _server.get_con_from_hdl(c);
+                if (_ptr->room == ptr->room) {
+                    _server.send(c,msg);
+                }
+            }
+        } catch (websocketpp::exception const &e) {
+            std::cout << "Error: " << "(" << e.what() << ")" << std::endl;
+        }
+    }
+
+    // Callback to handle new connections
+    void onOpen(connection_hdl conn) {
+        connection_ptr ptr = _server.get_con_from_hdl(conn);
+        ptr->session = _session++;
+        _connections.insert(conn);
+    }
+
+    // Callback to handle closing connection
+    void onClose(connection_hdl conn) {
+        connection_ptr ptr = _server.get_con_from_hdl(conn);
+        _connections.erase(conn);
+    }
+
+private:
+    typedef std::set<connection_hdl,std::owner_less<connection_hdl>> con_list;
+
+    // Create a server and list for connections.
+    server _server;
+    con_list _connections;
+
+    // Every client has own id to help identify.
+    int _session;
+
+    // Rooms
+    //std::map<int, string::string> _rooms;
+    //std::vector<std::map<std::string, std::string>> _rooms;
+};
+
+// Basic signal hander
 void signalHandler( int signum ) {
    std::cout << "Interrupt signal (" << signum << ") received.\n";
    exit(signum);
 }
 
-
 int main() {
     // Handle signals.
     signal(SIGINT, signalHandler);
 
-    // Create a server endpoint
-    server _server;
-
-    try {
-        // Set logging settings
-        _server.set_access_channels(websocketpp::log::alevel::none);
-        _server.clear_access_channels(websocketpp::log::alevel::none);
-        _server.init_asio();
-
-        // Message handler
-        _server.set_message_handler(bind(&on_message,&_server,::_1,::_2));
-        _server.listen(80);
-
-	// Tell something :)
-	std::cout << "Starting GameTactic CPP Echo server...\n";
-        std::cout << "--------------------------------------------------------------------------------\n";
-	std::cout << "|                 Copyright 2019 Niko Granö <niko@ironlions.fi>                |\n";
-	std::cout << "|                             Licensed under GPLv3.                            |\n";
-        std::cout << "|                             https://gametactic.eu                            |\n";
-	std::cout << "--------------------------------------------------------------------------------\n\n";
-
-        // Start the server.
-        // First start itself, second start asio.
-        _server.start_accept();
-        _server.run();
-    } catch (websocketpp::exception const & e) {
-        std::cout << e.what() << std::endl;
-    } catch (...) {
-        std::cout << "Error: " << std::endl;
-    }
+    // Start app.
+    EchoServer srv;
 }