rcpp_framework/core/http/web_root.cpp

#include "web_root.h"

#include <functional>
#include <string>

#include "request.h"

#include "core/file_cache.h"

#include <iostream>

#include <stdio.h>
#include <stdlib.h>

void WebRoot::setup() {
	setup_error_handlers();
	setup_middleware();
}

void WebRoot::setup_error_handlers() {
	default_error_handler_func = WebRoot::default_fallback_error_handler;
	error_handler_map[404] = WebRoot::default_404_error_handler;
}

void WebRoot::setup_middleware() {
	// Middlewares get processed in the order they are in the _middlewares array

	// If you want sessions add this to your inherited class.
	// _middlewares.push_back(Ref<SessionSetupMiddleware>(new SessionSetupMiddleware()));

	// This one looks up users based on sessions
	// _middlewares.push_back(Ref<UserSessionSetupMiddleware>(new UserSessionSetupMiddleware()));

	// Same as the previous, but if you want the RBAC system to work use one of these
	// _middlewares.push_back(Ref<RBACUserSessionSetupMiddleware>(new RBACUserSessionSetupMiddleware()));
	// _middlewares.push_back(Ref<RBACDefaultUserSessionSetupMiddleware>(new RBACDefaultUserSessionSetupMiddleware()));
}

void WebRoot::default_fallback_error_handler(Request *request, int error_code) {
	request->compiled_body = default_generic_error_body;
	request->send();
}

void WebRoot::default_404_error_handler(Request *request, int error_code) {
	request->compiled_body = default_error_404_body;
	request->send();
}

void WebRoot::handle_request_main(Request *request) {
	process_middlewares(request);

	// handle files first
	if (try_send_wwwroot_file(request)) {
		return;
	}

	// normal routing
	WebRouterNode::handle_request_main(request);
}

void WebRoot::handle_error_send_request(Request *request, const int error_code) {
	std::function<void(Request *, int)> func = error_handler_map[error_code];

	if (!func) {
		if (!default_error_handler_func) {
			WebNode::handle_error_send_request(request, error_code);
			return;
		}

		default_error_handler_func(request, error_code);
		return;
	}

	func(request, error_code);
}

void WebRoot::process_middlewares(Request *request) {
	for (int i = 0; i < _middlewares.size(); ++i) {
		if (_middlewares[i]->on_before_handle_request_main(request)) {
			// handled
			return;
		}
	}
}

bool WebRoot::try_send_wwwroot_file(Request *request) {
	const String &path = request->get_path_full();

	if (FileCache::get_singleton()->wwwroot_has_file(path)) {
		send_file(path, request);

		return true;
	}

	return false;
}

void WebRoot::send_file(const String &path, Request *request) {
	String fp = FileCache::get_singleton()->wwwroot + path;

	request->send_file(fp);
}

void WebRoot::register_request_update(Request *request) {
	std::lock_guard<std::mutex> lock(_update_registered_requests_mutex);

	_update_registered_requests.push_back(request);
}
void WebRoot::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 WebRoot::update() {
	for (std::size_t i = 0; i < _update_registered_requests.size(); ++i) {
		Request *r = _update_registered_requests[i];

		r->update();
	}
}

WebRoot::WebRoot() :
		WebRouterNode() {
}

WebRoot::~WebRoot() {
	error_handler_map.clear();
	_middlewares.clear();
}

String WebRoot::default_error_404_body = "<html><body>404 :(</body></html>";
String WebRoot::default_generic_error_body = "<html><body>Internal server error! :(</body></html>";
Split WebApplication into WebServer and WebRoot. WebServer inherits from NodeTree, and WebRoot inherits from WebNode. Also removed some of the brynet classes. 2022-01-07 20:11:06 +01:00			`#include "web_root.h"`
Basic structure. Also added tinydir. 2020-11-25 00:20:41 +01:00
			`#include <functional>`
			`#include <string>`

			`#include "request.h"`

Moved every brynet based http class in core to a new bry_http folder. 2021-05-27 17:06:42 +02:00			`#include "core/file_cache.h"`
Basic structure. Also added tinydir. 2020-11-25 00:20:41 +01:00
			`#include <iostream>`

			`#include <stdio.h>`
			`#include <stdlib.h>`

