feat: extract into temp directory

infinyon · Oct 1, 2022 · f8e2e84 · f8e2e84
1 parent 0b3dc9e
commit f8e2e84
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 17 deletions.
diff --git a/crates/smdk/Cargo.toml b/crates/smdk/Cargo.toml
@@ -26,6 +26,7 @@ cargo_metadata = "0.15.0"
 cargo-generate = "0.16.0"
 convert_case = "0.6.0"
 include_dir = "0.7.2"
+tempdir = "0.3.7"
 
 fluvio = { path = "../fluvio", default-features = false }
 fluvio-protocol = { path = "../fluvio-protocol", features=["record","api"] }

diff --git a/crates/smdk/src/generate.rs b/crates/smdk/src/generate.rs
@@ -2,8 +2,9 @@ use anyhow::{Error, Result};
 use clap::Parser;
 use cargo_generate::{GenerateArgs, TemplatePath, generate};
 use include_dir::{Dir, include_dir};
+use tempdir::TempDir;
 
-static SMART_MODULE_TEMPLATE: Dir<'_> =
+static SMART_MODULE_TEMPLATE: Dir<'static> =
     include_dir!("$CARGO_MANIFEST_DIR/../../smartmodule/cargo_template");
 
 /// Generate new SmartModule project
@@ -22,30 +23,32 @@ impl GenerateOpt {
     pub(crate) fn process(self) -> Result<()> {
         println!("Generating new SmartModule project: {}", self.name);
 
-        let template_path = match self.template {
-            Some(git) => TemplatePath {
-                git: Some(git),
+        let template_path = if self.template.is_some() {
+            TemplatePath {
+                git: self.template,
                 auto_path: None,
                 subfolder: None,
                 test: false,
                 branch: None,
                 tag: None,
                 path: None,
                 favorite: None,
-            },
-            None => {
-                let path = SMART_MODULE_TEMPLATE.path().to_str().unwrap().to_string();
+            }
+        } else {
+            let temp_dir = TempDir::new("smartmodule_template")?;
+            let path = temp_dir.path().to_str().unwrap().to_string();
+
+            SMART_MODULE_TEMPLATE.extract(temp_dir).map_err(Error::from)?;
 
-                TemplatePath {
-                    git: None,
-                    auto_path: None,
-                    subfolder: None,
-                    test: false,
-                    branch: None,
-                    tag: None,
-                    path: Some(path),
-                    favorite: None,
-                }
+            TemplatePath {
+                git: None,
+                auto_path: None,
+                subfolder: None,
+                test: false,
+                branch: None,
+                tag: None,
+                path: Some(path),
+                favorite: None,
             }
         };