feat: Adding log folder initalization

Author: sousandrei

.gitignore

@@ -6,6 +6,7 @@ assets/vmlinux
 
 firecracker
 tmp
+logs
 todo
 
 .vscode

src/cli.yml

@@ -3,40 +3,47 @@ version: '1.0'
 author: Andrei Sousa <sousandrei@gmail.com>
 about: Simple AWS Firecracker orchestrator
 args:
-  - assets_dir:
-      short: a
-      long: assets_dir
-      value_name: DIR
-      help: Folder containing the assets (kernel, and drive)
-      default_value: './assets'
-  - tmp_dir:
-      short: t
-      long: tmp_dir
-      value_name: DIR
-      help: Sets a custom tmp dir to use
-      takes_value: true
-      default_value: './tmp'
-  - kernel_name:
-      short: k
-      long: kernel_name
-      value_name: KERNEL_NAME
-      help: Name of the kernel file
-      default_value: 'vmlinux'
-  - drive_name:
-      short: d
-      long: drive_name
-      value_name: BOOT_DRIVE_NAME
-      help: Name of the boot drive file
-      default_value: 'rootfs.ext4'
-  - firecracker_path:
-      short: f
-      long: firecracker_path
-      value_name: FIRECRACKER_PATH
-      help: Path of the firecracker binary
-      default_value: './assets/firecracker'
-  - port:
-      short: p
-      long: port
-      value_name: PORT
-      help: Port to run http server
-      default_value: '3000'
+    - assets_dir:
+          short: a
+          long: assets_dir
+          value_name: DIR
+          help: Folder containing the assets (kernel, and drive)
+          default_value: './assets'
+    - tmp_dir:
+          short: t
+          long: tmp_dir
+          value_name: DIR
+          help: Sets a custom tmp dir to use
+          takes_value: true
+          default_value: './tmp'
+    - log_dir:
+          short: l
+          long: log_dir
+          value_name: DIR
+          help: Sets a custom tmp dir to use
+          takes_value: true
+          default_value: './logs'
+    - kernel_name:
+          short: k
+          long: kernel_name
+          value_name: KERNEL_NAME
+          help: Name of the kernel file
+          default_value: 'vmlinux'
+    - drive_name:
+          short: d
+          long: drive_name
+          value_name: BOOT_DRIVE_NAME
+          help: Name of the boot drive file
+          default_value: 'rootfs.ext4'
+    - firecracker_path:
+          short: f
+          long: firecracker_path
+          value_name: FIRECRACKER_PATH
+          help: Path of the firecracker binary
+          default_value: './assets/firecracker'
+    - port:
+          short: p
+          long: port
+          value_name: PORT
+          help: Port to run http server
+          default_value: '3000'

src/tmp.rs src/folders.rssrc/tmp.rs renamed to src/folders.rs

@@ -24,7 +24,7 @@ pub fn create_folder(path: PathBuf) -> Result<(), std::io::Error> {
     match fs::read_dir(&path) {
         Ok(entries) => {
             if entries.peekable().peek().is_some() {
-                warn!("Warning: tmp dir not empty [{}]", path.display());
+                warn!("Warning: dir not empty [{}]", path.display());
             }
         }
         Err(e) => error!("Cannot read dir [{}]", e),

src/io.rs

@@ -8,8 +8,8 @@ use tracing::info;
 
 mod api;
 mod error;
+mod folders;
 mod io;
-mod tmp;
 mod vm;
 
 #[derive(Serialize, Deserialize, Debug)]
@@ -21,6 +21,7 @@ pub struct Vm {
 pub struct State {
     vms: Arc<Mutex<Vec<Vm>>>,
     tmp_dir: String,
+    log_dir: String,
     assets_dir: String,
     drive_name: String,
     kernel_name: String,
@@ -31,6 +32,7 @@ impl Clone for State {
         State {
             vms: self.vms.clone(),
             tmp_dir: self.tmp_dir.clone(),
+            log_dir: self.log_dir.clone(),
             assets_dir: self.assets_dir.clone(),
             drive_name: self.drive_name.clone(),
             kernel_name: self.kernel_name.clone(),
@@ -52,12 +54,14 @@ async fn main() -> Result<(), Box<dyn std::error::Error + Send + Sync>> {
     let state = State {
         vms: Arc::new(Mutex::new(Vec::new())),
         tmp_dir: String::from(matches.value_of("tmp_dir").unwrap()),
+        log_dir: String::from(matches.value_of("log_dir").unwrap()),
         assets_dir: String::from(matches.value_of("assets_dir").unwrap()),
         drive_name: String::from(matches.value_of("drive_name").unwrap()),
         kernel_name: String::from(matches.value_of("kernel_name").unwrap()),
     };
 
-    tmp::init(&state.tmp_dir)?;
+    folders::init(&state.tmp_dir)?;
+    folders::init(&state.log_dir)?;
 
     let port = matches.value_of("port").unwrap();
     let port: u16 = port.parse().unwrap();