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

Reformat uploaded beatmap embed and send background

Browse source
This commit is contained in:
Natsu Kagami 2024-06-20 01:52:39 +02:00 • committed by Natsu Kagami
parent 2315f40cf1
commit e9a6e55569
2 changed files with 69 additions and 31 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

19
youmubot-osu/src/discord/embeds.rs
View file

@ -4,6 +4,7 @@ use crate::{
models::{Beatmap, Difficulty, Mode, Mods, Rank, Score, User},
};
use serenity::{
all::CreateAttachment,
builder::{CreateEmbed, CreateEmbedAuthor, CreateEmbedFooter},
utils::MessageBuilder,
};
@ -63,7 +64,7 @@ pub fn beatmap_offline_embed(
b: &'_ crate::discord::oppai_cache::BeatmapContent,
m: Mode,
mods: Mods,
) -> Result<CreateEmbed> {
) -> Result<(CreateEmbed, Vec<CreateAttachment>)> {
let bm = b.content.clone();
let metadata = b.metadata.clone();
let (info, pp) = b.get_possible_pp_with(m, mods)?;
@ -107,8 +108,7 @@ pub fn beatmap_offline_embed(
total_length,
}
.apply_mods(mods, info.stars);
Ok(
CreateEmbed::new()
let mut embed = CreateEmbed::new()
.title(beatmap_title(
&metadata.artist,
&metadata.title,
@ -128,8 +128,17 @@ pub fn beatmap_offline_embed(
),
false,
)
.field("Information", diff.format_info(m, mods, None), false), // .description(beatmap_description(b))
)
.field("Information", diff.format_info(m, mods, None), false);
let mut attachments = Vec::new();
if let Some(bg) = &b.beatmap_background {
embed = embed.thumbnail(format!("attachment://{}", bg.filename));
attachments.push(CreateAttachment::bytes(
bg.content.clone().into_vec(),
bg.filename.clone(),
));
}
Ok((embed, attachments))
}
// Some helper functions here

35
youmubot-osu/src/discord/hook.rs
View file

@ -101,16 +101,45 @@ pub fn dot_osu_hook<'a>(
osu_embeds.extend(osz_embeds);
if !osu_embeds.is_empty() {
let embed_len = osu_embeds.len();
if embed_len == 1 {
let (embed, attachments) = osu_embeds.into_iter().next().unwrap();
msg.channel_id
.send_message(
ctx,
CreateMessage::new()
.reference_message(msg)
.content(format!("{} attached beatmaps found", osu_embeds.len()))
.add_embeds(osu_embeds),
.embed(embed)
.add_files(attachments)
.content("Attached beatmap".to_owned()),
)
.await
.ok();
.pls_ok();
} else {
let osu_embeds = Arc::new(osu_embeds);
paginate_reply(
paginate_from_fn(|page, ctx, msg| {
let osu_embeds = osu_embeds.clone();
Box::pin(async move {
let (embed, attachments) = &osu_embeds[page as usize];
let mut edit = EditMessage::new()
.content(format!("Attached beatmaps ({}/{})", page + 1, embed_len))
.embed(embed.clone());
for att in attachments {
edit = edit.new_attachment(att.clone());
}
msg.edit(&ctx, edit).await?;
Ok(true)
})
})
.with_page_count(embed_len),
ctx,
msg,
std::time::Duration::from_secs(180),
)
.await
.pls_ok();
}
}
Ok(())