145 lines
3.6 KiB
C++
145 lines
3.6 KiB
C++
#include "web_application.h"
|
|
|
|
#include <functional>
|
|
#include <string>
|
|
|
|
#include "request.h"
|
|
|
|
#include "core/file_cache.h"
|
|
|
|
#include <iostream>
|
|
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
|
|
void WebApplication::load_settings() {
|
|
}
|
|
|
|
void WebApplication::setup_routes() {
|
|
default_error_handler_func = WebApplication::default_fallback_error_handler;
|
|
|
|
error_handler_map[404] = WebApplication::default_404_error_handler;
|
|
}
|
|
|
|
void WebApplication::setup_middleware() {
|
|
//If you want sessions add this to your inherited class. Should probably be the first one.
|
|
//middlewares.push_back(HandlerInstance(::SessionManager::session_setup_middleware));
|
|
|
|
middlewares.push_back(HandlerInstance([this](Object *instance, Request *request){ this->default_routing_middleware(instance, request); }));
|
|
}
|
|
|
|
void WebApplication::default_routing_middleware(Object *instance, Request *request) {
|
|
std::string path = request->get_path_full();
|
|
|
|
if (FileCache::get_singleton()->wwwroot_has_file(path)) {
|
|
send_file(path, request);
|
|
|
|
return;
|
|
}
|
|
|
|
HandlerInstance handler_data;
|
|
|
|
//std::function<void(Object *, Request *)> func;
|
|
|
|
//if (path == "/") {
|
|
if (request->get_path_segment_count() == 0) {
|
|
//quick shortcut
|
|
handler_data = index_func;
|
|
} else {
|
|
const std::string main_route = request->get_current_path_segment();
|
|
|
|
handler_data = main_route_map[main_route];
|
|
|
|
request->push_path();
|
|
}
|
|
|
|
if (!handler_data.handler_func) {
|
|
send_error(404, request);
|
|
|
|
return;
|
|
}
|
|
|
|
request->handler_instance = handler_data;
|
|
request->next_stage();
|
|
}
|
|
|
|
void WebApplication::default_fallback_error_handler(int error_code, Request *request) {
|
|
request->compiled_body = default_generic_error_body;
|
|
|
|
request->send();
|
|
}
|
|
|
|
void WebApplication::default_404_error_handler(int error_code, Request *request) {
|
|
request->compiled_body = default_error_404_body;
|
|
request->send();
|
|
}
|
|
|
|
void WebApplication::handle_request(Request *request) {
|
|
request->middleware_stack = &middlewares;
|
|
|
|
//note that middlewares handle the routing -> WebApplication::default_routing_middleware by default
|
|
request->next_stage();
|
|
}
|
|
|
|
void WebApplication::send_error(int error_code, Request *request) {
|
|
std::function<void(int, Request *)> func = error_handler_map[error_code];
|
|
|
|
if (!func) {
|
|
default_error_handler_func(error_code, request);
|
|
return;
|
|
}
|
|
|
|
func(error_code, request);
|
|
}
|
|
|
|
void WebApplication::send_file(const std::string &path, Request *request) {
|
|
std::string fp = FileCache::get_singleton()->wwwroot + path;
|
|
|
|
request->send_file(fp);
|
|
}
|
|
|
|
void WebApplication::migrate() {
|
|
}
|
|
|
|
void WebApplication::register_request_update(Request *request) {
|
|
std::lock_guard<std::mutex> lock(_update_registered_requests_mutex);
|
|
|
|
_update_registered_requests.push_back(request);
|
|
}
|
|
void WebApplication::unregister_request_update(Request *request) {
|
|
std::lock_guard<std::mutex> lock(_update_registered_requests_mutex);
|
|
|
|
std::size_t s = _update_registered_requests.size();
|
|
for (std::size_t i = 0; i < s; ++i) {
|
|
Request *r = _update_registered_requests[i];
|
|
|
|
if (r == request) {
|
|
_update_registered_requests[i] = _update_registered_requests[s - 1];
|
|
|
|
_update_registered_requests.pop_back();
|
|
|
|
return;
|
|
}
|
|
}
|
|
}
|
|
|
|
void WebApplication::update() {
|
|
for (std::size_t i = 0; i < _update_registered_requests.size(); ++i) {
|
|
Request *r = _update_registered_requests[i];
|
|
|
|
r->update();
|
|
}
|
|
}
|
|
|
|
WebApplication::WebApplication() {
|
|
}
|
|
|
|
WebApplication::~WebApplication() {
|
|
main_route_map.clear();
|
|
error_handler_map.clear();
|
|
middlewares.clear();
|
|
}
|
|
|
|
std::string WebApplication::default_error_404_body = "<html><body>404 :(</body></html>";
|
|
std::string WebApplication::default_generic_error_body = "<html><body>Internal server error! :(</body></html>";
|