Garmelon/euphoxide
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Send Snapshot on Connected event

Browse source
This commit is contained in:
Joscha 2023-01-23 21:38:48 +01:00
parent cd4b65d9cc
commit 29b7d0ed7a
1 changed files with 12 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

18
src/bot/instance.rs
View file

@ -170,6 +170,15 @@ pub struct Snapshot {
pub state: State,
}
impl Snapshot {
fn from_conn(conn: &Conn) -> Self {
Self {
conn_tx: conn.tx().clone(),
state: conn.state().clone(),
}
}
}
// Most of the time, the largest variant (`Packet`) is sent. The size of this
// enum is not critical anyways since it's not constructed that often.
#[allow(clippy::large_enum_variant)]
@ -187,7 +196,7 @@ pub struct Snapshot {
#[derive(Debug)]
pub enum Event {
Connecting(InstanceConfig),
Connected(InstanceConfig, ConnTx),
Connected(InstanceConfig, Snapshot),
Packet(InstanceConfig, ParsedPacket, Snapshot),
Disconnected(InstanceConfig),
Stopped(InstanceConfig),
@ -388,7 +397,7 @@ impl Instance {
.map_err(RunError::CouldNotConnect)?;
Self::set_cookies(config, cookies);
on_event(Event::Connected(config.clone(), conn.tx().clone()));
on_event(Event::Connected(config.clone(), Snapshot::from_conn(&conn)));
let conn_tx = conn.tx().clone();
select! {
@ -404,10 +413,7 @@ impl Instance {
) -> Result<(), RunError> {
loop {
let packet = conn.recv().await.map_err(RunError::Conn)?;
let snapshot = Snapshot {
conn_tx: conn.tx().clone(),
state: conn.state().clone(),
};
let snapshot = Snapshot::from_conn(conn);
match &packet.content {
Ok(Data::SnapshotEvent(snapshot)) => {