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

Send announcements not as part of scanning

Browse source
This commit is contained in:
Natsu Kagami 2020-09-14 22:33:23 -04:00
parent bc888e816b
commit 0fd5379446
No known key found for this signature in database
GPG key ID: F17543D4B9424B94
2 changed files with 67 additions and 50 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

54
youmubot-osu/src/discord/announcer.rs
View file

@ -40,10 +40,6 @@ impl youmubot_prelude::Announcer for Announcer {
let channels = &channels;
let c = c.clone();
async move {
let d = d.read().await;
let osu = d.get::<OsuClient>().unwrap();
let cache = d.get::<BeatmapMetaCache>().unwrap();
let oppai = d.get::<BeatmapCache>().unwrap();
let channels = channels.channels_of(c.clone(), user_id).await;
if channels.is_empty() {
return (user_id, osu_user); // We don't wanna update an user without any active server
@ -53,14 +49,11 @@ impl youmubot_prelude::Announcer for Announcer {
.map(|m| {
handle_user_mode(
c.clone(),
&osu,
&cache,
&oppai,
&osu_user,
user_id,
&channels[..],
channels.clone(),
*m,
&*d,
d.clone(),
)
})
.collect::<stream::FuturesOrdered<_>>()
@ -96,52 +89,73 @@ impl youmubot_prelude::Announcer for Announcer {
/// Handles an user/mode scan, announces all possible new scores, return the new pp value.
async fn handle_user_mode(
c: Arc<CacheAndHttp>,
osu: &Osu,
cache: &BeatmapMetaCache,
oppai: &BeatmapCache,
osu_user: &OsuUser,
user_id: UserId,
channels: &[ChannelId],
channels: Vec<ChannelId>,
mode: Mode,
d: &TypeMap,
d: AppData,
) -> Result<Option<f64>, Error> {
let (scores, user) = {
let data = d.read().await;
let osu = data.get::<OsuClient>().unwrap();
let scores = scan_user(osu, osu_user, mode).await?;
let user = osu
.user(UserID::ID(osu_user.id), |f| f.mode(mode))
.await?
.ok_or(Error::msg("user not found"))?;
(scores, user)
};
let pp = user.pp;
spawn_future(async move {
scores
.into_iter()
.map(|(rank, score)| async move {
.map(|(rank, score)| {
let d = d.clone();
async move {
let data = d.read().await;
let cache = data.get::<BeatmapMetaCache>().unwrap();
let oppai = data.get::<BeatmapCache>().unwrap();
let beatmap = cache.get_beatmap_default(score.beatmap_id).await?;
let content = oppai.get_beatmap(beatmap.beatmap_id).await?;
let r: Result<_> = Ok((rank, score, BeatmapWithMode(beatmap, mode), content));
r
}
})
.collect::<stream::FuturesOrdered<_>>()
.filter_map(|v| future::ready(v.ok()))
.for_each(|(rank, score, beatmap, content)| {
.for_each(move |(rank, score, beatmap, content)| {
let channels = channels.clone();
let d = d.clone();
let c = c.clone();
let user = &user;
let user = user.clone();
async move {
let data = d.read().await;
for channel in (&channels).iter() {
if let Err(e) = channel
.send_message(c.http(), |c| {
c.content(format!("New top record from {}!", user_id.mention()))
.embed(|e| {
score_embed(&score, &beatmap, &content, &user, Some(rank), e)
score_embed(
&score,
&beatmap,
&content,
&user,
Some(rank),
e,
)
})
})
.await
{
dbg!(e);
}
save_beatmap(d, *channel, &beatmap).ok();
save_beatmap(&*data, *channel, &beatmap).ok();
}
}
})
.await;
Ok(user.pp)
});
Ok(pp)
}
async fn scan_user(osu: &Osu, u: &OsuUser, mode: Mode) -> Result<Vec<(u8, Score)>, Error> {

3
youmubot-prelude/src/lib.rs
View file

@ -24,6 +24,9 @@ pub use anyhow::{Error, Result};
/// Re-export useful future and stream utils
pub use futures_util::{future, stream, FutureExt, StreamExt, TryFutureExt, TryStreamExt};
/// Re-export the spawn function
pub use tokio::spawn as spawn_future;
/// The global app data.
pub type AppData = Arc<RwLock<TypeMap>>;