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_;
};
}
}
}