Set rust

[rbdr/blog] / src / generator / txt.rs

diff --git a/src/generator/txt.rs b/src/generator/txt.rs
index ec96420c98baa3792cd0903c365d7fc3426a02ab..f505480489b1533643424fec9d8d6f46185f56ca 100644 (file)
--- a/src/generator/txt.rs
+++ b/src/generator/txt.rs
@@ -1,12 +1,18 @@
-use std::io::Result;
+use std::fs::write;
+use std::io::{Error, ErrorKind::Other, Result};
 use std::path::PathBuf;
 use crate::template::{find, parse, TemplateContext};
 
-pub fn generate(_: &PathBuf, template_directory: &PathBuf, _: &PathBuf, context: &TemplateContext) -> Result<()> {
-    match find(template_directory, "index.txt") {
+const FILENAME: &str = "index.txt";
+
+pub fn generate(_: &PathBuf, template_directory: &PathBuf, target: &PathBuf, context: &TemplateContext) -> Result<()> {
+    match find(template_directory, FILENAME) {
         Some(template) => {
-            let parsed_template = parse(&template);
-            println!("TOKEN {}", parsed_template.render(context));
+            let parsed_template = parse(&template)
+                .ok_or_else(|| Error::new(Other, "Unable to parse TXT template"))?;
+            let rendered_template = parsed_template.render(context)?;
+            let location = target.join(FILENAME);
+            write(location, rendered_template)?;
         },
         None => {}
     }