geode project add

Author: camila314

src/project.rs

@@ -1,3 +1,7 @@
+use serde_json::json;
+use clap::ValueEnum;
+use serde::Deserialize;
+use crate::logging::ask_value;
 use crate::mod_file::{PlatformName, ToGeodeString};
 use crate::util::mod_file::DependencyImportance;
 use crate::{done, fail, fatal, index, info, warn, NiceUnwrap};
@@ -18,6 +22,15 @@ use std::{
 	fs,
 	path::{Path, PathBuf},
 };
+use serde_json::Value;
+
+#[derive(Debug, Deserialize, Hash, PartialEq, Eq, Clone, Copy, ValueEnum)]
+#[serde(rename_all = "lowercase")]
+pub enum ResourceType {
+	Sprite,
+	Font,
+	File
+}
 
 #[derive(Subcommand, Debug)]
 #[clap(rename_all = "kebab-case")]
@@ -51,6 +64,13 @@ pub enum Project {
 		#[clap(long, num_args(0..))]
 		externals: Vec<String>,
 	},
+
+	/// Add a resource to the mod.json file
+	Add {
+		/// Type of resource to add
+		resource: ResourceType,
+		files: Vec<PathBuf>
+	}
 }
 
 fn find_build_directory(root: &Path) -> Option<PathBuf> {
@@ -517,6 +537,80 @@ pub fn check_dependencies(
 	}
 }
 
+fn add_resource(dir: &Path, resource: ResourceType, files: Vec<PathBuf>) {
+	let mut mod_json: HashMap<String, Value> = serde_json::from_reader(fs::File::open(dir.join("mod.json")).ok().nice_unwrap("Must be inside a project with a mod.json"))
+		.nice_unwrap("Unable to read mod.json");
+
+
+	let mut do_thing = |name: &str, othername: &str| {
+		let resource = mod_json.get("resources")
+			.and_then(|x| x.get(name))
+			.and_then(|x| x.as_array())
+			.unwrap_or(&vec![])
+			.clone();
+
+		let mut new_resource: Vec<Value> = resource.into_iter().chain(files.clone().into_iter().filter_map(|x| {
+			if !x.exists() { 
+				warn!("{} {} does not exist", othername, x.display());
+				return None
+			} else {
+				Some(Value::String(x.as_os_str().to_str().unwrap().to_string()))
+			}
+		})).collect();
+
+		let mut duplicates: Vec<_> = new_resource.iter().filter(|x| new_resource.iter().filter(|y| y == x).count() > 1).collect();
+		duplicates.dedup();
+		duplicates.into_iter().for_each(|x| warn!("Duplicate {}: {}", othername, x));
+
+		new_resource.dedup();
+
+		*mod_json.entry("resources".to_string()).or_insert(json!({}))
+			.as_object_mut().nice_unwrap("resources is not an object")
+			.entry(name.to_string()).or_insert(json!([]))
+			.as_array_mut().nice_unwrap(&format!("{} is not an array", name)) = new_resource;
+	};
+
+	match resource {
+		ResourceType::Sprite => do_thing("sprites", "Sprite"),
+		ResourceType::File => do_thing("files", "File"),
+
+		ResourceType::Font => {
+			let fonts = mod_json.get("resources")
+				.and_then(|x| x.get("fonts"))
+				.and_then(|x| x.as_array())
+				.unwrap_or(&vec![])
+				.clone();
+
+			let mut new_fonts: Vec<Value> = fonts.into_iter().chain(files.into_iter().filter_map(|x| {
+				if !x.exists() { 
+					warn!("Font {} does not exist", x.display());
+					return None
+				} else {
+					Some(json!({
+						"path": x.as_os_str().to_str().unwrap().to_string(),
+						"size": ask_value("Font Size", None, true).parse::<u32>().ok().nice_unwrap("Invalid font size!")
+					}))
+				}
+			})).collect();
+
+			let mut duplicates: Vec<_> = new_fonts.iter().filter_map(|x| x.get("path")).filter(|x| new_fonts.iter().filter(|y| y.get("path").map(|y| y == *x).unwrap_or(false)).count() > 1).collect();
+			duplicates.dedup();
+			duplicates.into_iter().for_each(|x| warn!("Duplicate Font: {}", x));
+
+			new_fonts.dedup();
+
+			*mod_json.entry("resources".to_string()).or_insert(json!({}))
+				.as_object_mut().nice_unwrap("resources is not an object")
+				.entry("fonts".to_string()).or_insert(json!([]))
+				.as_array_mut().nice_unwrap("fonts is not an array") = new_fonts;
+		}
+};
+
+	fs::write(dir.join("mod.json"), serde_json::to_string_pretty(&mod_json).unwrap()).nice_unwrap("Failed to save mod.json");
+
+	done!("Resource added to mod.json");
+}
+
 pub fn subcommand(cmd: Project) {
 	match cmd {
 		Project::New { path } => template::build_template(path),
@@ -531,5 +625,10 @@ pub fn subcommand(cmd: Project) {
 			platform,
 			externals,
 		),
+		Project::Add { resource, files } => add_resource(
+			&std::env::current_dir().unwrap(),
+			resource,
+			files
+		),
 	}
 }