tiger_lib/
everything.rs

1//! Stores everything known about the game and mod being validated.
2//!
3//! References to [`Everything`] are passed down through nearly all of the validation logic, so
4//! that individual functions can access all the defined game items.
5
6use std::borrow::Cow;
7use std::fmt::Debug;
8use std::path::{Path, PathBuf};
9#[cfg(any(feature = "ck3", feature = "vic3"))]
10use std::sync::RwLock;
11
12use anyhow::Result;
13use rayon::{scope, Scope};
14use strum::IntoEnumIterator;
15use thiserror::Error;
16
17use crate::block::Block;
18#[cfg(any(feature = "ck3", feature = "vic3"))]
19use crate::block::BV;
20#[cfg(feature = "ck3")]
21use crate::ck3::data::{
22    characters::Characters,
23    climate::Climate,
24    doctrines::Doctrines,
25    gameconcepts::GameConcepts,
26    interaction_cats::CharacterInteractionCategories,
27    maa::MenAtArmsTypes,
28    prov_history::ProvinceHistories,
29    prov_terrain::{ProvinceProperties, ProvinceTerrains},
30    provinces::Ck3Provinces,
31    title_history::TitleHistories,
32    titles::Titles,
33    traits::Traits,
34    wars::Wars,
35};
36#[cfg(feature = "ck3")]
37use crate::ck3::tables::misc::*;
38use crate::config_load::{check_for_legacy_ignore, load_filter};
39use crate::context::ScopeContext;
40#[cfg(any(feature = "ck3", feature = "vic3"))]
41use crate::data::data_binding::DataBindings;
42use crate::data::{
43    assets::Assets,
44    defines::Defines,
45    gui::Gui,
46    localization::Localization,
47    on_actions::OnActions,
48    scripted_effects::{Effect, Effects},
49    scripted_triggers::{Trigger, Triggers},
50};
51#[cfg(feature = "jomini")]
52use crate::data::{
53    coa::Coas, events::Events, music::Musics, script_values::ScriptValues,
54    scripted_lists::ScriptedLists, scripted_modifiers::ScriptedModifiers,
55};
56use crate::db::{Db, DbKind};
57use crate::dds::DdsFiles;
58use crate::fileset::{FileEntry, FileKind, Fileset};
59use crate::game::Game;
60#[cfg(any(feature = "ck3", feature = "vic3"))]
61use crate::helpers::TigerHashSet;
62#[cfg(feature = "hoi4")]
63use crate::hoi4::data::{
64    events::Hoi4Events, gfx::Gfx, music::Hoi4Musics, provinces::Hoi4Provinces,
65};
66#[cfg(feature = "hoi4")]
67use crate::hoi4::tables::misc::*;
68#[cfg(feature = "imperator")]
69use crate::imperator::data::{decisions::Decisions, provinces::ImperatorProvinces};
70#[cfg(feature = "imperator")]
71use crate::imperator::tables::misc::*;
72use crate::item::{Item, ItemLoader};
73use crate::lowercase::Lowercase;
74use crate::macros::MACRO_MAP;
75#[cfg(feature = "vic3")]
76use crate::parse::json::parse_json_file;
77use crate::parse::ParserMemory;
78use crate::pdxfile::PdxFile;
79#[cfg(any(feature = "ck3", feature = "vic3"))]
80use crate::report::err;
81use crate::report::{report, set_output_style, ErrorKey, OutputStyle, Severity};
82use crate::rivers::Rivers;
83use crate::token::{Loc, Token};
84use crate::variables::Variables;
85#[cfg(feature = "vic3")]
86use crate::vic3::data::{
87    buy_packages::BuyPackage, history::History, provinces::Vic3Provinces,
88    strategic_regions::StrategicRegion, terrain::TerrainMask,
89};
90#[cfg(feature = "vic3")]
91use crate::vic3::tables::misc::*;
92
93#[derive(Debug, Error)]
94#[allow(clippy::enum_variant_names)]
95pub enum FilesError {
96    #[error("Could not read game files at {path}")]
97    VanillaUnreadable { path: PathBuf, source: walkdir::Error },
98    #[error("Could not read mod files at {path}")]
99    ModUnreadable { path: PathBuf, source: walkdir::Error },
100    #[error("Could not read config file at {path}")]
101    ConfigUnreadable { path: PathBuf },
102}
103
104/// A record of everything known about the game and mod being validated.
105///
106/// References to [`Everything`] are passed down through nearly all of the validation logic, so
107/// that individual functions can access all the defined game items.
108///
109/// The validator has two main phases: parsing and validation.
110/// * During parsing, the script files are read, parsed, and loaded into the various databases.
111///   `Everything` is mutable during this period.
112/// * During validation, `Everything` is immutable and cross-checking between item types can be done safely.
113#[derive(Debug)]
114pub struct Everything {
115    /// Config from file
116    config: Block,
117
118    /// The global parser state, carrying information between files.
119    /// Currently only used by the pdxfile parser, to handle the `reader_export` directory,
120    /// which is specially processed before all other files.
121    pub parser: ParserMemory,
122
123    /// A cache of define values (from common/defines) that are missing and that have already been
124    /// warned about as missing. This is to avoid duplicate warnings.
125    #[cfg(any(feature = "ck3", feature = "vic3"))]
126    warned_defines: RwLock<TigerHashSet<String>>,
127
128    /// Tracks all the files (vanilla and mods) that are relevant to the current validation.
129    pub(crate) fileset: Fileset,
130
131    /// Tracks specifically the .dds files, and their formats and sizes.
132    pub(crate) dds: DdsFiles,
133
134    /// A general database of item types. Most items go here. The ones that need special handling
135    /// go in the separate databases listed below.
136    pub(crate) database: Db,
137
138    pub(crate) localization: Localization,
139
140    #[cfg(feature = "jomini")]
141    pub(crate) scripted_lists: ScriptedLists,
142
143    pub(crate) defines: Defines,
144
145    #[cfg(feature = "jomini")]
146    pub(crate) events: Events,
147    #[cfg(feature = "hoi4")]
148    pub(crate) events_hoi4: Hoi4Events,
149    #[cfg(feature = "imperator")]
150    pub(crate) decisions_imperator: Decisions,
151
152    #[cfg(feature = "jomini")]
153    pub(crate) scripted_modifiers: ScriptedModifiers,
154    pub(crate) on_actions: OnActions,
155
156    #[cfg(feature = "ck3")]
157    pub(crate) interaction_cats: CharacterInteractionCategories,
158
159    #[cfg(feature = "ck3")]
160    pub(crate) provinces_ck3: Ck3Provinces,
161    #[cfg(feature = "vic3")]
162    pub(crate) provinces_vic3: Vic3Provinces,
163    #[cfg(feature = "imperator")]
164    pub(crate) provinces_imperator: ImperatorProvinces,
165    #[cfg(feature = "hoi4")]
166    pub(crate) provinces_hoi4: Hoi4Provinces,
167
168    #[cfg(feature = "ck3")]
169    pub(crate) province_histories: ProvinceHistories,
170    #[cfg(feature = "ck3")]
171    pub(crate) province_properties: ProvinceProperties,
172    #[cfg(feature = "ck3")]
173    pub(crate) province_terrains: ProvinceTerrains,
174
175    #[cfg(feature = "ck3")]
176    pub(crate) gameconcepts: GameConcepts,
177
178    #[cfg(feature = "ck3")]
179    pub(crate) titles: Titles,
180
181    #[cfg(feature = "ck3")]
182    pub(crate) characters: Characters,
183
184    #[cfg(feature = "jomini")]
185    pub(crate) script_values: ScriptValues,
186
187    pub(crate) triggers: Triggers,
188    pub(crate) effects: Effects,
189
190    #[cfg(feature = "ck3")]
191    pub(crate) traits: Traits,
192
193    #[cfg(feature = "ck3")]
194    pub(crate) title_history: TitleHistories,
195
196    #[cfg(feature = "ck3")]
197    pub(crate) doctrines: Doctrines,
198
199    #[cfg(feature = "ck3")]
200    pub(crate) menatarmstypes: MenAtArmsTypes,
201
202    pub(crate) gui: Gui,
203    #[cfg(any(feature = "ck3", feature = "vic3"))]
204    pub(crate) data_bindings: DataBindings,
205
206    #[cfg(feature = "hoi4")]
207    pub(crate) gfx: Gfx,
208    pub(crate) assets: Assets,
209    #[cfg(feature = "hoi4")]
210    pub(crate) music_hoi4: Hoi4Musics,
211    #[cfg(feature = "jomini")]
212    pub(crate) music: Musics,
213
214    #[cfg(feature = "jomini")]
215    pub(crate) coas: Coas,
216
217    #[cfg(feature = "vic3")]
218    pub(crate) history: History,
219
220    #[cfg(feature = "ck3")]
221    pub(crate) wars: Wars,
222
223    pub(crate) variables: Variables,
224}
225
226impl Everything {
227    /// Create a new `Everything` instance, ready for validating a mod.
228    ///
229    /// `vanilla_dir` is the path to the base game files. If it's `None`, then no vanilla files
230    /// will be loaded. This will seriously affect validation, but it's ok if you just want to load
231    /// and examine the mod files.
232    ///
233    /// `mod_root` is the path to the mod files. The config file will also be looked for there.
234    ///
235    /// `replace_paths` is from the similarly named field in the `.mod` file.
236    pub fn new(
237        config_filepath: Option<&Path>,
238        vanilla_dir: Option<&Path>,
239        workshop_dir: Option<&Path>,
240        paradox_dir: Option<&Path>,
241        mod_root: &Path,
242        replace_paths: Vec<PathBuf>,
243    ) -> Result<Self> {
244        let mut fileset = Fileset::new(vanilla_dir, mod_root.to_path_buf(), replace_paths);
245
246        let config_file_name = match Game::game() {
247            #[cfg(feature = "ck3")]
248            Game::Ck3 => "ck3-tiger.conf",
249            #[cfg(feature = "vic3")]
250            Game::Vic3 => "vic3-tiger.conf",
251            #[cfg(feature = "imperator")]
252            Game::Imperator => "imperator-tiger.conf",
253            #[cfg(feature = "hoi4")]
254            Game::Hoi4 => "hoi4-tiger.conf",
255        };
256
257        let config_file = match config_filepath {
258            Some(path) => path.to_path_buf(),
259            None => mod_root.join(config_file_name),
260        };
261
262        let config = if config_file.is_file() {
263            Self::read_config(config_file_name, &config_file)
264                .ok_or(FilesError::ConfigUnreadable { path: config_file })?
265        } else {
266            Block::new(Loc::for_file(config_file.clone(), FileKind::Mod, config_file.clone()))
267        };
268
269        fileset.config(config.clone(), workshop_dir, paradox_dir)?;
270
271        fileset.scan_all()?;
272        fileset.finalize();
273
274        Ok(Everything {
275            parser: ParserMemory::default(),
276            fileset,
277            dds: DdsFiles::default(),
278            config,
279            #[cfg(any(feature = "ck3", feature = "vic3"))]
280            warned_defines: RwLock::new(TigerHashSet::default()),
281            database: Db::default(),
282            localization: Localization::default(),
283            #[cfg(feature = "jomini")]
284            scripted_lists: ScriptedLists::default(),
285            defines: Defines::default(),
286            #[cfg(feature = "jomini")]
287            events: Events::default(),
288            #[cfg(feature = "hoi4")]
289            events_hoi4: Hoi4Events::default(),
290            #[cfg(feature = "imperator")]
291            decisions_imperator: Decisions::default(),
292            #[cfg(feature = "jomini")]
293            scripted_modifiers: ScriptedModifiers::default(),
294            on_actions: OnActions::default(),
295            #[cfg(feature = "ck3")]
296            interaction_cats: CharacterInteractionCategories::default(),
297            #[cfg(feature = "ck3")]
298            provinces_ck3: Ck3Provinces::default(),
299            #[cfg(feature = "vic3")]
300            provinces_vic3: Vic3Provinces::default(),
301            #[cfg(feature = "imperator")]
302            provinces_imperator: ImperatorProvinces::default(),
303            #[cfg(feature = "hoi4")]
304            provinces_hoi4: Hoi4Provinces::default(),
305            #[cfg(feature = "ck3")]
306            province_histories: ProvinceHistories::default(),
307            #[cfg(feature = "ck3")]
308            province_properties: ProvinceProperties::default(),
309            #[cfg(feature = "ck3")]
310            province_terrains: ProvinceTerrains::default(),
311            #[cfg(feature = "ck3")]
312            gameconcepts: GameConcepts::default(),
313            #[cfg(feature = "ck3")]
314            titles: Titles::default(),
315            #[cfg(feature = "ck3")]
316            characters: Characters::default(),
317            #[cfg(feature = "jomini")]
318            script_values: ScriptValues::default(),
319            triggers: Triggers::default(),
320            effects: Effects::default(),
321            #[cfg(feature = "ck3")]
322            traits: Traits::default(),
323            #[cfg(feature = "ck3")]
324            title_history: TitleHistories::default(),
325            #[cfg(feature = "ck3")]
326            doctrines: Doctrines::default(),
327            #[cfg(feature = "ck3")]
328            menatarmstypes: MenAtArmsTypes::default(),
329            gui: Gui::default(),
330            #[cfg(any(feature = "ck3", feature = "vic3"))]
331            data_bindings: DataBindings::default(),
332            #[cfg(feature = "hoi4")]
333            gfx: Gfx::default(),
334            assets: Assets::default(),
335            #[cfg(feature = "hoi4")]
336            music_hoi4: Hoi4Musics::default(),
337            #[cfg(feature = "jomini")]
338            music: Musics::default(),
339            #[cfg(feature = "jomini")]
340            coas: Coas::default(),
341            #[cfg(feature = "vic3")]
342            history: History::default(),
343            #[cfg(feature = "ck3")]
344            wars: Wars::default(),
345            variables: Variables::new(),
346        })
347    }
348
349    fn read_config(name: &str, path: &Path) -> Option<Block> {
350        let entry = FileEntry::new(PathBuf::from(name), FileKind::Mod, path.to_path_buf());
351        PdxFile::read_optional_bom(&entry, &ParserMemory::default())
352    }
353
354    pub fn load_config_filtering_rules(&self) {
355        check_for_legacy_ignore(&self.config);
356        load_filter(&self.config);
357    }
358
359    /// Load the `OutputStyle` settings from the config.
360    /// Note that the settings from the config can still be overridden
361    /// by supplying the --no-color flag.
362    fn load_output_styles(&self, default_color: bool) -> OutputStyle {
363        // Treat a missing output_style block and an empty output_style block exactly the same.
364        let block = match self.config.get_field_block("output_style") {
365            Some(block) => Cow::Borrowed(block),
366            None => Cow::Owned(Block::new(self.config.loc)),
367        };
368        if !block.get_field_bool("enable").unwrap_or(default_color) {
369            return OutputStyle::no_color();
370        }
371        let mut style = OutputStyle::default();
372        for severity in Severity::iter() {
373            if let Some(error_block) =
374                block.get_field_block(format!("{severity}").to_ascii_lowercase().as_str())
375            {
376                if let Some(color) = error_block.get_field_value("color") {
377                    style.set(severity, color.as_str());
378                }
379            }
380        }
381        style
382    }
383
384    pub fn load_output_settings(&self, default_colors: bool) {
385        set_output_style(self.load_output_styles(default_colors));
386    }
387
388    #[cfg(feature = "vic3")]
389    fn load_json<F>(&mut self, itype: Item, add_json: F)
390    where
391        F: Fn(&mut Db, Block) + Sync + Send,
392    {
393        for block in self.fileset.filter_map_under(&PathBuf::from(itype.path()), |entry| {
394            if entry.filename().to_string_lossy().ends_with(".json") {
395                parse_json_file(entry)
396            } else {
397                None
398            }
399        }) {
400            add_json(&mut self.database, block);
401        }
402    }
403
404    #[cfg(feature = "ck3")]
405    fn load_reader_export(&mut self) {
406        let path = PathBuf::from("reader_export");
407        for entry in self.fileset.get_files_under(&path) {
408            if entry.filename().to_string_lossy().ends_with(".txt") {
409                PdxFile::reader_export(entry, &mut self.parser.pdxfile);
410            }
411        }
412    }
413
414    fn load_pdx_files(&mut self, loader: &ItemLoader) {
415        let path = PathBuf::from(loader.itype().path());
416        let recursive = loader.recursive();
417        let expect_count = path.components().count() + 1;
418        for mut block in self.fileset.filter_map_under(&path, |entry| {
419            // It's <= expect_count because some loader paths are files not directories
420            if (recursive || entry.path().components().count() <= expect_count)
421                && entry.filename().to_string_lossy().ends_with(loader.extension())
422            {
423                PdxFile::read_encoded(entry, loader.encoding(), &self.parser)
424            } else {
425                None
426            }
427        }) {
428            if loader.whole_file() {
429                let fname = block.loc.filename();
430                // unwrap is safe here because of the ends_with check above.
431                let key = fname.strip_suffix(loader.extension()).unwrap();
432                let key = Token::new(key, block.loc);
433                (loader.adder())(&mut self.database, key, block);
434            } else {
435                for (key, block) in block.drain_definitions_warn() {
436                    (loader.adder())(&mut self.database, key, block);
437                }
438            }
439        }
440    }
441
442    fn load_all_normal_pdx_files(&mut self) {
443        for loader in inventory::iter::<ItemLoader> {
444            if loader.for_game(Game::game()) {
445                self.load_pdx_files(loader);
446            }
447        }
448    }
449
450    fn load_all_generic(&mut self) {
451        scope(|s| {
452            s.spawn(|_| self.fileset.handle(&mut self.dds, &self.parser));
453            s.spawn(|_| self.fileset.handle(&mut self.localization, &self.parser));
454            s.spawn(|_| self.fileset.handle(&mut self.defines, &self.parser));
455            s.spawn(|_| self.fileset.handle(&mut self.triggers, &self.parser));
456            s.spawn(|_| self.fileset.handle(&mut self.effects, &self.parser));
457            s.spawn(|_| self.fileset.handle(&mut self.assets, &self.parser));
458            s.spawn(|_| self.fileset.handle(&mut self.gui, &self.parser));
459            s.spawn(|_| self.fileset.handle(&mut self.on_actions, &self.parser));
460        });
461
462        self.load_all_normal_pdx_files();
463    }
464
465    #[cfg(feature = "ck3")]
466    fn load_all_ck3(&mut self) {
467        scope(|s| {
468            s.spawn(|_| self.fileset.handle(&mut self.events, &self.parser));
469            s.spawn(|_| self.fileset.handle(&mut self.interaction_cats, &self.parser));
470            s.spawn(|_| self.fileset.handle(&mut self.province_histories, &self.parser));
471            s.spawn(|_| self.fileset.handle(&mut self.province_properties, &self.parser));
472            s.spawn(|_| self.fileset.handle(&mut self.province_terrains, &self.parser));
473            s.spawn(|_| self.fileset.handle(&mut self.gameconcepts, &self.parser));
474            s.spawn(|_| self.fileset.handle(&mut self.titles, &self.parser));
475            s.spawn(|_| self.fileset.handle(&mut self.characters, &self.parser));
476            s.spawn(|_| self.fileset.handle(&mut self.traits, &self.parser));
477            s.spawn(|_| self.fileset.handle(&mut self.title_history, &self.parser));
478            s.spawn(|_| self.fileset.handle(&mut self.doctrines, &self.parser));
479            s.spawn(|_| self.fileset.handle(&mut self.menatarmstypes, &self.parser));
480            s.spawn(|_| self.fileset.handle(&mut self.music, &self.parser));
481            s.spawn(|_| self.fileset.handle(&mut self.data_bindings, &self.parser));
482            s.spawn(|_| self.fileset.handle(&mut self.provinces_ck3, &self.parser));
483            s.spawn(|_| self.fileset.handle(&mut self.scripted_lists, &self.parser));
484            s.spawn(|_| self.fileset.handle(&mut self.wars, &self.parser));
485            s.spawn(|_| self.fileset.handle(&mut self.coas, &self.parser));
486            s.spawn(|_| self.fileset.handle(&mut self.scripted_modifiers, &self.parser));
487            s.spawn(|_| self.fileset.handle(&mut self.script_values, &self.parser));
488        });
489        crate::ck3::data::buildings::Building::finalize(&mut self.database);
490    }
491
492    #[cfg(feature = "vic3")]
493    fn load_all_vic3(&mut self) {
494        self.fileset.handle(&mut self.events, &self.parser);
495        self.fileset.handle(&mut self.history, &self.parser);
496        self.fileset.handle(&mut self.provinces_vic3, &self.parser);
497        self.fileset.handle(&mut self.data_bindings, &self.parser);
498        self.fileset.handle(&mut self.coas, &self.parser);
499        self.fileset.handle(&mut self.scripted_lists, &self.parser);
500        self.fileset.handle(&mut self.scripted_modifiers, &self.parser);
501        self.fileset.handle(&mut self.script_values, &self.parser);
502        self.fileset.handle(&mut self.music, &self.parser);
503        self.load_json(Item::TerrainMask, TerrainMask::add_json);
504    }
505
506    #[cfg(feature = "imperator")]
507    fn load_all_imperator(&mut self) {
508        self.fileset.handle(&mut self.events, &self.parser);
509        self.fileset.handle(&mut self.decisions_imperator, &self.parser);
510        self.fileset.handle(&mut self.provinces_imperator, &self.parser);
511        self.fileset.handle(&mut self.coas, &self.parser);
512        self.fileset.handle(&mut self.scripted_lists, &self.parser);
513        self.fileset.handle(&mut self.scripted_modifiers, &self.parser);
514        self.fileset.handle(&mut self.script_values, &self.parser);
515        self.fileset.handle(&mut self.music, &self.parser);
516    }
517
518    #[cfg(feature = "hoi4")]
519    fn load_all_hoi4(&mut self) {
520        self.fileset.handle(&mut self.events_hoi4, &self.parser);
521        self.fileset.handle(&mut self.gfx, &self.parser);
522        self.fileset.handle(&mut self.provinces_hoi4, &self.parser);
523        self.fileset.handle(&mut self.music_hoi4, &self.parser);
524    }
525
526    fn scan_all_generic(&mut self) {
527        self.triggers.scan_variables(&mut self.variables);
528        self.effects.scan_variables(&mut self.variables);
529        self.on_actions.scan_variables(&mut self.variables);
530    }
531
532    #[cfg(feature = "ck3")]
533    fn scan_all_ck3(&mut self) {
534        self.events.scan_variables(&mut self.variables);
535        self.interaction_cats.scan_variables(&mut self.variables);
536        self.province_histories.scan_variables(&mut self.variables);
537        self.titles.scan_variables(&mut self.variables);
538        self.characters.scan_variables(&mut self.variables);
539        self.traits.scan_variables(&mut self.variables);
540        self.title_history.scan_variables(&mut self.variables);
541        self.doctrines.scan_variables(&mut self.variables);
542        self.menatarmstypes.scan_variables(&mut self.variables);
543        self.music.scan_variables(&mut self.variables);
544        self.scripted_lists.scan_variables(&mut self.variables);
545        self.coas.scan_variables(&mut self.variables);
546        self.scripted_modifiers.scan_variables(&mut self.variables);
547        self.script_values.scan_variables(&mut self.variables);
548    }
549
550    #[cfg(feature = "vic3")]
551    fn scan_all_vic3(&mut self) {
552        self.events.scan_variables(&mut self.variables);
553        self.history.scan_variables(&mut self.variables);
554        self.coas.scan_variables(&mut self.variables);
555        self.scripted_lists.scan_variables(&mut self.variables);
556        self.scripted_modifiers.scan_variables(&mut self.variables);
557        self.script_values.scan_variables(&mut self.variables);
558        self.music.scan_variables(&mut self.variables);
559    }
560
561    #[cfg(feature = "imperator")]
562    fn scan_all_imperator(&mut self) {
563        self.events.scan_variables(&mut self.variables);
564        self.decisions_imperator.scan_variables(&mut self.variables);
565        self.coas.scan_variables(&mut self.variables);
566        self.scripted_lists.scan_variables(&mut self.variables);
567        self.scripted_modifiers.scan_variables(&mut self.variables);
568        self.script_values.scan_variables(&mut self.variables);
569        self.music.scan_variables(&mut self.variables);
570    }
571
572    #[cfg(feature = "hoi4")]
573    fn scan_all_hoi4(&mut self) {
574        self.events_hoi4.scan_variables(&mut self.variables);
575        self.music_hoi4.scan_variables(&mut self.variables);
576    }
577
578    pub fn load_all(&mut self) {
579        #[cfg(feature = "ck3")]
580        self.load_reader_export();
581        self.load_all_generic();
582        match Game::game() {
583            #[cfg(feature = "ck3")]
584            Game::Ck3 => self.load_all_ck3(),
585            #[cfg(feature = "vic3")]
586            Game::Vic3 => self.load_all_vic3(),
587            #[cfg(feature = "imperator")]
588            Game::Imperator => self.load_all_imperator(),
589            #[cfg(feature = "hoi4")]
590            Game::Hoi4 => self.load_all_hoi4(),
591        }
592        self.database.add_subitems();
593
594        self.scan_all_generic();
595        match Game::game() {
596            #[cfg(feature = "ck3")]
597            Game::Ck3 => self.scan_all_ck3(),
598            #[cfg(feature = "vic3")]
599            Game::Vic3 => self.scan_all_vic3(),
600            #[cfg(feature = "imperator")]
601            Game::Imperator => self.scan_all_imperator(),
602            #[cfg(feature = "hoi4")]
603            Game::Hoi4 => self.scan_all_hoi4(),
604        }
605        self.database.scan_variables(&mut self.variables);
606    }
607
608    fn validate_all_generic<'a>(&'a self, s: &Scope<'a>) {
609        s.spawn(|_| self.fileset.validate(self));
610        s.spawn(|_| self.defines.validate(self));
611        s.spawn(|_| self.triggers.validate(self));
612        s.spawn(|_| self.effects.validate(self));
613        s.spawn(|_| self.assets.validate(self));
614        s.spawn(|_| self.gui.validate(self));
615        s.spawn(|_| self.on_actions.validate(self));
616        s.spawn(|_| self.dds.validate());
617    }
618
619    #[cfg(feature = "ck3")]
620    fn validate_all_ck3<'a>(&'a self, s: &Scope<'a>) {
621        s.spawn(|_| self.events.validate(self));
622        s.spawn(|_| self.interaction_cats.validate(self));
623        s.spawn(|_| self.province_histories.validate(self));
624        s.spawn(|_| self.province_properties.validate(self));
625        s.spawn(|_| self.province_terrains.validate(self));
626        s.spawn(|_| self.gameconcepts.validate(self));
627        s.spawn(|_| self.titles.validate(self));
628        s.spawn(|_| self.characters.validate(self));
629        s.spawn(|_| self.traits.validate(self));
630        s.spawn(|_| self.title_history.validate(self));
631        s.spawn(|_| self.doctrines.validate(self));
632        s.spawn(|_| self.menatarmstypes.validate(self));
633        s.spawn(|_| self.data_bindings.validate(self));
634        s.spawn(|_| self.provinces_ck3.validate(self));
635        s.spawn(|_| self.wars.validate(self));
636        s.spawn(|_| self.coas.validate(self));
637        s.spawn(|_| self.scripted_lists.validate(self));
638        s.spawn(|_| self.scripted_modifiers.validate(self));
639        s.spawn(|_| self.script_values.validate(self));
640        s.spawn(|_| self.music.validate(self));
641        s.spawn(|_| Climate::validate_all(&self.database, self));
642    }
643
644    #[cfg(feature = "vic3")]
645    fn validate_all_vic3<'a>(&'a self, s: &Scope<'a>) {
646        s.spawn(|_| self.events.validate(self));
647        s.spawn(|_| self.history.validate(self));
648        s.spawn(|_| self.provinces_vic3.validate(self));
649        s.spawn(|_| self.data_bindings.validate(self));
650        s.spawn(|_| self.coas.validate(self));
651        s.spawn(|_| self.scripted_lists.validate(self));
652        s.spawn(|_| self.scripted_modifiers.validate(self));
653        s.spawn(|_| self.script_values.validate(self));
654        s.spawn(|_| self.music.validate(self));
655        s.spawn(|_| StrategicRegion::crosscheck(self));
656        s.spawn(|_| BuyPackage::crosscheck(self));
657    }
658
659    #[cfg(feature = "imperator")]
660    fn validate_all_imperator<'a>(&'a self, s: &Scope<'a>) {
661        s.spawn(|_| self.events.validate(self));
662        s.spawn(|_| self.decisions_imperator.validate(self));
663        s.spawn(|_| self.provinces_imperator.validate(self));
664        s.spawn(|_| self.coas.validate(self));
665        s.spawn(|_| self.scripted_lists.validate(self));
666        s.spawn(|_| self.scripted_modifiers.validate(self));
667        s.spawn(|_| self.script_values.validate(self));
668        s.spawn(|_| self.music.validate(self));
669    }
670
671    #[cfg(feature = "hoi4")]
672    fn validate_all_hoi4<'a>(&'a self, s: &Scope<'a>) {
673        s.spawn(|_| self.events_hoi4.validate(self));
674        s.spawn(|_| self.provinces_hoi4.validate(self));
675        s.spawn(|_| self.gfx.validate(self));
676        s.spawn(|_| self.music_hoi4.validate(self));
677    }
678
679    pub fn validate_all(&self) {
680        scope(|s| {
681            self.validate_all_generic(s);
682            match Game::game() {
683                #[cfg(feature = "ck3")]
684                Game::Ck3 => self.validate_all_ck3(s),
685                #[cfg(feature = "vic3")]
686                Game::Vic3 => self.validate_all_vic3(s),
687                #[cfg(feature = "imperator")]
688                Game::Imperator => self.validate_all_imperator(s),
689                #[cfg(feature = "hoi4")]
690                Game::Hoi4 => self.validate_all_hoi4(s),
691            }
692        });
693        self.database.validate(self);
694
695        self.localization.validate_pass2(self);
696    }
697
698    pub fn check_rivers(&mut self) {
699        let mut rivers = Rivers::default();
700        self.fileset.handle(&mut rivers, &self.parser);
701        rivers.validate(self);
702    }
703
704    #[cfg(feature = "ck3")]
705    pub fn check_pod(&mut self) {
706        self.province_histories.check_pod_faiths(self, &self.titles);
707        self.characters.check_pod_flags(self);
708        self.localization.check_pod_loca(self);
709    }
710
711    pub fn check_unused(&mut self) {
712        self.localization.check_unused(self);
713        self.fileset.check_unused_dds(self);
714    }
715
716    #[allow(dead_code)]
717    pub(crate) fn item_has_property(&self, itype: Item, key: &str, property: &str) -> bool {
718        self.database.has_property(itype, key, property, self)
719    }
720
721    #[cfg(feature = "ck3")] // vic3 happens not to use
722    pub(crate) fn item_lc_has_property(
723        &self,
724        itype: Item,
725        key: &Lowercase,
726        property: &str,
727    ) -> bool {
728        self.database.lc_has_property(itype, key, property, self)
729    }
730
731    #[cfg(feature = "ck3")]
732    fn item_exists_ck3(&self, itype: Item, key: &str) -> bool {
733        match itype {
734            Item::ActivityState => ACTIVITY_STATES.contains(&key),
735            Item::ArtifactHistory => ARTIFACT_HISTORY.contains(&key),
736            Item::ArtifactRarity => ARTIFACT_RARITIES.contains(&&*key.to_ascii_lowercase()),
737            Item::Character => self.characters.exists(key),
738            Item::CharacterInteractionCategory => self.interaction_cats.exists(key),
739            Item::Coa => self.coas.exists(key),
740            Item::CoaTemplate => self.coas.template_exists(key),
741            Item::DangerType => DANGER_TYPES.contains(&key),
742            Item::DlcFeature => DLC_FEATURES_CK3.contains(&key),
743            Item::Doctrine => self.doctrines.exists(key),
744            Item::DoctrineCategory => self.doctrines.category_exists(key),
745            Item::DoctrineParameter => self.doctrines.parameter_exists(key),
746            Item::Event => self.events.exists(key),
747            Item::EventNamespace => self.events.namespace_exists(key),
748            Item::GameConcept => self.gameconcepts.exists(key),
749            Item::GeneAttribute => self.assets.attribute_exists(key),
750            Item::GeneticConstraint => self.traits.constraint_exists(key),
751            Item::MenAtArms => self.menatarmstypes.exists(key),
752            Item::MenAtArmsBase => self.menatarmstypes.base_exists(key),
753            Item::Music => self.music.exists(key),
754            Item::PrisonType => PRISON_TYPES.contains(&key),
755            Item::Province => self.provinces_ck3.exists(key),
756            Item::RewardItem => REWARD_ITEMS.contains(&key),
757            Item::ScriptedList => self.scripted_lists.exists(key),
758            Item::ScriptedModifier => self.scripted_modifiers.exists(key),
759            Item::ScriptValue => self.script_values.exists(key),
760            Item::Sexuality => SEXUALITIES.contains(&key),
761            Item::Skill => SKILLS.contains(&key),
762            Item::Sound => self.valid_sound(key),
763            Item::Title => self.titles.exists(key),
764            Item::TitleHistory => self.title_history.exists(key),
765            Item::Trait => self.traits.exists(key),
766            Item::TraitFlag => self.traits.flag_exists(key),
767            Item::TraitTrack => self.traits.track_exists(key),
768            Item::TraitCategory => TRAIT_CATEGORIES.contains(&key),
769            _ => self.database.exists(itype, key),
770        }
771    }
772
773    #[cfg(feature = "vic3")]
774    fn item_exists_vic3(&self, itype: Item, key: &str) -> bool {
775        match itype {
776            Item::Approval => APPROVALS.contains(&key),
777            Item::Attitude => ATTITUDES.contains(&&*key.to_lowercase()),
778            Item::CharacterRole => CHARACTER_ROLES.contains(&key),
779            Item::Coa => self.coas.exists(key),
780            Item::CoaTemplate => self.coas.template_exists(key),
781            Item::CountryTier => COUNTRY_TIERS.contains(&key),
782            Item::DlcFeature => DLC_FEATURES_VIC3.contains(&key),
783            Item::Event => self.events.exists(key),
784            Item::EventCategory => EVENT_CATEGORIES.contains(&key),
785            Item::EventNamespace => self.events.namespace_exists(key),
786            Item::GeneAttribute => self.assets.attribute_exists(key),
787            Item::InfamyThreshold => INFAMY_THRESHOLDS.contains(&key),
788            Item::Level => LEVELS.contains(&key),
789            Item::Music => self.music.exists(key),
790            Item::RelationsThreshold => RELATIONS.contains(&key),
791            Item::ScriptedList => self.scripted_lists.exists(key),
792            Item::ScriptedModifier => self.scripted_modifiers.exists(key),
793            Item::ScriptValue => self.script_values.exists(key),
794            Item::SecretGoal => SECRET_GOALS.contains(&key),
795            Item::Sound => self.valid_sound(key),
796            Item::Strata => STRATA.contains(&key),
797            Item::TerrainKey => TERRAIN_KEYS.contains(&key),
798            Item::TransferOfPower => TRANSFER_OF_POWER.contains(&key),
799            Item::Wargoal => WARGOALS.contains(&key),
800            _ => self.database.exists(itype, key),
801        }
802    }
803
804    #[cfg(feature = "imperator")]
805    fn item_exists_imperator(&self, itype: Item, key: &str) -> bool {
806        match itype {
807            Item::Coa => self.coas.exists(key),
808            Item::CoaTemplate => self.coas.template_exists(key),
809            Item::DlcName => DLC_NAME_IMPERATOR.contains(&key),
810            Item::Decision => self.decisions_imperator.exists(key),
811            Item::Event => self.events.exists(key),
812            Item::EventNamespace => self.events.namespace_exists(key),
813            Item::GeneAttribute => self.assets.attribute_exists(key),
814            Item::Music => self.music.exists(key),
815            Item::Province => self.provinces_imperator.exists(key),
816            Item::ScriptedList => self.scripted_lists.exists(key),
817            Item::ScriptedModifier => self.scripted_modifiers.exists(key),
818            Item::ScriptValue => self.script_values.exists(key),
819            Item::Sound => self.valid_sound(key),
820            _ => self.database.exists(itype, key),
821        }
822    }
823
824    #[cfg(feature = "hoi4")]
825    fn item_exists_hoi4(&self, itype: Item, key: &str) -> bool {
826        match itype {
827            Item::AiStrategyType => AI_STRATEGY_TYPES.contains(&key),
828            Item::Event => self.events_hoi4.exists(key),
829            Item::EventNamespace => self.events_hoi4.namespace_exists(key),
830            Item::Music => self.music_hoi4.exists(key),
831            Item::MusicAsset => self.assets.music_exists(key),
832            Item::Pdxmesh => self.gfx.mesh_exists(key),
833            Item::Province => self.provinces_hoi4.exists(key),
834            Item::Sprite => self.gfx.sprite_exists(key),
835            _ => self.database.exists(itype, key),
836        }
837    }
838
839    pub(crate) fn item_exists(&self, itype: Item, key: &str) -> bool {
840        match itype {
841            Item::Asset => self.assets.asset_exists(key),
842            Item::BlendShape => self.assets.blend_shape_exists(key),
843            Item::Define => self.defines.exists(key),
844            Item::Entity => self.assets.entity_exists(key),
845            Item::Entry => self.fileset.entry_exists(key),
846            Item::File => self.fileset.exists(key),
847            Item::GuiLayer => self.gui.layer_exists(key),
848            Item::GuiTemplate => self.gui.template_exists(key),
849            Item::GuiType => self.gui.type_exists(&Lowercase::new(key)),
850            Item::Localization => self.localization.exists(key),
851            Item::OnAction => self.on_actions.exists(key),
852            #[cfg(feature = "jomini")]
853            Item::Pdxmesh => self.assets.mesh_exists(key),
854            Item::ScriptedEffect => self.effects.exists(key),
855            Item::ScriptedTrigger => self.triggers.exists(key),
856            Item::TextFormat => self.gui.textformat_exists(key),
857            Item::TextIcon => self.gui.texticon_exists(key),
858            Item::TextureFile => self.assets.texture_exists(key),
859            Item::WidgetName => self.gui.name_exists(key),
860            Item::Directory | Item::Shortcut => true, // TODO
861            _ => match Game::game() {
862                #[cfg(feature = "ck3")]
863                Game::Ck3 => self.item_exists_ck3(itype, key),
864                #[cfg(feature = "vic3")]
865                Game::Vic3 => self.item_exists_vic3(itype, key),
866                #[cfg(feature = "imperator")]
867                Game::Imperator => self.item_exists_imperator(itype, key),
868                #[cfg(feature = "hoi4")]
869                Game::Hoi4 => self.item_exists_hoi4(itype, key),
870            },
871        }
872    }
873
874    /// Return true iff the item `key` is found with a case insensitive match.
875    /// This function is **incomplete**. It only contains the item types for which case insensitive
876    /// matches are needed; this is currently the ones used in `src/ck3/tables/modif.rs`.
877    #[cfg(feature = "ck3")]
878    fn item_exists_lc_ck3(&self, itype: Item, key: &Lowercase) -> bool {
879        match itype {
880            Item::MenAtArmsBase => self.menatarmstypes.base_exists_lc(key),
881            Item::Trait => self.traits.exists_lc(key),
882            Item::TraitTrack => self.traits.track_exists_lc(key),
883            _ => self.database.exists_lc(itype, key),
884        }
885    }
886
887    /// Return true iff the item `key` is found with a case insensitive match.
888    /// This function is **incomplete**. It only contains the item types for which case insensitive
889    /// matches are needed; this is currently the ones used in `src/vic3/tables/modif.rs`.
890    #[cfg(feature = "vic3")]
891    fn item_exists_lc_vic3(&self, itype: Item, key: &Lowercase) -> bool {
892        match itype {
893            Item::TerrainKey => TERRAIN_KEYS.contains(&key.as_str()),
894            _ => self.database.exists_lc(itype, key),
895        }
896    }
897
898    /// Return true iff the item `key` is found with a case insensitive match.
899    /// This function is **incomplete**. It only contains the item types for which case insensitive
900    /// matches are needed; this is currently the ones used in `src/imperator/tables/modif.rs`.
901    #[cfg(feature = "imperator")]
902    fn item_exists_lc_imperator(&self, itype: Item, key: &Lowercase) -> bool {
903        #[allow(clippy::match_single_binding)]
904        match itype {
905            _ => self.database.exists_lc(itype, key),
906        }
907    }
908
909    /// Return true iff the item `key` is found with a case insensitive match.
910    /// This function is **incomplete**. It only contains the item types for which case insensitive
911    /// matches are needed.
912    #[cfg(feature = "hoi4")]
913    fn item_exists_lc_hoi4(&self, itype: Item, key: &Lowercase) -> bool {
914        #[allow(clippy::match_single_binding)]
915        match itype {
916            Item::EventNamespace => self.events_hoi4.namespace_exists_lc(key),
917            _ => self.database.exists_lc(itype, key),
918        }
919    }
920    /// Return true iff the item `key` is found with a case insensitive match.
921    /// This function is **incomplete**. It only contains the item types for which case insensitive
922    /// matches are needed; this is currently the ones used in modif lookups.
923    pub(crate) fn item_exists_lc(&self, itype: Item, key: &Lowercase) -> bool {
924        #[allow(clippy::match_single_binding)]
925        match itype {
926            _ => match Game::game() {
927                #[cfg(feature = "ck3")]
928                Game::Ck3 => self.item_exists_lc_ck3(itype, key),
929                #[cfg(feature = "vic3")]
930                Game::Vic3 => self.item_exists_lc_vic3(itype, key),
931                #[cfg(feature = "imperator")]
932                Game::Imperator => self.item_exists_lc_imperator(itype, key),
933                #[cfg(feature = "hoi4")]
934                Game::Hoi4 => self.item_exists_lc_hoi4(itype, key),
935            },
936        }
937    }
938
939    pub(crate) fn mark_used(&self, itype: Item, key: &str) {
940        match itype {
941            Item::File => self.fileset.mark_used(key),
942            Item::Localization => self.localization.mark_used(key),
943            _ => (),
944        }
945    }
946
947    pub(crate) fn verify_exists(&self, itype: Item, token: &Token) {
948        self.verify_exists_implied(itype, token.as_str(), token);
949    }
950
951    pub(crate) fn verify_exists_max_sev(&self, itype: Item, token: &Token, max_sev: Severity) {
952        self.verify_exists_implied_max_sev(itype, token.as_str(), token, max_sev);
953    }
954
955    pub(crate) fn verify_exists_implied_max_sev(
956        &self,
957        itype: Item,
958        key: &str,
959        token: &Token,
960        max_sev: Severity,
961    ) {
962        match itype {
963            Item::Entry => self.fileset.verify_entry_exists(key, token, max_sev),
964            Item::File => self.fileset.verify_exists_implied(key, token, max_sev),
965            Item::Localization => self.localization.verify_exists_implied(key, token, max_sev),
966            Item::Music => match Game::game() {
967                #[cfg(feature = "ck3")]
968                Game::Ck3 => self.music.verify_exists_implied(key, token, max_sev),
969                #[cfg(feature = "vic3")]
970                Game::Vic3 => self.music.verify_exists_implied(key, token, max_sev),
971                #[cfg(feature = "imperator")]
972                Game::Imperator => self.music.verify_exists_implied(key, token, max_sev),
973                #[cfg(feature = "hoi4")]
974                Game::Hoi4 => self.music_hoi4.verify_exists_implied(key, token, max_sev),
975            },
976            Item::Province => match Game::game() {
977                #[cfg(feature = "ck3")]
978                Game::Ck3 => self.provinces_ck3.verify_exists_implied(key, token, max_sev),
979                #[cfg(feature = "vic3")]
980                Game::Vic3 => self.provinces_vic3.verify_exists_implied(key, token, max_sev),
981                #[cfg(feature = "imperator")]
982                Game::Imperator => {
983                    self.provinces_imperator.verify_exists_implied(key, token, max_sev);
984                }
985                #[cfg(feature = "hoi4")]
986                Game::Hoi4 => {
987                    self.provinces_hoi4.verify_exists_implied(key, token, max_sev);
988                }
989            },
990            Item::TextureFile => {
991                if let Some(entry) = self.assets.get_texture(key) {
992                    // TODO: avoid allocating a string here
993                    self.fileset.mark_used(&entry.path().to_string_lossy());
994                } else {
995                    let msg = format!("no texture file {key} anywhere under {}", itype.path());
996                    report(ErrorKey::MissingFile, itype.severity().at_most(max_sev))
997                        .conf(itype.confidence())
998                        .msg(msg)
999                        .loc(token)
1000                        .push();
1001                }
1002            }
1003            #[cfg(feature = "vic3")]
1004            Item::Wargoal if key == "war_reparations" => {
1005                let msg = "wargoal `war_reparations` was removed in 1.9";
1006                let info = "the new way is to use `enforce_treaty_article` with a `money_transfer` article";
1007                err(ErrorKey::MissingItem).strong().msg(msg).info(info).loc(token).push();
1008            }
1009            _ => {
1010                if !self.item_exists(itype, key) {
1011                    let path = itype.path();
1012                    let msg = if path.is_empty() {
1013                        format!("unknown {itype} {key}")
1014                    } else {
1015                        format!("{itype} {key} not defined in {path}")
1016                    };
1017                    report(ErrorKey::MissingItem, itype.severity().at_most(max_sev))
1018                        .conf(itype.confidence())
1019                        .msg(msg)
1020                        .loc(token)
1021                        .push();
1022                }
1023            }
1024        }
1025    }
1026
1027    #[allow(dead_code)]
1028    pub(crate) fn verify_exists_implied_max_sev_lc(
1029        &self,
1030        itype: Item,
1031        key: &Lowercase,
1032        token: &Token,
1033        max_sev: Severity,
1034    ) {
1035        if !self.item_exists_lc(itype, key) {
1036            let path = itype.path();
1037            let msg = if path.is_empty() {
1038                format!("unknown {itype} {key}")
1039            } else {
1040                format!("{itype} {key} not defined in {path}")
1041            };
1042            report(ErrorKey::MissingItem, itype.severity().at_most(max_sev))
1043                .conf(itype.confidence())
1044                .msg(msg)
1045                .loc(token)
1046                .push();
1047        }
1048    }
1049
1050    pub(crate) fn verify_exists_implied(&self, itype: Item, key: &str, token: &Token) {
1051        self.verify_exists_implied_max_sev(itype, key, token, Severity::Error);
1052    }
1053
1054    #[cfg(feature = "ck3")]
1055    pub(crate) fn verify_icon(&self, define: &str, token: &Token, suffix: &str) {
1056        if let Some(icon_path) = self.get_defined_string_warn(token, define) {
1057            let pathname = format!("{icon_path}/{token}{suffix}");
1058            // It's `Severity::Warning` because a missing icon is only a UI issue.
1059            self.verify_exists_implied_max_sev(Item::File, &pathname, token, Severity::Warning);
1060        }
1061    }
1062
1063    #[cfg(feature = "ck3")]
1064    pub(crate) fn mark_used_icon(&self, define: &str, token: &Token, suffix: &str) {
1065        if let Some(icon_path) = self.get_defined_string_warn(token, define) {
1066            let pathname = format!("{icon_path}/{token}{suffix}");
1067            self.fileset.mark_used(&pathname);
1068        }
1069    }
1070
1071    #[allow(dead_code)]
1072    pub(crate) fn validate_use(&self, itype: Item, key: &Token, block: &Block) {
1073        self.database.validate_use(itype, key, block, self);
1074    }
1075
1076    #[cfg(feature = "ck3")] // happens not to be used by vic3
1077    pub(crate) fn validate_call(
1078        &self,
1079        itype: Item,
1080        key: &Token,
1081        block: &Block,
1082        sc: &mut ScopeContext,
1083    ) {
1084        self.database.validate_call(itype, key, block, self, sc);
1085    }
1086
1087    /// Validate the use of a localization within a specific `ScopeContext`.
1088    /// This allows validation of the named scopes used within the localization's datafunctions.
1089    pub(crate) fn validate_localization_sc(&self, key: &str, sc: &mut ScopeContext) {
1090        self.localization.validate_use(key, self, sc);
1091    }
1092
1093    #[allow(dead_code)]
1094    pub(crate) fn get_item<T: DbKind>(
1095        &self,
1096        itype: Item,
1097        key: &str,
1098    ) -> Option<(&Token, &Block, &T)> {
1099        self.database.get_item(itype, key)
1100    }
1101
1102    pub(crate) fn get_key_block(&self, itype: Item, key: &str) -> Option<(&Token, &Block)> {
1103        self.database.get_key_block(itype, key)
1104    }
1105
1106    pub(crate) fn get_trigger(&self, key: &Token) -> Option<&Trigger> {
1107        #[cfg(feature = "ck3")]
1108        if Game::is_ck3() {
1109            if let Some(trigger) = self.triggers.get(key.as_str()) {
1110                return Some(trigger);
1111            }
1112            if let Some(trigger) = self.events.get_trigger(key) {
1113                return Some(trigger);
1114            }
1115            return None;
1116        }
1117        self.triggers.get(key.as_str())
1118    }
1119
1120    pub(crate) fn get_effect(&self, key: &Token) -> Option<&Effect> {
1121        #[cfg(feature = "ck3")]
1122        if Game::is_ck3() {
1123            if let Some(effect) = self.effects.get(key.as_str()) {
1124                return Some(effect);
1125            }
1126            if let Some(effect) = self.events.get_effect(key) {
1127                return Some(effect);
1128            }
1129            return None;
1130        }
1131        self.effects.get(key.as_str())
1132    }
1133
1134    #[cfg(feature = "ck3")] // happens not to be used by vic3
1135    pub(crate) fn get_defined_string(&self, key: &str) -> Option<&Token> {
1136        self.defines.get_bv(key).and_then(BV::get_value)
1137    }
1138
1139    #[cfg(any(feature = "ck3", feature = "vic3"))]
1140    pub(crate) fn get_defined_array(&self, key: &str) -> Option<&Block> {
1141        self.defines.get_bv(key).and_then(BV::get_block)
1142    }
1143
1144    #[allow(clippy::missing_panics_doc)] // only panics on poisoned mutex
1145    #[cfg(feature = "ck3")] // happens not to be used by vic3
1146    pub(crate) fn get_defined_string_warn(&self, token: &Token, key: &str) -> Option<&Token> {
1147        let result = self.get_defined_string(key);
1148        let mut cache = self.warned_defines.write().unwrap();
1149        if result.is_none() && !cache.contains(key) {
1150            let msg = format!("{key} not defined in common/defines/");
1151            err(ErrorKey::MissingItem).msg(msg).loc(token).push();
1152            cache.insert(key.to_string());
1153        }
1154        result
1155    }
1156
1157    #[allow(clippy::missing_panics_doc)] // only panics on poisoned mutex
1158    #[cfg(any(feature = "ck3", feature = "vic3"))]
1159    pub(crate) fn get_defined_array_warn(&self, token: &Token, key: &str) -> Option<&Block> {
1160        let result = self.get_defined_array(key);
1161        let mut cache = self.warned_defines.write().unwrap();
1162        if result.is_none() && !cache.contains(key) {
1163            let msg = format!("{key} not defined in common/defines/");
1164            err(ErrorKey::MissingItem).msg(msg).loc(token).push();
1165            cache.insert(key.to_string());
1166        }
1167        result
1168    }
1169
1170    #[cfg(feature = "ck3")]
1171    pub fn iter_keys_ck3<'a>(&'a self, itype: Item) -> Box<dyn Iterator<Item = &'a Token> + 'a> {
1172        match itype {
1173            Item::Coa => Box::new(self.coas.iter_keys()),
1174            Item::CoaTemplate => Box::new(self.coas.iter_template_keys()),
1175            Item::Character => Box::new(self.characters.iter_keys()),
1176            Item::CharacterInteractionCategory => Box::new(self.interaction_cats.iter_keys()),
1177            Item::Doctrine => Box::new(self.doctrines.iter_keys()),
1178            Item::DoctrineCategory => Box::new(self.doctrines.iter_category_keys()),
1179            Item::DoctrineParameter => Box::new(self.doctrines.iter_parameter_keys()),
1180            Item::Event => Box::new(self.events.iter_keys()),
1181            Item::EventNamespace => Box::new(self.events.iter_namespace_keys()),
1182            Item::GameConcept => Box::new(self.gameconcepts.iter_keys()),
1183            Item::GeneAttribute => Box::new(self.assets.iter_attribute_keys()),
1184            Item::GeneticConstraint => Box::new(self.traits.iter_constraint_keys()),
1185            Item::MenAtArms => Box::new(self.menatarmstypes.iter_keys()),
1186            Item::MenAtArmsBase => Box::new(self.menatarmstypes.iter_base_keys()),
1187            Item::Music => Box::new(self.music.iter_keys()),
1188            Item::Province => Box::new(self.provinces_ck3.iter_keys()),
1189            Item::ScriptedList => Box::new(self.scripted_lists.iter_keys()),
1190            Item::ScriptedModifier => Box::new(self.scripted_modifiers.iter_keys()),
1191            Item::ScriptValue => Box::new(self.script_values.iter_keys()),
1192            Item::Title => Box::new(self.titles.iter_keys()),
1193            Item::TitleHistory => Box::new(self.title_history.iter_keys()),
1194            Item::Trait => Box::new(self.traits.iter_keys()),
1195            Item::TraitFlag => Box::new(self.traits.iter_flag_keys()),
1196            Item::TraitTrack => Box::new(self.traits.iter_track_keys()),
1197            _ => Box::new(self.database.iter_keys(itype)),
1198        }
1199    }
1200
1201    #[cfg(feature = "vic3")]
1202    fn iter_keys_vic3<'a>(&'a self, itype: Item) -> Box<dyn Iterator<Item = &'a Token> + 'a> {
1203        match itype {
1204            Item::Coa => Box::new(self.coas.iter_keys()),
1205            Item::CoaTemplate => Box::new(self.coas.iter_template_keys()),
1206            Item::Event => Box::new(self.events.iter_keys()),
1207            Item::EventNamespace => Box::new(self.events.iter_namespace_keys()),
1208            Item::Music => Box::new(self.music.iter_keys()),
1209            Item::GeneAttribute => Box::new(self.assets.iter_attribute_keys()),
1210            Item::ScriptedList => Box::new(self.scripted_lists.iter_keys()),
1211            Item::ScriptedModifier => Box::new(self.scripted_modifiers.iter_keys()),
1212            Item::ScriptValue => Box::new(self.script_values.iter_keys()),
1213            _ => Box::new(self.database.iter_keys(itype)),
1214        }
1215    }
1216
1217    #[cfg(feature = "imperator")]
1218    fn iter_keys_imperator<'a>(&'a self, itype: Item) -> Box<dyn Iterator<Item = &'a Token> + 'a> {
1219        match itype {
1220            Item::Coa => Box::new(self.coas.iter_keys()),
1221            Item::CoaTemplate => Box::new(self.coas.iter_template_keys()),
1222            Item::Decision => Box::new(self.decisions_imperator.iter_keys()),
1223            Item::Event => Box::new(self.events.iter_keys()),
1224            Item::EventNamespace => Box::new(self.events.iter_namespace_keys()),
1225            Item::GeneAttribute => Box::new(self.assets.iter_attribute_keys()),
1226            Item::Music => Box::new(self.music.iter_keys()),
1227            Item::Province => Box::new(self.provinces_imperator.iter_keys()),
1228            Item::ScriptedList => Box::new(self.scripted_lists.iter_keys()),
1229            Item::ScriptedModifier => Box::new(self.scripted_modifiers.iter_keys()),
1230            Item::ScriptValue => Box::new(self.script_values.iter_keys()),
1231            _ => Box::new(self.database.iter_keys(itype)),
1232        }
1233    }
1234
1235    #[cfg(feature = "hoi4")]
1236    fn iter_keys_hoi4<'a>(&'a self, itype: Item) -> Box<dyn Iterator<Item = &'a Token> + 'a> {
1237        match itype {
1238            Item::Event => Box::new(self.events_hoi4.iter_keys()),
1239            Item::EventNamespace => Box::new(self.events_hoi4.iter_namespace_keys()),
1240            Item::Music => Box::new(self.music_hoi4.iter_keys()),
1241            Item::MusicAsset => Box::new(self.assets.iter_music_keys()),
1242            Item::Pdxmesh => Box::new(self.gfx.iter_mesh_keys()),
1243            Item::Province => Box::new(self.provinces_hoi4.iter_keys()),
1244            Item::Sprite => Box::new(self.gfx.iter_sprite_keys()),
1245            _ => Box::new(self.database.iter_keys(itype)),
1246        }
1247    }
1248
1249    pub fn iter_keys<'a>(&'a self, itype: Item) -> Box<dyn Iterator<Item = &'a Token> + 'a> {
1250        match itype {
1251            Item::Asset => Box::new(self.assets.iter_asset_keys()),
1252            Item::BlendShape => Box::new(self.assets.iter_blend_shape_keys()),
1253            Item::Define => Box::new(self.defines.iter_keys()),
1254            Item::Entity => Box::new(self.assets.iter_entity_keys()),
1255            Item::File => Box::new(self.fileset.iter_keys()),
1256            Item::GuiLayer => Box::new(self.gui.iter_layer_keys()),
1257            Item::GuiTemplate => Box::new(self.gui.iter_template_keys()),
1258            Item::GuiType => Box::new(self.gui.iter_type_keys()),
1259            Item::Localization => Box::new(self.localization.iter_keys()),
1260            Item::OnAction => Box::new(self.on_actions.iter_keys()),
1261            #[cfg(feature = "jomini")]
1262            Item::Pdxmesh => Box::new(self.assets.iter_mesh_keys()),
1263            Item::ScriptedEffect => Box::new(self.effects.iter_keys()),
1264            Item::ScriptedTrigger => Box::new(self.triggers.iter_keys()),
1265            Item::TextFormat => Box::new(self.gui.iter_textformat_keys()),
1266            Item::TextIcon => Box::new(self.gui.iter_texticon_keys()),
1267            Item::TextureFile => Box::new(self.assets.iter_texture_keys()),
1268            Item::WidgetName => Box::new(self.gui.iter_names()),
1269            _ => match Game::game() {
1270                #[cfg(feature = "ck3")]
1271                Game::Ck3 => self.iter_keys_ck3(itype),
1272                #[cfg(feature = "vic3")]
1273                Game::Vic3 => self.iter_keys_vic3(itype),
1274                #[cfg(feature = "imperator")]
1275                Game::Imperator => self.iter_keys_imperator(itype),
1276                #[cfg(feature = "hoi4")]
1277                Game::Hoi4 => self.iter_keys_hoi4(itype),
1278            },
1279        }
1280    }
1281
1282    #[cfg(feature = "jomini")]
1283    fn valid_sound(&self, name: &str) -> bool {
1284        // TODO: verify that file:/ values work
1285        if let Some(filename) = name.strip_prefix("file:/") {
1286            self.fileset.exists(filename)
1287        } else {
1288            let sounds_set = match Game::game() {
1289                #[cfg(feature = "ck3")]
1290                Game::Ck3 => &crate::ck3::tables::sounds::SOUNDS_SET,
1291                #[cfg(feature = "vic3")]
1292                Game::Vic3 => &crate::vic3::tables::sounds::SOUNDS_SET,
1293                #[cfg(feature = "imperator")]
1294                Game::Imperator => &crate::imperator::tables::sounds::SOUNDS_SET,
1295            };
1296            sounds_set.contains(&Lowercase::new(name))
1297        }
1298    }
1299
1300    /// Return true iff a script value of the given name is defined.
1301    #[allow(clippy::unused_self)]
1302    #[allow(unused_variables)] // hoi4 does not use `name`
1303    pub(crate) fn script_value_exists(&self, name: &str) -> bool {
1304        if Game::is_jomini() {
1305            #[cfg(feature = "jomini")]
1306            return self.script_values.exists(name);
1307        }
1308        false
1309    }
1310
1311    pub(crate) fn event_check_scope(&self, id: &Token, sc: &mut ScopeContext) {
1312        if Game::is_hoi4() {
1313            #[cfg(feature = "hoi4")]
1314            self.events_hoi4.check_scope(id, sc);
1315        } else {
1316            #[cfg(feature = "jomini")]
1317            self.events.check_scope(id, sc);
1318        }
1319    }
1320
1321    pub(crate) fn event_validate_call(&self, id: &Token, sc: &mut ScopeContext) {
1322        if Game::is_hoi4() {
1323            #[cfg(feature = "hoi4")]
1324            self.events_hoi4.validate_call(id, self, sc);
1325        } else {
1326            #[cfg(feature = "jomini")]
1327            self.events.validate_call(id, self, sc);
1328        }
1329    }
1330}
1331
1332impl Drop for Everything {
1333    fn drop(&mut self) {
1334        // For the sake of the benchmark code, restore MACRO_MAP to a clean slate
1335        MACRO_MAP.clear();
1336    }
1337}