poe_data_tools 1.0.0

A library for working with Path of Exile game data
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
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82

use std::fmt::Display;

use anyhow::{Result, anyhow};
use winnow::{
    Parser,
    binary::{le_u32, le_u64, length_repeat, length_take},
};

use super::types::*;
use crate::file_parsers::{bundle::parser::bundle, shared::winnow::WinnowParser};

fn bundle_info<'a>() -> impl WinnowParser<&'a [u8], BundleInfo> {
    winnow::trace!(
        "bundle_info",
        (
            length_take(le_u32).try_map(|bytes: &[_]| String::from_utf8(bytes.to_vec())),
            le_u32,
        )
            .map(|(name, uncompressed_size)| BundleInfo {
                name,
                uncompressed_size,
            })
    )
}

fn file_info<'a>() -> impl WinnowParser<&'a [u8], FileInfo> {
    winnow::trace!(
        "file_info",
        (le_u64, le_u32, le_u32, le_u32) //
            .map(|(hash, bundle_index, offset, size)| FileInfo {
                hash,
                bundle_index,
                offset,
                size,
            })
    )
}

fn path_rep<'a>() -> impl WinnowParser<&'a [u8], PathRep> {
    winnow::trace!(
        "path_rep",
        (le_u64, le_u32, le_u32, le_u32) //
            .map(|(hash, offset, size, recursive_size)| PathRep {
                hash,
                offset,
                size,
                recursive_size,
            })
    )
}

/// Wrapper for anyhow::Error that implements std Error so winnow can properly handle them
#[derive(Debug)]
pub struct AnyhowError(anyhow::Error);
impl std::error::Error for AnyhowError {}

impl Display for AnyhowError {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        write!(f, "{}", self.0)
    }
}

pub fn parse_bundle_index_bytes(contents: &[u8]) -> Result<BundleIndexFile> {
    let mut parser = (
        length_repeat(le_u32, bundle_info()),
        length_repeat(le_u32, file_info()),
        length_repeat(le_u32, path_rep()),
        bundle().try_map(|b| b.read_all().map_err(AnyhowError)),
    )
        .map(|(bundles, files, paths, path_rep_bundle)| BundleIndexFile {
            bundles,
            files,
            paths,
            path_rep_bundle,
        });

    let file = parser
        .parse(contents)
        .map_err(|e| anyhow!("Failed to parse file: {e:?}"))?;

    Ok(file)
}