beet 0.0.8

Malleable Application Framework
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
86
87
88
89
90
91
92
93

//! Example of a basic server with hand-rolled routing and templating
use beet::prelude::*;
use serde::Deserialize;

fn main() {
	App::new()
		.add_plugins((
			MinimalPlugins,
			LogPlugin::default(),
			ServerPlugin::default(),
		))
		.init_resource::<VisitCounter>()
		.add_observer(handler)
		.run();
}

#[derive(Deserialize)]
struct MyParams {
	name: String,
}

#[derive(Default, Resource)]
struct VisitCounter(u32);

fn handler(
	ev: On<Insert, Request>,
	mut commands: Commands,
	requests: Query<&RequestMeta>,
	time: Res<Time>,
	mut visit_counter: ResMut<VisitCounter>,
) -> Result {
	let request = requests.get(ev.event_target())?;
	let path = request.path_string();
	// our diy router, only match root path
	if path != "/" {
		commands
			.entity(ev.event_target())
			.insert(Response::from_status_body(
				StatusCode::NotFound,
				format!("Path not found: {}", path),
				"text/plain",
			));
		return Ok(());
	}
	visit_counter.0 += 1;
	let num_visits = visit_counter.0;

	let name = if let Ok(params) =
		QueryParams::<MyParams>::from_request_meta(&request)
	{
		params.name.clone()
	} else {
		"User".to_string()
	};

	let uptime = time.elapsed_secs();

	let special_message = if num_visits % 7 == 0 {
		format!("<p>Congratulations you are visitor number {num_visits}!</p>")
	} else {
		default()
	};
	// simple templating with format!
	let response_text = format!(
		r#"
<!DOCTYPE html>
<html>
  <head>
    <title>Beet Server</title>
    <style>
      body {{
      font-family: system-ui, sans-serif;
     	  background-color: black;
     	  color: white;
      }}
    </style>
  </head>
  <body>
    <pre>
  Greetings {name}!
  Visit Count: {num_visits}
  Uptime: {uptime:.2} seconds
    </pre>
  {special_message}
  </body>
</html>
"#,
	);
	commands
		.entity(ev.event_target())
		.insert(Response::ok_body(response_text, "text/html"));
	Ok(())
}