ptth/crates/ptth_relay/src/routing.rs

use hyper::Method;

#[derive (Debug, PartialEq)]
pub enum Route <'a> {
	ClientAuditLog,
	ClientRelayIsUp,
	ClientServerGet {
		listen_code: &'a str,
		path: &'a str,
	},
	ClientServerList,
	ClientUnregisteredServers,
	Debug,
	DebugEndlessSink,
	DebugEndlessSource (Option <usize>),
	DebugGenKey,
	DebugMysteriousError,
	ErrorBadUriFormat,
	ErrorCantPost,
	ErrorMethodNotAllowed,
	ErrorRoutingFailed,
	RegisterServer,
	Root,
	Scraper {
		rest: &'a str,
	},
	ServerHttpListen {
		listen_code: &'a str
	},
	ServerHttpResponse {
		request_code: &'a str,
	},
}

pub fn route_url <'a> (method: &Method, path: &'a str) -> Route <'a> {
	if let Some (listen_code) = path.strip_prefix ("/7ZSFUKGV/http_listen/") {
		if method != Method::GET {
			return Route::ErrorMethodNotAllowed;
		}
		
		Route::ServerHttpListen {
			listen_code
		}
	}
	else if let Some (request_code) = path.strip_prefix ("/7ZSFUKGV/http_response/") {
		if method != Method::POST {
			return Route::ErrorMethodNotAllowed;
		}
		Route::ServerHttpResponse {
			request_code
		}
	}
	else if path == "/frontend/register" {
		if method != Method::POST {
			return Route::ErrorMethodNotAllowed;
		}
		Route::RegisterServer
	}
	else if let Some (rest) = path.strip_prefix ("/frontend/servers/") {
		// DRY T4H76LB3
		
		if rest.is_empty () {
			if method != Method::GET {
				return Route::ErrorMethodNotAllowed;
			}
			Route::ClientServerList
		}
		else if let Some (idx) = rest.find ('/') {
			let listen_code = &rest [0..idx];
			
			Route::ClientServerGet {
				listen_code, 
				path: &rest [idx..],
			}
		}
		else {
			if method != Method::GET {
				return Route::ErrorMethodNotAllowed;
			}
			Route::ErrorBadUriFormat
		}
	}
	else if path == "/frontend/unregistered_servers" {
		if method != Method::GET {
			return Route::ErrorMethodNotAllowed;
		}
		Route::ClientUnregisteredServers
	}
	else if path == "/frontend/audit_log" {
		if method != Method::GET {
			return Route::ErrorMethodNotAllowed;
		}
		Route::ClientAuditLog
	}
	else if let Some (rest) = path.strip_prefix ("/frontend/debug/") {
		if rest.is_empty () {
			if method != Method::GET {
				return Route::ErrorMethodNotAllowed;
			}
			Route::Debug
		}
		else if rest == "endless_source" {
			if method != Method::GET {
				return Route::ErrorMethodNotAllowed;
			}
			Route::DebugEndlessSource (None)
		}
		else if rest == "endless_source_throttled" {
			if method != Method::GET {
				return Route::ErrorMethodNotAllowed;
			}
			Route::DebugEndlessSource (Some (1024 / 64))
		}
		else if rest == "endless_sink" {
			if method != Method::POST {
				return Route::ErrorMethodNotAllowed;
			}
			Route::DebugEndlessSink
		}
		else if rest == "gen_key" {
			if method != Method::GET {
				return Route::ErrorMethodNotAllowed;
			}
			Route::DebugGenKey
		}
		else {
			Route::ErrorRoutingFailed
		}
	}
	else if path == "/" {
		if method != Method::GET {
			return Route::ErrorMethodNotAllowed;
		}
		Route::Root
	}
	else if path == "/frontend/relay_up_check" {
		if method != Method::GET {
			return Route::ErrorMethodNotAllowed;
		}
		Route::ClientRelayIsUp
	}
	else if path == "/frontend/test_mysterious_error" {
		if method != Method::GET {
			return Route::ErrorMethodNotAllowed;
		}
		Route::DebugMysteriousError
	}
	else if let Some (rest) = path.strip_prefix ("/scraper/") {
		if method != Method::GET {
			return Route::ErrorMethodNotAllowed;
		}
		Route::Scraper {
			rest
		}
	}
	else {
		Route::ErrorRoutingFailed
	}
}

#[cfg (test)]
mod tests {
	use super::*;
	
	#[test]
	fn routing () {
		for (input, expected) in vec! [
			("/", Route::Root),
		].into_iter () {
			let actual = route_url (&Method::GET, input);
			assert_eq! (actual, expected);
		}
		
		for (input, expected) in vec! [
			("/", Route::ErrorMethodNotAllowed),
		].into_iter () {
			let actual = route_url (&Method::POST, input);
			assert_eq! (actual, expected);
		}
	}
}