nki/youmubot
1
0
Fork 0
mirror of https://github.com/natsukagami/youmubot.git synced 2025-04-19 00:38:54 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Improve 1-page cases

Browse source
This commit is contained in:
Natsu Kagami 2021-06-19 22:49:49 +09:00
parent 4916b90abd
commit 8297472635
Signed by: nki
GPG key ID: 7306B3D3C3AD6E51
1 changed files with 6 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
youmubot-prelude/src/pagination.rs
View file

@ -102,6 +102,12 @@ async fn paginate_with_first_message(
mut message: Message, mut message: Message,
timeout: std::time::Duration, timeout: std::time::Duration,
) -> Result<()> { ) -> Result<()> {
pager.prerender(&ctx, &mut message).await?;
pager.render(0, ctx, &mut message).await?;
// Just quit if there is only one page
if pager.len().filter(|&v| v == 1).is_some() {
return Ok(());
}
// React to the message // React to the message
let large_count = pager.len().filter(|&p| p > 10).is_some(); let large_count = pager.len().filter(|&p| p > 10).is_some();
if large_count { if large_count {
@ -120,8 +126,6 @@ async fn paginate_with_first_message(
.react(&ctx, ReactionType::try_from(FAST_FORWARD)?) .react(&ctx, ReactionType::try_from(FAST_FORWARD)?)
.await?; .await?;
} }
pager.prerender(&ctx, &mut message).await?;
pager.render(0, ctx, &mut message).await?;
// Build a reaction collector // Build a reaction collector
let mut reaction_collector = message.await_reactions(&ctx).removed(true).await; let mut reaction_collector = message.await_reactions(&ctx).removed(true).await;
let mut page = 0; let mut page = 0;