Add number of pdf pages automation

3 files changed, 12 insertions, 5 deletions

diff --git a/Cargo.toml b/Cargo.toml
index 9bd7333..43d73e8 100644
--- a/Cargo.toml
+++ b/Cargo.toml
@@ -11,3 +11,4 @@ serde = {version = "1.0", features = ["derive"]}
 serde_json = "1.0"
 anyhow = "1.0"
 chrono = {version = "0.4", features = ["serde"]}
+pdf = "0.8"
diff --git a/src/works/stories.rs b/src/works/stories.rs
index c2b4b8f..510f30b 100644
--- a/src/works/stories.rs
+++ b/src/works/stories.rs
@@ -13,9 +13,10 @@ struct Story {
     title: String,
     description: String,
     path_to_document: String,
-    number_of_pages: u16,
     language: Language,
     #[serde(default)]
+    number_of_pages: u32,
+    #[serde(default)]
     last_update: NaiveDate,
 }
 
@@ -47,6 +48,7 @@ impl super::Works for Stories {
         let mut stories: Stories = serde_json::from_reader(std::io::BufReader::new(&file))?;
         for story in stories.0.iter_mut() {
             story.get_last_modified()?;
+            story.get_pages_num()?;
         }
         // add error context
         Ok(stories)
@@ -99,7 +101,7 @@ impl Story {
         anyhow::Ok(text).with_context(|| format!("Failed to extract text for {}", self.title))
     }
 
-    // get date of last modification of pdf file
+    // get date of last modification 
     fn get_last_modified(&mut self) -> anyhow::Result<chrono::NaiveDate> {
         use std::os::linux::fs::MetadataExt;
         let pdf_last_modification = std::fs::metadata(self.get_pdf_path())?.st_mtime();
@@ -108,6 +110,13 @@ impl Story {
         self.last_update = date;
         Ok(date)
     }
+
+    // get number of pages of pdf file
+    fn get_pages_num(&mut self) -> anyhow::Result<u32> {
+        let file = pdf::file::File::open(self.get_pdf_path())?;
+        self.number_of_pages = file.num_pages();
+        Ok(file.num_pages())
+    }
 }
 
 // language of story content
diff --git a/stories.json b/stories.json
index e9b10a9..fd01c0e 100644
--- a/stories.json
+++ b/stories.json
@@ -3,21 +3,18 @@
         "title": "Mit gutem Gewissen",
         "description": "Der gefährliche Zauber von Leon wird von seinem wegweisenden Freund durchschaut. Kann er ihn stoppen?",
         "path_to_document": "mit_gutem_gewissen.pdf",
-        "number_of_pages": 7,
         "language": "German"
     },
     {
         "title": "Das schweigende Spiegelbild",
         "description": "Das Spiegelbild verzweifelt in seiner Hilflosigkeit gegenüber seinem Ich.",
         "path_to_document": "das_schweigende_spiegelbild.pdf",
-        "number_of_pages": 2,
         "language": "German"
     },
     {
         "title": "The fallen dragons of Bavaria",
         "description": "In his new life in Germany Kazuyuki is needed by dragons.",
         "path_to_document": "the_fallen_dragons_of_bavaria.pdf",
-        "number_of_pages": 3,
         "language": "English"
     }
 ]