store the template in the template files too

esp-rs · Feb 24, 2025 · 8b32d36 · 8b32d36
1 parent 25d3d23
commit 8b32d36
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 6 deletions.
diff --git a/build.rs b/build.rs
@@ -31,10 +31,6 @@ fn main() {
             continue;
         }
 
-        if path.file_name() == Some(std::ffi::OsStr::new("template.yaml")) {
-            continue;
-        }
-
         println!("{:?} {}", path, relative_path);
         let content = std::fs::read_to_string(path).unwrap();
 

diff --git a/src/main.rs b/src/main.rs
@@ -20,8 +20,13 @@ mod template_files;
 mod tui;
 
 static TEMPLATE: LazyLock<Template> = LazyLock::new(|| {
-    let options = include_str!("../template/template.yaml");
-    serde_yaml::from_str(options).unwrap()
+    serde_yaml::from_str(
+        template_files::TEMPLATE_FILES
+            .iter()
+            .find_map(|(k, v)| if *k == "template.yaml" { Some(v) } else { None })
+            .unwrap(),
+    )
+    .unwrap()
 });
 
 #[derive(Parser, Debug)]