Adds a message_type
field to MessageFromIRC
.
The `message_type` allows switching to `notice` type messages. It defaults to `privmsg` and any other (string) values will have the same effect than using `privmsg`. No code should need to be updated after this change.
This commit is contained in:
parent
29b5035fdf
commit
23081ad5d5
|
@ -30,7 +30,9 @@ use std::env;
|
|||
#[derive(Serialize, Deserialize, Debug)]
|
||||
struct MessageToIRC {
|
||||
target: String,
|
||||
body: String
|
||||
body: String,
|
||||
#[serde(default = "default_irc_message_type")]
|
||||
message_type: String
|
||||
}
|
||||
|
||||
#[derive(Serialize, Deserialize, Debug)]
|
||||
|
@ -40,6 +42,10 @@ struct MessageFromIRC {
|
|||
body: String
|
||||
}
|
||||
|
||||
fn default_irc_message_type() -> String {
|
||||
"privmsg".to_string()
|
||||
}
|
||||
|
||||
fn main() {
|
||||
if let Err(_) = env::var("RUST_LOG") {
|
||||
env::set_var("RUST_LOG", "info");
|
||||
|
@ -81,8 +87,13 @@ fn main() {
|
|||
move |_channel: &mut Channel, _deliver: Deliver, _headers: BasicProperties, body: Vec<u8>| {
|
||||
let msg: Result<MessageToIRC, serde_json::Error> = serde_json::from_slice(&body);
|
||||
if let Ok(msg) = msg {
|
||||
if &msg.message_type == "notice" {
|
||||
server.send_notice(&msg.target, &msg.body).unwrap();
|
||||
}
|
||||
else {
|
||||
server.send_privmsg(&msg.target, &msg.body).unwrap();
|
||||
}
|
||||
}
|
||||
},
|
||||
"queue-publish", "", false, true, false, false, Table::new());
|
||||
println!("Starting consumer {:?}", consumer_name);
|
||||
|
|
Loading…
Reference in a new issue