#include "application.h" #include #include #include "request.h" #include "file_cache.h" #include #include #include void Application::setup_routes() { default_error_handler_func = Application::default_fallback_error_handler; error_handler_map[404] = Application::default_404_error_handler; } void Application::setup_middleware() { middlewares.push_back(HandlerInstance(Application::default_routing_middleware)); } void Application::default_routing_middleware(Object *instance, Request *request) { std::string path = request->http_parser->getPath(); if (FileCache::get_singleton()->wwwroot_has_file(path)) { send_file(path, request); return; } HandlerInstance handler_data; //std::function func; if (path == "/") { //quick shortcut handler_data = index_func; } else { std::string main_route = ""; uint32_t endpos = 1; for (; endpos < path.size(); ++endpos) { if (path[endpos] == '/') { break; } } main_route = path.substr(1, endpos - 1); handler_data = main_route_map[main_route]; } if (!handler_data.handler_func) { send_error(404, request); return; } request->handler_instance = handler_data; request->next_stage(); } void Application::default_fallback_error_handler(int error_code, Request *request) { request->response->setBody(default_generic_error_body); request->send(); } void Application::default_404_error_handler(int error_code, Request *request) { request->response->setBody(default_error_404_body); request->send(); } void Application::handle_request(Request *request) { request->middleware_stack = &middlewares; //note that middlewares handle the routing -> Application::default_routing_middleware by default request->next_stage(); } void Application::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 Application::send_file(const std::string &path, Request *request) { std::string fp = FileCache::get_singleton()->wwwroot + path; FILE *f = fopen(fp.c_str(), "rb"); if (!f) { printf("Error: Registered file doesn't exists anymore! %s\n", path.c_str()); send_error(404, request); return; } fseek(f, 0, SEEK_END); long fsize = ftell(f); fseek(f, 0, SEEK_SET); /* same as rewind(f); */ std::string body; body.resize(fsize); fread(&body[0], 1, fsize, f); fclose(f); //TODO set mimetype? request->response->setBody(body); request->send(); } Application::Application() { _instance = this; } Application::~Application() { main_route_map.clear(); error_handler_map.clear(); middlewares.clear(); } Application *Application::get_instance() { return _instance; } HandlerInstance Application::index_func; std::map Application::main_route_map; std::vector Application::middlewares; std::map > Application::error_handler_map; std::function Application::default_error_handler_func = nullptr; Application *Application::_instance = nullptr; std::string Application::default_error_404_body = "404 :("; std::string Application::default_generic_error_body = "Internal server error! :(";