Update main.cc
file is now in .clang format
This commit is contained in:
Bhavya Gupta
GitHub
parent
87f944a758
commit
55957e3fae
110
docker/main.cc
110
docker/main.cc
@ -25,76 +25,84 @@ constexpr auto error_html = R"(<html>
|
||||
</html>
|
||||
)";
|
||||
|
||||
void sigint_handler(int s) { exit(1); }
|
||||
void sigint_handler(int s) {
|
||||
exit(1);
|
||||
}
|
||||
|
||||
std::string time_local() {
|
||||
auto p = std::chrono::system_clock::now();
|
||||
auto t = std::chrono::system_clock::to_time_t(p);
|
||||
auto p = std::chrono::system_clock::now();
|
||||
auto t = std::chrono::system_clock::to_time_t(p);
|
||||
|
||||
std::stringstream ss;
|
||||
ss << std::put_time(std::localtime(&t), "%d/%b/%Y:%H:%M:%S %z");
|
||||
return ss.str();
|
||||
std::stringstream ss;
|
||||
ss << std::put_time(std::localtime(&t), "%d/%b/%Y:%H:%M:%S %z");
|
||||
return ss.str();
|
||||
}
|
||||
|
||||
std::string log(const httplib::Request &req, const httplib::Response &res) {
|
||||
auto remote_user = "-"; // TODO:
|
||||
auto request = std::format("{} {} {}", req.method, req.path, req.version);
|
||||
auto body_bytes_sent = res.get_header_value("Content-Length");
|
||||
auto http_referer = "-"; // TODO:
|
||||
auto http_user_agent = req.get_header_value("User-Agent", "-");
|
||||
auto remote_user = "-"; // TODO:
|
||||
auto request = std::format("{} {} {}", req.method, req.path, req.version);
|
||||
auto body_bytes_sent = res.get_header_value("Content-Length");
|
||||
auto http_referer = "-"; // TODO:
|
||||
auto http_user_agent = req.get_header_value("User-Agent", "-");
|
||||
|
||||
// NOTE: From NGINX default access log format
|
||||
return std::format(R"({} - {} [{}] "{}" {} {} "{}" "{}")", req.remote_addr,
|
||||
remote_user, time_local(), request, res.status,
|
||||
body_bytes_sent, http_referer, http_user_agent);
|
||||
// NOTE: From NGINX default access log format
|
||||
return std::format(
|
||||
R"({} - {} [{}] "{}" {} {} "{}" "{}")", req.remote_addr, remote_user, time_local(),
|
||||
request, res.status, body_bytes_sent, http_referer, http_user_agent);
|
||||
}
|
||||
|
||||
int main(int argc, const char **argv) {
|
||||
// Default values for host, port, and base directory
|
||||
std::string base_dir = "./html";
|
||||
std::string host = "0.0.0.0";
|
||||
int port = 80;
|
||||
// Default values for host, port, and base directory
|
||||
std::string base_dir = "./html";
|
||||
std::string host = "0.0.0.0";
|
||||
int port = 80;
|
||||
|
||||
// Parse command-line arguments (if provided)
|
||||
if (argc > 1) base_dir = argv[1];
|
||||
if (argc > 2) host = argv[2];
|
||||
if (argc > 3) port = std::stoi(argv[3]);
|
||||
// Parse command-line arguments (if provided)
|
||||
if (argc > 1)
|
||||
base_dir = argv[1];
|
||||
if (argc > 2)
|
||||
host = argv[2];
|
||||
if (argc > 3)
|
||||
port = std::stoi(argv[3]);
|
||||
|
||||
// Check if the base directory exists
|
||||
if (!std::filesystem::exists(base_dir)) {
|
||||
std::cerr << "Error: Directory " << base_dir << " does not exist." << std::endl;
|
||||
return 1; // Exit if the directory is invalid
|
||||
}
|
||||
// Check if the base directory exists
|
||||
if (!std::filesystem::exists(base_dir)) {
|
||||
std::cerr << "Error: Directory " << base_dir << " does not exist." << std::endl;
|
||||
return 1; // Exit if the directory is invalid
|
||||
}
|
||||
|
||||
signal(SIGINT, sigint_handler);
|
||||
signal(SIGINT, sigint_handler);
|
||||
|
||||
httplib::Server svr;
|
||||
httplib::Server svr;
|
||||
|
||||
// Set up multithreading with 8 threads
|
||||
svr.new_task_queue = [] { return new httplib::ThreadPool(8); };
|
||||
// Set up multithreading with 8 threads
|
||||
svr.new_task_queue = [] {
|
||||
return new httplib::ThreadPool(8);
|
||||
};
|
||||
|
||||
// Error handling for 404 and other status codes
|
||||
svr.set_error_handler([](const httplib::Request & /*req*/, httplib::Response &res) {
|
||||
auto body = std::format(error_html, res.status, httplib::status_message(res.status), CPPHTTPLIB_VERSION);
|
||||
res.set_content(body, "text/html");
|
||||
});
|
||||
// Error handling for 404 and other status codes
|
||||
svr.set_error_handler([](const httplib::Request & /*req*/, httplib::Response &res) {
|
||||
auto body = std::format(
|
||||
error_html, res.status, httplib::status_message(res.status), CPPHTTPLIB_VERSION);
|
||||
res.set_content(body, "text/html");
|
||||
});
|
||||
|
||||
// Set up logging
|
||||
svr.set_logger([](const httplib::Request &req, const httplib::Response &res) {
|
||||
std::cout << log(req, res) << std::endl;
|
||||
});
|
||||
// Set up logging
|
||||
svr.set_logger([](const httplib::Request &req, const httplib::Response &res) {
|
||||
std::cout << log(req, res) << std::endl;
|
||||
});
|
||||
|
||||
// Mount the base directory for serving static files
|
||||
svr.set_mount_point("/", base_dir);
|
||||
// Mount the base directory for serving static files
|
||||
svr.set_mount_point("/", base_dir);
|
||||
|
||||
// Start server and handle failures
|
||||
std::cout << std::format("Serving HTTP on {0} port {1} ...", host, port) << std::endl;
|
||||
auto ret = svr.listen(host.c_str(), port);
|
||||
// Start server and handle failures
|
||||
std::cout << std::format("Serving HTTP on {0} port {1} ...", host, port) << std::endl;
|
||||
auto ret = svr.listen(host.c_str(), port);
|
||||
|
||||
if (!ret) {
|
||||
std::cerr << "Error: Failed to start server on " << host << ":" << port << std::endl;
|
||||
return 1; // Exit with failure code if listen fails
|
||||
}
|
||||
if (!ret) {
|
||||
std::cerr << "Error: Failed to start server on " << host << ":" << port << std::endl;
|
||||
return 1; // Exit with failure code if listen fails
|
||||
}
|
||||
|
||||
return 0;
|
||||
return 0;
|
||||
}
|
||||
|
||||
Loading…
Reference in New Issue
Block a user