1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
|
//! Module to handle commands sent directly to the bot, as opposed to something
//! that scans the messages for Žižek-trigger-words.
/// The leading command the bot is looking for before any internal command
pub const COMMAND_PREFIX: &'static str = "/zizek ";
const ZIZEK_ADD_QUOTE_NOTICE: &'static str = "I think I'll add that to my repertoire.";
const ZIZEK_KNOWN_QUOTE_NOTICE: &'static str = "Hey, I know that one, that's a good one!";
const ZIZEK_EMPTY_NOTICE: &'static str =
"You know, it's quite funny to me. The moment you call I've ran out of material.";
const CMD_ADD: &'static str = "add";
use std::sync::{Arc, Mutex};
use matrix_bot_api::handlers::{extract_command, HandleResult, MessageHandler};
use matrix_bot_api::{ActiveBot, Message, MessageType};
use crate::zizek_db::ZizekDb;
/// Command handler struct that can be passed to the matrix bot as a handler
pub struct CommandHandler
{
zizek_db: Arc<Mutex<ZizekDb>>,
}
impl CommandHandler
{
/// Create a new command handler with access to the zizek quote database
pub fn new(zizek_db: Arc<Mutex<ZizekDb>>) -> Self { Self { zizek_db } }
fn handle_zizek(&self, bot: &ActiveBot, message: &Message)
{
let zizek_lock = self.zizek_db.lock().unwrap();
let quote = zizek_lock.random_quote().map_or(ZIZEK_EMPTY_NOTICE, |s| &s);
bot.send_message(quote, &message.room, MessageType::TextMessage);
}
fn handle_add_quote(&mut self, bot: &ActiveBot, message: &Message)
{
let mut zizek_lock = self.zizek_db.lock().unwrap();
let quote = &message.body[COMMAND_PREFIX.len() + CMD_ADD.len() + 1..];
if zizek_lock.add_quote(quote) {
bot.send_message(
ZIZEK_ADD_QUOTE_NOTICE,
&message.room,
MessageType::TextMessage,
);
}
else {
bot.send_message(
ZIZEK_KNOWN_QUOTE_NOTICE,
&message.room,
MessageType::TextMessage,
);
}
}
}
impl MessageHandler for CommandHandler
{
fn handle_message(&mut self, bot: &ActiveBot, message: &Message) -> HandleResult
{
if let Some(command) = extract_command(&message.body, COMMAND_PREFIX) {
match command {
"" => {
self.handle_zizek(bot, message);
HandleResult::StopHandling
},
CMD_ADD => {
self.handle_add_quote(bot, message);
HandleResult::StopHandling
},
_ => {
println!("Žižek cannot be handled.");
HandleResult::ContinueHandling
},
}
}
else {
HandleResult::ContinueHandling
}
}
}
|
