router.h

Go to the documentation of this file.

 
#ifndef MB_ROUTER_H
#define MB_ROUTER_H
 
#include <memory>
#include <vector>
#include <nlohmann/json.hpp>
#include <httplib.h>
 
#include "route_registry.h"
#include "models/entity.h"
#include "../utils/utils.h"
#include  "types.h"
 
namespace mb {
    class SSEMgr;
 
    class Router {
    public:
        Router();
 
        ~Router();
 
        bool init();
 
        bool listen();
 
        void close();
 
        httplib::Server &server();
 
        SSEMgr& sseMgr() const;
 
        // ----------- HTTP METHODS ----------- //
        void Get(const std::string &path, const HandlerFn &handler, const Middlewares &middlewares = {});
 
        void Post(const std::string &path, const HandlerWithContentReaderFn &handler,
                  const Middlewares &middlewares = {});
 
        void Post(const std::string &path, const HandlerFn &handler, const Middlewares &middlewares = {});
 
        void Patch(const std::string &path, const HandlerWithContentReaderFn &handler,
                   const Middlewares &middlewares = {});
 
        void Patch(const std::string &path, const HandlerFn &handler, const Middlewares &middlewares = {});
 
        void Delete(const std::string &path, const HandlerFn &handler, const Middlewares &middlewares = {});
 
        // ----------- SCHEMA CACHE METHODS ----------- //
        const json &schemaCache(const std::string &table_name) const;
 
        bool hasSchemaCache(const std::string &table_name) const;
 
        Entity schemaCacheEntity(const std::string &table_name) const;
 
        void addSchemaCache(const nlohmann::json &entity_schema);
 
        void updateSchemaCache(const std::string &old_entity_name, const json &new_schema);
 
        void removeSchemaCache(const std::string &entity_name);
 
        // ----------- UTILS METHODS ----------- //
        std::string decompressResponseBody(const std::string &body, const std::string &encoding);
 
    private:
        void globalRouteHandler(const std::string &method, const std::string &path);
 
        void globalRouteHandlerWithReader(const std::string &method, const std::string &path);
 
        void generateMiscEndpoints();
 
        void registerEntityRoutes();
 
    void registerSchemaRoutes();
 
    void registerAuthRoutes();
 
    static std::string getMimeType(const std::string &path);
 
        // ----------- REQ/RES METHODS ----------- //
        static std::function<void(const MantisRequest &, MantisResponse &)> handleAdminDashboardRoute() ;
 
        static std::function<void(const MantisRequest &, MantisResponse &)> fileServingHandler();
 
        static std::function<void(const MantisRequest &, MantisResponse &)> healthCheckHandler();
 
        std::function<HandlerResponse(const httplib::Request &, httplib::Response &)> preRoutingHandler();
 
        std::function<void(const httplib::Request &, httplib::Response &)> postRoutingHandler();
 
        std::function<void(const httplib::Request &, httplib::Response &)> optionsHandler();
 
        std::function<void(const httplib::Request &, const httplib::Response &)> routingLogger();
 
        std::function<void(const httplib::Request &, httplib::Response &)> routingErrorHandler();
 
        std::function<void(MantisRequest &, MantisResponse &)> handleAuthLogin();
 
        std::function<void(MantisRequest &, MantisResponse &)> handleAdminLogin();
 
        std::function<void(MantisRequest &, MantisResponse &)> handleAuthRefresh();
 
        std::function<void(MantisRequest &, MantisResponse &)> handleAuthLogout();
 
        std::function<void(MantisRequest &, MantisResponse &)> handleSetupAdmin();
 
        static std::function<void(const MantisRequest &, MantisResponse &)> handleLogs();
 
 
        // Member Variables
        MantisBase &mApp;
        httplib::Server svr;
        RouteRegistry m_routeRegistry;
        std::unique_ptr<SSEMgr> m_sseMgr;
        std::vector<MiddlewareFn> m_preRoutingMiddlewares;
        std::vector<HandlerFn> m_postRoutingMiddlewares;
        std::unordered_map<std::string, Entity> m_entityMap;
    };
} // mb
 
#endif // MB_ROUTER_H