90 lines
2 KiB
Rust
90 lines
2 KiB
Rust
use std::time::Duration;
|
|
|
|
use euphoxide::{
|
|
api::{Data, Message, Nick, Send},
|
|
client::conn::{ClientConn, ClientConnHandle},
|
|
};
|
|
|
|
async fn set_nick(conn: &ClientConnHandle) -> anyhow::Result<()> {
|
|
conn.send_only(Nick {
|
|
name: "examplebot".to_string(),
|
|
})
|
|
.await?;
|
|
|
|
Ok(())
|
|
}
|
|
|
|
async fn send_pong(conn: &ClientConnHandle, msg: Message) -> anyhow::Result<()> {
|
|
conn.send_only(Send {
|
|
content: "Pong!".to_string(),
|
|
parent: Some(msg.id),
|
|
})
|
|
.await?;
|
|
|
|
Ok(())
|
|
}
|
|
|
|
async fn send_pyramid(conn: &ClientConnHandle, msg: Message) -> anyhow::Result<()> {
|
|
let mut parent = msg.id;
|
|
|
|
for _ in 0..3 {
|
|
let first = conn
|
|
.send(Send {
|
|
content: "brick".to_string(),
|
|
parent: Some(parent),
|
|
})
|
|
.await?;
|
|
|
|
conn.send_only(Send {
|
|
content: "brick".to_string(),
|
|
parent: Some(parent),
|
|
})
|
|
.await?;
|
|
|
|
parent = first.await?.0.id;
|
|
tokio::time::sleep(Duration::from_secs(1)).await;
|
|
}
|
|
|
|
conn.send_only(Send {
|
|
content: "brick".to_string(),
|
|
parent: Some(parent),
|
|
})
|
|
.await?;
|
|
|
|
Ok(())
|
|
}
|
|
|
|
async fn on_data(conn: ClientConnHandle, data: Data) {
|
|
let result = match data {
|
|
Data::SnapshotEvent(_) => set_nick(&conn).await,
|
|
Data::SendEvent(event) if event.0.content == "!ping" => send_pong(&conn, event.0).await,
|
|
Data::SendEvent(event) if event.0.content == "!pyramid" => {
|
|
send_pyramid(&conn, event.0).await
|
|
}
|
|
_ => Ok(()),
|
|
};
|
|
|
|
if let Err(err) = result {
|
|
println!("Error while responding: {err}");
|
|
}
|
|
}
|
|
|
|
async fn run() -> anyhow::Result<()> {
|
|
let (mut conn, _) = ClientConn::connect("test", None).await?;
|
|
|
|
while let Some(packet) = conn.recv().await? {
|
|
let data = packet.into_data()?;
|
|
tokio::task::spawn(on_data(conn.handle(), data));
|
|
}
|
|
|
|
Ok(())
|
|
}
|
|
|
|
#[tokio::main]
|
|
async fn main() {
|
|
loop {
|
|
if let Err(err) = run().await {
|
|
println!("Error while running: {err}");
|
|
}
|
|
}
|
|
}
|