Cleaned up WebRoot. 2022-01-08 12:35:44 +01:00			`void WebRoot::setup() {`
			`setup_error_handlers();`
			`setup_middleware();`
Started working on settings. 2020-12-08 16:22:18 +01:00			`}`

Cleaned up WebRoot. 2022-01-08 12:35:44 +01:00			`void WebRoot::setup_error_handlers() {`
Split WebApplication into WebServer and WebRoot. WebServer inherits from NodeTree, and WebRoot inherits from WebNode. Also removed some of the brynet classes. 2022-01-07 20:11:06 +01:00			`default_error_handler_func = WebRoot::default_fallback_error_handler;`
			`error_handler_map[404] = WebRoot::default_404_error_handler;`
Basic structure. Also added tinydir. 2020-11-25 00:20:41 +01:00			`}`

Split WebApplication into WebServer and WebRoot. WebServer inherits from NodeTree, and WebRoot inherits from WebNode. Also removed some of the brynet classes. 2022-01-07 20:11:06 +01:00			`void WebRoot::setup_middleware() {`
Implemented middleware handling. Also converted middleware handlers to the new stystem. 2022-01-08 12:25:52 +01:00			`// Middlewares get processed in the order they are in the _middlewares array`
Actually make session_setup_middleware work. 2021-08-04 21:58:25 +02:00
Move middleware processing to a helper method. 2022-01-08 13:09:12 +01:00			`// If you want sessions add this to your inherited class.`
Implemented middleware handling. Also converted middleware handlers to the new stystem. 2022-01-08 12:25:52 +01:00			`// _middlewares.push_back(Ref<SessionSetupMiddleware>(new SessionSetupMiddleware()));`
Basic structure. Also added tinydir. 2020-11-25 00:20:41 +01:00
Implemented middleware handling. Also converted middleware handlers to the new stystem. 2022-01-08 12:25:52 +01:00			`// This one looks up users based on sessions`
			`// _middlewares.push_back(Ref<UserSessionSetupMiddleware>(new UserSessionSetupMiddleware()));`
Basic structure. Also added tinydir. 2020-11-25 00:20:41 +01:00
Implemented middleware handling. Also converted middleware handlers to the new stystem. 2022-01-08 12:25:52 +01:00			`// Same as the previous, but if you want the RBAC system to work use one of these`
			`// _middlewares.push_back(Ref<RBACUserSessionSetupMiddleware>(new RBACUserSessionSetupMiddleware()));`
			`// _middlewares.push_back(Ref<RBACDefaultUserSessionSetupMiddleware>(new RBACDefaultUserSessionSetupMiddleware()));`
Moved routing to a middleware, so it's easily customizable, and also it is now able to do async things. 2020-11-26 12:10:21 +01:00			`}`

Make WebRoot use the new error handling system. 2022-01-08 11:51:39 +01:00			`void WebRoot::default_fallback_error_handler(Request *request, int error_code) {`
Made Request a base inheritable class, and made DRequest inherit from it. Also added BryRequest for the brynet backend. Also made DWebApplication inherit from WebApplication. 2021-07-06 20:06:38 +02:00			`request->compiled_body = default_generic_error_body;`
Moved routing to a middleware, so it's easily customizable, and also it is now able to do async things. 2020-11-26 12:10:21 +01:00			`request->send();`
			`}`

Make WebRoot use the new error handling system. 2022-01-08 11:51:39 +01:00			`void WebRoot::default_404_error_handler(Request *request, int error_code) {`
Made Request a base inheritable class, and made DRequest inherit from it. Also added BryRequest for the brynet backend. Also made DWebApplication inherit from WebApplication. 2021-07-06 20:06:38 +02:00			`request->compiled_body = default_error_404_body;`
Moved routing to a middleware, so it's easily customizable, and also it is now able to do async things. 2020-11-26 12:10:21 +01:00			`request->send();`
			`}`

