Compare commits

...
2 Commits
Author SHA1 Message Date
eigenein 87beaf2076 🚑️ Fix for 64-bit player IDs 2024-01-29 20:05:45 +01:00
eigenein 581192b8c7 ✨ Add PlayerInfo::rank 2023-08-02 21:38:20 +02:00
7 changed files with 34 additions and 10 deletions

No files matched your search

+3
View File
@@ -93,6 +93,9 @@ pub struct PlayerInfo {
#[prost(message, required, tag = "7")] #[prost(message, required, tag = "7")]
pub avatar: Avatar, pub avatar: Avatar,
#[prost(uint32, optional, tag = "9")]
pub rank: Option<u32>,
} }
#[derive(Clone, Copy, Debug, PartialEq, Eq, Hash, PartialOrd, Ord, Enumeration, Serialize)] #[derive(Clone, Copy, Debug, PartialEq, Eq, Hash, PartialOrd, Ord, Enumeration, Serialize)]
+2 -4
View File
@@ -63,8 +63,7 @@ fn assert_magic<T: Into<u32> + PartialEq>(actual: T, expected: T) -> Result {
#[inline] #[inline]
fn read_length_delimited(reader: &mut impl Read) -> Result<Vec<u8>> { fn read_length_delimited(reader: &mut impl Read) -> Result<Vec<u8>> {
let length = reader.read_u8()? as usize; let length = reader.read_u8()? as usize;
let mut buffer = Vec::new(); let mut buffer = vec![0; length];
buffer.resize(length, 0);
reader.read_exact(&mut buffer)?; reader.read_exact(&mut buffer)?;
Ok(buffer) Ok(buffer)
} }
@@ -76,8 +75,7 @@ fn read_string(reader: &mut impl Read) -> Result<String> {
#[inline] #[inline]
fn read_pickled<T: DeserializeOwned>(reader: &mut impl Read, length: usize) -> Result<T> { fn read_pickled<T: DeserializeOwned>(reader: &mut impl Read, length: usize) -> Result<T> {
let mut buffer = Vec::new(); let mut buffer = vec![0; length];
buffer.resize(length, 0);
reader.read_exact(&mut buffer)?; reader.read_exact(&mut buffer)?;
Ok(serde_pickle::from_slice(&buffer, Default::default())?) Ok(serde_pickle::from_slice(&buffer, Default::default())?)
} }
+3 -3
View File
@@ -5,7 +5,7 @@ use serde::{Deserialize, Serialize};
#[derive(Debug, Serialize, Deserialize)] #[derive(Debug, Serialize, Deserialize)]
pub struct BasePlayerCreate { pub struct BasePlayerCreate {
#[serde(default, rename(deserialize = "playersBattleCategoriesIds"))] #[serde(default, rename(deserialize = "playersBattleCategoriesIds"))]
pub players_battle_categories_ids: Option<HashMap<u32, (u8, u32)>>, pub players_battle_categories_ids: Option<HashMap<u64, (u8, u32)>>,
#[serde(rename(deserialize = "battleLevel"))] #[serde(rename(deserialize = "battleLevel"))]
pub battle_level: u8, pub battle_level: u8,
@@ -34,10 +34,10 @@ pub struct BasePlayerCreate {
pub average_mm_ratings: Option<Vec<Option<f64>>>, pub average_mm_ratings: Option<Vec<Option<f64>>>,
#[serde(default, rename(deserialize = "playerWaitTimes"))] #[serde(default, rename(deserialize = "playerWaitTimes"))]
pub player_wait_times: Option<HashMap<u32, f64>>, pub player_wait_times: Option<HashMap<u64, f64>>,
#[serde(rename(deserialize = "accountDatabaseIds"))] #[serde(rename(deserialize = "accountDatabaseIds"))]
pub account_database_ids: Option<Vec<u32>>, pub account_database_ids: Option<Vec<u64>>,
#[serde(default, rename(deserialize = "turboBattlesStats"))] #[serde(default, rename(deserialize = "turboBattlesStats"))]
pub turbo_battles_statistics: Option<TurboBattlesStatistics>, pub turbo_battles_statistics: Option<TurboBattlesStatistics>,
+4 -3
View File
@@ -35,7 +35,9 @@ impl Packet {
/// ///
/// Parsed packet, or [`None`], when no packets are left. /// Parsed packet, or [`None`], when no packets are left.
pub fn from_reader(reader: &mut impl Read) -> Result<Option<Self>> { pub fn from_reader(reader: &mut impl Read) -> Result<Option<Self>> {
let Ok(length) = reader.read_u32::<LittleEndian>() else { return Ok(None) }; let Ok(length) = reader.read_u32::<LittleEndian>() else {
return Ok(None);
};
let type_ = reader.read_u32::<LittleEndian>()?; let type_ = reader.read_u32::<LittleEndian>()?;
let clock_secs = reader.read_f32::<LittleEndian>()?; let clock_secs = reader.read_f32::<LittleEndian>()?;
let raw_payload = Self::read_raw_payload(reader, length as usize)?; let raw_payload = Self::read_raw_payload(reader, length as usize)?;
@@ -57,8 +59,7 @@ impl Packet {
} }
fn read_raw_payload(reader: &mut impl Read, length: usize) -> Result<Vec<u8>> { fn read_raw_payload(reader: &mut impl Read, length: usize) -> Result<Vec<u8>> {
let mut buffer = Vec::new(); let mut buffer = vec![0; length];
buffer.resize(length, 0);
reader.read_exact(&mut buffer)?; reader.read_exact(&mut buffer)?;
Ok(buffer) Ok(buffer)
} }
+22
View File
@@ -139,3 +139,25 @@ fn room_type_ok() -> Result<()> {
assert_eq!(battle_results.room_type(), RoomType::MadGames); assert_eq!(battle_results.room_type(), RoomType::MadGames);
Ok(()) Ok(())
} }
/// Training room on Strv K with a rank.
#[test]
fn rank_ok() -> Result<()> {
let battle_results = Replay::open(File::open(
"replays/20230802_2133__helaas_pindakaas_S31_Strv_K_15193856522497923.wotbreplay",
)?)?
.read_battle_results()?;
assert_eq!(battle_results.players[0].info.rank, Some(12));
Ok(())
}
/// Training room on Regressor with no rank.
#[test]
fn no_rank_ok() -> Result<()> {
let battle_results = Replay::open(File::open(
"replays/20230802_2136__helaas_pindakaas_Oth38_50TP_Tyszkiewicza_S1_15188303129784363.wotbreplay",
)?)?
.read_battle_results()?;
assert_eq!(battle_results.players[0].info.rank, None);
Ok(())
}