Split replaytool into multiple files

1 files changed, 81 insertions, 0 deletions

diff --git a/server/replaytool/src/record.rs b/server/replaytool/src/record.rs
new file mode 100644
index 00000000..8c12fc94
--- /dev/null
+++ b/server/replaytool/src/record.rs
@@ -0,0 +1,81 @@
+/*
+    Hurry Curry! - a game about cooking
+    Copyright (C) 2025 Hurry Curry! Contributors
+
+    This program is free software: you can redistribute it and/or modify
+    it under the terms of the GNU Affero General Public License as published by
+    the Free Software Foundation, version 3 of the License only.
+
+    This program is distributed in the hope that it will be useful,
+    but WITHOUT ANY WARRANTY; without even the implied warranty of
+    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+    GNU Affero General Public License for more details.
+
+    You should have received a copy of the GNU Affero General Public License
+    along with this program.  If not, see <https://www.gnu.org/licenses/>.
+
+*/
+
+use anyhow::Context;
+use async_compression::tokio::write::ZstdEncoder;
+use futures_util::StreamExt;
+use hurrycurry_protocol::PacketC;
+use log::{debug, info};
+use std::{path::Path, time::Instant};
+use tokio::{
+    fs::File,
+    io::{AsyncWriteExt, BufWriter},
+};
+use tokio_tungstenite::tungstenite;
+
+use crate::Event;
+
+pub async fn record(output: &Path, url: &str) -> anyhow::Result<()> {
+    let mut file = BufWriter::new(ZstdEncoder::new(BufWriter::new(
+        File::create(&output).await?,
+    )));
+    info!("connecting to {url:?}...");
+    let (mut sock, _) = tokio_tungstenite::connect_async(url).await?;
+    info!("starting recording.");
+    let start = Instant::now();
+
+    while let Some(Ok(message)) = sock.next().await {
+        match message {
+            tungstenite::Message::Text(line) => {
+                let packet: PacketC = serde_json::from_str(&line).context("invalid packet")?;
+                debug!("<- {packet:?}");
+
+                let is_end = matches!(packet, PacketC::SetIngame { state: false, .. });
+
+                file.write_all(
+                    format!(
+                        "{}\n",
+                        serde_json::to_string(&Event {
+                            ts: start.elapsed().as_secs_f64(),
+                            packet
+                        })
+                        .unwrap()
+                    )
+                    .as_bytes(),
+                )
+                .await?;
+
+                if is_end {
+                    info!("stopping replay...");
+                    break;
+                }
+            }
+            tungstenite::Message::Close(_) => break,
+            _ => (),
+        }
+    }
+    file.flush().await?;
+    let mut file = file.into_inner();
+    file.flush().await?;
+    let mut file = file.into_inner();
+    file.flush().await?;
+    let mut file = file.into_inner();
+    file.flush().await?;
+    info!("done");
+    Ok(())
+}