You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
124 lines
2.7 KiB
C++
124 lines
2.7 KiB
C++
#include "webcc/router.h"
|
|
|
|
#include <algorithm>
|
|
|
|
#include "boost/algorithm/string.hpp"
|
|
|
|
#include "webcc/logger.h"
|
|
#include "router.h"
|
|
|
|
namespace webcc {
|
|
|
|
bool Router::Route(string_view url, ViewPtr view, const Strings& methods) {
|
|
assert(view);
|
|
|
|
// TODO: More error check
|
|
/*
|
|
* Repeat Checker by Unknown Object at 2022-02-12
|
|
*/
|
|
//------------------RC Start--------------------------
|
|
int cnt = 0;
|
|
for (cnt = 0; cnt < routes_.size(); cnt++)
|
|
if(ToString(routes_[cnt].url) == ToString(url))
|
|
break;
|
|
if (cnt != routes_.size()) //repeat found
|
|
{
|
|
routes_[cnt].view = view;
|
|
routes_[cnt].methods = methods;
|
|
return true;
|
|
}
|
|
//-------------------RC End---------------------------
|
|
|
|
routes_.push_back({ ToString(url), {}, view, methods });
|
|
|
|
return true;
|
|
}
|
|
|
|
bool Router::Route(const UrlRegex& regex_url, ViewPtr view,
|
|
const Strings& methods) {
|
|
assert(view);
|
|
|
|
// TODO: More error check
|
|
|
|
try {
|
|
routes_.push_back({ "", regex_url(), view, methods });
|
|
|
|
} catch (const std::regex_error& e) {
|
|
LOG_ERRO("Not a valid regular expression: %s", e.what());
|
|
return false;
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
ViewPtr Router::FindView(const std::string& method, const std::string& url,
|
|
UrlArgs* args) {
|
|
assert(args != nullptr);
|
|
|
|
for (auto& route : routes_) {
|
|
if (std::find(route.methods.begin(), route.methods.end(), method) ==
|
|
route.methods.end()) {
|
|
continue;
|
|
}
|
|
|
|
if (route.url.empty()) {
|
|
std::smatch match;
|
|
|
|
if (std::regex_match(url, match, route.url_regex)) {
|
|
// Any sub-matches?
|
|
// Start from 1 because match[0] is the whole string itself.
|
|
for (size_t i = 1; i < match.size(); ++i) {
|
|
args->push_back(match[i].str());
|
|
}
|
|
|
|
return route.view;
|
|
}
|
|
} else {
|
|
if (boost::iequals(route.url, url)) {
|
|
return route.view;
|
|
}
|
|
}
|
|
}
|
|
|
|
return ViewPtr();
|
|
}
|
|
|
|
bool Router::MatchView(const std::string& method, const std::string& url,
|
|
bool* stream) {
|
|
assert(stream != nullptr);
|
|
*stream = false;
|
|
|
|
for (auto& route : routes_) {
|
|
if (std::find(route.methods.begin(), route.methods.end(), method) ==
|
|
route.methods.end()) {
|
|
continue;
|
|
}
|
|
|
|
if (route.url.empty()) {
|
|
std::smatch match;
|
|
|
|
if (std::regex_match(url, match, route.url_regex)) {
|
|
*stream = route.view->Stream(method);
|
|
return true;
|
|
}
|
|
} else {
|
|
if (boost::iequals(route.url, url)) {
|
|
*stream = route.view->Stream(method);
|
|
return true;
|
|
}
|
|
}
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
size_t Router::GetViewCount() {
|
|
return routes_.size();
|
|
}
|
|
|
|
ViewPtr& Router::AccessView(size_t index) {
|
|
return routes_.at(index).view;
|
|
}
|
|
|
|
} // namespace webcc
|