Split WebApplication into WebServer and WebRoot. WebServer inherits from NodeTree, and WebRoot inherits from WebNode. Also removed some of the brynet classes. 2022-01-07 20:11:06 +01:00			`void WebRoot::handle_request_main(Request *request) {`
Move middleware processing to a helper method. 2022-01-08 13:09:12 +01:00			`process_middlewares(request);`
Started reworking routing. 2022-01-08 10:50:07 +01:00
Implemented the new node based routing. 2022-01-08 11:27:26 +01:00			`// handle files first`
			`if (try_send_wwwroot_file(request)) {`
			`return;`
			`}`
Started reworking routing. 2022-01-08 10:50:07 +01:00
Implemented the new node based routing. 2022-01-08 11:27:26 +01:00			`// normal routing`
			`WebRouterNode::handle_request_main(request);`
			`}`
Make WebRoot use the new error handling system. 2022-01-08 11:51:39 +01:00
Reworked error handling. 2022-01-08 11:47:39 +01:00			`void WebRoot::handle_error_send_request(Request *request, const int error_code) {`
Make WebRoot use the new error handling system. 2022-01-08 11:51:39 +01:00			`std::function<void(Request *, int)> func = error_handler_map[error_code];`

			`if (!func) {`
			`if (!default_error_handler_func) {`
			`WebNode::handle_error_send_request(request, error_code);`
			`return;`
			`}`
Reworked error handling. 2022-01-08 11:47:39 +01:00
Make WebRoot use the new error handling system. 2022-01-08 11:51:39 +01:00			`default_error_handler_func(request, error_code);`
			`return;`
			`}`

			`func(request, error_code);`
			`}`
Implemented the new node based routing. 2022-01-08 11:27:26 +01:00
Move middleware processing to a helper method. 2022-01-08 13:09:12 +01:00			`void WebRoot::process_middlewares(Request *request) {`
			`for (int i = 0; i < _middlewares.size(); ++i) {`
			`if (_middlewares[i]->on_before_handle_request_main(request)) {`
			`// handled`
			`return;`
			`}`
			`}`
			`}`

Implemented the new node based routing. 2022-01-08 11:27:26 +01:00			`bool WebRoot::try_send_wwwroot_file(Request *request) {`
			`const String &path = request->get_path_full();`
Started reworking routing. 2022-01-08 10:50:07 +01:00
			`if (FileCache::get_singleton()->wwwroot_has_file(path)) {`
			`send_file(path, request);`

Implemented the new node based routing. 2022-01-08 11:27:26 +01:00			`return true;`
Started reworking routing. 2022-01-08 10:50:07 +01:00			`}`

Implemented the new node based routing. 2022-01-08 11:27:26 +01:00			`return false;`
Basic structure. Also added tinydir. 2020-11-25 00:20:41 +01:00			`}`

Replaced std::strings with String in WebRoot. 2022-01-08 12:41:15 +01:00			`void WebRoot::send_file(const String &path, Request *request) {`
			`String fp = FileCache::get_singleton()->wwwroot + path;`
Implement Themes, and fix middlewares. 2020-11-25 21:00:22 +01:00
Fix file chunking with downloads. Also fixed a bunch of std::shared_ptr misuses, and made Application::send_file use Request's file chunking. Also the framework will properly handle client disconnects now. 2021-02-09 01:25:24 +01:00			`request->send_file(fp);`
			`}`
Fix crash if a file gets deleted but still registered. 2020-11-25 13:22:43 +01:00
Split WebApplication into WebServer and WebRoot. WebServer inherits from NodeTree, and WebRoot inherits from WebNode. Also removed some of the brynet classes. 2022-01-07 20:11:06 +01:00			`void WebRoot::register_request_update(Request *request) {`
Fix file chunking with downloads. Also fixed a bunch of std::shared_ptr misuses, and made Application::send_file use Request's file chunking. Also the framework will properly handle client disconnects now. 2021-02-09 01:25:24 +01:00			`std::lock_guard<std::mutex> lock(_update_registered_requests_mutex);`
Fix crash if a file gets deleted but still registered. 2020-11-25 13:22:43 +01:00
Fix file chunking with downloads. Also fixed a bunch of std::shared_ptr misuses, and made Application::send_file use Request's file chunking. Also the framework will properly handle client disconnects now. 2021-02-09 01:25:24 +01:00			`_update_registered_requests.push_back(request);`
			`}`
