tiger_lib/imperator/data/
unit_abilities.rs

1use crate::block::Block;
2use crate::context::ScopeContext;
3use crate::db::{Db, DbKind};
4use crate::everything::Everything;
5use crate::game::GameFlags;
6use crate::imperator::modif::ModifKinds;
7use crate::item::{Item, ItemLoader};
8use crate::modif::validate_modifs;
9use crate::scopes::Scopes;
10use crate::token::Token;
11use crate::tooltipped::Tooltipped;
12use crate::validate::validate_modifiers_with_base;
13use crate::validator::Validator;
14
15#[derive(Clone, Debug)]
16pub struct UnitAbility {}
17
18inventory::submit! {
19    ItemLoader::Normal(GameFlags::Imperator, Item::UnitAbility, UnitAbility::add)
20}
21
22impl UnitAbility {
23    pub fn add(db: &mut Db, key: Token, block: Block) {
24        db.add(Item::UnitAbility, key, block, Box::new(Self {}));
25    }
26}
27
28impl DbKind for UnitAbility {
29    fn validate(&self, key: &Token, block: &Block, data: &Everything) {
30        let mut vd = Validator::new(block, data);
31        let mut sc = ScopeContext::new(Scopes::Unit, key);
32        sc.define_name("province", Scopes::Province, key);
33
34        vd.field_bool("enable");
35        vd.field_bool("toggle");
36        vd.field_bool("army_only");
37        vd.field_bool("navy_only");
38        vd.field_bool("map");
39        vd.field_bool("is_road_building");
40        vd.field_bool("cancel_on_combat_end");
41        vd.field_bool("is_desecrate");
42        vd.field_bool("confirm");
43
44        vd.field_numeric("duration");
45
46        vd.field_item("soundeffect", Item::Sound);
47
48        vd.field_choice("idle_entity_state", &["recruiting", "army_drill", "raiding"]);
49        vd.field_choice("move_entity_state", &["build_road", "force_march"]);
50
51        vd.field_list("available_states");
52
53        vd.field_validated_block("modifier", |block, data| {
54            let vd = Validator::new(block, data);
55            validate_modifs(block, data, ModifKinds::Country, vd);
56        });
57
58        vd.field_trigger("hidden", Tooltipped::No, &mut sc);
59        vd.field_trigger("allow", Tooltipped::Yes, &mut sc);
60        vd.field_trigger("ai_will_revoke", Tooltipped::Yes, &mut sc);
61        vd.field_trigger("finished_when", Tooltipped::Yes, &mut sc);
62        vd.field_effect("start_effect", Tooltipped::Yes, &mut sc);
63        vd.field_effect("finish_effect", Tooltipped::Yes, &mut sc);
64        vd.field_effect("on_entering_province", Tooltipped::Yes, &mut sc);
65        vd.field_validated_block_sc("ai_will_do", &mut sc, validate_modifiers_with_base);
66    }
67}