Universalize method to create context from works and pass entire works object

2 files changed, 9 insertions, 13 deletions

diff --git a/src/works/mod.rs b/src/works/mod.rs
index 3dfd4a2..f062565 100644
--- a/src/works/mod.rs
+++ b/src/works/mod.rs
@@ -1,8 +1,8 @@
-use serde::Deserialize;
+use serde::{Deserialize, Serialize};
 
 pub mod stories;
 
-pub trait Works: for<'a> Deserialize<'a> {
+pub trait Works: for<'a> Deserialize<'a> + Serialize {
     fn new_from_file(file: std::fs::File) -> Self {
         match serde_json::from_reader(std::io::BufReader::new(file)) {
             Ok(work) => work,
@@ -10,5 +10,9 @@ pub trait Works: for<'a> Deserialize<'a> {
         }
     }
 
-    fn create_tera_context(&self) -> tera::Context;
+    fn create_tera_context(&self) -> tera::Context {
+        let mut context = tera::Context::new();
+        context.insert("works", &self);
+        context
+    }
 }
diff --git a/src/works/stories.rs b/src/works/stories.rs
index 450f3bc..73d89ad 100644
--- a/src/works/stories.rs
+++ b/src/works/stories.rs
@@ -1,6 +1,6 @@
 use serde::{Deserialize, Serialize};
 
-#[derive(Debug, Deserialize)]
+#[derive(Debug, Deserialize, Serialize)]
 pub struct Stories(pub Vec<Story>);
 
 #[derive(Debug, Deserialize, Serialize)]
@@ -13,15 +13,7 @@ pub struct Story {
     pub language: Language,
 }
 
-impl super::Works for Stories {
-    fn create_tera_context(&self) -> tera::Context {
-        let mut context = tera::Context::new();
-        for story in self.0.iter() {
-            context.insert(format!("story-{}", story.id), &story);
-        }
-        context
-    }
-}
+impl super::Works for Stories {}
 
 #[derive(Deserialize, Serialize, Debug)]
 pub enum Language {