Add a function to write container.xml

1 files changed, 2 insertions, 1 deletions

diff --git a/src/main.rs b/src/main.rs
index 6e9e6dd..6376315 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -5,7 +5,7 @@ mod models;
 use std::collections::HashMap;
 use std::path::Path;
 
-use crate::epub::{download_all_files, write_mimetype};
+use crate::epub::{download_all_files, write_container_xml, write_mimetype};
 use crate::http_client::build_authenticated_client;
 use crate::models::{Chapter, EpubResponse, FileEntry, Paginated, SpineItem, TocNode};
 use anyhow::{Context, Result, ensure};
@@ -122,6 +122,7 @@ async fn main() -> Result<()> {
     let dest_root = Path::new(&dest_root);
     download_all_files(&client, &file_entries, dest_root).await?;
     write_mimetype(dest_root).await?;
+    write_container_xml(dest_root, &opf_entry.full_path).await?;
 
     // Sanity check: Every entry in spine exists in chapters.
     let chapters: HashMap<String, Chapter> =