heapswap 0.0.1

Heapswap's server-side engine
Documentation 
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

use futures_util::StreamExt;
use std::sync::Arc;
use tokio::sync::{Mutex, RwLock};
use warp::ws::{WebSocket, Ws};
use warp::{Filter, Rejection, Reply};
use y_sync::awareness::Awareness;
use y_sync::net::BroadcastGroup;
use yrs::{Doc, Text, Transact};
use yrs_warp::ws::{WarpSink, WarpStream};
use yrs_warp::AwarenessRef;

const STATIC_FILES_DIR: &str = "examples/code-mirror/frontend/dist";

#[tokio::main]
async fn main() {
	// We're using a single static document shared among all the peers.
	let awareness: AwarenessRef = {
		let doc = Doc::new();
		{
			// pre-initialize code mirror document with some text
			let txt = doc.get_or_insert_text("codemirror");
			let mut txn = doc.transact_mut();
			txt.push(
				&mut txn,
				r#"function hello() {
  console.log('hello world');
}"#,
			);
		}
		Arc::new(RwLock::new(Awareness::new(doc)))
	};

	// open a broadcast group that listens to awareness and document updates
	// and has a pending message buffer of up to 32 updates
	let bcast = Arc::new(BroadcastGroup::new(awareness.clone(), 32).await);

	let static_files = warp::get().and(warp::fs::dir(STATIC_FILES_DIR));

	let ws = warp::path("my-room")
		.and(warp::ws())
		.and(warp::any().map(move || bcast.clone()))
		.and_then(ws_handler);

	let routes = ws.or(static_files);

	warp::serve(routes).run(([0, 0, 0, 0], 8000)).await;
}

async fn ws_handler(
	ws: Ws,
	bcast: Arc<BroadcastGroup>,
) -> Result<impl Reply, Rejection> {
	Ok(ws.on_upgrade(move |socket| peer(socket, bcast)))
}

async fn peer(
	ws: WebSocket,
	bcast: Arc<BroadcastGroup>,
) {
	let (sink, stream) = ws.split();
	let sink = Arc::new(Mutex::new(WarpSink::from(sink)));
	let stream = WarpStream::from(stream);
	let sub = bcast.subscribe(sink, stream);
	match sub.completed().await {
		Ok(_) => println!("broadcasting for channel finished successfully"),
		Err(e) => {
			eprintln!("broadcasting for channel finished abruptly: {}", e)
		}
	}
}