aipack 0.7.7-WIP

use super::{DirContext, RepoKind};
use crate::pack::PackRef;
use crate::support::files::list_dirs;
use crate::support::paths;
use crate::{Error, Result};
use simple_fs::SPath;

#[derive(Debug)]
pub struct PackDir {
	pub repo_kind: RepoKind,
	pub namespace: String,
	pub name: String,
	pub path: SPath,
}

impl PackDir {
	pub fn new(repo_kind: RepoKind, namespace: impl Into<String>, path: SPath) -> Self {
		let namespace = namespace.into();
		let name = path.name().to_string();
		Self {
			repo_kind,
			namespace,
			path,
			name,
		}
	}
}

impl PackDir {
	pub fn pretty_path(&self) -> String {
		let last_five = paths::path_last_components(&self.path, 5);
		let prefix = match self.repo_kind {
			RepoKind::WksCustom => "",
			RepoKind::BaseCustom => "~/",
			RepoKind::BaseInstalled => "~/",
		};
		format!("{}{}", prefix, last_five)
	}
}
impl std::fmt::Display for PackDir {
	fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
		write!(f, "{}@{}", self.namespace, self.name)
	}
}

/// Find all of the pack dirs, in order or precedencese
pub fn find_pack_dirs(dir_context: &DirContext, pack_ref: &PackRef) -> Result<Vec<PackDir>> {
	let aipack_paths = dir_context.aipack_paths();

	let repo_dirs = aipack_paths.get_pack_repo_dirs()?;

	let mut pack_dirs = Vec::new();

	for repo_dir in repo_dirs {
		let ns_dirs = list_dirs(repo_dir.path(), 1, true);
		let found_ns_dir = ns_dirs.into_iter().find(|ns_dir| ns_dir.name() == pack_ref.namespace);
		let repo_kind = repo_dir.kind;

		if let Some(ns_dir) = found_ns_dir {
			let aipack_dirs = list_dirs(&ns_dir, 1, true);
			let found_pack_dir = aipack_dirs.into_iter().find(|aipack_dir| aipack_dir.name() == pack_ref.name);
			if let Some(pack_dir_path) = found_pack_dir {
				// NOTE: Since direct match, just return this one
				pack_dirs.push(PackDir::new(repo_kind, &pack_ref.name, pack_dir_path));
				break;
			}
		}
	}

	Ok(pack_dirs)
}

pub fn find_to_run_pack_dir(dir_context: &DirContext, pack_ref: &PackRef) -> Result<PackDir> {
	// -- the the pack dir and reverse it
	let mut reversed_pack_dir = find_pack_dirs(dir_context, pack_ref)?;
	// So that we get the first last.
	reversed_pack_dir.reverse();

	// -- Get the pack dir
	let Some(pack_dir) = reversed_pack_dir.pop() else {
		return Err(Error::custom(format!("No aipack matches for {pack_ref}.")));
	};

	// -- check that theey all have the same namespace, in this case ok to return the pack_dir above
	if !reversed_pack_dir.is_empty() {
		for other_pack_dir in reversed_pack_dir {
			if other_pack_dir.namespace != pack_dir.namespace {
				// Get the propert list
				let pack_dirs = find_pack_dirs(dir_context, pack_ref)?;
				// -- in case > 1, for now, no support
				return Err(Error::custom(format!(
					"{pack_ref} matches multiple AI packs across different namespaces.\n\nRun aip run ... with one of the full AI pack references below:\n\n{}\n",
					pack_dirs.iter().map(|p| p.to_string()).collect::<Vec<String>>().join("\n")
				)));
			}
		}
	}

	Ok(pack_dir)
}

/// Lookup for pack_irs for optional namspace and pack_name
///
/// IMPORTANT: Should be used only for `aip list ...`
///
/// - if no namespace, then, will return all of the matching PackDir with this pack_name
/// - if a namespace, will return only the first matching one (following the custom/installed preferences)
pub fn lookup_pack_dirs(dir_context: &DirContext, ns: Option<&str>, pack_name: Option<&str>) -> Result<Vec<PackDir>> {
	let aipack_paths = dir_context.aipack_paths();

	let repo_dirs = aipack_paths.get_pack_repo_dirs()?;

	let mut pack_dirs = Vec::new();

	for repo_dir in repo_dirs {
		let repo_kind = repo_dir.kind;
		match (ns, pack_name) {
			(Some(ns_name), Some(pack_name)) => {
				let ns_dirs = list_dirs(repo_dir.path(), 1, true);
				let found_ns_dir = ns_dirs.into_iter().find(|ns_dir| ns_dir.name() == ns_name);

				if let Some(ns_dir) = found_ns_dir {
					let aipack_dirs = list_dirs(&ns_dir, 1, true);
					let found_pack_dir = aipack_dirs.into_iter().find(|aipack_dir| aipack_dir.name() == pack_name);
					if let Some(aipack_dir) = found_pack_dir {
						// NOTE: Since direct match, just return this one
						pack_dirs.push(PackDir::new(repo_kind, ns_name, aipack_dir));
						break;
					}
				}
			}
			(ns, pack_name) => {
				let pack_dir_paths = list_dirs(repo_dir.path(), 2, true);
				for pack_path in pack_dir_paths {
					let path_pack_name = pack_path.name();
					let path_ns = pack_path.parent_name();

					// compute if we should include or not (if input is none, then, match, hence the unwrap_or true)
					let pass = match (ns, pack_name) {
						(None, None) => true,
						(Some(ns), None) => ns == path_ns,
						(None, Some(pack_name)) => pack_name == path_pack_name,
						(Some(ns), Some(pack_name)) => ns == path_ns && pack_name == path_pack_name,
					};

					if pass {
						let path_ns = path_ns.to_string();
						pack_dirs.push(PackDir::new(repo_kind, path_ns, pack_path));
					}
				}
			}
		}
	}

	Ok(pack_dirs)
}

