use chuchi::extractor::{PathStr, Res};
use chuchi::header::{Mime, RequestHeader, ResponseHeader, StatusCode};
use chuchi::resources::Resources;
use chuchi::routes::Catcher;
use chuchi::util::PinnedFuture;
use chuchi::{get, post, Body, Request, Response};

#[macro_use]
mod util;

#[tokio::test]
async fn hello_world() {
	const BODY: &str = "Hello, World!";

	// build route
	#[get("/")]
	fn hello_world() -> &'static str {
		BODY
	}

	let addr = spawn_server!(|builder| {
		builder.add_route(hello_world);
	});

	// now do a request
	make_request!("GET", addr, "/")
		.await
		.assert_status(200)
		.assert_header("content-type", "text/plain; charset=utf-8")
		.assert_header("content-length", BODY.len().to_string())
		.assert_body_str(BODY)
		.await;
}

#[tokio::test]
async fn test_post() {
	const BODY: &str = "Hello, World!";

	// build route
	#[post("/")]
	fn post(req: &mut Request) -> Body {
		req.take_body()
	}

	let addr = spawn_server!(|builder| {
		builder.add_route(post);
	});

	// now do a request
	make_request!("POST", addr, "/", BODY)
		.await
		.assert_status(200)
		// this is the default content-type
		// we should probably change that
		.assert_not_header("content-type")
		// because we return a stream
		// we don't know how big it is
		.assert_not_header("content-length")
		.assert_body_str(BODY)
		.await;
}

#[tokio::test]
async fn test_params() {
	const BODY: &str = "Hello, name!";

	#[get("/hello/{name}")]
	async fn hello(name: &PathStr) -> String {
		format!("Hello, {}!", name)
	}

	#[get("/wild/{*rest}")]
	async fn wild(rest: &PathStr) -> String {
		format!("Wild: {}", rest)
	}

	#[get("/opt/{*?rest}")]
	async fn opt(rest: &PathStr) -> String {
		format!("Opt: {}", rest)
	}

	let addr = spawn_server!(|builder| {
		builder.add_route(hello);
		builder.add_route(wild);
		builder.add_route(opt);
	});

	// now do a request
	make_request!("GET", addr, "/hello/name")
		.await
		.assert_status(200)
		.assert_header("content-type", "text/plain; charset=utf-8")
		.assert_header("content-length", BODY.len().to_string())
		.assert_body_str(BODY)
		.await;

	make_request!("GET", addr, "/wild").await.assert_status(404);
	make_request!("GET", addr, "/wild/")
		.await
		.assert_status(404);

	make_request!("GET", addr, "/wild/123")
		.await
		.assert_status(200)
		.assert_body_str("Wild: 123")
		.await;

	make_request!("GET", addr, "/opt")
		.await
		.assert_status(200)
		.assert_body_str("Opt: ")
		.await;

	make_request!("GET", addr, "/opt/")
		.await
		.assert_status(200)
		.assert_body_str("Opt: ")
		.await;

	make_request!("GET", addr, "/opt/123")
		.await
		.assert_status(200)
		.assert_body_str("Opt: 123")
		.await;
}

#[tokio::test]
async fn test_catcher() {
	const BODY: &str = "Body not Found";

	struct NotFound;

	impl Catcher for NotFound {
		fn check(&self, _req: &RequestHeader, res: &ResponseHeader) -> bool {
			res.status_code() == &StatusCode::NOT_FOUND
		}

		fn call<'a>(
			&'a self,
			_req: &'a mut Request,
			resp: &'a mut Response,
			_data: &'a Resources,
		) -> PinnedFuture<'a, chuchi::Result<()>> {
			PinnedFuture::new(async move {
				*resp = Response::builder()
					.status_code(StatusCode::NOT_FOUND)
					.content_type(Mime::TEXT)
					.body(BODY)
					.build();

				Ok(())
			})
		}
	}

	let addr = spawn_server!(|builder| {
		builder.add_catcher(NotFound);
	});

	// now do a request
	make_request!("GET", addr, "/")
		.await
		.assert_status(404)
		// this is the default content-type
		// we should probably change that
		.assert_header("content-type", "text/plain; charset=utf-8")
		.assert_header("content-length", BODY.len().to_string())
		.assert_body_str(BODY)
		.await;
}

#[tokio::test]
async fn test_resource() {
	struct Data(Vec<u8>);

	// some random data
	let mut data = vec![];
	for i in 1..=1024 {
		data.push((i % 255) as u8);
	}

	// build route
	#[get("/")]
	fn get(data: Res<Data>) -> Vec<u8> {
		data.0.clone()
	}

	let addr = spawn_server!(|builder| {
		builder.add_resource(Data(data.clone()));
		builder.add_route(get);
	});

	// now do a request
	make_request!("GET", addr, "/")
		.await
		.assert_status(200)
		// this is the default content-type
		// we should probably change that
		.assert_header("content-type", "application/octet-stream")
		.assert_header("content-length", data.len().to_string())
		.assert_body_vec(&data)
		.await;
}