Crow/include/crow/routing.h

1346 lines
47 KiB
C
Raw Normal View History

2014-04-02 16:38:08 +00:00
#pragma once
#include <cstdint>
2014-04-02 20:31:32 +00:00
#include <utility>
#include <tuple>
2014-04-10 16:43:33 +00:00
#include <unordered_map>
#include <memory>
#include <boost/lexical_cast.hpp>
2014-10-23 17:45:34 +00:00
#include <vector>
2014-04-02 16:38:08 +00:00
#include "crow/common.h"
#include "crow/http_response.h"
#include "crow/http_request.h"
#include "crow/utility.h"
#include "crow/logging.h"
#include "crow/websocket.h"
2014-04-02 16:38:08 +00:00
2014-04-26 17:19:59 +00:00
namespace crow
2014-04-02 16:38:08 +00:00
{
/// A base class for all rules.
/// Used to provide a common interface for code dealing with different types of rules.
/// A Rule provides a URL, allowed HTTP methods, and handlers.
class BaseRule
2014-04-02 16:38:08 +00:00
{
public:
2015-02-18 15:57:01 +00:00
BaseRule(std::string rule)
: rule_(std::move(rule))
{
}
virtual ~BaseRule()
{
}
2020-10-14 21:09:35 +00:00
2014-04-15 13:46:28 +00:00
virtual void validate() = 0;
2016-08-28 05:46:31 +00:00
std::unique_ptr<BaseRule> upgrade()
{
if (rule_to_upgrade_)
return std::move(rule_to_upgrade_);
return {};
}
2014-04-14 15:31:51 +00:00
virtual void handle(const request&, response&, const routing_params&) = 0;
2020-10-14 21:09:35 +00:00
virtual void handle_upgrade(const request&, response& res, SocketAdaptor&&)
{
res = response(404);
res.end();
}
2016-08-28 05:46:31 +00:00
#ifdef CROW_ENABLE_SSL
2020-10-14 21:09:35 +00:00
virtual void handle_upgrade(const request&, response& res, SSLAdaptor&&)
{
res = response(404);
res.end();
}
2016-08-28 05:46:31 +00:00
#endif
uint32_t get_methods()
2014-10-07 12:51:24 +00:00
{
return methods_;
}
template <typename F>
void foreach_method(F f)
{
2021-02-21 00:14:30 +00:00
for(uint32_t method = 0, method_bit = 1; method < static_cast<uint32_t>(HTTPMethod::InternalMethodCount); method++, method_bit<<=1)
{
if (methods_ & method_bit)
f(method);
}
}
const std::string& rule() { return rule_; }
2014-10-07 12:51:24 +00:00
protected:
2021-02-21 00:14:30 +00:00
uint32_t methods_{1<<static_cast<int>(HTTPMethod::Get)};
2015-02-18 15:57:01 +00:00
std::string rule_;
std::string name_;
2016-08-28 05:46:31 +00:00
std::unique_ptr<BaseRule> rule_to_upgrade_;
2016-08-28 05:46:31 +00:00
2015-02-18 15:57:01 +00:00
friend class Router;
template <typename T>
friend struct RuleParameterTraits;
};
2015-02-18 15:57:01 +00:00
namespace detail
{
namespace routing_handler_call_helper
2014-04-14 15:31:51 +00:00
{
2015-02-18 15:57:01 +00:00
template <typename T, int Pos>
struct call_pair
{
using type = T;
static const int pos = Pos;
};
2014-04-14 15:31:51 +00:00
2015-02-18 15:57:01 +00:00
template <typename H1>
struct call_params
2014-04-14 15:31:51 +00:00
{
2015-02-18 15:57:01 +00:00
H1& handler;
const routing_params& params;
const request& req;
response& res;
};
2014-04-14 15:31:51 +00:00
2020-10-14 21:09:35 +00:00
template <typename F, int NInt, int NUint, int NDouble, int NString, typename S1, typename S2>
2015-02-18 15:57:01 +00:00
struct call
2014-04-15 13:46:28 +00:00
{
2015-02-18 15:57:01 +00:00
};
2014-04-15 13:46:28 +00:00
2020-10-14 21:09:35 +00:00
template <typename F, int NInt, int NUint, int NDouble, int NString, typename ... Args1, typename ... Args2>
2015-02-18 15:57:01 +00:00
struct call<F, NInt, NUint, NDouble, NString, black_magic::S<int64_t, Args1...>, black_magic::S<Args2...>>
2014-04-15 13:46:28 +00:00
{
2015-02-18 15:57:01 +00:00
void operator()(F cparams)
{
using pushed = typename black_magic::S<Args2...>::template push_back<call_pair<int64_t, NInt>>;
call<F, NInt+1, NUint, NDouble, NString,
black_magic::S<Args1...>, pushed>()(cparams);
}
};
2014-04-15 13:46:28 +00:00
2020-10-14 21:09:35 +00:00
template <typename F, int NInt, int NUint, int NDouble, int NString, typename ... Args1, typename ... Args2>
2015-02-18 15:57:01 +00:00
struct call<F, NInt, NUint, NDouble, NString, black_magic::S<uint64_t, Args1...>, black_magic::S<Args2...>>
2014-04-15 13:46:28 +00:00
{
2015-02-18 15:57:01 +00:00
void operator()(F cparams)
{
using pushed = typename black_magic::S<Args2...>::template push_back<call_pair<uint64_t, NUint>>;
call<F, NInt, NUint+1, NDouble, NString,
black_magic::S<Args1...>, pushed>()(cparams);
}
};
2014-04-15 13:46:28 +00:00
2020-10-14 21:09:35 +00:00
template <typename F, int NInt, int NUint, int NDouble, int NString, typename ... Args1, typename ... Args2>
2015-02-18 15:57:01 +00:00
struct call<F, NInt, NUint, NDouble, NString, black_magic::S<double, Args1...>, black_magic::S<Args2...>>
2014-04-14 15:31:51 +00:00
{
2015-02-18 15:57:01 +00:00
void operator()(F cparams)
{
2015-02-18 15:57:01 +00:00
using pushed = typename black_magic::S<Args2...>::template push_back<call_pair<double, NDouble>>;
call<F, NInt, NUint, NDouble+1, NString,
black_magic::S<Args1...>, pushed>()(cparams);
}
2015-02-18 15:57:01 +00:00
};
2020-10-14 21:09:35 +00:00
template <typename F, int NInt, int NUint, int NDouble, int NString, typename ... Args1, typename ... Args2>
2015-02-18 15:57:01 +00:00
struct call<F, NInt, NUint, NDouble, NString, black_magic::S<std::string, Args1...>, black_magic::S<Args2...>>
{
void operator()(F cparams)
{
2015-02-18 15:57:01 +00:00
using pushed = typename black_magic::S<Args2...>::template push_back<call_pair<std::string, NString>>;
call<F, NInt, NUint, NDouble, NString+1,
black_magic::S<Args1...>, pushed>()(cparams);
}
2015-02-18 15:57:01 +00:00
};
2020-10-14 21:09:35 +00:00
template <typename F, int NInt, int NUint, int NDouble, int NString, typename ... Args1>
2015-02-18 15:57:01 +00:00
struct call<F, NInt, NUint, NDouble, NString, black_magic::S<>, black_magic::S<Args1...>>
{
void operator()(F cparams)
{
2015-02-18 15:57:01 +00:00
cparams.handler(
cparams.req,
cparams.res,
2020-10-14 21:09:35 +00:00
cparams.params.template get<typename Args1::type>(Args1::pos)...
);
}
2015-02-18 15:57:01 +00:00
};
template <typename Func, typename ... ArgsWrapped>
struct Wrapped
{
template <typename ... Args>
2016-12-06 15:22:10 +00:00
void set_(Func f, typename std::enable_if<
2015-02-18 15:57:01 +00:00
!std::is_same<typename std::tuple_element<0, std::tuple<Args..., void>>::type, const request&>::value
, int>::type = 0)
{
handler_ = (
#ifdef CROW_CAN_USE_CPP14
2015-02-18 15:57:01 +00:00
[f = std::move(f)]
#else
[f]
#endif
2015-02-18 15:57:01 +00:00
(const request&, response& res, Args... args){
res = response(f(args...));
res.end();
});
}
template <typename Req, typename ... Args>
struct req_handler_wrapper
{
req_handler_wrapper(Func f)
: f(std::move(f))
{
}
void operator()(const request& req, response& res, Args... args)
{
res = response(f(req, args...));
res.end();
}
Func f;
};
template <typename ... Args>
2016-12-06 15:22:10 +00:00
void set_(Func f, typename std::enable_if<
2015-02-18 15:57:01 +00:00
std::is_same<typename std::tuple_element<0, std::tuple<Args..., void>>::type, const request&>::value &&
!std::is_same<typename std::tuple_element<1, std::tuple<Args..., void, void>>::type, response&>::value
, int>::type = 0)
{
handler_ = req_handler_wrapper<Args...>(std::move(f));
/*handler_ = (
[f = std::move(f)]
(const request& req, response& res, Args... args){
res = response(f(req, args...));
res.end();
});*/
}
template <typename ... Args>
2016-12-06 15:22:10 +00:00
void set_(Func f, typename std::enable_if<
2015-02-18 15:57:01 +00:00
std::is_same<typename std::tuple_element<0, std::tuple<Args..., void>>::type, const request&>::value &&
std::is_same<typename std::tuple_element<1, std::tuple<Args..., void, void>>::type, response&>::value
, int>::type = 0)
{
handler_ = std::move(f);
}
template <typename ... Args>
struct handler_type_helper
{
using type = std::function<void(const crow::request&, crow::response&, Args...)>;
2020-10-14 21:09:35 +00:00
using args_type = black_magic::S<typename black_magic::promote_t<Args>...>;
2015-02-18 15:57:01 +00:00
};
2015-02-18 15:57:01 +00:00
template <typename ... Args>
struct handler_type_helper<const request&, Args...>
{
using type = std::function<void(const crow::request&, crow::response&, Args...)>;
2020-10-14 21:09:35 +00:00
using args_type = black_magic::S<typename black_magic::promote_t<Args>...>;
2015-02-18 15:57:01 +00:00
};
template <typename ... Args>
struct handler_type_helper<const request&, response&, Args...>
{
using type = std::function<void(const crow::request&, crow::response&, Args...)>;
2020-10-14 21:09:35 +00:00
using args_type = black_magic::S<typename black_magic::promote_t<Args>...>;
2015-02-18 15:57:01 +00:00
};
typename handler_type_helper<ArgsWrapped...>::type handler_;
void operator()(const request& req, response& res, const routing_params& params)
{
detail::routing_handler_call_helper::call<
detail::routing_handler_call_helper::call_params<
decltype(handler_)>,
2020-10-14 21:09:35 +00:00
0, 0, 0, 0,
2015-02-18 15:57:01 +00:00
typename handler_type_helper<ArgsWrapped...>::args_type,
black_magic::S<>
>()(
detail::routing_handler_call_helper::call_params<
decltype(handler_)>
{handler_, params, req, res}
);
}
};
}
}
class CatchallRule
{
public:
CatchallRule(){}
template <typename Func>
typename std::enable_if<black_magic::CallHelper<Func, black_magic::S<>>::value, void>::type
operator()(Func&& f)
{
static_assert(!std::is_same<void, decltype(f())>::value,
"Handler function cannot have void return type; valid return types: string, int, crow::resposne, crow::json::wvalue");
handler_ = (
#ifdef CROW_CAN_USE_CPP14
[f = std::move(f)]
#else
[f]
#endif
(const request&, response& res){
res = response(f());
res.end();
});
}
template <typename Func>
typename std::enable_if<
!black_magic::CallHelper<Func, black_magic::S<>>::value &&
black_magic::CallHelper<Func, black_magic::S<crow::request>>::value,
void>::type
operator()(Func&& f)
{
static_assert(!std::is_same<void, decltype(f(std::declval<crow::request>()))>::value,
"Handler function cannot have void return type; valid return types: string, int, crow::resposne, crow::json::wvalue");
handler_ = (
#ifdef CROW_CAN_USE_CPP14
[f = std::move(f)]
#else
[f]
#endif
(const crow::request& req, crow::response& res){
res = response(f(req));
res.end();
});
}
template <typename Func>
typename std::enable_if<
!black_magic::CallHelper<Func, black_magic::S<>>::value &&
!black_magic::CallHelper<Func, black_magic::S<crow::request>>::value &&
black_magic::CallHelper<Func, black_magic::S<crow::response&>>::value,
void>::type
operator()(Func&& f)
{
static_assert(std::is_same<void, decltype(f(std::declval<crow::response&>()))>::value,
"Handler function with response argument should have void return type");
handler_ = (
#ifdef CROW_CAN_USE_CPP14
[f = std::move(f)]
#else
[f]
#endif
(const crow::request&, crow::response& res){
f(res);
});
}
template <typename Func>
typename std::enable_if<
!black_magic::CallHelper<Func, black_magic::S<>>::value &&
!black_magic::CallHelper<Func, black_magic::S<crow::request>>::value &&
!black_magic::CallHelper<Func, black_magic::S<crow::response&>>::value,
void>::type
operator()(Func&& f)
{
static_assert(std::is_same<void, decltype(f(std::declval<crow::request>(), std::declval<crow::response&>()))>::value,
"Handler function with response argument should have void return type");
handler_ = std::move(f);
}
bool has_handler()
{
return (handler_ != nullptr);
}
protected:
friend class Router;
private:
std::function<void(const crow::request&, crow::response&)> handler_;
};
/// A rule dealing with websockets.
/// Provides the interface for the user to put in the necessary handlers for a websocket to work.
///
class WebSocketRule : public BaseRule
{
2016-08-28 05:46:31 +00:00
using self_t = WebSocketRule;
public:
2016-08-28 05:46:31 +00:00
WebSocketRule(std::string rule)
: BaseRule(std::move(rule))
{
}
void validate() override
{
}
2016-08-28 05:46:31 +00:00
void handle(const request&, response& res, const routing_params&) override
{
res = response(404);
res.end();
}
2016-08-28 05:46:31 +00:00
2020-10-14 21:09:35 +00:00
void handle_upgrade(const request& req, response&, SocketAdaptor&& adaptor) override
{
new crow::websocket::Connection<SocketAdaptor>(req, std::move(adaptor), open_handler_, message_handler_, close_handler_, error_handler_, accept_handler_);
}
2016-08-28 05:46:31 +00:00
#ifdef CROW_ENABLE_SSL
void handle_upgrade(const request& req, response&, SSLAdaptor&& adaptor) override
{
new crow::websocket::Connection<SSLAdaptor>(req, std::move(adaptor), open_handler_, message_handler_, close_handler_, error_handler_, accept_handler_);
}
2016-08-28 05:46:31 +00:00
#endif
template <typename Func>
self_t& onopen(Func f)
{
open_handler_ = f;
return *this;
}
template <typename Func>
self_t& onmessage(Func f)
{
message_handler_ = f;
return *this;
}
template <typename Func>
self_t& onclose(Func f)
{
close_handler_ = f;
return *this;
}
template <typename Func>
self_t& onerror(Func f)
{
error_handler_ = f;
return *this;
}
template <typename Func>
self_t& onaccept(Func f)
{
accept_handler_ = f;
return *this;
}
protected:
std::function<void(crow::websocket::connection&)> open_handler_;
std::function<void(crow::websocket::connection&, const std::string&, bool)> message_handler_;
std::function<void(crow::websocket::connection&, const std::string&)> close_handler_;
std::function<void(crow::websocket::connection&)> error_handler_;
std::function<bool(const crow::request&)> accept_handler_;
};
2016-08-28 05:46:31 +00:00
/// Allows the user to assign parameters using functions.
///
/// `rule.name("name").methods(HTTPMethod::POST)`
template <typename T>
struct RuleParameterTraits
{
using self_t = T;
2020-10-14 21:09:35 +00:00
WebSocketRule& websocket()
{
2021-02-21 00:14:30 +00:00
auto p =new WebSocketRule(static_cast<self_t*>(this)->rule_);
static_cast<self_t*>(this)->rule_to_upgrade_.reset(p);
return *p;
}
2016-08-28 05:46:31 +00:00
self_t& name(std::string name) noexcept
{
2021-02-21 00:14:30 +00:00
static_cast<self_t*>(this)->name_ = std::move(name);
return static_cast<self_t&>(*this);
}
self_t& methods(HTTPMethod method)
{
2021-02-21 00:14:30 +00:00
static_cast<self_t*>(this)->methods_ = 1 << static_cast<int>(method);
return static_cast<self_t&>(*this);
}
template <typename ... MethodArgs>
self_t& methods(HTTPMethod method, MethodArgs ... args_method)
{
methods(args_method...);
2021-02-21 00:14:30 +00:00
static_cast<self_t*>(this)->methods_ |= 1 << static_cast<int>(method);
return static_cast<self_t&>(*this);
}
2016-08-28 05:46:31 +00:00
};
/// A rule that can change its parameters during runtime.
class DynamicRule : public BaseRule, public RuleParameterTraits<DynamicRule>
2015-02-18 15:57:01 +00:00
{
public:
2015-02-18 15:57:01 +00:00
DynamicRule(std::string rule)
: BaseRule(std::move(rule))
{
}
void validate() override
{
if (!erased_handler_)
{
throw std::runtime_error(name_ + (!name_.empty() ? ": " : "") + "no handler for url " + rule_);
2014-04-14 15:31:51 +00:00
}
2015-02-18 15:57:01 +00:00
}
void handle(const request& req, response& res, const routing_params& params) override
{
erased_handler_(req, res, params);
}
template <typename Func>
void operator()(Func f)
{
#ifdef CROW_MSVC_WORKAROUND
using function_t = utility::function_traits<decltype(&Func::operator())>;
#else
2015-02-18 15:57:01 +00:00
using function_t = utility::function_traits<Func>;
#endif
2015-02-18 15:57:01 +00:00
erased_handler_ = wrap(std::move(f), black_magic::gen_seq<function_t::arity>());
}
// enable_if Arg1 == request && Arg2 == response
// enable_if Arg1 == request && Arg2 != resposne
// enable_if Arg1 != request
2015-03-15 08:51:55 +00:00
#ifdef CROW_MSVC_WORKAROUND
template <typename Func, size_t ... Indices>
#else
2015-02-18 15:57:01 +00:00
template <typename Func, unsigned ... Indices>
2015-03-15 08:51:55 +00:00
#endif
2020-10-14 21:09:35 +00:00
std::function<void(const request&, response&, const routing_params&)>
2015-02-18 15:57:01 +00:00
wrap(Func f, black_magic::seq<Indices...>)
{
#ifdef CROW_MSVC_WORKAROUND
using function_t = utility::function_traits<decltype(&Func::operator())>;
#else
2015-02-18 15:57:01 +00:00
using function_t = utility::function_traits<Func>;
#endif
2015-03-12 21:53:45 +00:00
if (!black_magic::is_parameter_tag_compatible(
2020-10-14 21:09:35 +00:00
black_magic::get_parameter_tag_runtime(rule_.c_str()),
2015-03-12 21:53:45 +00:00
black_magic::compute_parameter_tag_from_args_list<
typename function_t::template arg<Indices>...>::value))
{
2015-03-12 21:53:45 +00:00
throw std::runtime_error("route_dynamic: Handler type is mismatched with URL parameters: " + rule_);
}
2015-02-18 15:57:01 +00:00
auto ret = detail::routing_handler_call_helper::Wrapped<Func, typename function_t::template arg<Indices>...>();
2016-12-06 15:22:10 +00:00
ret.template set_<
2015-02-18 15:57:01 +00:00
typename function_t::template arg<Indices>...
>(std::move(f));
return ret;
}
template <typename Func>
void operator()(std::string name, Func&& f)
{
name_ = std::move(name);
(*this).template operator()<Func>(std::forward(f));
}
private:
std::function<void(const request&, response&, const routing_params&)> erased_handler_;
};
/// Default rule created when CROW_ROUTE is called.
2015-02-18 15:57:01 +00:00
template <typename ... Args>
class TaggedRule : public BaseRule, public RuleParameterTraits<TaggedRule<Args...>>
2015-02-18 15:57:01 +00:00
{
public:
2014-04-26 17:19:59 +00:00
using self_t = TaggedRule<Args...>;
2015-02-18 15:57:01 +00:00
TaggedRule(std::string rule)
2015-02-18 15:57:01 +00:00
: BaseRule(std::move(rule))
{
}
2014-04-26 17:19:59 +00:00
2016-08-28 05:46:31 +00:00
void validate() override
2014-04-15 13:46:28 +00:00
{
2015-02-18 15:57:01 +00:00
if (!handler_)
2014-04-21 18:27:53 +00:00
{
throw std::runtime_error(name_ + (!name_.empty() ? ": " : "") + "no handler for url " + rule_);
}
2014-04-15 13:46:28 +00:00
}
template <typename Func>
typename std::enable_if<black_magic::CallHelper<Func, black_magic::S<Args...>>::value, void>::type
operator()(Func&& f)
2014-04-02 20:31:32 +00:00
{
static_assert(black_magic::CallHelper<Func, black_magic::S<Args...>>::value ||
2020-10-14 21:09:35 +00:00
black_magic::CallHelper<Func, black_magic::S<crow::request, Args...>>::value ,
2015-03-12 21:53:45 +00:00
"Handler type is mismatched with URL parameters");
2020-10-14 21:09:35 +00:00
static_assert(!std::is_same<void, decltype(f(std::declval<Args>()...))>::value,
2014-04-26 17:19:59 +00:00
"Handler function cannot have void return type; valid return types: string, int, crow::resposne, crow::json::wvalue");
handler_ = (
#ifdef CROW_CAN_USE_CPP14
[f = std::move(f)]
#else
[f]
#endif
(const request&, response& res, Args ... args){
2015-02-18 15:57:01 +00:00
res = response(f(args...));
res.end();
});
}
template <typename Func>
typename std::enable_if<
!black_magic::CallHelper<Func, black_magic::S<Args...>>::value &&
2020-10-14 21:09:35 +00:00
black_magic::CallHelper<Func, black_magic::S<crow::request, Args...>>::value,
void>::type
operator()(Func&& f)
{
static_assert(black_magic::CallHelper<Func, black_magic::S<Args...>>::value ||
2020-10-14 21:09:35 +00:00
black_magic::CallHelper<Func, black_magic::S<crow::request, Args...>>::value,
2015-03-12 21:53:45 +00:00
"Handler type is mismatched with URL parameters");
2020-10-14 21:09:35 +00:00
static_assert(!std::is_same<void, decltype(f(std::declval<crow::request>(), std::declval<Args>()...))>::value,
2014-04-26 17:19:59 +00:00
"Handler function cannot have void return type; valid return types: string, int, crow::resposne, crow::json::wvalue");
handler_ = (
#ifdef CROW_CAN_USE_CPP14
[f = std::move(f)]
#else
[f]
#endif
(const crow::request& req, crow::response& res, Args ... args){
2015-02-18 15:57:01 +00:00
res = response(f(req, args...));
res.end();
});
}
template <typename Func>
typename std::enable_if<
!black_magic::CallHelper<Func, black_magic::S<Args...>>::value &&
!black_magic::CallHelper<Func, black_magic::S<crow::request, Args...>>::value &&
black_magic::CallHelper<Func, black_magic::S<crow::response&, Args...>>::value,
2020-12-02 10:06:29 +00:00
void>::type
operator()(Func&& f)
{
static_assert(black_magic::CallHelper<Func, black_magic::S<Args...>>::value ||
black_magic::CallHelper<Func, black_magic::S<crow::response&, Args...>>::value
,
"Handler type is mismatched with URL parameters");
static_assert(std::is_same<void, decltype(f(std::declval<crow::response&>(), std::declval<Args>()...))>::value,
"Handler function with response argument should have void return type");
2020-12-02 11:07:43 +00:00
handler_ = (
2020-12-02 10:06:29 +00:00
#ifdef CROW_CAN_USE_CPP14
[f = std::move(f)]
#else
[f]
#endif
2020-12-02 17:29:45 +00:00
(const crow::request&, crow::response& res, Args ... args){
2020-12-02 10:06:29 +00:00
f(res, args...);
});
}
template <typename Func>
typename std::enable_if<
!black_magic::CallHelper<Func, black_magic::S<Args...>>::value &&
!black_magic::CallHelper<Func, black_magic::S<crow::request, Args...>>::value &&
2020-12-02 10:06:29 +00:00
!black_magic::CallHelper<Func, black_magic::S<crow::response&, Args...>>::value,
void>::type
operator()(Func&& f)
{
static_assert(black_magic::CallHelper<Func, black_magic::S<Args...>>::value ||
black_magic::CallHelper<Func, black_magic::S<crow::request, Args...>>::value ||
black_magic::CallHelper<Func, black_magic::S<crow::request, crow::response&, Args...>>::value
2020-10-14 21:09:35 +00:00
,
2015-03-12 21:53:45 +00:00
"Handler type is mismatched with URL parameters");
2020-10-14 21:09:35 +00:00
static_assert(std::is_same<void, decltype(f(std::declval<crow::request>(), std::declval<crow::response&>(), std::declval<Args>()...))>::value,
"Handler function with response argument should have void return type");
2015-02-18 15:57:01 +00:00
handler_ = std::move(f);
}
template <typename Func>
void operator()(std::string name, Func&& f)
{
name_ = std::move(name);
(*this).template operator()<Func>(std::forward(f));
}
2014-04-02 20:31:32 +00:00
void handle(const request& req, response& res, const routing_params& params) override
{
2015-02-18 15:57:01 +00:00
detail::routing_handler_call_helper::call<
detail::routing_handler_call_helper::call_params<
2020-10-14 21:09:35 +00:00
decltype(handler_)>,
0, 0, 0, 0,
black_magic::S<Args...>,
black_magic::S<>
>()(
2015-02-18 15:57:01 +00:00
detail::routing_handler_call_helper::call_params<
decltype(handler_)>
{handler_, params, req, res}
);
2014-04-02 20:31:32 +00:00
}
private:
2015-02-18 15:57:01 +00:00
std::function<void(const crow::request&, crow::response&, Args...)> handler_;
2014-04-26 17:19:59 +00:00
};
2014-04-02 16:38:08 +00:00
const int RULE_SPECIAL_REDIRECT_SLASH = 1;
/// A search tree.
2014-04-10 16:43:33 +00:00
class Trie
{
public:
struct Node
{
2014-04-14 15:31:51 +00:00
unsigned rule_index{};
2021-02-21 00:14:30 +00:00
std::array<unsigned, static_cast<int>(ParamType::MAX)> param_childrens{};
std::unordered_map<std::string, unsigned> children;
2014-08-06 21:18:21 +00:00
bool IsSimpleNode() const
{
2020-10-14 21:09:35 +00:00
return
2014-08-06 21:18:21 +00:00
!rule_index &&
std::all_of(
2020-10-14 21:09:35 +00:00
std::begin(param_childrens),
std::end(param_childrens),
2014-08-06 21:18:21 +00:00
[](unsigned x){ return !x; });
}
2014-04-10 16:43:33 +00:00
};
2014-08-06 21:18:21 +00:00
Trie() : nodes_(1)
{
}
2014-04-15 13:46:28 +00:00
///Check whether or not the trie is empty.
bool is_empty()
{
return nodes_.size() > 1;
}
private:
2014-08-06 21:18:21 +00:00
void optimizeNode(Node* node)
{
for(auto x : node->param_childrens)
{
if (!x)
continue;
Node* child = &nodes_[x];
optimizeNode(child);
}
if (node->children.empty())
return;
bool mergeWithChild = true;
for(auto& kv : node->children)
{
Node* child = &nodes_[kv.second];
if (!child->IsSimpleNode())
{
mergeWithChild = false;
break;
}
}
if (mergeWithChild)
{
decltype(node->children) merged;
for(auto& kv : node->children)
{
Node* child = &nodes_[kv.second];
for(auto& child_kv : child->children)
{
merged[kv.first + child_kv.first] = child_kv.second;
}
}
node->children = std::move(merged);
optimizeNode(node);
}
else
{
for(auto& kv : node->children)
{
Node* child = &nodes_[kv.second];
optimizeNode(child);
}
}
}
void optimize()
{
optimizeNode(head());
}
2014-04-15 13:46:28 +00:00
public:
2014-08-06 21:18:21 +00:00
void validate()
{
if (!head()->IsSimpleNode())
throw std::runtime_error("Internal error: Trie header should be simple!");
optimize();
}
2014-04-10 16:43:33 +00:00
2014-09-15 16:28:15 +00:00
std::pair<unsigned, routing_params> find(const std::string& req_url, const Node* node = nullptr, unsigned pos = 0, routing_params* params = nullptr) const
2014-08-06 21:18:21 +00:00
{
routing_params empty;
if (params == nullptr)
params = &empty;
2014-04-10 16:43:33 +00:00
2014-08-06 21:18:21 +00:00
unsigned found{};
routing_params match_params;
2014-04-10 16:43:33 +00:00
2014-08-06 21:18:21 +00:00
if (node == nullptr)
node = head();
2014-09-15 16:28:15 +00:00
if (pos == req_url.size())
2014-08-06 21:18:21 +00:00
return {node->rule_index, *params};
auto update_found = [&found, &match_params](std::pair<unsigned, routing_params>& ret)
{
if (ret.first && (!found || found > ret.first))
{
found = ret.first;
match_params = std::move(ret.second);
}
};
2021-02-21 00:14:30 +00:00
if (node->param_childrens[static_cast<int>(ParamType::INT)])
2014-08-06 21:18:21 +00:00
{
2014-09-15 16:28:15 +00:00
char c = req_url[pos];
2014-08-06 21:18:21 +00:00
if ((c >= '0' && c <= '9') || c == '+' || c == '-')
{
char* eptr;
errno = 0;
2014-09-15 16:28:15 +00:00
long long int value = strtoll(req_url.data()+pos, &eptr, 10);
if (errno != ERANGE && eptr != req_url.data()+pos)
2014-08-06 21:18:21 +00:00
{
params->int_params.push_back(value);
2021-02-21 00:14:30 +00:00
auto ret = find(req_url, &nodes_[node->param_childrens[static_cast<int>(ParamType::INT)]], eptr - req_url.data(), params);
2014-08-06 21:18:21 +00:00
update_found(ret);
params->int_params.pop_back();
}
}
}
2021-02-21 00:14:30 +00:00
if (node->param_childrens[static_cast<int>(ParamType::UINT)])
2014-08-06 21:18:21 +00:00
{
2014-09-15 16:28:15 +00:00
char c = req_url[pos];
2014-08-06 21:18:21 +00:00
if ((c >= '0' && c <= '9') || c == '+')
{
char* eptr;
errno = 0;
2014-09-15 16:28:15 +00:00
unsigned long long int value = strtoull(req_url.data()+pos, &eptr, 10);
if (errno != ERANGE && eptr != req_url.data()+pos)
2014-08-06 21:18:21 +00:00
{
params->uint_params.push_back(value);
2021-02-21 00:14:30 +00:00
auto ret = find(req_url, &nodes_[node->param_childrens[static_cast<int>(ParamType::UINT)]], eptr - req_url.data(), params);
2014-08-06 21:18:21 +00:00
update_found(ret);
params->uint_params.pop_back();
}
}
}
2021-02-21 00:14:30 +00:00
if (node->param_childrens[static_cast<int>(ParamType::DOUBLE)])
2014-08-06 21:18:21 +00:00
{
2014-09-15 16:28:15 +00:00
char c = req_url[pos];
2014-08-06 21:18:21 +00:00
if ((c >= '0' && c <= '9') || c == '+' || c == '-' || c == '.')
{
char* eptr;
errno = 0;
2014-09-15 16:28:15 +00:00
double value = strtod(req_url.data()+pos, &eptr);
if (errno != ERANGE && eptr != req_url.data()+pos)
2014-08-06 21:18:21 +00:00
{
params->double_params.push_back(value);
2021-02-21 00:14:30 +00:00
auto ret = find(req_url, &nodes_[node->param_childrens[static_cast<int>(ParamType::DOUBLE)]], eptr - req_url.data(), params);
2014-08-06 21:18:21 +00:00
update_found(ret);
params->double_params.pop_back();
}
}
}
2021-02-21 00:14:30 +00:00
if (node->param_childrens[static_cast<int>(ParamType::STRING)])
2014-08-06 21:18:21 +00:00
{
size_t epos = pos;
2014-09-15 16:28:15 +00:00
for(; epos < req_url.size(); epos ++)
2014-08-06 21:18:21 +00:00
{
2014-09-15 16:28:15 +00:00
if (req_url[epos] == '/')
2014-08-06 21:18:21 +00:00
break;
}
if (epos != pos)
{
2014-09-15 16:28:15 +00:00
params->string_params.push_back(req_url.substr(pos, epos-pos));
2021-02-21 00:14:30 +00:00
auto ret = find(req_url, &nodes_[node->param_childrens[static_cast<int>(ParamType::STRING)]], epos, params);
2014-08-06 21:18:21 +00:00
update_found(ret);
params->string_params.pop_back();
}
}
2021-02-21 00:14:30 +00:00
if (node->param_childrens[static_cast<int>(ParamType::PATH)])
2014-08-06 21:18:21 +00:00
{
2014-09-15 16:28:15 +00:00
size_t epos = req_url.size();
2014-08-06 21:18:21 +00:00
if (epos != pos)
{
2014-09-15 16:28:15 +00:00
params->string_params.push_back(req_url.substr(pos, epos-pos));
2021-02-21 00:14:30 +00:00
auto ret = find(req_url, &nodes_[node->param_childrens[static_cast<int>(ParamType::PATH)]], epos, params);
2014-08-06 21:18:21 +00:00
update_found(ret);
params->string_params.pop_back();
}
}
2014-08-06 21:18:21 +00:00
for(auto& kv : node->children)
{
const std::string& fragment = kv.first;
const Node* child = &nodes_[kv.second];
2014-04-10 16:43:33 +00:00
2014-09-15 16:28:15 +00:00
if (req_url.compare(pos, fragment.size(), fragment) == 0)
2014-08-06 21:18:21 +00:00
{
2014-09-15 16:28:15 +00:00
auto ret = find(req_url, child, pos + fragment.size(), params);
2014-08-06 21:18:21 +00:00
update_found(ret);
}
}
2014-04-10 16:43:33 +00:00
2014-08-06 21:18:21 +00:00
return {found, match_params};
}
void add(const std::string& url, unsigned rule_index)
{
unsigned idx{0};
for(unsigned i = 0; i < url.size(); i ++)
{
char c = url[i];
if (c == '<')
{
static struct ParamTraits
{
ParamType type;
std::string name;
} paramTraits[] =
{
{ ParamType::INT, "<int>" },
{ ParamType::UINT, "<uint>" },
{ ParamType::DOUBLE, "<float>" },
{ ParamType::DOUBLE, "<double>" },
{ ParamType::STRING, "<str>" },
{ ParamType::STRING, "<string>" },
{ ParamType::PATH, "<path>" },
};
for(auto& x:paramTraits)
{
if (url.compare(i, x.name.size(), x.name) == 0)
{
2021-02-21 00:14:30 +00:00
if (!nodes_[idx].param_childrens[static_cast<int>(x.type)])
2014-08-06 21:18:21 +00:00
{
auto new_node_idx = new_node();
2021-02-21 00:14:30 +00:00
nodes_[idx].param_childrens[static_cast<int>(x.type)] = new_node_idx;
2014-08-06 21:18:21 +00:00
}
2021-02-21 00:14:30 +00:00
idx = nodes_[idx].param_childrens[static_cast<int>(x.type)];
2014-08-06 21:18:21 +00:00
i += x.name.size();
break;
}
}
i --;
}
else
{
std::string piece(&c, 1);
if (!nodes_[idx].children.count(piece))
{
auto new_node_idx = new_node();
nodes_[idx].children.emplace(piece, new_node_idx);
}
idx = nodes_[idx].children[piece];
}
}
if (nodes_[idx].rule_index)
throw std::runtime_error("handler already exists for " + url);
nodes_[idx].rule_index = rule_index;
}
private:
void debug_node_print(Node* n, int level)
{
2021-02-21 00:14:30 +00:00
for(int i = 0; i < static_cast<int>(ParamType::MAX); i ++)
2014-08-06 21:18:21 +00:00
{
if (n->param_childrens[i])
{
CROW_LOG_DEBUG << std::string(2*level, ' ') /*<< "("<<n->param_childrens[i]<<") "*/;
2021-02-21 00:14:30 +00:00
switch(static_cast<ParamType>(i))
2014-08-06 21:18:21 +00:00
{
case ParamType::INT:
CROW_LOG_DEBUG << "<int>";
break;
case ParamType::UINT:
CROW_LOG_DEBUG << "<uint>";
break;
case ParamType::DOUBLE:
CROW_LOG_DEBUG << "<float>";
break;
case ParamType::STRING:
CROW_LOG_DEBUG << "<str>";
break;
case ParamType::PATH:
CROW_LOG_DEBUG << "<path>";
break;
default:
CROW_LOG_DEBUG << "<ERROR>";
break;
}
debug_node_print(&nodes_[n->param_childrens[i]], level+1);
}
}
for(auto& kv : n->children)
{
CROW_LOG_DEBUG << std::string(2*level, ' ') /*<< "(" << kv.second << ") "*/ << kv.first;
debug_node_print(&nodes_[kv.second], level+1);
}
}
public:
void debug_print()
{
debug_node_print(head(), 0);
}
2014-04-10 16:43:33 +00:00
private:
2014-08-06 21:18:21 +00:00
const Node* head() const
{
return &nodes_.front();
}
Node* head()
{
return &nodes_.front();
}
unsigned new_node()
{
nodes_.resize(nodes_.size()+1);
return nodes_.size() - 1;
}
2014-04-10 16:43:33 +00:00
std::vector<Node> nodes_;
};
/// Handles matching requests to existing rules and upgrade requests.
2014-04-02 16:38:08 +00:00
class Router
{
public:
Router()
{
}
2015-02-18 15:57:01 +00:00
DynamicRule& new_rule_dynamic(const std::string& rule)
{
auto ruleObject = new DynamicRule(rule);
all_rules_.emplace_back(ruleObject);
2015-02-18 15:57:01 +00:00
return *ruleObject;
}
template <uint64_t N>
typename black_magic::arguments<N>::type::template rebind<TaggedRule>& new_rule_tagged(const std::string& rule)
{
using RuleT = typename black_magic::arguments<N>::type::template rebind<TaggedRule>;
auto ruleObject = new RuleT(rule);
all_rules_.emplace_back(ruleObject);
2015-02-18 15:57:01 +00:00
return *ruleObject;
}
CatchallRule* catchall_rule()
{
return catchall_rule_;
}
2015-02-18 15:57:01 +00:00
void internal_add_rule_object(const std::string& rule, BaseRule* ruleObject)
{
bool has_trailing_slash = false;
std::string rule_without_trailing_slash;
if (rule.size() > 1 && rule.back() == '/')
{
has_trailing_slash = true;
rule_without_trailing_slash = rule;
rule_without_trailing_slash.pop_back();
}
ruleObject->foreach_method([&](int method)
{
per_methods_[method].rules.emplace_back(ruleObject);
per_methods_[method].trie.add(rule, per_methods_[method].rules.size() - 1);
2020-10-14 21:09:35 +00:00
// directory case:
// request to '/about' url matches '/about/' rule
if (has_trailing_slash)
{
per_methods_[method].trie.add(rule_without_trailing_slash, RULE_SPECIAL_REDIRECT_SLASH);
}
});
}
2014-08-06 21:18:21 +00:00
void validate()
{
for(auto& rule:all_rules_)
2014-08-06 21:18:21 +00:00
{
if (rule)
{
auto upgraded = rule->upgrade();
if (upgraded)
rule = std::move(upgraded);
2014-08-06 21:18:21 +00:00
rule->validate();
internal_add_rule_object(rule->rule(), rule.get());
}
}
for(auto& per_method:per_methods_)
{
per_method.trie.validate();
2014-08-06 21:18:21 +00:00
}
}
2014-04-10 16:43:33 +00:00
//TODO maybe add actual_method
2020-10-14 21:09:35 +00:00
template <typename Adaptor>
void handle_upgrade(const request& req, response& res, Adaptor&& adaptor)
{
if (req.method >= HTTPMethod::InternalMethodCount)
return;
2020-10-15 10:59:15 +00:00
2021-02-21 00:14:30 +00:00
auto& per_method = per_methods_[static_cast<int>(req.method)];
auto& rules = per_method.rules;
2020-10-15 10:59:15 +00:00
unsigned rule_index = per_method.trie.find(req.url).first;
2016-08-28 05:46:31 +00:00
if (!rule_index)
{
for (auto& per_method: per_methods_)
{
2020-10-15 10:59:15 +00:00
if (per_method.trie.find(req.url).first)
{
2020-10-15 10:59:15 +00:00
CROW_LOG_DEBUG << "Cannot match method " << req.url << " " << method_name(req.method);
res = response(405);
res.end();
return;
}
}
2020-10-15 10:59:15 +00:00
CROW_LOG_INFO << "Cannot match rules " << req.url;
2016-08-28 05:46:31 +00:00
res = response(404);
res.end();
return;
}
if (rule_index >= rules.size())
2016-08-28 05:46:31 +00:00
throw std::runtime_error("Trie internal structure corrupted!");
if (rule_index == RULE_SPECIAL_REDIRECT_SLASH)
{
CROW_LOG_INFO << "Redirecting to a url with trailing slash: " << req.url;
res = response(301);
// TODO absolute url building
if (req.get_header_value("Host").empty())
{
res.add_header("Location", req.url + "/");
}
else
{
res.add_header("Location", "http://" + req.get_header_value("Host") + req.url + "/");
}
res.end();
return;
}
2021-02-21 00:14:30 +00:00
CROW_LOG_DEBUG << "Matched rule (upgrade) '" << rules[rule_index]->rule_ << "' " << static_cast<uint32_t>(req.method) << " / " << rules[rule_index]->get_methods();
2016-08-28 05:46:31 +00:00
// any uncaught exceptions become 500s
try
{
rules[rule_index]->handle_upgrade(req, res, std::move(adaptor));
2016-08-28 05:46:31 +00:00
}
catch(std::exception& e)
{
CROW_LOG_ERROR << "An uncaught exception occurred: " << e.what();
res = response(500);
res.end();
2020-10-14 21:09:35 +00:00
return;
2016-08-28 05:46:31 +00:00
}
catch(...)
{
CROW_LOG_ERROR << "An uncaught exception occurred. The type was unknown so no information was available.";
res = response(500);
res.end();
2020-10-14 21:09:35 +00:00
return;
2016-08-28 05:46:31 +00:00
}
}
2016-08-28 05:46:31 +00:00
2014-08-06 21:18:21 +00:00
void handle(const request& req, response& res)
{
2021-03-13 10:51:27 +00:00
HTTPMethod method_actual = req.method;
if (req.method >= HTTPMethod::InternalMethodCount)
return;
2021-03-13 10:51:27 +00:00
else if (req.method == HTTPMethod::HEAD)
{
method_actual = HTTPMethod::GET;
res.is_head_response = true;
2021-03-13 10:51:27 +00:00
}
else if (req.method == HTTPMethod::OPTIONS)
{
std::string allow = "OPTIONS, HEAD, ";
2021-03-13 10:51:27 +00:00
if (req.url == "/*")
{
for(int i = 0; i < static_cast<int>(HTTPMethod::InternalMethodCount); i ++)
{
if (per_methods_[i].trie.is_empty())
{
allow += method_name(static_cast<HTTPMethod>(i)) + ", ";
}
}
allow = allow.substr(0, allow.size()-2);
res = response(204);
res.set_header("Allow", allow);
res.manual_length_header = true;
res.end();
return;
}
else
{
for(int i = 0; i < static_cast<int>(HTTPMethod::InternalMethodCount); i ++)
{
if (per_methods_[i].trie.find(req.url).first)
{
allow += method_name(static_cast<HTTPMethod>(i)) + ", ";
}
}
if (allow != "OPTIONS, HEAD, ")
{
allow = allow.substr(0, allow.size()-2);
res = response(204);
res.set_header("Allow", allow);
res.manual_length_header = true;
res.end();
return;
}
else
{
CROW_LOG_DEBUG << "Cannot match rules " << req.url;
res = response(404);
res.end();
return;
}
}
}
2021-04-03 03:40:14 +00:00
auto& per_method = per_methods_[static_cast<int>(method_actual)];
auto& trie = per_method.trie;
auto& rules = per_method.rules;
auto found = trie.find(req.url);
2014-08-06 21:18:21 +00:00
unsigned rule_index = found.first;
if (!rule_index)
{
for (auto& per_method: per_methods_)
{
2020-10-15 10:59:15 +00:00
if (per_method.trie.find(req.url).first)
{
CROW_LOG_DEBUG << "Cannot match method " << req.url << " " << method_name(method_actual);
res = response(405);
res.end();
return;
}
}
if (catchall_rule_->has_handler())
{
CROW_LOG_DEBUG << "Cannot match rules " << req.url << ". Redirecting to Catchall rule";
catchall_rule_->handler_(req, res);
}
else
{
CROW_LOG_DEBUG << "Cannot match rules " << req.url;
res = response(404);
}
res.end();
2014-08-06 21:18:21 +00:00
return;
}
if (rule_index >= rules.size())
2014-08-06 21:18:21 +00:00
throw std::runtime_error("Trie internal structure corrupted!");
if (rule_index == RULE_SPECIAL_REDIRECT_SLASH)
{
CROW_LOG_INFO << "Redirecting to a url with trailing slash: " << req.url;
res = response(301);
// TODO absolute url building
if (req.get_header_value("Host").empty())
{
res.add_header("Location", req.url + "/");
}
else
{
res.add_header("Location", "http://" + req.get_header_value("Host") + req.url + "/");
}
res.end();
return;
}
2021-02-21 00:14:30 +00:00
CROW_LOG_DEBUG << "Matched rule '" << rules[rule_index]->rule_ << "' " << static_cast<uint32_t>(req.method) << " / " << rules[rule_index]->get_methods();
2014-08-06 21:18:21 +00:00
// any uncaught exceptions become 500s
try
{
rules[rule_index]->handle(req, res, found.second);
}
catch(std::exception& e)
{
CROW_LOG_ERROR << "An uncaught exception occurred: " << e.what();
res = response(500);
res.end();
2020-10-14 21:09:35 +00:00
return;
}
catch(...)
{
CROW_LOG_ERROR << "An uncaught exception occurred. The type was unknown so no information was available.";
res = response(500);
res.end();
2020-10-14 21:09:35 +00:00
return;
}
2014-08-06 21:18:21 +00:00
}
void debug_print()
{
2021-02-21 00:14:30 +00:00
for(int i = 0; i < static_cast<int>(HTTPMethod::InternalMethodCount); i ++)
{
2021-02-21 00:14:30 +00:00
CROW_LOG_DEBUG << method_name(static_cast<HTTPMethod>(i));
per_methods_[i].trie.debug_print();
}
2014-08-06 21:18:21 +00:00
}
2014-04-02 20:31:32 +00:00
private:
CatchallRule* catchall_rule_ = new CatchallRule();
struct PerMethod
{
std::vector<BaseRule*> rules;
Trie trie;
// rule index 0, 1 has special meaning; preallocate it to avoid duplication.
PerMethod() : rules(2) {}
};
2021-02-21 00:14:30 +00:00
std::array<PerMethod, static_cast<int>(HTTPMethod::InternalMethodCount)> per_methods_;
std::vector<std::unique_ptr<BaseRule>> all_rules_;
2014-04-02 16:38:08 +00:00
};
}