feat: initial, basic implementation

2 files changed, 112 insertions, 0 deletions

diff --git a/src/error.rs b/src/error.rs
new file mode 100644
index 0000000..04541a5
--- /dev/null
+++ b/src/error.rs
@@ -0,0 +1,33 @@
+use std::result::Result as StdResult;
+
+use fluent::FluentResource;
+
+pub type Result<T> = StdResult<T, Error>;
+
+pub enum Error {
+	IoError(String)
+}
+
+impl From<std::io::Error> for Error {
+	fn from(value: std::io::Error) -> Self {
+		todo!()
+	}
+}
+
+impl From<(FluentResource, Vec<fluent_syntax::parser::ParserError>)> for Error {
+	fn from(value: (FluentResource, Vec<fluent_syntax::parser::ParserError>)) -> Self {
+		todo!()
+	}
+}
+
+impl From<Vec<fluent::FluentError>> for Error {
+	fn from(value: Vec<fluent::FluentError>) -> Self {
+		todo!()
+	}
+}
+
+impl From<unic_langid::LanguageIdentifierError> for Error {
+	fn from(value: unic_langid::LanguageIdentifierError) -> Self {
+		todo!()
+	}
+}
diff --git a/src/lib.rs b/src/lib.rs
new file mode 100644
index 0000000..49cc13a
--- /dev/null
+++ b/src/lib.rs
@@ -0,0 +1,79 @@
+use std::{collections::HashMap, sync::Arc};
+use fluent::{bundle::FluentBundle, FluentResource};
+use intl_memoizer::concurrent::IntlLangMemoizer;
+use unic_langid::LanguageIdentifier;
+use crate::error::Result;
+
+pub mod error;
+
+type TypedFluentBundle = FluentBundle<Arc<FluentResource>, IntlLangMemoizer>;
+pub struct Localiser {
+	bundles: HashMap<LanguageIdentifier, TypedFluentBundle>,
+	default_language: LanguageIdentifier
+}
+
+impl Localiser {
+	pub fn try_load(path: String, default_language: String) -> Result<Self> {
+		let mut bundles = HashMap::new();
+		let paths = std::fs::read_dir(path)?
+			.filter_map(|res| res.ok())
+			.map(|dir_entry| dir_entry.path())
+			.filter_map(|path| {
+				if path.extension().map_or(false, |ext| ext == "ftl") || path.is_dir() {
+					Some(path)
+				} else {
+					None
+				}
+			}).collect::<Vec<_>>();
+
+		//TODO load default first and call bundle.add_resource_overriding(default_bundle) on others
+		let default_language = default_language.parse::<LanguageIdentifier>()?;
+
+		for path in paths {
+			// validate filename as language code
+			let language_code = path.file_stem()
+				.and_then(|f| f.to_str())
+				.map(|f| f.parse::<LanguageIdentifier>())
+				.and_then(|id| match id {
+					Ok(id) => Some(id),
+					Err(_) => None
+				});
+
+			if language_code.is_none() {
+				continue;
+			}
+
+			let language_code = language_code.unwrap();
+
+			let mut bundle: TypedFluentBundle = fluent::bundle::FluentBundle::new_concurrent(vec![language_code.clone()]);
+			if path.is_dir() { //is a directory
+				for res in Self::path_to_resources(&path)? {
+					bundle.add_resource(res)?;
+				}
+			} else { //is a single file
+				bundle.add_resource(Self::file_to_resource(&path)?)?;
+			}
+
+			bundles.insert(language_code, bundle);
+		}
+
+		Ok(Self {
+			bundles,
+			default_language
+		})
+	}
+
+	fn path_to_resources(path: &std::path::PathBuf) -> Result<Vec<Arc<FluentResource>>> {
+		let mut res = Vec::new();
+		for entry in walkdir::WalkDir::new(path).follow_links(true).into_iter().filter_map(|e| e.ok()) {
+			let entry_path = entry.path().to_path_buf();
+			res.push(Self::file_to_resource(&entry_path)?);
+		}
+		Ok(res)
+	}
+
+	fn file_to_resource(path: &std::path::PathBuf) -> Result<Arc<FluentResource>> {
+		let content = std::fs::read_to_string(path)?;
+		Ok(Arc::new(FluentResource::try_new(content)?))
+	}
+}