nki/youmubot
1
0
Fork 0
mirror of https://github.com/natsukagami/youmubot.git synced 2025-05-24 17:20:49 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Store osu! user username

Browse source
This commit is contained in:
Natsu Kagami 2024-02-04 23:24:28 +01:00
parent 85954639d7
commit f7ea23ed3b
Signed by: nki
GPG key ID: 55A032EB38B49ADB
4 changed files with 34 additions and 11 deletions
Download patch file Download diff file Expand all files Collapse all files

24
youmubot-osu/src/discord/mod.rs
View file

@ -2,7 +2,7 @@ use crate::{
discord::beatmap_cache::BeatmapMetaCache,
discord::display::ScoreListStyle,
discord::oppai_cache::{BeatmapCache, BeatmapInfo},
models::{Beatmap, Mode, Mods, User},
models::{self, Beatmap, Mode, Mods, User},
request::{BeatmapRequestKind, UserID},
Client as OsuHttpClient,
};
@ -210,12 +210,13 @@ pub async fn save(ctx: &Context, msg: &Message, mut args: Args) -> CommandResult
.await?;
return Ok(());
}
add_user(msg.author.id, u.id, &data).await?;
let username = u.username.clone();
add_user(msg.author.id, u, &data).await?;
msg.reply(
&ctx,
MessageBuilder::new()
.push("user has been set to ")
.push_mono_safe(u.username)
.push_mono_safe(username)
.build(),
)
.await?;
@ -238,16 +239,16 @@ pub async fn forcesave(ctx: &Context, msg: &Message, mut args: Args) -> CommandR
let osu = data.get::<OsuClient>().unwrap();
let target = args.single::<serenity::model::id::UserId>()?;
let user = args.quoted().trimmed().single::<String>()?;
let user: Option<User> = osu.user(UserID::Auto(user), |f| f).await?;
let username = args.quoted().trimmed().single::<String>()?;
let user: Option<User> = osu.user(UserID::Auto(username.clone()), |f| f).await?;
match user {
Some(u) => {
add_user(target, u.id, &data).await?;
add_user(target, u, &data).await?;
msg.reply(
&ctx,
MessageBuilder::new()
.push("user has been set to ")
.push_mono_safe(u.username)
.push_mono_safe(username)
.build(),
)
.await?;
@ -259,10 +260,15 @@ pub async fn forcesave(ctx: &Context, msg: &Message, mut args: Args) -> CommandR
Ok(())
}
async fn add_user(target: serenity::model::id::UserId, user_id: u64, data: &TypeMap) -> Result<()> {
async fn add_user(
target: serenity::model::id::UserId,
user: models::User,
data: &TypeMap,
) -> Result<()> {
let u = OsuUser {
user_id: target,
id: user_id,
username: user.username.into(),
id: user.id,
failures: 0,
last_update: chrono::Utc::now(),
pp: [None, None, None, None],