rust-embed

rust macro which loads files into the rust binary at compile time during release and loads the file from the fs during dev.

a283b903 — Tangent 128

Impl the RustEmbed trait on asset types

Files changed (2) hide show

impl/src/lib.rs CHANGED Viewed

@@ -44,6 +44,14 @@ fn generate_assets(ident: &syn::Ident, folder_path: String) -> quote::Tokens {
               get_files(String::from(#folder_path)).map(|e| std::borrow::Cow::from(e.rel_path))
           }
       }
+      impl rust_embed::RustEmbed for #ident {
+        fn get(&self, file_path: &str) -> Option<std::borrow::Cow<'static, [u8]>> {
+          #ident::get(file_path)
+        }
+        fn iter(&self) -> Box<dyn Iterator<Item = std::borrow::Cow<'static, str>>> {
+          Box::new(#ident::iter())
+        }
+      }
   }
 }
@@ -80,6 +88,14 @@ fn generate_assets(ident: &syn::Ident, folder_path: String) -> quote::Tokens {
               items.iter().map(|x| std::borrow::Cow::from(*x))
           }
       }
+      impl rust_embed::RustEmbed for #ident {
+        fn get(&self, file_path: &str) -> Option<std::borrow::Cow<'static, [u8]>> {
+          #ident::get(file_path)
+        }
+        fn iter(&self) -> Box<dyn Iterator<Item = std::borrow::Cow<'static, str>>> {
+          Box::new(#ident::iter())
+        }
+      }
   }
 }

tests/lib.rs CHANGED Viewed

@@ -31,3 +31,31 @@ fn iter_works() {
   }
   assert_eq!(num_files, 6);
 }
+#[test]
+fn trait_works_generic() {
+  trait_works_generic_helper(&Asset);
+}
+fn trait_works_generic_helper(folder: &impl rust_embed::RustEmbed) {
+  let mut num_files = 0;
+  for file in folder.iter() {
+    assert!(folder.get(file.as_ref()).is_some());
+    num_files += 1;
+  }
+  assert_eq!(num_files, 6);
+  assert!(folder.get("gg.html").is_none(), "gg.html should not exist");
+}
+#[test]
+fn trait_works_object() {
+  trait_works_object_helper(&Asset);
+}
+fn trait_works_object_helper(folder: &dyn rust_embed::RustEmbed) {
+  let mut num_files = 0;
+  for file in folder.iter() {
+    assert!(folder.get(file.as_ref()).is_some());
+    num_files += 1;
+  }
+  assert_eq!(num_files, 6);
+  assert!(folder.get("gg.html").is_none(), "gg.html should not exist");
+}