Install assets on first run

build.rs

@@ -134,20 +134,4 @@ fn main() {
     clap_app.gen_completions(BIN_NAME, Shell::Bash, &out_dir);
     clap_app.gen_completions(BIN_NAME, Shell::Zsh, &out_dir);
     clap_app.gen_completions(BIN_NAME, Shell::Fish, &out_dir);
-
-    // Install Default Plugins and Layouts
-    let assets = vec![
-        "plugins/status-bar.wasm",
-        "plugins/strider.wasm",
-        "layouts/default.yaml",
-        "layouts/strider.yaml",
-    ];
-    let project_dirs = ProjectDirs::from("org", "Zellij Contributors", "Zellij").unwrap();
-    let data_dir = project_dirs.data_dir();
-    fs::create_dir_all(data_dir.join("plugins")).unwrap();
-    fs::create_dir_all(data_dir.join("layouts")).unwrap();
-    for asset in assets {
-        fs::copy(Path::new("assets/").join(asset), data_dir.join(asset))
-            .expect("Failed to copy asset files");
-    }
 }

src/common/install.rs

@@ -0,0 +1,12 @@
+#[macro_export]
+macro_rules! asset_map {
+    ($($path:literal),+) => {
+        {
+            let mut assets = std::collections::HashMap::new();
+            $(
+                assets.insert($path, include_bytes!(concat!("../assets/", $path)).to_vec());
+            )+
+            assets
+        }
+    }
+}

src/common/mod.rs

@@ -2,6 +2,7 @@ pub mod command_is_executing;
 pub mod errors;
 pub mod input;
 pub mod ipc;
+pub mod install;
 pub mod os_input_output;
 pub mod pty_bus;
 pub mod screen;

src/main.rs

@@ -11,8 +11,9 @@ use common::{
     command_is_executing, errors, os_input_output, pty_bus, screen, start, utils, wasm_vm,
     ApiCommand,
 };
+use directories_next::ProjectDirs;
 
-use std::io::Write;
+use std::{fs, io::Write};
 use std::os::unix::net::UnixStream;
 
 use structopt::StructOpt;
@@ -27,6 +28,24 @@ use crate::utils::{
 };
 
 pub fn main() {
+    // First run installation of default plugins & layouts
+    let project_dirs = ProjectDirs::from("org", "Zellij Contributors", "Zellij").unwrap();
+    let data_dir = project_dirs.data_dir();
+    let assets = asset_map!{
+        "plugins/status-bar.wasm",
+        "plugins/strider.wasm",
+        "layouts/default.yaml",
+        "layouts/strider.yaml"
+    };
+
+    for (path, bytes) in assets {
+        let path = data_dir.join(path);
+        fs::create_dir_all(path.parent().unwrap()).unwrap();
+        if !path.exists() {
+            fs::write(path, bytes).expect("Failed to install default assets!");
+        }
+    }
+
     let opts = CliArgs::from_args();
     if let Some(split_dir) = opts.split {
         match split_dir {