|
|
|
@ -9,7 +9,8 @@ use crate::context::Context; |
|
|
|
use crate::control::Request as ControlRequest;
|
|
|
|
use crate::executor::Job;
|
|
|
|
use crate::handlers::{
|
|
|
|
RoomListRequestHandler, SetPrivilegedUsersHandler, SetRoomListHandler,
|
|
|
|
RoomListRequestHandler, RoomMessageRequestHandler, SetPrivilegedUsersHandler,
|
|
|
|
SetRoomListHandler,
|
|
|
|
};
|
|
|
|
use crate::message_handler::MessageHandler;
|
|
|
|
|
|
|
|
@ -26,6 +27,12 @@ impl From<ServerResponse> for Message { |
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
impl From<ControlRequest> for Message {
|
|
|
|
fn from(request: ControlRequest) -> Self {
|
|
|
|
Self::ControlRequest(request)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
/// Pairs together a message and its handler as chosen by the dispatcher.
|
|
|
|
/// Implements Job so as to be scheduled on an executor.
|
|
|
|
struct DispatchedMessage<M, H> {
|
|
|
|
@ -84,6 +91,12 @@ impl Dispatcher { |
|
|
|
handler: RoomListRequestHandler::default(),
|
|
|
|
}))
|
|
|
|
}
|
|
|
|
Message::ControlRequest(ControlRequest::RoomMessageRequest(request)) => {
|
|
|
|
Some(Box::new(DispatchedMessage {
|
|
|
|
message: request,
|
|
|
|
handler: RoomMessageRequestHandler::default(),
|
|
|
|
}))
|
|
|
|
}
|
|
|
|
Message::ControlRequest(request) => {
|
|
|
|
warn!("Unhandled control request: {:?}", request);
|
|
|
|
None
|
|
|
|
@ -101,10 +114,6 @@ mod tests { |
|
|
|
|
|
|
|
use super::*;
|
|
|
|
|
|
|
|
fn into_message<T: Into<ServerResponse>>(t: T) -> Message {
|
|
|
|
t.into().into()
|
|
|
|
}
|
|
|
|
|
|
|
|
#[test]
|
|
|
|
fn does_not_dispatch_unhandled_response() {
|
|
|
|
assert!(Dispatcher::new()
|
|
|
|
@ -144,6 +153,19 @@ mod tests { |
|
|
|
.is_some());
|
|
|
|
}
|
|
|
|
|
|
|
|
#[test]
|
|
|
|
fn dispatches_room_message_request() {
|
|
|
|
assert!(Dispatcher::new()
|
|
|
|
.dispatch(Message::ControlRequest(
|
|
|
|
control::RoomMessageRequest {
|
|
|
|
room_name: "bleep".to_string(),
|
|
|
|
message: "yo!".to_string(),
|
|
|
|
}
|
|
|
|
.into()
|
|
|
|
))
|
|
|
|
.is_some());
|
|
|
|
}
|
|
|
|
|
|
|
|
#[test]
|
|
|
|
fn dispatches_room_list_request() {
|
|
|
|
assert!(Dispatcher::new()
|
|
|
|
|