Gather stylesheet file entries for each chapter

This prepares them to be added to their xhtml content.

2 files changed, 13 insertions, 2 deletions

diff --git a/src/epub.rs b/src/epub.rs
index 406135b..946c9ee 100644
--- a/src/epub.rs
+++ b/src/epub.rs
@@ -91,6 +91,11 @@ pub fn create_epub_archive(
         .iter()
         .map(|e| (e.url.path(), &e.full_path))
         .collect::<HashMap<_, _>>();
+    // Prepare url to local path mapping to insert related assets based on their URL.
+    let url_to_file = file_entries
+        .iter()
+        .map(|e| (&e.url, e))
+        .collect::<HashMap<_, _>>();
 
     // Add the rest of the files according to file_entries.
     let options: FileOptions<()> =
@@ -100,7 +105,13 @@ pub fn create_epub_archive(
         let mut src_file = std::fs::File::open(entry.full_path.to_path(epub_root))?;
         let mut buffer = Vec::new();
         src_file.read_to_end(&mut buffer)?;
-        if chapters.contains_key(&entry.ourn) {
+        if let Some(chapter) = chapters.get(&entry.ourn) {
+            let stylesheet_entries = chapter
+                .related_assets
+                .stylesheets
+                .iter()
+                .filter_map(|u| url_to_file.get(u))
+                .collect::<Vec<_>>();
             let mut html = String::from_utf8(buffer)?;
             let chapter_dir = entry.full_path.parent().unwrap_or(RelativePath::new(""));
             for (url_path, local_path) in &url_path_to_local {
diff --git a/src/models.rs b/src/models.rs
index 77b3730..704ad84 100644
--- a/src/models.rs
+++ b/src/models.rs
@@ -30,7 +30,7 @@ pub struct Chapter {
 
 /// Sub-model of Chapter - related_assets.
 #[derive(Debug, Deserialize)]
-struct ChapRelAssets {
+pub struct ChapRelAssets {
     pub stylesheets: Vec<Url>,
 }