nebula_atom.h

/*
* Nebula Atom
 
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* This program 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 General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program.  If not, see <http://www.gnu.org/licenses/>.
*/
 
#ifndef NAF_CORE_NEBULAATOM
#define NAF_CORE_NEBULAATOM
 
 
#include <string>
#include <vector>
#include <iostream>
#include <memory>
#include <exception>
 
#include <Poco/Net/ServerSocket.h>
#include <Poco/Net/HTTPServer.h>
#include <Poco/Net/HTTPServerParams.h>
#include <Poco/Net/NetSSL.h>
#include "Poco/Format.h"
#include "Poco/Exception.h"
#include "Poco/Net/SecureStreamSocket.h"
#include "Poco/Net/SecureServerSocket.h"
#include "Poco/Net/X509Certificate.h"
#include "Poco/Net/SSLManager.h"
#include "Poco/Net/SSLException.h"
#include "Poco/Net/KeyConsoleHandler.h"
#include "Poco/Net/AcceptCertificateHandler.h"
#include "Poco/Net/ConsoleCertificateHandler.h"
#include <Poco/Crypto/RSAKey.h>
#include <Poco/Net/AcceptCertificateHandler.h>
#include <Poco/Net/Context.h>
#include <Poco/Net/NetException.h>
#include <Poco/Net/NetSSL.h>
#include <Poco/Net/SSLManager.h>
 
#include "core/server.h"
#include "core/handler_factory.h"
#include "query/database_manager.h"
#include "tools/settings_manager.h"
#include "tools/output_logger.h"
#include "tools/sessions_manager.h"
#include "security/permissions_manager.h"
 
 
namespace NAF
{
    namespace Core
    {
        class NebulaAtom;
    }
}
 
using namespace Poco;
using namespace Poco::Net;
using namespace Poco::Util;
 
class NAF::Core::NebulaAtom
{
    public:
        NebulaAtom(bool use_ssl = false);
        virtual ~NebulaAtom();
 
        bool get_use_ssl() const { return use_ssl_; }
        Core::Server::Ptr get_server() const { return server_; }
        HandlerFactory* get_handler_factory()
        {
            auto& var = handler_factory_;
            return var;
        }
        std::vector<std::string>& get_console_parameters()
        {
            auto& var = console_parameters_;
            return var;
        }
 
        int Init_();
        int Init_(int argc, char** argv);
        void CustomHandlerCreator_(HandlerFactory::FunctionHandlerCreator handler_creator);
        void AddHandler_(std::string route, HandlerFactory::FunctionHandler handler);
        void SetupSettings_();
 
    protected:
        void AddBasicSettings_();
 
    private:
        const bool use_ssl_;
        Core::Server::Ptr server_;
        HandlerFactory* handler_factory_;
        std::vector<std::string> console_parameters_;
};
 
#endif // NAF_CORE_NEBULAATOM