#include "web_application.h" #include #include #include "request.h" #include "core/file_cache.h" #include #include #include 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 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 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 lock(_update_registered_requests_mutex); _update_registered_requests.push_back(request); } void WebApplication::unregister_request_update(Request *request) { std::lock_guard 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 = "404 :("; std::string WebApplication::default_generic_error_body = "Internal server error! :(";