beet_net 0.0.8

Cross-platform networking utilities
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85

use crate::prelude::*;
use beet_core::prelude::*;

/// A 'server' that accepts the cli arguments and environment variables as a request,
/// exiting with the output.
#[derive(Component)]
#[component(on_add=on_add)]
pub struct CliServer;

fn on_add(mut world: DeferredWorld, cx: HookContext) {
	let entity = cx.entity;
	world.commands().queue(move |world: &mut World| -> Result {
		world.entity_mut(entity).run_async_local(
			async move |entity| -> Result {
				let req = Request::from_cli_args(CliArgs::parse_env())?;
				let res = entity.oneshot(req).await;
				let (parts, body) = res.into_parts();
				let body = body.into_string().await?;
				let exit = match parts.status_to_exit_code() {
					Ok(()) => {
						body.xprint_display();
						AppExit::Success
					}
					Err(code) => {
						let body = if body.is_empty() {
							body
						} else {
							format!("Body:\n{}", body)
						};
						error!("Command failed\nStatus code: {code}\n{}", body);
						// TODO map http status to
						AppExit::Error(code)
					}
				};
				entity.world().write_message(exit);
				Ok(())
			},
		);
		Ok(())
	});
}

#[cfg(test)]
mod tests {
	use super::*;

	#[beet_core::test]
	#[cfg(feature = "http")]
	async fn cli_server_works() {
		App::new()
			.add_plugins((MinimalPlugins, ServerPlugin))
			.spawn_then((
				CliServer,
				ExchangeSpawner::new_handler(|_, _| {
					StatusCode::ImATeapot.into()
				}),
			))
			.run_async()
			.await
			.xpect_eq(AppExit::Error(1.try_into().unwrap()));
	}

	#[test]
	fn into_request_simple_path() {
		Request::from_cli_str("foo bar")
			.unwrap()
			.path_string()
			.xpect_eq("/foo/bar");
	}

	#[test]
	fn into_request_with_query() {
		let req = Request::from_cli_str("api users --id=123").unwrap();
		req.path_string().xpect_eq("/api/users");
		req.get_param("id").xpect_some();
	}

	#[test]
	fn into_request_empty() {
		Request::from_cli_str("")
			.unwrap()
			.path_string()
			.xpect_eq("/");
	}
}