arnekeller/wynnrobot
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Fix cleverbot error handling

Browse Source
This commit is contained in:
Sakuhl 2018-03-07 14:53:31 +01:00
parent d6c59bb1d1
commit 4669d37a71
1 changed files with 20 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
src/main.rs
View File

@ -156,6 +156,18 @@ struct CBResponse {
output: String
}
fn cleverbot(msg: &Message) -> Result<String, Error> {
let _ = msg.channel_id.broadcast_typing();
let resp = reqwest::get(&format!("http://www.cleverbot.com/getreply?key=CC7wgO9gQl7yPjB54eDLHmM5Jmg&input={}&cb_settings_tweak1=0&cb_settings_tweak2=100&cb_settings_tweak3=100{}",
msg.content.replace("<@392763365409292298>", ""), (*CS).lock().clone().map(|x| format!("&cs={}", x)).unwrap_or_else(String::new)))?
.error_for_status()?
.text()?;
thread::sleep_ms(500);
let resp: CBResponse = serde_json::from_str(&resp)?;
msg.channel_id.say(format!("{} {}", msg.author.mention(), resp.output))?;
Ok(resp.cs)
}
impl Handler {
fn process_message(&self, msg: &Message, prefix: &str) -> Result<(), Error> {
if msg.content == "<@392763365409292298> has big gay" {
@ -171,16 +183,14 @@ impl Handler {
return Ok(());
}
if msg.content.contains("<@392763365409292298>") {
let _ = msg.channel_id.broadcast_typing();
let resp = reqwest::get(&format!("http://www.cleverbot.com/getreply?key=CC7wgO9gQl7yPjB54eDLHmM5Jmg&input={}&cb_settings_tweak1=0&cb_settings_tweak2=100&cb_settings_tweak3=100{}",
msg.content.replace("<@392763365409292298>", ""), (*CS).lock().clone().map(|x| format!("&cs={}", x)).unwrap_or_else(String::new)))?
.error_for_status()?
.text()?;
thread::sleep_ms(500);
let resp: CBResponse = serde_json::from_str(&resp)?;
*(*CS).lock() = Some(resp.cs);
msg.channel_id.say(format!("{} {}", msg.author.mention(), resp.output))?;
if let Ok(cs) = cleverbot(msg) {
*(*CS).lock() = Some(cs);
return Ok(());
} else {
*(*CS).lock() = None;
self.process_message(msg, prefix)?;
return Ok(());
}
}
if !msg.content.starts_with(&prefix) {
return Ok(());