// region:    --- Tests

#[cfg(test)]
mod tests {
	type Result<T> = core::result::Result<T, Box<dyn std::error::Error>>; // For tests.

	use super::*;
	use crate::_test_support::assert_contains;
	use crate::runtime::Runtime;
	use crate::support::AsStrsExt;
	use crate::support::paths::path_last_components;

	/// Note - In this tests, we do not use the PackDir::pretty() to not have to change those test if pretty changes.

	#[tokio::test]
	async fn test_pack_dir_lookup_pack_dirs_all() -> Result<()> {
		// -- Setup & Fixtures
		let runtime = Runtime::new_test_runtime_sandbox_01()?;
		let expected = vec![
			"ns_a@pack_a_1 - .aipack/pack/custom/ns_a/pack_a_1",
			"ns_a@pack_a_2 - .aipack/pack/custom/ns_a/pack_a_2",
			"ns_b@pack_b_1 - .aipack/pack/custom/ns_b/pack_b_1",
			"ns_b@pack_b_1 - .aipack-base/pack/custom/ns_b/pack_b_1",
			"ns_b@pack_b_2 - .aipack-base/pack/custom/ns_b/pack_b_2",
			"ns_b@pack_b_2 - .aipack-base/pack/installed/ns_b/pack_b_2",
			"ns_d@pack_d_1 - .aipack-base/pack/installed/ns_d/pack_d_1",
		];

		// -- Exec
		let pack_dirs = lookup_pack_dirs(runtime.dir_context(), None, None)?;

		// -- Check
		let pack_dir_refs = pack_dir_into_strs(pack_dirs);
		let pack_dir_refs = pack_dir_refs.x_as_strs();
		for pack_dir_ref in pack_dir_refs {
			assert_contains(&expected, pack_dir_ref);
		}

		Ok(())
	}

	#[tokio::test]
	async fn test_pack_dir_lookup_pack_dirs_ns() -> Result<()> {
		// -- Setup & Fixtures
		let runtime = Runtime::new_test_runtime_sandbox_01()?;
		let expected = vec![
			"ns_b@pack_b_1 - .aipack/pack/custom/ns_b/pack_b_1",
			"ns_b@pack_b_1 - .aipack-base/pack/custom/ns_b/pack_b_1",
			"ns_b@pack_b_2 - .aipack-base/pack/custom/ns_b/pack_b_2",
			"ns_b@pack_b_2 - .aipack-base/pack/installed/ns_b/pack_b_2",
		];

		// -- Exec
		let pack_dirs = lookup_pack_dirs(runtime.dir_context(), Some("ns_b"), None)?;

		// -- Check
		let pack_dir_refs = pack_dir_into_strs(pack_dirs);
		let pack_dir_refs = pack_dir_refs.x_as_strs();
		for pack_dir_ref in pack_dir_refs {
			assert_contains(&expected, pack_dir_ref);
		}

		Ok(())
	}

	#[tokio::test]
	async fn test_pack_dir_lookup_pack_dirs_pack_name() -> Result<()> {
		// -- Setup & Fixtures
		let runtime = Runtime::new_test_runtime_sandbox_01()?;
		let expected = vec![
			"ns_b@pack_b_1 - .aipack/pack/custom/ns_b/pack_b_1",
			"ns_b@pack_b_1 - .aipack-base/pack/custom/ns_b/pack_b_1",
		];

		// -- Exec
		let pack_dirs = lookup_pack_dirs(runtime.dir_context(), None, Some("pack_b_1"))?;

		// -- Check
		let pack_dir_refs = pack_dir_into_strs(pack_dirs);
		let pack_dir_refs = pack_dir_refs.x_as_strs();
		for pack_dir_ref in pack_dir_refs {
			assert_contains(&expected, pack_dir_ref);
		}

		Ok(())
	}

	// region:    --- Support

	fn pack_dir_into_strs(pack_dirs: Vec<PackDir>) -> Vec<String> {
		pack_dirs
			.into_iter()
			.map(|p| format!("{}@{} - {}", p.namespace, p.name, path_last_components(&p.path, 5)))
			.collect::<Vec<_>>()
	}
	// endregion: --- Support
}

// endregion: --- Tests