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
|
#include <csignal>
#include <forward_list>
#include <functional>
#include <chrono>
#include <thread>
#include <boost/regex.hpp>
#include <boost/program_options.hpp>
#include "mongoose.h"
#include "db.h"
#include "web.h"
typedef void(web_handler_t)(const boost::cmatch&, struct mg_connection *conn);
std::forward_list<std::pair<boost::regex,std::function<web_handler_t>>> web_handler;
struct mg_context *ctx;
int begin_request_handler(struct mg_connection *conn) {
boost::cmatch match;
const struct mg_request_info *request_info = mg_get_request_info(conn);
std::cout << request_info->request_method << " "
<< request_info->uri << std::endl;
for (auto item = web_handler.begin(); item != web_handler.end(); ++item) {
if (boost::regex_match(request_info->uri, match, (*item).first)) {
(*item).second(match, conn);
return 1;
}
}
return 0;
}
struct sigaction old_action;
// Stop the server.
void sigint_handler(int s) {
mg_stop(ctx);
db_close();
exit(0);
}
namespace po = boost::program_options;
int main(int argc, char **argv) {
// Program options
po::options_description desc("Program options");
desc.add_options()
("wwwroot", po::value<std::string>()->required(), "Path to www root directory")
("port", po::value<int>()->default_value(8080), "HTTP Port")
("help", "Print help message")
;
po::variables_map vm;
po::store(po::parse_command_line(argc, argv, desc), vm);
try {
po::notify(vm);
} catch(boost::program_options::required_option& e) {
std::cerr << "ERROR: " << e.what() << std::endl << std::endl;
return 1;
}
if (vm.count("help")) {
std::cout << "Usage: " << argv[0] << std::endl << std::endl;
std::cout << desc << std::endl;
return 0;
}
const char *port = strdup(std::to_string(vm["port"].as<int>()).c_str());
const char *wwwroot = strdup(vm["wwwroot"].as<std::string>().c_str());
// Mongoose options
const char *options[] = {
"num_threads", "8",
"listening_ports", port,
"document_root", wwwroot,
NULL};
// Mongoose callbacks
struct mg_callbacks callbacks;
memset(&callbacks, 0, sizeof(callbacks));
callbacks.begin_request = begin_request_handler;
// Routing
web_handler.push_front(std::make_pair(
web_handle_api_value_R,
web_handle_api_value));
web_handler.push_front(std::make_pair(
web_handle_api_value_timestamp_R,
web_handle_api_value_timestamp));
web_handler.push_front(std::make_pair(
web_handle_api_range_R,
web_handle_api_range));
web_handler.push_front(std::make_pair(
web_handle_api_range_size_R,
web_handle_api_range_size));
web_handler.push_front(std::make_pair(
web_handle_api_list_R,
web_handle_api_list));
// Signals: handle C-c
struct sigaction action;
memset(&action, 0, sizeof(action));
action.sa_handler = &sigint_handler;
sigaction(SIGINT, &action, &old_action);
// Start the web server.
ctx = mg_start(&callbacks, NULL, options);
while (1)
std::this_thread::sleep_for(std::chrono::seconds(1));
return 1;
}
|