Split WebApplication into WebServer and WebRoot. WebServer inherits from NodeTree, and WebRoot inherits from WebNode. Also removed some of the brynet classes. 2022-01-07 20:11:06 +01:00			`void WebRoot::unregister_request_update(Request *request) {`
Fix file chunking with downloads. Also fixed a bunch of std::shared_ptr misuses, and made Application::send_file use Request's file chunking. Also the framework will properly handle client disconnects now. 2021-02-09 01:25:24 +01:00			`std::lock_guard<std::mutex> lock(_update_registered_requests_mutex);`
Basic structure. Also added tinydir. 2020-11-25 00:20:41 +01:00
Fix file chunking with downloads. Also fixed a bunch of std::shared_ptr misuses, and made Application::send_file use Request's file chunking. Also the framework will properly handle client disconnects now. 2021-02-09 01:25:24 +01:00			`std::size_t s = _update_registered_requests.size();`
			`for (std::size_t i = 0; i < s; ++i) {`
			`Request *r = _update_registered_requests[i];`
Basic structure. Also added tinydir. 2020-11-25 00:20:41 +01:00
Fix file chunking with downloads. Also fixed a bunch of std::shared_ptr misuses, and made Application::send_file use Request's file chunking. Also the framework will properly handle client disconnects now. 2021-02-09 01:25:24 +01:00			`if (r == request) {`
			`_update_registered_requests[i] = _update_registered_requests[s - 1];`
Basic structure. Also added tinydir. 2020-11-25 00:20:41 +01:00
Fix file chunking with downloads. Also fixed a bunch of std::shared_ptr misuses, and made Application::send_file use Request's file chunking. Also the framework will properly handle client disconnects now. 2021-02-09 01:25:24 +01:00			`_update_registered_requests.pop_back();`
Fix crash if a file gets deleted but still registered. 2020-11-25 13:22:43 +01:00
Fix file chunking with downloads. Also fixed a bunch of std::shared_ptr misuses, and made Application::send_file use Request's file chunking. Also the framework will properly handle client disconnects now. 2021-02-09 01:25:24 +01:00			`return;`
			`}`
			`}`
Basic structure. Also added tinydir. 2020-11-25 00:20:41 +01:00			`}`

Split WebApplication into WebServer and WebRoot. WebServer inherits from NodeTree, and WebRoot inherits from WebNode. Also removed some of the brynet classes. 2022-01-07 20:11:06 +01:00			`void WebRoot::update() {`
Fix file chunking with downloads. Also fixed a bunch of std::shared_ptr misuses, and made Application::send_file use Request's file chunking. Also the framework will properly handle client disconnects now. 2021-02-09 01:25:24 +01:00			`for (std::size_t i = 0; i < _update_registered_requests.size(); ++i) {`
			`Request *r = _update_registered_requests[i];`

			`r->update();`
			`}`
Initial TableBuilder implementation, and simple migrations. 2020-12-01 23:37:10 +01:00			`}`

Implemented the new node based routing. 2022-01-08 11:27:26 +01:00			`WebRoot::WebRoot() :`
			`WebRouterNode() {`
Basic structure. Also added tinydir. 2020-11-25 00:20:41 +01:00			`}`

Split WebApplication into WebServer and WebRoot. WebServer inherits from NodeTree, and WebRoot inherits from WebNode. Also removed some of the brynet classes. 2022-01-07 20:11:06 +01:00			`WebRoot::~WebRoot() {`
Basic structure. Also added tinydir. 2020-11-25 00:20:41 +01:00			`error_handler_map.clear();`
Cleaned up WebRoot. 2022-01-08 12:35:44 +01:00			`_middlewares.clear();`
Basic structure. Also added tinydir. 2020-11-25 00:20:41 +01:00			`}`

Replaced std::strings with String in WebRoot. 2022-01-08 12:41:15 +01:00			`String WebRoot::default_error_404_body = "<html><body>404 :(</body></html>";`
			`String WebRoot::default_generic_error_body = "<html><body>Internal server error! :(</body></html>";`