Hexor/desubot
1
0
Fork 0
mirror of https://github.com/house-of-vanity/desubot.git synced 2025-07-08 13:04:06 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Add markov chain sentence generator.

Browse Source
This commit is contained in:
AB
2020-12-08 12:31:21 +03:00
parent d457e8af58
commit 2a8dcc8ffb
5 changed files with 142 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

33
src/commands.rs
View File

@ -1,8 +1,10 @@
use crate::db;
use crate::errors::Error;
use html_escape::encode_text;
use markov::Chain;
use telegram_bot::prelude::*;
use telegram_bot::{Api, Message, ParseMode};
use crate::errors::Error;
use rand::Rng;
pub(crate) async fn here(api: Api, message: Message) -> Result<(), Error> {
let members: Vec<telegram_bot::User> = db::get_members(message.chat.id()).unwrap();
@ -39,7 +41,10 @@ pub(crate) async fn top(api: Api, message: Message) -> Result<(), Error> {
let mut msg = "<b>Your top using words:</b>\n<pre>".to_string();
let mut counter = 1;
for word in top.iter() {
msg = format!("{} <b>{}</b> {} - {}\n", msg, counter, word.word, word.count);
msg = format!(
"{} <b>{}</b> {} - {}\n",
msg, counter, word.word, word.count
);
counter += 1;
}
msg = format!("{}{}", msg, "</pre>");
@ -52,4 +57,26 @@ pub(crate) async fn top(api: Api, message: Message) -> Result<(), Error> {
}
//api.send(message.chat.text("Text to message chat")).await?;
//api.send(message.from.text("Private text")).await?;
Ok(())}
Ok(())
}
pub(crate) async fn markov_all(api: Api, message: Message) -> Result<(), Error> {
let messages = db::get_random_messages().await?;
let mut chain = Chain::new();
chain.feed(messages);
let mut sentences = chain.generate();
let mut msg = String::new();
for _ in 1..rand::thread_rng().gen_range(2, 10) {
msg = format!("{} {}", msg, sentences.pop().unwrap());
}
match api
.send(message.text_reply(msg.trim()).parse_mode(ParseMode::Html))
.await
{
Ok(_) => debug!("/top command sent to {}", message.chat.id()),
Err(_) => warn!("/top command sent failed to {}", message.chat.id()),
}
//api.send(message.chat.text("Text to message chat")).await?;
//api.send(message.from.text("Private text")).await?;
Ok(())
}