1use anyhow::{anyhow, Result};
21use directories::ProjectDirs;
22use ron::ser::{PrettyConfig, to_string_pretty};
23use serde_derive::{Serialize, Deserialize};
24
25use std::collections::HashMap;
26use std::io::{BufReader, BufWriter, Read, Write};
27use std::fs::{DirBuilder, File};
28use std::path::{Path, PathBuf};
29
30use rpfm_extensions::optimizer::OptimizerOptions;
31
32use rpfm_ipc::settings_keys::*;
33
34use rpfm_lib::error::RLibError;
35use rpfm_lib::games::{GameInfo, LUA_AUTOGEN_FOLDER, supported_games::*};
36use rpfm_lib::schema::{DefinitionPatch, SCHEMA_FOLDER};
37
38use crate::*;
39
40const SETTINGS_FILE_NAME: &str = "settings.json";
41
42const CONFIG_REDIRECT_FILE_NAME: &str = "config_folder.txt";
44
45const DEPENDENCIES_FOLDER: &str = "dependencies";
46
47const SCRIPTS_FOLDER: &str = "scripts";
49
50const TABLE_PATCHES_FOLDER: &str = "table_patches";
51const TABLE_PROFILES_FOLDER: &str = "table_profiles";
52const TRANSLATIONS_LOCAL_FOLDER: &str = "translations_local";
53const TRANSLATIONS_REMOTE_FOLDER: &str = "translations_remote";
54
55#[macro_export]
69macro_rules! set_batch {
70 ($( $rtype:ident, $id:literal, $source:expr), *) => {
71 {
72 let mut set = SETTINGS.write().unwrap();
73 set.set_block_write(true);
74 $(
75 let _ = set.$rtype($id, $source);
76 )*
77 set.set_block_write(false);
78 let _ = set.write();
79 }
80 };
81}
82
83#[derive(Clone, Debug, Default, Serialize, Deserialize)]
99pub struct Settings {
100
101 #[serde(skip_serializing, skip_deserializing)]
104 pub block_write: bool,
105
106 pub bool: HashMap<String, bool>,
108 pub i32: HashMap<String, i32>,
110 pub f32: HashMap<String, f32>,
112 pub string: HashMap<String, String>,
115 pub raw_data: HashMap<String, Vec<u8>>,
117 pub vec_string: HashMap<String, Vec<String>>
119}
120
121impl Settings {
126
127 pub fn init(as_new: bool) -> Result<Self> {
136 let mut settings = if !as_new {
137 match Settings::read() {
138 Ok(settings) => settings,
139 Err(error) => {
140
141 if let Ok(config) = config_path() {
144 let settings_path = config.join(SETTINGS_FILE_NAME);
145 if settings_path.exists() {
146 let backup_path = config.join(format!("{SETTINGS_FILE_NAME}.bak"));
147 let _ = std::fs::copy(&settings_path, &backup_path);
148 }
149 }
150
151 rpfm_telemetry::warn!("Failed to read settings file, using defaults. Error: {error}");
152 Settings::default()
153 }
154 }
155 } else {
156 Settings::default()
157 };
158
159 settings.set_block_write(true);
160
161 settings.initialize_string(MYMOD_BASE_PATH, "");
162 settings.initialize_string(SECONDARY_PATH, "");
163
164 let supported_games = SupportedGames::default();
165 for game in &supported_games.games() {
166 let game_key = game.key();
167
168 let current_path = settings.string(game_key);
170 if current_path.is_empty() {
171 if current_path.contains("\\") {
172 let _ = settings.set_string(game_key, ¤t_path.replace("\\", "/"));
173 }
174
175 let game_path = if let Ok(Some(game_path)) = game.find_game_install_location() {
176 game_path.to_string_lossy().replace("\\", "/")
177 } else {
178 String::new()
179 };
180
181 if !game_path.is_empty() {
183 let _ = settings.set_string(game_key, &game_path);
184 } else {
185 settings.initialize_string(game_key, &game_path);
186 }
187 }
188
189 if game_key != KEY_EMPIRE &&
190 game_key != KEY_NAPOLEON &&
191 game_key != KEY_ARENA {
192
193 let ak_key = game_key.to_owned() + ASSEMBLY_KIT_SUFFIX;
195 let current_path = settings.string(&ak_key);
196
197 if current_path.is_empty() {
198 let ak_path = if let Ok(Some(ak_path)) = game.find_assembly_kit_install_location() {
199 ak_path.join("assembly_kit").to_string_lossy().replace("\\", "/")
200 } else {
201 String::new()
202 };
203
204 if current_path.contains("\\") {
206 let _ = settings.set_string(&ak_key, ¤t_path.replace("\\", "/"));
207 }
208
209 if !ak_path.is_empty() && game_key != KEY_SHOGUN_2 {
211 let _ = settings.set_string(&ak_key, &ak_path);
212 } else {
213 settings.initialize_string(&ak_key, &ak_path);
214 }
215 }
216 }
217 }
218
219 settings.initialize_bool(IMPORT_FROM_QT, false);
221
222 settings.initialize_string(DEFAULT_GAME, KEY_WARHAMMER_3);
224 settings.initialize_string(LANGUAGE, "English_en");
225 settings.initialize_string(THEME, THEME_OS);
226 settings.initialize_i32(AUTOSAVE_AMOUNT, 10);
228 settings.initialize_i32(AUTOSAVE_INTERVAL, 5);
229
230 settings.initialize_bool(START_MAXIMIZED, false);
241 settings.initialize_bool(ALLOW_EDITING_OF_CA_PACKFILES, false);
242 settings.initialize_bool(CHECK_UPDATES_ON_START, true);
243 settings.initialize_bool(CHECK_SCHEMA_UPDATES_ON_START, true);
244 settings.initialize_bool(CHECK_LUA_AUTOGEN_UPDATES_ON_START, true);
245 settings.initialize_bool(CHECK_OLD_AK_UPDATES_ON_START, true);
246 settings.initialize_bool(USE_LAZY_LOADING, true);
247 settings.initialize_bool(DISABLE_UUID_REGENERATION_ON_DB_TABLES, true);
248 settings.initialize_bool(PACKFILE_TREEVIEW_RESIZE_TO_FIT, false);
249 settings.initialize_bool(EXPAND_TREEVIEW_WHEN_ADDING_ITEMS, true);
250 settings.initialize_bool(USE_RIGHT_SIZE_MARKERS, false);
251 settings.initialize_bool(DISABLE_FILE_PREVIEWS, false);
252 settings.initialize_bool(INCLUDE_BASE_FOLDER_ON_ADD_FROM_FOLDER, true);
253 settings.initialize_bool(DELETE_EMPTY_FOLDERS_ON_DELETE, true);
254 settings.initialize_bool(AUTOSAVE_FOLDER_SIZE_WARNING_TRIGGERED, false);
255 settings.initialize_bool(IGNORE_GAME_FILES_IN_AK, false);
256 settings.initialize_bool(ENABLE_MULTIFOLDER_FILEPICKER, false);
257 settings.initialize_bool(ENABLE_PACK_CONTENTS_DRAG_AND_DROP, true);
258 settings.initialize_bool(CLEAN_UI, false);
259 settings.initialize_bool(SINGLE_PACK_MODE, true);
260 settings.initialize_bool(GLOBAL_SEARCH_COLLAPSE_RESULTS, false);
261
262 settings.initialize_bool(ADJUST_COLUMNS_TO_CONTENT, true);
264 settings.initialize_bool(EXTEND_LAST_COLUMN_ON_TABLES, true);
265 settings.initialize_bool(DISABLE_COMBOS_ON_TABLES, false);
266 settings.initialize_bool(TIGHT_TABLE_MODE, false);
267 settings.initialize_bool(TABLE_RESIZE_ON_EDIT, false);
268 settings.initialize_bool(TABLES_USE_OLD_COLUMN_ORDER, true);
269 settings.initialize_bool(TABLES_USE_OLD_COLUMN_ORDER_FOR_TSV, true);
270 settings.initialize_bool(ENABLE_LOOKUPS, true);
271 settings.initialize_bool(ENABLE_ICONS, true);
272 settings.initialize_bool(ENABLE_DIFF_MARKERS, true);
273 settings.initialize_bool(HIDE_UNUSED_COLUMNS, true);
274 settings.initialize_bool(SHOW_TABLE_TOOLBAR, false);
275
276 settings.initialize_bool(CHECK_FOR_MISSING_TABLE_DEFINITIONS, false);
278 settings.initialize_bool(ENABLE_DEBUG_MENU, false);
279 settings.initialize_bool(ENABLE_UNIT_EDITOR, false);
280 settings.initialize_bool(ENABLE_ESF_EDITOR, false);
281 settings.initialize_bool(USE_DEBUG_VIEW_UNIT_VARIANT, false);
282 settings.initialize_bool(ENABLE_RENDERER, true);
283
284 settings.initialize_bool(DIAGNOSTICS_TRIGGER_ON_OPEN, true);
286 settings.initialize_bool(DIAGNOSTICS_TRIGGER_ON_TABLE_EDIT, true);
287
288 settings.initialize_bool(ENABLE_USAGE_TELEMETRY, true);
290 settings.initialize_bool(ENABLE_CRASH_REPORTS, true);
291
292 if settings.string(ANONYMOUS_TELEMETRY_ID).is_empty() {
294 let _ = settings.set_string(ANONYMOUS_TELEMETRY_ID, &uuid::Uuid::new_v4().to_string());
295 }
296
297 settings.initialize_string(AI_API_URL, "https://api.openai.com/v1/chat/completions");
298 settings.initialize_string(AI_API_KEY, "");
299 settings.initialize_string(AI_MODEL, "gpt-4o-mini");
300 settings.initialize_string(DEEPL_API_KEY, "");
301
302 settings.initialize_vec_string(RECENT_FILE_LIST, &[]);
303
304 let opt = OptimizerOptions::default();
320 settings.initialize_bool(PACK_REMOVE_ITM_FILES, *opt.pack_remove_itm_files());
321 settings.initialize_bool(PACK_APPLY_COMPRESSION, *opt.pack_apply_compression());
322 settings.initialize_bool(PACK_REMOVE_DUPLICATED_FILES, *opt.pack_remove_duplicated_files());
323 settings.initialize_bool(DB_IMPORT_DATACORES_INTO_TWAD_KEY_DELETES, *opt.db_import_datacores_into_twad_key_deletes());
324 settings.initialize_bool(DB_OPTIMIZE_DATACORED_TABLES, *opt.db_optimize_datacored_tables());
325 settings.initialize_bool(TABLE_REMOVE_DUPLICATED_ENTRIES, *opt.table_remove_duplicated_entries());
326 settings.initialize_bool(TABLE_REMOVE_ITM_ENTRIES, *opt.table_remove_itm_entries());
327 settings.initialize_bool(TABLE_REMOVE_ITNR_ENTRIES, *opt.table_remove_itnr_entries());
328 settings.initialize_bool(TABLE_REMOVE_EMPTY_FILE, *opt.table_remove_empty_file());
329 settings.initialize_bool(TEXT_REMOVE_UNUSED_XML_MAP_FOLDERS, *opt.text_remove_unused_xml_map_folders());
330 settings.initialize_bool(TEXT_REMOVE_UNUSED_XML_PREFAB_FOLDER, *opt.text_remove_unused_xml_prefab_folder());
331 settings.initialize_bool(TEXT_REMOVE_AGF_FILES, *opt.text_remove_agf_files());
332 settings.initialize_bool(TEXT_REMOVE_MODEL_STATISTICS_FILES, *opt.text_remove_model_statistics_files());
333 settings.initialize_bool(PTS_REMOVE_UNUSED_ART_SETS, *opt.pts_remove_unused_art_sets());
334 settings.initialize_bool(PTS_REMOVE_UNUSED_VARIANTS, *opt.pts_remove_unused_variants());
335 settings.initialize_bool(PTS_REMOVE_EMPTY_MASKS, *opt.pts_remove_empty_masks());
336 settings.initialize_bool(PTS_REMOVE_EMPTY_FILE, *opt.pts_remove_empty_file());
337
338 settings.set_block_write(false);
339
340 settings.write()?;
341
342 Ok(settings)
343 }
344
345 pub fn read() -> Result<Self> {
350 let mut data = vec![];
351 let mut file = BufReader::new(File::open(config_path()?.join(SETTINGS_FILE_NAME))?);
352 file.read_to_end(&mut data)?;
353
354 serde_json::from_slice(&data).map_err(From::from)
355 }
356
357 pub fn write(&self) -> Result<()> {
359 if self.block_write {
360 return Ok(());
361 }
362
363 let mut file = BufWriter::new(File::create(config_path()?.join(SETTINGS_FILE_NAME))?);
364 file.write_all(serde_json::to_string_pretty(self)?.as_bytes()).map_err(From::from)
365 }
366
367 pub fn set_block_write(&mut self, status: bool) {
369 self.block_write = status;
370 }
371
372 pub fn bool(&self, setting: &str) -> bool {
374 self.bool.get(setting).copied().unwrap_or_default()
375 }
376
377 pub fn i32(&self, setting: &str) -> i32 {
379 self.i32.get(setting).copied().unwrap_or_default()
380 }
381
382 pub fn f32(&self, setting: &str) -> f32 {
384 self.f32.get(setting).copied().unwrap_or_default()
385 }
386
387 pub fn string(&self, setting: &str) -> String {
389 self.string.get(setting).map(|x| x.to_owned()).unwrap_or_default()
390 }
391
392 pub fn path_buf(&self, setting: &str) -> PathBuf {
396 self.string.get(setting).map(PathBuf::from).unwrap_or_default()
397 }
398
399 pub fn raw_data(&self, setting: &str) -> Vec<u8> {
401 self.raw_data.get(setting).map(|x| x.to_vec()).unwrap_or_default()
402 }
403
404 pub fn vec_string(&self, setting: &str) -> Vec<String> {
406 self.vec_string.get(setting).map(|x| x.to_vec()).unwrap_or_default()
407 }
408
409 pub fn set_bool(&mut self, setting: &str, value: bool) -> Result<()> {
411 self.bool.insert(setting.to_owned(), value);
412 self.write()
413 }
414
415 pub fn set_i32(&mut self, setting: &str, value: i32) -> Result<()> {
417 self.i32.insert(setting.to_owned(), value);
418 self.write()
419 }
420
421 pub fn set_f32(&mut self, setting: &str, value: f32) -> Result<()> {
423 self.f32.insert(setting.to_owned(), value);
424 self.write()
425 }
426
427 pub fn set_string(&mut self, setting: &str, value: &str) -> Result<()> {
429 self.string.insert(setting.to_owned(), value.to_owned());
430 self.write()
431 }
432
433 pub fn set_path_buf(&mut self, setting: &str, value: &Path) -> Result<()> {
436 self.string.insert(setting.to_owned(), value.to_string_lossy().to_string());
437 self.write()
438 }
439
440 pub fn set_raw_data(&mut self, setting: &str, value: &[u8]) -> Result<()> {
442 self.raw_data.insert(setting.to_owned(), value.to_vec());
443 self.write()
444 }
445
446 pub fn set_vec_string(&mut self, setting: &str, value: &[String]) -> Result<()> {
448 self.vec_string.insert(setting.to_owned(), value.to_vec());
449 self.write()
450 }
451
452 pub fn initialize_bool(&mut self, setting: &str, value: bool) {
455 if !self.bool.contains_key(setting) {
456 self.bool.insert(setting.to_owned(), value);
457 }
458 }
459
460 pub fn initialize_i32(&mut self, setting: &str, value: i32) {
462 if !self.i32.contains_key(setting) {
463 self.i32.insert(setting.to_owned(), value);
464 }
465 }
466
467 pub fn initialize_f32(&mut self, setting: &str, value: f32) {
469 if !self.f32.contains_key(setting) {
470 self.f32.insert(setting.to_owned(), value);
471 }
472 }
473
474 pub fn initialize_string(&mut self, setting: &str, value: &str) {
476 if !self.string.contains_key(setting) {
477 self.string.insert(setting.to_owned(), value.to_owned());
478 }
479 }
480
481 pub fn initialize_path_buf(&mut self, setting: &str, value: &Path) {
483 if !self.string.contains_key(setting) {
484 self.string.insert(setting.to_owned(), value.to_string_lossy().to_string());
485 }
486 }
487
488 pub fn initialize_raw_data(&mut self, setting: &str, value: &[u8]) {
490 if !self.raw_data.contains_key(setting) {
491 self.raw_data.insert(setting.to_owned(), value.to_vec());
492 }
493 }
494
495 pub fn initialize_vec_string(&mut self, setting: &str, value: &[String]) {
497 if !self.vec_string.contains_key(setting) {
498 self.vec_string.insert(setting.to_owned(), value.to_vec());
499 }
500 }
501
502 pub fn optimizer_options(&self) -> OptimizerOptions {
506 let mut options = OptimizerOptions::default();
507
508 options.set_pack_remove_itm_files(self.bool(PACK_REMOVE_ITM_FILES));
509 options.set_pack_apply_compression(self.bool(PACK_APPLY_COMPRESSION));
510 options.set_pack_remove_duplicated_files(self.bool(PACK_REMOVE_DUPLICATED_FILES));
511 options.set_db_import_datacores_into_twad_key_deletes(self.bool(DB_IMPORT_DATACORES_INTO_TWAD_KEY_DELETES));
512 options.set_db_optimize_datacored_tables(self.bool(DB_OPTIMIZE_DATACORED_TABLES));
513 options.set_table_remove_duplicated_entries(self.bool(TABLE_REMOVE_DUPLICATED_ENTRIES));
514 options.set_table_remove_itm_entries(self.bool(TABLE_REMOVE_ITM_ENTRIES));
515 options.set_table_remove_itnr_entries(self.bool(TABLE_REMOVE_ITNR_ENTRIES));
516 options.set_table_remove_empty_file(self.bool(TABLE_REMOVE_EMPTY_FILE));
517 options.set_text_remove_unused_xml_map_folders(self.bool(TEXT_REMOVE_UNUSED_XML_MAP_FOLDERS));
518 options.set_text_remove_unused_xml_prefab_folder(self.bool(TEXT_REMOVE_UNUSED_XML_PREFAB_FOLDER));
519 options.set_text_remove_agf_files(self.bool(TEXT_REMOVE_AGF_FILES));
520 options.set_text_remove_model_statistics_files(self.bool(TEXT_REMOVE_MODEL_STATISTICS_FILES));
521 options.set_pts_remove_unused_art_sets(self.bool(PTS_REMOVE_UNUSED_ART_SETS));
522 options.set_pts_remove_unused_variants(self.bool(PTS_REMOVE_UNUSED_VARIANTS));
523 options.set_pts_remove_empty_masks(self.bool(PTS_REMOVE_EMPTY_MASKS));
524 options.set_pts_remove_empty_file(self.bool(PTS_REMOVE_EMPTY_FILE));
525
526 options
527 }
528
529 pub fn assembly_kit_path(&self, game: &GameInfo) -> Result<PathBuf> {
531 let version = *game.raw_db_version();
532 match version {
533
534 2 | 1 => {
536 let mut base_path = self.path_buf(&format!("{}_assembly_kit", game.key()));
537 base_path.push("raw_data/db");
538 Ok(base_path)
539 }
540
541 0 => {
542 let base_path = old_ak_files_path()?.join(game.key());
543 Ok(base_path)
544 },
545
546 _ => Err(RLibError::AssemblyKitUnsupportedVersion(version).into())
548 }
549 }
550}
551
552pub fn default_config_path() -> Result<PathBuf> {
560
561 if cfg!(debug_assertions) {
563 std::env::current_dir().map_err(From::from)
564 } else {
565 match ProjectDirs::from(ORG_DOMAIN, ORG_NAME, APP_NAME) {
566 Some(proj_dirs) => Ok(proj_dirs.config_dir().to_path_buf()),
567 None => Err(anyhow!("Failed to get the config path."))
568 }
569 }
570}
571
572pub fn config_path() -> Result<PathBuf> {
576 match custom_config_path()? {
577 Some(path) => Ok(path),
578 None => default_config_path(),
579 }
580}
581
582pub fn custom_config_path() -> Result<Option<PathBuf>> {
586 let redirect_file = default_config_path()?.join(CONFIG_REDIRECT_FILE_NAME);
587 if !redirect_file.is_file() {
588 return Ok(None);
589 }
590
591 let raw = std::fs::read_to_string(&redirect_file)?;
592 let trimmed = raw.trim();
593 if trimmed.is_empty() {
594 Ok(None)
595 } else {
596 Ok(Some(PathBuf::from(trimmed)))
597 }
598}
599
600pub fn set_custom_config_path(path: Option<&Path>) -> Result<()> {
605
606 let default_path = default_config_path()?;
608 DirBuilder::new().recursive(true).create(&default_path)?;
609 let redirect_file = default_path.join(CONFIG_REDIRECT_FILE_NAME);
610
611 match path {
612 Some(path) if !path.as_os_str().is_empty() => {
613 DirBuilder::new().recursive(true).create(path)?;
614 std::fs::write(&redirect_file, path.to_string_lossy().as_bytes())?;
615 }
616 _ => if redirect_file.is_file() {
617 std::fs::remove_file(&redirect_file)?;
618 }
619 }
620
621 init_config_path()
622}
623
624pub fn error_path() -> Result<PathBuf> {
626 Ok(config_path()?.join("error"))
627}
628
629#[must_use = "Many things depend on this folder existing. So better check this worked."]
633pub fn init_config_path() -> Result<()> {
634
635 let config_path = config_path()?;
636 DirBuilder::new().recursive(true).create(&config_path)?;
637 DirBuilder::new().recursive(true).create(backup_autosave_path()?)?;
638 DirBuilder::new().recursive(true).create(error_path()?)?;
639 DirBuilder::new().recursive(true).create(schemas_path()?)?;
640 DirBuilder::new().recursive(true).create(table_patches_path()?)?;
641 DirBuilder::new().recursive(true).create(table_profiles_path()?)?;
642 DirBuilder::new().recursive(true).create(scripts_path()?)?;
643 DirBuilder::new().recursive(true).create(old_ak_files_path()?)?;
644
645 let games = SupportedGames::default();
647 for game in games.games_sorted() {
648 let path = table_patches_path().unwrap().join(game.schema_file_name());
649 if !path.is_file() {
650 let base: HashMap<String, DefinitionPatch> = HashMap::new();
651 let mut file = BufWriter::new(File::create(path)?);
652 let config = PrettyConfig::default();
653 file.write_all(to_string_pretty(&base, config)?.as_bytes())?;
654 }
655 }
656
657 Ok(())
671}
672
673pub fn schemas_path() -> Result<PathBuf> {
675 Ok(config_path()?.join(SCHEMA_FOLDER))
676}
677
678pub fn table_patches_path() -> Result<PathBuf> {
680 Ok(config_path()?.join(TABLE_PATCHES_FOLDER))
681}
682
683pub fn table_profiles_path() -> Result<PathBuf> {
686 Ok(config_path()?.join(TABLE_PROFILES_FOLDER))
687}
688
689pub fn lua_autogen_base_path() -> Result<PathBuf> {
691 Ok(config_path()?.join(LUA_AUTOGEN_FOLDER))
692}
693
694pub fn lua_autogen_game_path(game: &GameInfo) -> Result<PathBuf> {
696 match game.lua_autogen_folder() {
697 Some(folder) => Ok(config_path()?.join(LUA_AUTOGEN_FOLDER).join(folder)),
698 None => Err(anyhow!("Lua Autogen not available for this game."))
699 }
700}
701
702pub fn backup_autosave_path() -> Result<PathBuf> {
704 Ok(config_path()?.join("autosaves"))
705}
706
707pub fn dependencies_cache_path() -> Result<PathBuf> {
709 Ok(config_path()?.join(DEPENDENCIES_FOLDER))
710}
711
712pub fn scripts_path() -> Result<PathBuf> {
715 Ok(config_path()?.join(SCRIPTS_FOLDER))
716}
717
718pub fn old_ak_files_path() -> Result<PathBuf> {
722 Ok(config_path()?.join("old_ak_files"))
723}
724
725pub fn translations_local_path() -> Result<PathBuf> {
728 Ok(config_path()?.join(TRANSLATIONS_LOCAL_FOLDER))
729}
730
731pub fn translations_remote_path() -> Result<PathBuf> {
734 Ok(config_path()?.join(TRANSLATIONS_REMOTE_FOLDER))
735}
736
737pub fn clear_config_path(path: &Path) -> Result<()> {
743 if path.exists() && path.is_dir() && path.starts_with(config_path()?) {
744 std::fs::remove_dir_all(path)?;
745 init_config_path()
746 } else {
747 Err(anyhow!("Path is not a valid directory to clear or does not exist"))
748 }
749}