aoe2-probe 0.3.2

A rust library for editing aoe2scenario files from AoE2 DE.
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

use std::sync::Arc;

use crate::{
    parse::Token,
    utils::{map::*, DynString},
};

use super::AI2;

pub struct Files;

impl Files {
    pub fn template() -> Token {
        let mut root = PatchedMap::with_capacity(7);
        root.push_back("script_file_path", DynString::with_capacity(0_u16, ""));
        root.push_back("script_file_content", DynString::with_capacity(0_u32, ""));
        root.push_back("ai_files_present", 0_u32);
        root.push_back("unknown", vec![0_u8.into(); 4]);
        root.push_back("number_of_ai_files", vec![0_u32.into()]);

        root.patches.insert(
            "number_of_ai_files".to_string(),
            Arc::new(|map: &mut PatchedMap, template: &mut Token| {
                if map.contains("ai_files_present") {
                    let count = *map["ai_files_present"].try_u32();
                    let unit = template.try_vec()[0].clone();
                    let vec = template.try_mut_vec();
                    vec.clear();

                    for _ in 0..count {
                        vec.push(unit.clone());
                    }
                }
            }),
        );

        root.push_back("ai_files", vec![AI2::template(); 1]);

        root.patches.insert(
            "ai_files".to_string(),
            Arc::new(|map: &mut PatchedMap, template: &mut Token| {
                if map.contains("number_of_ai_files") {
                    if map["number_of_ai_files"].try_vec().len() > 0 {
                        let count = *map["number_of_ai_files"].try_vec()[0].try_u32();
                        let unit = template.try_vec()[0].clone();
                        let vec = template.try_mut_vec();
                        vec.clear();

                        for _ in 0..count {
                            vec.push(unit.clone());
                        }
                    } else {
                        let vec = template.try_mut_vec();
                        vec.clear();
                    }
                }
            }),
        );

        root.into()
    }
}