mirror of
https://github.com/redstrate/Physis.git
synced 2025-04-20 03:37:47 +00:00
Add tests for ConfigFile, fix writing and add modification functions
This commit is contained in:
parent
0621ce03d7
commit
0a6209dfd5
3 changed files with 86 additions and 11 deletions
BIN
resources/tests/FFXIV.cfg
Normal file
BIN
resources/tests/FFXIV.cfg
Normal file
Binary file not shown.
BIN
resources/tests/FFXIV.modified.cfg
Normal file
BIN
resources/tests/FFXIV.modified.cfg
Normal file
Binary file not shown.
97
src/cfg.rs
97
src/cfg.rs
|
@ -10,7 +10,7 @@ use crate::gamedata::MemoryBuffer;
|
|||
#[derive(Debug)]
|
||||
pub struct ConfigMap {
|
||||
/// A map of setting name to value.
|
||||
pub keys: HashMap<String, String>,
|
||||
pub keys: Vec<(String, String)>,
|
||||
}
|
||||
|
||||
/// Represents a config file, which is made up of categories and settings. Categories may have zero to one settings.
|
||||
|
@ -36,20 +36,17 @@ impl ConfigFile {
|
|||
let mut current_category: Option<String> = None;
|
||||
|
||||
for (_, line) in reader.lines().enumerate() {
|
||||
// now parse the line!
|
||||
|
||||
let unwrap = line.unwrap();
|
||||
if !unwrap.is_empty() {
|
||||
if !unwrap.is_empty() && unwrap != "\0" {
|
||||
if unwrap.contains('<') || unwrap.contains('>') {
|
||||
let name = &unwrap[1..unwrap.len() - 1];
|
||||
println!("{}", name);
|
||||
current_category = Some(String::from(name));
|
||||
cfg.categories.push(String::from(name));
|
||||
} else {
|
||||
let parts = unwrap.split_once('\t').unwrap();
|
||||
cfg.settings.entry(current_category.clone().unwrap()).or_insert_with(|| ConfigMap{ keys: HashMap::new() });
|
||||
cfg.settings.entry(current_category.clone().unwrap()).or_insert_with(|| ConfigMap{ keys: Vec::new() });
|
||||
|
||||
cfg.settings.get_mut(¤t_category.clone().unwrap()).unwrap().keys.insert(parts.0.to_string(), parts.1.to_string());
|
||||
cfg.settings.get_mut(¤t_category.clone().unwrap()).unwrap().keys.push((parts.0.to_string(), parts.1.to_string()));
|
||||
}
|
||||
}
|
||||
}
|
||||
|
@ -66,18 +63,96 @@ impl ConfigFile {
|
|||
let mut writer = BufWriter::new(cursor);
|
||||
|
||||
for category in &self.categories {
|
||||
writer.write_all(format!("\n<{}>", category).as_ref()).ok()?;
|
||||
writer.write_all(format!("\r\n<{}>\r\n", category).as_ref()).ok()?;
|
||||
|
||||
if self.settings.contains_key(category) {
|
||||
for key in &self.settings[category].keys {
|
||||
writer.write_all(format!("\n{}\t{}", key.0, key.1).as_ref()).ok()?;
|
||||
writer.write_all(format!("{}\t{}\r\n", key.0, key.1).as_ref()).ok()?;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
writer.write_all(b"\n").ok()?;
|
||||
}
|
||||
writer.write_all(b"\0").ok()?;
|
||||
}
|
||||
|
||||
|
||||
Some(buffer)
|
||||
}
|
||||
|
||||
pub fn has_key(&self, select_key: &str) -> bool {
|
||||
for (category, keys) in &self.settings {
|
||||
for (key, value) in &keys.keys {
|
||||
if select_key == key {
|
||||
return true;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
false
|
||||
}
|
||||
|
||||
pub fn has_category(&self, select_category: &str) -> bool {
|
||||
for (category, keys) in &self.settings {
|
||||
if select_category == category {
|
||||
return true;
|
||||
}
|
||||
}
|
||||
|
||||
false
|
||||
}
|
||||
|
||||
pub fn set_value(&mut self, select_key: &str, new_value: &str) {
|
||||
for (category, keys) in &mut self.settings {
|
||||
for (key, value) in &mut keys.keys {
|
||||
if select_key == key {
|
||||
*value = new_value.to_string();
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use std::fs::{read, write};
|
||||
use std::path::PathBuf;
|
||||
|
||||
use super::*;
|
||||
|
||||
fn common_setup() -> ConfigFile {
|
||||
let mut d = PathBuf::from(env!("CARGO_MANIFEST_DIR"));
|
||||
d.push("resources/tests");
|
||||
d.push("FFXIV.cfg");
|
||||
|
||||
ConfigFile::from_existing(&read(d).unwrap()).unwrap()
|
||||
}
|
||||
|
||||
fn common_setup_modified() -> MemoryBuffer {
|
||||
let mut d = PathBuf::from(env!("CARGO_MANIFEST_DIR"));
|
||||
d.push("resources/tests");
|
||||
d.push("FFXIV.modified.cfg");
|
||||
|
||||
read(d).unwrap()
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn basic_parsing() {
|
||||
let cfg = common_setup();
|
||||
|
||||
assert!(cfg.has_key("TextureFilterQuality"));
|
||||
assert!(cfg.has_category("Cutscene Settings"));
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn basic_writing() {
|
||||
let mut cfg = common_setup();
|
||||
let modified_cfg = common_setup_modified();
|
||||
|
||||
cfg.set_value("CutsceneMovieOpening", "1");
|
||||
|
||||
let cfg_buffer = cfg.write_to_buffer().unwrap();
|
||||
|
||||
assert_eq!(modified_cfg, cfg_buffer);
|
||||
}
|
||||
}
|
||||
|
|
Loading…
Add table
Reference in a new issue