1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
|
/* @file
* @brief Concrete implementation of Router
*/
#include "yahttp.hpp"
#include "router.hpp"
namespace YaHTTP {
typedef funcptr::tuple<int,int> TDelim;
// router is defined here.
YaHTTP::Router Router::router;
void Router::map(const std::string& method, const std::string& url, THandlerFunction handler, const std::string& name) {
std::string method2 = method;
bool isopen=false;
// add into vector
for(std::string::const_iterator i = url.begin(); i != url.end(); i++) {
if (*i == '<' && isopen) throw Error("Invalid URL mask, cannot have < after <");
if (*i == '<') isopen = true;
if (*i == '>' && !isopen) throw Error("Invalid URL mask, cannot have > without < first");
if (*i == '>') isopen = false;
}
std::transform(method2.begin(), method2.end(), method2.begin(), ::toupper);
routes.push_back(funcptr::make_tuple(method2, url, handler, name));
};
bool Router::route(Request *req, THandlerFunction& handler) {
std::map<std::string, TDelim> params;
int pos1,pos2;
bool matched = false;
std::string rname;
// iterate routes
for(TRouteList::iterator i = routes.begin(); !matched && i != routes.end(); i++) {
int k1,k2,k3;
std::string pname;
std::string method, url;
funcptr::tie(method, url, handler, rname) = *i;
if (method.empty() == false && req->method != method) continue; // no match on method
// see if we can't match the url
params.clear();
// simple matcher func
for(k1=0, k2=0; k1 < static_cast<int>(url.size()) && k2 < static_cast<int>(req->url.path.size()); ) {
if (url[k1] == '<') {
pos1 = k2;
k3 = k1+1;
// start of parameter
while(k1 < static_cast<int>(url.size()) && url[k1] != '>') k1++;
pname = std::string(url.begin()+k3, url.begin()+k1);
// then we also look it on the url
if (pname[0]=='*') {
pname = pname.substr(1);
// this matches whatever comes after it, basically end of string
pos2 = req->url.path.size();
if (pname != "")
params[pname] = funcptr::tie(pos1,pos2);
k1 = url.size();
k2 = req->url.path.size();
break;
} else {
// match until url[k1]
while(k2 < static_cast<int>(req->url.path.size()) && req->url.path[k2] != url[k1+1]) k2++;
pos2 = k2;
params[pname] = funcptr::tie(pos1,pos2);
}
k2--;
}
else if (url[k1] != req->url.path[k2]) {
break;
}
k1++; k2++;
}
// ensure.
if (url[k1] != req->url.path[k2])
matched = false;
else
matched = true;
}
if (!matched) { return false; } // no route
req->parameters.clear();
for(std::map<std::string, TDelim>::iterator i = params.begin(); i != params.end(); i++) {
int p1,p2;
funcptr::tie(p1,p2) = i->second;
std::string value(req->url.path.begin() + p1, req->url.path.begin() + p2);
value = Utility::decodeURL(value);
req->parameters[i->first] = value;
}
req->routeName = rname;
return true;
};
void Router::printRoutes(std::ostream &os) {
for(TRouteList::iterator i = routes.begin(); i != routes.end(); i++) {
#ifdef HAVE_CXX11
std::streamsize ss = os.width();
std::ios::fmtflags ff = os.setf(std::ios::left);
os.width(10);
os << std::get<0>(*i);
os.width(50);
os << std::get<1>(*i);
os.width(ss);
os.setf(ff);
os << " " << std::get<3>(*i);
os << std::endl;
#else
os << i->get<0>() << " " << i->get<1>() << " " << i->get<3>() << std::endl;
#endif
}
};
std::pair<std::string,std::string> Router::urlFor(const std::string &name, const strstr_map_t& arguments) {
std::ostringstream path;
std::string mask,method,result;
int k1,k2,k3;
bool found = false;
for(TRouteList::iterator i = routes.begin(); !found && i != routes.end(); i++) {
#ifdef HAVE_CXX11
if (std::get<3>(*i) == name) { mask = std::get<1>(*i); method = std::get<0>(*i); found = true; }
#else
if (i->get<3>() == name) { mask = i->get<1>(); method = i->get<0>(); found = true; }
#endif
}
if (!found)
throw Error("Route not found");
for(k1=0,k3=0;k1<static_cast<int>(mask.size());k1++) {
if (mask[k1] == '<') {
std::string pname;
strstr_map_t::const_iterator pptr;
k2=k1;
while(k1<static_cast<int>(mask.size()) && mask[k1]!='>') k1++;
path << mask.substr(k3,k2-k3);
if (mask[k2+1] == '*')
pname = std::string(mask.begin() + k2 + 2, mask.begin() + k1);
else
pname = std::string(mask.begin() + k2 + 1, mask.begin() + k1);
if ((pptr = arguments.find(pname)) != arguments.end())
path << Utility::encodeURL(pptr->second);
k3 = k1+1;
}
else if (mask[k1] == '*') {
// ready
k3++;
continue;
}
}
path << mask.substr(k3);
result = path.str();
return std::make_pair(method, result);
}
};
|
