router.hpp

Go to the documentation of this file.

#pragma once

#include <regex>
#include <string>
#include <vector>
#include <utility>
#include <functional>
#include <unordered_set>
#include <unordered_map>
#include <optional>

#include <boost/beast/http/verb.hpp>
#include <boost/asio/ip/tcp.hpp>

#include "beast_common.hpp"

namespace http {
    enum class Method {
        GET = boost::beast::http::verb::get,
        HEAD = boost::beast::http::verb::head,
        POST = boost::beast::http::verb::post,
        PUT = boost::beast::http::verb::put,
        DELETE_ = boost::beast::http::verb::delete_,
        CONNECT = boost::beast::http::verb::connect,
        OPTIONS = boost::beast::http::verb::options,
        TRACE = boost::beast::http::verb::trace
    };

    class Router {
        public:
            typedef std::unordered_map<std::string, std::string> matches;

            struct Context {
                const boost::asio::ip::tcp::socket& socket;
                const dynamic_request& request;
                dynamic_response& response;
                const matches matches;

                //Since we store references here better to not let it get copied around
                //Of course nothing prevents user from storing particular reference somewhere,
                //but it is not my problem. I did all I could.
                Context(const Context& that) = delete;
                Context(Context&& other) noexcept = delete;
            };
            typedef std::function<void(Context&&)> router_handler;

            class Route {
                public:
                    Route(const char* path, router_handler fn);
                    std::optional<matches> match(const char* uri) const;
                    std::optional<matches> match(boost::beast::string_view uri) const;

                    router_handler callback;

                private:
                    std::regex path;
                    std::vector<std::string> keys;
                    const char* component_re = ":([^\\/]+)?";
                    const char* replace_re = "(?:([^\\/]+?))";

            };

            Router() noexcept;

            Router& add_route(Method method, const char* route, router_handler fn);

            bool dispatch(const boost::asio::ip::tcp::socket& socket, const dynamic_request& request, dynamic_response& response) const noexcept;

            bool is_method_used(boost::beast::http::verb method) const noexcept;

        private:
            std::unordered_set<int> avail_methods;

            typedef std::vector<Route> handler_map_t;
            handler_map_t get_handlers;
            handler_map_t head_handlers;
            handler_map_t post_handlers;
            handler_map_t put_handlers;
            handler_map_t delete_handlers;
            handler_map_t options_handlers;
            handler_map_t trace_handlers;

            handler_map_t* map_method(boost::beast::http::verb method) noexcept;
            const handler_map_t* map_method(boost::beast::http::verb method) const noexcept;
    };
}