diff options
Diffstat (limited to 'server/data/src/index.rs')
-rw-r--r-- | server/data/src/index.rs | 94 |
1 files changed, 94 insertions, 0 deletions
diff --git a/server/data/src/index.rs b/server/data/src/index.rs new file mode 100644 index 00000000..a5ec8d97 --- /dev/null +++ b/server/data/src/index.rs @@ -0,0 +1,94 @@ +/* + Hurry Curry! - a game about cooking + Copyright (C) 2025 Hurry Curry! Contributors + + This program is free software: you can redistribute it and/or modify + it under the terms of the GNU Affero General Public License as published by + the Free Software Foundation, version 3 of the License only. + + This program is distributed in the hope that it will be useful, + but WITHOUT ANY WARRANTY; without even the implied warranty of + MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + GNU Affero General Public License for more details. + + You should have received a copy of the GNU Affero General Public License + along with this program. If not, see <https://www.gnu.org/licenses/>. + +*/ + +use anyhow::{Context, Result, anyhow, bail}; +use hurrycurry_protocol::{Gamedata, MapMetadata}; +use serde::Deserialize; +use std::{ + collections::{HashMap, HashSet}, + fs::{File, read_to_string}, + path::PathBuf, + str::FromStr, + sync::Mutex, +}; + +use crate::{MapDecl, Serverdata, book::book, build_data}; + +#[derive(Debug, Deserialize, Default)] +pub struct DataIndex { + pub maps: HashMap<String, MapMetadata>, + pub recipes: HashSet<String>, +} + +pub static DATA_DIR: Mutex<Option<PathBuf>> = Mutex::new(None); +fn data_dir() -> PathBuf { + DATA_DIR + .lock() + .unwrap() + .to_owned() + .unwrap_or_else(|| PathBuf::from_str("data").unwrap()) +} + +impl DataIndex { + pub fn load() -> Result<Self> { + let mut s = Self::default(); + s.reload()?; + Ok(s) + } + pub fn reload(&mut self) -> Result<()> { + *self = serde_yml::from_reader(File::open(data_dir().join("index.yaml"))?)?; + Ok(()) + } + + pub fn read_map(&self, name: &str) -> Result<String> { + // Scary! + if name.contains("..") || name.starts_with("/") || name.contains("//") { + bail!("illegal map path"); + } + let path = data_dir().join(format!("maps/{name}.yaml")); + Ok(read_to_string(path)?) + } + pub fn read_recipes(&self, name: &str) -> Result<String> { + if !self.recipes.contains(name) { + bail!("unknown recipes: {name:?}"); + } + let path = data_dir().join(format!("recipes/{name}.yaml")); + Ok(read_to_string(path)?) + } + pub fn generate(&self, map: &str) -> Result<(Gamedata, Serverdata)> { + let map_in: MapDecl = serde_yml::from_str( + &self + .read_map(map) + .context(anyhow!("Failed to read map file ({map})"))?, + ) + .context(anyhow!("Failed to parse map file ({map})"))?; + let recipes_in = serde_yml::from_str( + &self + .read_recipes(map_in.recipes.as_deref().unwrap_or("default")) + .context("Failed read recipe file")?, + ) + .context("Failed to parse recipe file")?; + + build_data(&self.maps, map.to_string(), map_in, recipes_in) + } + pub fn generate_with_book(&self, map: &str) -> Result<(Gamedata, Serverdata)> { + let (gd, mut sd) = self.generate(map)?; + sd.book = book(&gd, &sd).context("within book")?; + Ok((gd, sd)) + } +} |