feat: support dxratingnet format

This commit is contained in:
mokurin000
2025-08-01 18:16:16 +08:00
parent 953feee4c4
commit 4e07eaf2e0
6 changed files with 201 additions and 6 deletions

View File

@@ -43,6 +43,13 @@ pub enum Commands {
Rating {
#[arg(short, long)]
user_id: u32,
/// JSON format.
///
/// - `origin`: official json response
/// - `dx_rating_net`: DxRatingNet Format
#[arg(short, long, default_value_t = RatingFormat::default())]
format: RatingFormat,
},
// below requires login
@@ -71,3 +78,14 @@ pub enum Commands {
timestamp: u64,
},
}
#[derive(Default, EnumString)]
#[strum(serialize_all = "snake_case")]
pub enum RatingFormat {
#[default]
/// Official API response
Origin,
/// dxrating.net format
DxRatingNet,
}

View File

@@ -15,16 +15,16 @@ use sdgb_api::{
MaiVersionExt, Sdgb1_40, Sdgb1_50,
methods::APIMethod,
model::{
GetUserDataApi, GetUserDataApiResp, GetUserPreviewApi, GetUserPreviewApiResp,
GetUserRatingApi, GetUserRatingApiResp, Ping, PingResp, UserLogoutApi,
UserLogoutApiResp,
DxCalculatedEntries, GetUserDataApi, GetUserDataApiResp, GetUserPreviewApi,
GetUserPreviewApiResp, GetUserRatingApi, GetUserRatingApiResp, Ping, PingResp,
UserLogoutApi, UserLogoutApiResp,
},
},
};
use spdlog::{error, info, warn};
use crate::{
commands::Cli,
commands::{Cli, RatingFormat},
utils::{human_readable_display, json_display, login_action},
};
@@ -77,7 +77,7 @@ async fn main() -> Result<(), Box<dyn snafu::Error>> {
// TODO: refactor via enum_dispatch
match command {
commands::Commands::Rating { user_id } => {
commands::Commands::Rating { user_id, format } => {
let rating: GetUserRatingApiResp = Sdgb1_50::request(
&client,
APIMethod::GetUserRatingApi,
@@ -86,7 +86,13 @@ async fn main() -> Result<(), Box<dyn snafu::Error>> {
)
.await?;
human_readable_display(rating, human_readable)?;
match (human_readable, format) {
(true, _) => println!("{rating}"),
(false, RatingFormat::Origin) => json_display(rating)?,
(false, RatingFormat::DxRatingNet) => json_display(
DxCalculatedEntries::from_user_rating_lossy(&rating.user_rating),
)?,
}
}
commands::Commands::Logout { user_id, timestamp } => {
let logout: UserLogoutApiResp = Sdgb1_50::request(