WSServer.hpp

Go to the documentation of this file.

/*
    Copyright (C) 2014-2016 Leosac
 
    This file is part of Leosac.
 
    Leosac is free software: you can redistribute it and/or modify
    it under the terms of the GNU Affero General Public License as published by
    the Free Software Foundation, either version 3 of the License, or
    (at your option) any later version.
 
    Leosac is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
    GNU Affero General Public License for more details.
 
    You should have received a copy of the GNU Affero General Public License
    along with this program. If not, see <http://www.gnu.org/licenses/>.
*/
 
#pragma once
 
#include "LeosacFwd.hpp"
#include "Messages.hpp"
#include "Service.hpp"
#include "WebSockFwd.hpp"
#include "api/APIAuth.hpp"
#include "api/APISession.hpp"
#include "api/CRUDResourceHandler.hpp"
#include "api/MethodHandler.hpp"
#include "core/APIStatusCode.hpp"
#include "core/audit/AuditFwd.hpp"
#include "tools/db/db_fwd.hpp"
#include <boost/optional.hpp>
#include <set>
#include <type_traits>
#include <websocketpp/config/asio_no_tls.hpp>
#include <websocketpp/server.hpp>
#include <zmqpp/zmqpp.hpp>
 
namespace Leosac
{
namespace Module
{
namespace WebSockAPI
{
using json = nlohmann::json;
 
class WSServer
{
  public:
    WSServer(WebSockAPIModule &module, DBPtr database);
    ~WSServer();
 
    using Server           = websocketpp::server<websocketpp::config::asio>;
    using ConnectionAPIMap = std::map<websocketpp::connection_hdl, APIPtr,
                                      std::owner_less<websocketpp::connection_hdl>>;
 
    void run(const std::string &interface, uint16_t port);
 
    Server srv_;
 
    void start_shutdown();
 
    bool register_asio_handler(const Service::WSHandler &handler,
                               const std::string &name);
 
 
    void register_crud_handler_external(const std::string &resource_name,
                                        CRUDResourceHandler::Factory factory);
    void unregister_handler(const std::string &name);
 
    APIAuth &auth();
 
    DBPtr db();
 
    DBServicePtr dbsrv();
 
    CoreUtilsPtr core_utils();
 
    void clear_user_sessions(Auth::UserPtr user, APIPtr exception);
 
  private:
    void on_open(websocketpp::connection_hdl hdl);
 
    void on_close(websocketpp::connection_hdl hdl);
 
    void attempt_unregister_ws_service();
 
    void on_message(websocketpp::connection_hdl hdl, Server::message_ptr msg);
 
    boost::optional<ServerMessage> handle_request(APIPtr api_handle,
                                                  const ClientMessage &msg,
                                                  Audit::IAuditEntryPtr);
 
    ClientMessage parse_request(const json &in);
 
    void send_message(websocketpp::connection_hdl hdl, const ServerMessage &msg);
 
    void register_crud_handler(const std::string &resource_name,
                               CRUDResourceHandler::Factory factory);
 
    boost::optional<json> dispatch_request(APIPtr api_handle,
                                           const ClientMessage &in,
                                           Audit::IAuditEntryPtr);
 
    bool has_handler(const std::string &name) const;
 
    void finalize_audit(const Audit::IWSAPICallPtr &audit, ServerMessage &msg);
 
    ConnectionAPIMap connection_session_;
    APIAuth auth_;
 
    std::map<std::string, json (APISession::*)(const json &)> handlers_;
 
    std::map<std::string, MethodHandler::Factory> individual_handlers_;
 
    std::map<std::string, CRUDResourceHandler::Factory> crud_handlers_;
 
    std::map<std::string, Service::WSHandler> asio_handlers_;
 
    DBServicePtr dbsrv_;
 
    WebSockAPIModule &module_;
 
    std::unique_ptr<boost::asio::io_service::work> work_;
};
}
}
}