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

//! WebSocket client example
//!
//! This example demonstrates how to connect to a WebSocket server and
//! send/receive messages.
//!
//! Make sure the server is running first:
//! ```sh
//! cargo run --example socket_server --features tungstenite
//! ```
//!
//! Run with:
//! ```sh
//! cargo run --example socket_client --features tungstenite
//! ```
//!

use beet::net::prelude::sockets::Message;
use beet::net::prelude::sockets::*;
use beet::prelude::*;
use beet_net::sockets::common_handlers::PingTime;

fn main() {
	App::new()
		.add_plugins((
			MinimalPlugins,
			LogPlugin::default(),
			AsyncPlugin::default(),
		))
		.spawn_then((
			Socket::insert_on_connect("ws://127.0.0.1:9000"),
			OnSpawn::observe(on_ready),
			OnSpawn::observe(my_handler),
			OnSpawn::observe(common_handlers::log_send),
			OnSpawn::observe(common_handlers::echo_pingtime),
			OnSpawn::observe(common_handlers::log_recv),
		))
		.run();
	info!("Done");
}


fn on_ready(ev: On<SocketReady>, mut commands: Commands) {
	commands
		.entity(ev.target())
		// we'll send a PingTime ping to get the RTT
		.trigger_target(MessageSend(PingTime::default().into_message()))
		// we'll also send a text message that the server is expecting
		.trigger_target(MessageSend(Message::Text(
			"the cat sat on the".into(),
		)));
}

fn my_handler(ev: On<MessageRecv>, mut commands: Commands) {
	match ev.event().inner() {
		// on receiving the expected message we'll send a close message
		Message::Text(txt) if txt == "hat" => {
			commands
				.entity(ev.target())
				.trigger_target(MessageSend(Message::Close(None)));
		}
		// exit the app when the close is acknowledged
		Message::Close(_) => {
			println!("done!");
			commands.write_message(AppExit::Success);
		}
		_ => {}
	}
}