1use crate::block::Block;
2use crate::context::ScopeContext;
3use crate::db::{Db, DbKind};
4use crate::everything::Everything;
5use crate::game::GameFlags;
6use crate::item::{Item, ItemLoader};
7use crate::modif::validate_modifs;
8use crate::scopes::Scopes;
9use crate::token::Token;
10use crate::tooltipped::Tooltipped;
11use crate::validate::validate_duration;
12use crate::validator::Validator;
13use crate::vic3::modif::ModifKinds;
14use crate::vic3::tables::misc::COMPANY_CATEGORIES;
15
16#[derive(Clone, Debug)]
17pub struct CompanyType {}
18#[derive(Clone, Debug)]
19pub struct DynamicCompanyName {}
20#[derive(Clone, Debug)]
21pub struct CompanyCharterType {}
22
23inventory::submit! {
24 ItemLoader::Normal(GameFlags::Vic3, Item::CompanyType, CompanyType::add)
25}
26inventory::submit! {
27 ItemLoader::Normal(GameFlags::Vic3, Item::DynamicCompanyName, DynamicCompanyName::add)
28}
29inventory::submit! {
30 ItemLoader::Normal(GameFlags::Vic3, Item::CompanyCharterType, CompanyCharterType::add)
31}
32
33impl CompanyType {
34 pub fn add(db: &mut Db, key: Token, block: Block) {
35 db.add(Item::CompanyType, key, block, Box::new(Self {}));
36 }
37}
38
39impl DynamicCompanyName {
40 pub fn add(db: &mut Db, key: Token, block: Block) {
41 db.add(Item::DynamicCompanyName, key, block, Box::new(Self {}));
42 }
43}
44
45impl CompanyCharterType {
46 pub fn add(db: &mut Db, key: Token, block: Block) {
47 db.add(Item::CompanyCharterType, key, block, Box::new(Self {}));
48 }
49}
50
51impl DbKind for CompanyType {
52 fn validate(&self, key: &Token, block: &Block, data: &Everything) {
53 let mut vd = Validator::new(block, data);
54
55 data.verify_exists(Item::Localization, key);
56
57 vd.field_item("icon", Item::File);
58 vd.field_item("background", Item::File);
59
60 vd.field_bool("flavored_company");
61 vd.field_bool("uses_dynamic_naming");
62 vd.field_list_items("preferred_headquarters", Item::StateRegion);
63
64 vd.field_item("replaces_company", Item::CompanyType);
65 vd.field_list_items("possible_prestige_goods", Item::PrestigeGoods);
66 vd.field_trigger_rooted("prestige_goods_trigger", Tooltipped::No, Scopes::Company);
67
68 if block.field_value_is("uses_dynamic_naming", "yes") {
69 vd.field_list_items("dynamic_company_type_names", Item::Localization);
70 } else {
71 vd.ban_field("dynamic_company_type_names", || "uses_dynamic_naming = yes");
72 }
73
74 vd.multi_field_list_items("building_types", Item::BuildingType);
75 vd.multi_field_list_items("extension_building_types", Item::BuildingType);
76
77 vd.field_trigger_rooted("potential", Tooltipped::No, Scopes::Country);
78 vd.field_trigger_rooted("attainable", Tooltipped::Yes, Scopes::Country);
79 vd.field_trigger_rooted("possible", Tooltipped::Yes, Scopes::Country);
80
81 vd.multi_warn_field_validated_block("prosperity_modifier", |block, data| {
82 let vd = Validator::new(block, data);
83 validate_modifs(block, data, ModifKinds::Country, vd);
84 });
85
86 vd.field_trigger_rooted("ai_will_do", Tooltipped::No, Scopes::Country);
87 vd.field_validated_block("ai_construction_targets", |block, data| {
88 let mut vd = Validator::new(block, data);
89 vd.validate_item_key_blocks(Item::BuildingType, |_, block, data| {
90 let mut vd = Validator::new(block, data);
91 vd.field_integer("level");
92 vd.field_trigger_rooted("state_trigger", Tooltipped::No, Scopes::State);
93 });
94 });
95 vd.field_script_value_rooted("ai_weight", Scopes::Country);
96
97 vd.field_choice("category", COMPANY_CATEGORIES);
100 vd.field_list_items("unlocking_principles", Item::Principle);
101 }
102}
103
104impl DbKind for DynamicCompanyName {
105 fn validate(&self, key: &Token, block: &Block, data: &Everything) {
106 let mut vd = Validator::new(block, data);
107
108 data.verify_exists(Item::Localization, key);
109
110 vd.field_bool("uses_plural_naming");
111 vd.field_bool("use_for_flavored_companies");
112 vd.field_script_value_rooted("weight", Scopes::Country);
113 }
114}
115
116impl DbKind for CompanyCharterType {
117 fn validate(&self, key: &Token, block: &Block, data: &Everything) {
118 let mut vd = Validator::new(block, data);
119
120 data.verify_exists(Item::Localization, key);
121 let loca = format!("{key}_desc");
122 data.verify_exists_implied(Item::Localization, &loca, key);
123 let loca = format!("{key}_type_desc");
124 data.verify_exists_implied(Item::Localization, &loca, key);
125 let loca = format!("{key}_effects");
126 data.verify_exists_implied(Item::Localization, &loca, key);
127
128 vd.field_choice(
130 "type",
131 &["industry", "investment", "monopoly", "new_industry", "trade", "colonization"],
132 );
133 vd.field_item("icon", Item::File);
134 let mut sc = ScopeContext::new(Scopes::Company | Scopes::Country, key);
136 vd.field_validated_block_sc("cooldown", &mut sc, validate_duration);
137
138 vd.field_trigger_rooted("possible", Tooltipped::No, Scopes::Company);
139 vd.field_trigger_rooted("ai_possible", Tooltipped::No, Scopes::Company);
140 vd.field_script_value_no_breakdown_rooted("ai_weight", Scopes::Company);
141
142 vd.field_bool("additional_input");
145 }
146}