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
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148

use anyhow::anyhow;
use winnow::{
    Parser,
    ascii::space1,
    combinator::{alt, delimited, opt, preceded as P, repeat},
    stream::{Offset, Stream},
    token::{any, literal},
};

use super::types::*;
use crate::file_parsers::{
    VersionedResult, VersionedResultExt,
    shared::winnow::{
        WinnowParser, quoted_str, single_quoted_str, spaces_or_comments, unquoted_str, version_line,
    },
};

fn entry<'a>() -> impl WinnowParser<&'a str, Entry> {
    winnow::trace!(
        "entry",
        (
            unquoted_str,
            (spaces_or_comments(), literal("="), spaces_or_comments()),
            alt((
                quoted_str,
                single_quoted_str,
                stack_take('{', '}').map(String::from),
                unquoted_str,
            )),
        )
            .map(|(key, _, value)| Entry { key, value })
    )
}

fn parse_struct<'a>() -> impl WinnowParser<&'a str, Struct> {
    winnow::trace!(
        "struct",
        (
            unquoted_str.verify(|s: &str| s != "client"),
            delimited(
                P(opt(spaces_or_comments()), literal("{")),
                repeat(0.., P(spaces_or_comments(), entry())),
                P(opt(spaces_or_comments()), literal("}")),
            ),
        )
            .map(|(name, entries)| Struct { name, entries })
    )
}

fn client_structs<'a>() -> impl WinnowParser<&'a str, Vec<Struct>> {
    winnow::trace!(
        "struct",
        P(
            "client",
            delimited(
                P(opt(spaces_or_comments()), literal("{")),
                repeat(0.., P(spaces_or_comments(), parse_struct())),
                P(opt(spaces_or_comments()), literal("}")),
            ),
        )
    )
}

fn stack_take<'a>(incrementer: char, decrementer: char) -> impl WinnowParser<&'a str, &'a str> {
    move |input: &mut &'a str| -> winnow::Result<&'a str> {
        let checkpoint = input.checkpoint();

        // First char must always increment
        literal(incrementer).parse_next(input)?;
        let mut counter = 1;
        while counter > 0 {
            counter += alt((
                incrementer.value(1), //
                decrementer.value(-1),
                any.value(0),
            ))
            .parse_next(input)?;
        }

        // Do some shenanigans to take first slice of input
        let mut start = "";
        start.reset(&checkpoint);

        let offset = input.offset_from(&start);
        let out = &start[..offset];

        Ok(out)
    }
}

enum StructKind {
    Base(Struct),
    Client(Vec<Struct>),
}

pub fn parse_ao_str(mut contents: &str) -> VersionedResult<AOFile> {
    let version = version_line()
        .parse_next(&mut contents)
        .map_err(|e| anyhow!("Failed to parse ao file: {e:?}"))?;

    let parser = (
        opt(P(spaces_or_comments(), literal("abstract"))),
        repeat::<_, _, Vec<_>, _, _>(
            1..,
            P(
                (spaces_or_comments(), literal("extends"), space1),
                quoted_str,
            ),
        ),
        repeat(
            0..,
            P(
                spaces_or_comments(),
                alt((
                    parse_struct().map(StructKind::Base),
                    client_structs().map(StructKind::Client),
                )),
            ),
        )
        .fold(
            || (vec![], vec![]),
            |(mut structs, mut client_structs), s| {
                match s {
                    StructKind::Base(s) => structs.push(s),
                    StructKind::Client(s) => client_structs.extend(s),
                };
                (structs, client_structs)
            },
        ),
        opt(spaces_or_comments()),
    )
        .map(
            |(is_abstract, extends, (structs, client_structs), _)| AOFile {
                version,
                is_abstract: is_abstract.is_some(),
                extends: extends.into_iter().filter(|e| e == "nothing").collect(),
                structs,
                client_structs,
            },
        );

    let mut parser = winnow::trace!("ao_file", parser);

    parser
        .parse(contents)
        .map_err(|e| anyhow!("Failed to parse ao file: {e:?}"))
        .with_version(Some(version))
}