You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
81 lines
2.1 KiB
81 lines
2.1 KiB
use anyhow::{bail, Context as AnyhowContext};
|
|
use solstice_proto::server::{PeerAddressRequest, ServerRequest};
|
|
|
|
use crate::context::Context;
|
|
use crate::message_handler::MessageHandler;
|
|
use crate::peer::PeerState;
|
|
|
|
#[derive(Debug, Default)]
|
|
pub struct PeerConnectRequestHandler;
|
|
|
|
impl MessageHandler for PeerConnectRequestHandler {
|
|
type Message = String;
|
|
|
|
fn run(
|
|
self,
|
|
context: &mut Context,
|
|
user_name: &String,
|
|
) -> anyhow::Result<()> {
|
|
let peer = match context.state.peers.peers.get_mut(user_name) {
|
|
Some(peer) => peer,
|
|
None => {
|
|
// Address is not even known, we need to ask for it first.
|
|
context
|
|
.state
|
|
.peers
|
|
.peers
|
|
.insert(user_name.clone(), PeerState::FetchingAddress);
|
|
|
|
let request = ServerRequest::PeerAddressRequest(PeerAddressRequest {
|
|
user_name: user_name.clone(),
|
|
});
|
|
context
|
|
.server_request_tx
|
|
.blocking_send(request)
|
|
.context("sending peer address request")?;
|
|
|
|
return Ok(());
|
|
}
|
|
};
|
|
|
|
bail!("Unimplemented. Peer = {:?}", peer);
|
|
}
|
|
|
|
fn name() -> String {
|
|
"PeerConnectRequestHandler".to_string()
|
|
}
|
|
}
|
|
|
|
#[cfg(test)]
|
|
mod tests {
|
|
use solstice_proto::server::{PeerAddressRequest, ServerRequest};
|
|
|
|
use crate::context::ContextBundle;
|
|
use crate::message_handler::MessageHandler;
|
|
use crate::peer::PeerState;
|
|
|
|
use super::PeerConnectRequestHandler;
|
|
|
|
#[test]
|
|
fn run_unknown_peer() {
|
|
let mut bundle = ContextBundle::default();
|
|
|
|
PeerConnectRequestHandler::default()
|
|
.run(&mut bundle.context, &"aisha".to_string())
|
|
.unwrap();
|
|
|
|
match bundle.context.state.peers.peers.get("aisha") {
|
|
Some(PeerState::FetchingAddress) => (),
|
|
entry => panic!("Unexpected entry: {:?}", entry),
|
|
}
|
|
|
|
drop(bundle.context.server_request_tx);
|
|
match bundle.server_request_rx.blocking_recv() {
|
|
Some(ServerRequest::PeerAddressRequest(PeerAddressRequest {
|
|
user_name,
|
|
})) => assert_eq!(user_name, "aisha"),
|
|
Some(request) => panic!("Unexpected server request: {:?}", request),
|
|
None => panic!("Expected server request, got none"),
|
|
}
|
|
}
|
|
}
|