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

bistrofix

Browse Source
This commit is contained in:
Mikhail Kharchenko
2020-12-11 16:10:41 +03:00
parent ad5ae11f0d f2c8311704
commit 3890ba0c62
2 changed files with 6 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/commands.rs
View File

@ -82,7 +82,7 @@ pub(crate) async fn markov_all(api: Api, message: Message) -> Result<(), Error>
} }
pub(crate) async fn markov(api: Api, message: Message) -> Result<(), Error> { pub(crate) async fn markov(api: Api, message: Message) -> Result<(), Error> {
let messages = db::get_random_messages_group().await?; let messages = db::get_random_messages_group(&message).await?;
let mut chain = Chain::new(); let mut chain = Chain::new();
chain.feed(messages); chain.feed(messages);
let mut sentences = chain.generate(); let mut sentences = chain.generate();

7
src/db.rs
View File

@ -115,7 +115,10 @@ pub(crate) async fn get_random_messages() -> Result<Vec<String>, Error> {
Ok(messages) Ok(messages)
} }
pub(crate) async fn get_random_messages_group() -> Result<Vec<String>, Error> { pub(crate) async fn get_random_messages_group(
message: &telegram_bot::Message
) -> Result<Vec<String>, Error> {
let conf_id = i64::from(message.chat.id());
let conn = open()?; let conn = open()?;
let mut stmt = conn.prepare_cached(" let mut stmt = conn.prepare_cached("
SELECT m.text FROM messages m SELECT m.text FROM messages m
@ -123,7 +126,7 @@ pub(crate) async fn get_random_messages_group() -> Result<Vec<String>, Error> {
ORDER BY RANDOM() LIMIT 50 ORDER BY RANDOM() LIMIT 50
" "
)?; )?;
let mut rows = stmt.query_named(named_params![])?; let mut rows = stmt.query_named(named_params! {":conf_id": conf_id})?;
let mut messages = Vec::new(); let mut messages = Vec::new();
while let Some(row) = rows.next()? { while let Some(row) = rows.next()? {