wotbreplay-parser 0.3.0-alpha.2

World of Tanks Blitz replay parser
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47

use std::io::Read;

use prost::Message;
use serde::Deserialize;
use serde_bytes::ByteBuf;

use crate::error::Error;
use crate::models::battle_results::BattleResults;
use crate::result::Result;

/// Represents un-pickled `battle_results.dat`.
#[derive(Debug, Deserialize)]
#[serde(from = "(u64, ByteBuf)")]
pub struct BattleResultsDat {
    pub arena_unique_id: u64,

    /// Protobuf-serialized battle results.
    pub buffer: ByteBuf,
}

impl From<(u64, ByteBuf)> for BattleResultsDat {
    fn from((arena_unique_id, buffer): (u64, ByteBuf)) -> Self {
        Self {
            arena_unique_id,
            buffer,
        }
    }
}

impl BattleResultsDat {
    /// Parses the pickled structure from the reader which contains `battle_results.dat`.
    ///
    /// It can be used either on the `battle_results.dat` for a replay archive,
    /// or directly on `**/DAVAProject/battle_results/*/*_full.dat`.
    pub fn from_reader(reader: impl Read) -> Result<Self> {
        Ok(serde_pickle::from_reader(reader, Default::default())?)
    }
}

impl TryInto<BattleResults> for BattleResultsDat {
    type Error = Error;

    /// Decodes the battle results from the internal buffer.
    fn try_into(self) -> Result<BattleResults> {
        Ok(BattleResults::decode(self.buffer.as_ref())?)
    }
}