From 7ec3e09bd2011d81cc5d98c79945b4ed9def92a2 Mon Sep 17 00:00:00 2001
From: Dominik Nakamura <dnaka91@gmail.com>
Date: Sun, 5 Nov 2023 12:43:06 +0900
Subject: [PATCH] feat: create macro for easy code inclusion

Provide a macro that makes inclusion of generates Rust source code
easier. Also, generate all files into a subfolder to avoid any
collisions with other generated files.
---
 crates/stef-build/src/lib.rs      | 14 ++++++++-
 crates/stef-playground/src/lib.rs | 52 +++++++++++++++----------------
 crates/stef/src/lib.rs            |  7 +++++
 3 files changed, 46 insertions(+), 27 deletions(-)

diff --git a/crates/stef-build/src/lib.rs b/crates/stef-build/src/lib.rs
index cf4089f..aa921ff 100644
--- a/crates/stef-build/src/lib.rs
+++ b/crates/stef-build/src/lib.rs
@@ -30,6 +30,12 @@ pub enum Error {
         #[source]
         source: glob::GlobError,
     },
+    #[error("failed creating output directory at {path:?}")]
+    Create {
+        #[source]
+        source: std::io::Error,
+        path: PathBuf,
+    },
     #[error("failed reading schema file at {file:?}")]
     Read {
         #[source]
@@ -75,7 +81,13 @@ impl Compiler {
     pub fn compile(&self, schemas: &[impl AsRef<str>]) -> Result<()> {
         init_miette();
 
-        let out_dir = PathBuf::from(std::env::var_os("OUT_DIR").unwrap());
+        let out_dir = PathBuf::from(std::env::var_os("OUT_DIR").unwrap()).join("stef");
+
+        std::fs::create_dir_all(&out_dir).map_err(|source| Error::Create {
+            source,
+            path: out_dir.clone(),
+        })?;
+
         let mut inputs = Vec::new();
         let mut validated = Vec::new();
 
diff --git a/crates/stef-playground/src/lib.rs b/crates/stef-playground/src/lib.rs
index 4fcc2db..3677ae5 100644
--- a/crates/stef-playground/src/lib.rs
+++ b/crates/stef-playground/src/lib.rs
@@ -4,109 +4,109 @@
 #![allow(clippy::missing_errors_doc)]
 
 mod sample {
-    include!(concat!(env!("OUT_DIR"), "/sample.rs"));
+    stef::include!("sample");
 }
 
 mod schemas {
     mod alias_basic {
-        include!(concat!(env!("OUT_DIR"), "/alias_basic.rs"));
+        stef::include!("alias_basic");
     }
 
     mod attribute_multi {
-        include!(concat!(env!("OUT_DIR"), "/attribute_multi.rs"));
+        stef::include!("attribute_multi");
     }
 
     mod attribute_single {
-        include!(concat!(env!("OUT_DIR"), "/attribute_single.rs"));
+        stef::include!("attribute_single");
     }
 
     mod attribute_unit {
-        include!(concat!(env!("OUT_DIR"), "/attribute_unit.rs"));
+        stef::include!("attribute_unit");
     }
 
     mod attributes_min_ws {
-        include!(concat!(env!("OUT_DIR"), "/attributes_min_ws.rs"));
+        stef::include!("attributes_min_ws");
     }
 
     mod attributes {
-        include!(concat!(env!("OUT_DIR"), "/attributes.rs"));
+        stef::include!("attributes");
     }
 
     mod const_basic {
-        include!(concat!(env!("OUT_DIR"), "/const_basic.rs"));
+        stef::include!("const_basic");
     }
 
     mod const_string {
-        include!(concat!(env!("OUT_DIR"), "/const_string.rs"));
+        stef::include!("const_string");
     }
 
     mod enum_basic {
-        include!(concat!(env!("OUT_DIR"), "/enum_basic.rs"));
+        stef::include!("enum_basic");
     }
 
     mod enum_generics {
-        include!(concat!(env!("OUT_DIR"), "/enum_generics.rs"));
+        stef::include!("enum_generics");
     }
 
     mod enum_many_ws {
-        include!(concat!(env!("OUT_DIR"), "/enum_many_ws.rs"));
+        stef::include!("enum_many_ws");
     }
 
     mod enum_min_ws {
-        include!(concat!(env!("OUT_DIR"), "/enum_min_ws.rs"));
+        stef::include!("enum_min_ws");
     }
 
     // TODO: implement imports
     // mod import_basic {
-    //     include!(concat!(env!("OUT_DIR"), "/import_basic.rs"));
+    //     stef::include!("import_basic");
     // }
 
     mod module_basic {
-        include!(concat!(env!("OUT_DIR"), "/module_basic.rs"));
+        stef::include!("module_basic");
     }
 
     mod schema_basic {
-        include!(concat!(env!("OUT_DIR"), "/schema_basic.rs"));
+        stef::include!("schema_basic");
     }
 
     mod struct_basic {
-        include!(concat!(env!("OUT_DIR"), "/struct_basic.rs"));
+        stef::include!("struct_basic");
     }
 
     mod struct_generics {
-        include!(concat!(env!("OUT_DIR"), "/struct_generics.rs"));
+        stef::include!("struct_generics");
     }
 
     mod struct_many_ws {
-        include!(concat!(env!("OUT_DIR"), "/struct_many_ws.rs"));
+        stef::include!("struct_many_ws");
     }
 
     mod struct_min_ws {
-        include!(concat!(env!("OUT_DIR"), "/struct_min_ws.rs"));
+        stef::include!("struct_min_ws");
     }
 
     mod struct_tuple {
-        include!(concat!(env!("OUT_DIR"), "/struct_tuple.rs"));
+        stef::include!("struct_tuple");
     }
 
     mod types_basic {
-        include!(concat!(env!("OUT_DIR"), "/types_basic.rs"));
+        stef::include!("types_basic");
     }
 
     mod types_generic {
-        include!(concat!(env!("OUT_DIR"), "/types_generic.rs"));
+        stef::include!("types_generic");
     }
 
     mod types_nested {
-        include!(concat!(env!("OUT_DIR"), "/types_nested.rs"));
+        stef::include!("types_nested");
     }
 
     mod types_non_zero {
-        include!(concat!(env!("OUT_DIR"), "/types_non_zero.rs"));
+        stef::include!("types_non_zero");
     }
 
     mod types_ref {
-        include!(concat!(env!("OUT_DIR"), "/types_ref.rs"));
+        stef::include!("types_ref");
     }
 }
 
diff --git a/crates/stef/src/lib.rs b/crates/stef/src/lib.rs
index 743a353..6385b97 100644
--- a/crates/stef/src/lib.rs
+++ b/crates/stef/src/lib.rs
@@ -20,6 +20,13 @@ pub use buf::{Buf, BufMut, Bytes, Decode, Encode};
 pub mod buf;
 pub mod varint;
 
+#[macro_export]
+macro_rules! include {
+    ($name:literal) => {
+        include!(concat!(env!("OUT_DIR"), "/stef/", $name, ".rs"));
+    };
+}
+
 #[derive(Clone, Debug, PartialEq)]
 pub struct NonZero<T>(T);