Change trm to us up and down scripts instead of scripts and hooks

Cargo.lock

@@ -3362,6 +3362,7 @@ dependencies = [
  "dotenv",
  "embed-nu",
  "lazy_static",
+ "paste",
  "serde",
  "serde_json",
  "thiserror",

Cargo.toml

@@ -21,6 +21,7 @@ color-eyre = "0.6.2"
 dotenv = "0.15.0"
 embed-nu = "0.3.0"
 lazy_static = "1.4.0"
+paste = "1.0.9"
 serde = { version = "1.0.145", features = ["derive"] }
 serde_json = "1.0.86"
 thiserror = "1.0.37"

configs/crystal/configure-locale/down.nu

@@ -0,0 +1,4 @@
+# Reverts all system changes of `configure-locale`
+def main [cfg] {
+    echo "Executing up task `configure-locale` with config" $cfg
+}

configs/crystal/configure-locale/up.nu

@@ -0,0 +1,4 @@
+# Applies all system changes of `configure-locale`
+def main [cfg] {
+    echo "Executing up task `configure-locale` with config" $cfg
+}

configs/crystal/configure-network/down.nu

@@ -0,0 +1,4 @@
+# Reverts all system changes of `configure-network`
+def main [cfg] {
+    echo "Executing up task `configure-network` with config" $cfg
+}

configs/crystal/configure-network/up.nu

@@ -0,0 +1,4 @@
+# Applies all system changes of `configure-network`
+def main [cfg] {
+    echo "Executing up task `configure-network` with config" $cfg
+}

configs/crystal/configure-unakite/down.nu

@@ -0,0 +1,4 @@
+# Reverts all system changes of `configure-unakite`
+def main [cfg] {
+    echo "Executing up task `configure-unakite` with config" $cfg
+}

configs/crystal/configure-unakite/up.nu

@@ -0,0 +1,4 @@
+# Applies all system changes of `configure-unakite`
+def main [cfg] {
+    echo "Executing up task `configure-unakite` with config" $cfg
+}

configs/crystal/create-partitions/down.nu

@@ -0,0 +1,4 @@
+# Reverts all system changes of `create-partitions`
+def main [cfg] {
+    echo "Executing up task `create-partitions` with config" $cfg
+}

configs/crystal/create-partitions/up.nu

@@ -0,0 +1,4 @@
+# Applies all system changes of `create-partitions`
+def main [cfg] {
+    echo "Executing up task `create-partitions` with config" $cfg
+}

configs/crystal/hooks/configure-locale.post.nu

@@ -1,6 +0,0 @@
-
-def main [cfg] {
-    echo "Executing after Task with task config: " $cfg                
-    echo "The global config is: " $TRM_CONFIG
-}
-        

configs/crystal/hooks/configure-locale.pre.nu

@@ -1,6 +0,0 @@
-
-def main [cfg] {
-    echo "Executing before Task with task config: " $cfg                
-    echo "The global config is: " $TRM_CONFIG
-}
-        

configs/crystal/hooks/configure-network.post.nu

@@ -1,6 +0,0 @@
-
-def main [cfg] {
-    echo "Executing after Task with task config: " $cfg                
-    echo "The global config is: " $TRM_CONFIG
-}
-        

configs/crystal/hooks/configure-network.pre.nu

@@ -1,6 +0,0 @@
-
-def main [cfg] {
-    echo "Executing before Task with task config: " $cfg                
-    echo "The global config is: " $TRM_CONFIG
-}
-        

configs/crystal/hooks/configure-unakite.post.nu

@@ -1,6 +0,0 @@
-
-def main [cfg] {
-    echo "Executing after Task with task config: " $cfg                
-    echo "The global config is: " $TRM_CONFIG
-}
-        

configs/crystal/hooks/configure-unakite.pre.nu

@@ -1,6 +0,0 @@
-
-def main [cfg] {
-    echo "Executing before Task with task config: " $cfg                
-    echo "The global config is: " $TRM_CONFIG
-}
-        

configs/crystal/hooks/create-partitions.post.nu

@@ -1,6 +0,0 @@
-
-def main [cfg] {
-    echo "Executing after Task with task config: " $cfg                
-    echo "The global config is: " $TRM_CONFIG
-}
-        

configs/crystal/hooks/create-partitions.pre.nu

@@ -1,6 +0,0 @@
-
-def main [cfg] {
-    echo "Executing before Task with task config: " $cfg                
-    echo "The global config is: " $TRM_CONFIG
-}
-        

configs/crystal/hooks/install-base.post.nu

@@ -1,6 +0,0 @@
-
-def main [cfg] {
-    echo "Executing after Task with task config: " $cfg                
-    echo "The global config is: " $TRM_CONFIG
-}
-        

configs/crystal/hooks/install-base.pre.nu

@@ -1,6 +0,0 @@
-
-def main [cfg] {
-    echo "Executing before Task with task config: " $cfg                
-    echo "The global config is: " $TRM_CONFIG
-}
-        

configs/crystal/hooks/install-bootloader.post.nu

@@ -1,6 +0,0 @@
-
-def main [cfg] {
-    echo "Executing after Task with task config: " $cfg                
-    echo "The global config is: " $TRM_CONFIG
-}
-        

configs/crystal/hooks/install-bootloader.pre.nu

@@ -1,6 +0,0 @@
-
-def main [cfg] {
-    echo "Executing before Task with task config: " $cfg                
-    echo "The global config is: " $TRM_CONFIG
-}
-        

configs/crystal/hooks/install-desktop.post.nu

@@ -1,6 +0,0 @@
-
-def main [cfg] {
-    echo "Executing after Task with task config: " $cfg                
-    echo "The global config is: " $TRM_CONFIG
-}
-        

configs/crystal/hooks/install-desktop.pre.nu

@@ -1,6 +0,0 @@
-
-def main [cfg] {
-    echo "Executing before Task with task config: " $cfg                
-    echo "The global config is: " $TRM_CONFIG
-}
-        

configs/crystal/hooks/install-extra-packages.post.nu

@@ -1,6 +0,0 @@
-
-def main [cfg] {
-    echo "Executing after Task with task config: " $cfg                
-    echo "The global config is: " $TRM_CONFIG
-}
-        

configs/crystal/hooks/install-extra-packages.pre.nu

@@ -1,6 +0,0 @@
-
-def main [cfg] {
-    echo "Executing before Task with task config: " $cfg                
-    echo "The global config is: " $TRM_CONFIG
-}
-        

configs/crystal/hooks/install-flatpak.post.nu

@@ -1,6 +0,0 @@
-
-def main [cfg] {
-    echo "Executing after Task with task config: " $cfg                
-    echo "The global config is: " $TRM_CONFIG
-}
-        

configs/crystal/hooks/install-flatpak.pre.nu

@@ -1,6 +0,0 @@
-
-def main [cfg] {
-    echo "Executing before Task with task config: " $cfg                
-    echo "The global config is: " $TRM_CONFIG
-}
-        

configs/crystal/hooks/install-kernels.post.nu

@@ -1,6 +0,0 @@
-
-def main [cfg] {
-    echo "Executing after Task with task config: " $cfg                
-    echo "The global config is: " $TRM_CONFIG
-}
-        

configs/crystal/hooks/install-kernels.pre.nu

@@ -1,6 +0,0 @@
-
-def main [cfg] {
-    echo "Executing before Task with task config: " $cfg                
-    echo "The global config is: " $TRM_CONFIG
-}
-        

configs/crystal/hooks/install-timeshift.post.nu

@@ -1,6 +0,0 @@
-
-def main [cfg] {
-    echo "Executing after Task with task config: " $cfg                
-    echo "The global config is: " $TRM_CONFIG
-}
-        

configs/crystal/hooks/install-timeshift.pre.nu

@@ -1,6 +0,0 @@
-
-def main [cfg] {
-    echo "Executing before Task with task config: " $cfg                
-    echo "The global config is: " $TRM_CONFIG
-}
-        

configs/crystal/hooks/install-zramd.post.nu

@@ -1,6 +0,0 @@
-
-def main [cfg] {
-    echo "Executing after Task with task config: " $cfg                
-    echo "The global config is: " $TRM_CONFIG
-}
-        

configs/crystal/hooks/install-zramd.pre.nu

@@ -1,6 +0,0 @@
-
-def main [cfg] {
-    echo "Executing before Task with task config: " $cfg                
-    echo "The global config is: " $TRM_CONFIG
-}
-        

configs/crystal/hooks/setup-root-user.post.nu

@@ -1,6 +0,0 @@
-
-def main [cfg] {
-    echo "Executing after Task with task config: " $cfg                
-    echo "The global config is: " $TRM_CONFIG
-}
-        

configs/crystal/hooks/setup-root-user.pre.nu

@@ -1,6 +0,0 @@
-
-def main [cfg] {
-    echo "Executing before Task with task config: " $cfg                
-    echo "The global config is: " $TRM_CONFIG
-}
-        

configs/crystal/hooks/setup-users.post.nu

@@ -1,6 +0,0 @@
-
-def main [cfg] {
-    echo "Executing after Task with task config: " $cfg                
-    echo "The global config is: " $TRM_CONFIG
-}
-        

configs/crystal/hooks/setup-users.pre.nu

@@ -1,6 +0,0 @@
-
-def main [cfg] {
-    echo "Executing before Task with task config: " $cfg                
-    echo "The global config is: " $TRM_CONFIG
-}
-        

configs/crystal/install-base/down.nu

@@ -0,0 +1,4 @@
+# Reverts all system changes of `install-base`
+def main [cfg] {
+    echo "Executing up task `install-base` with config" $cfg
+}

configs/crystal/install-base/up.nu

@@ -0,0 +1,4 @@
+# Applies all system changes of `install-base`
+def main [cfg] {
+    echo "Executing up task `install-base` with config" $cfg
+}

configs/crystal/install-bootloader/down.nu

@@ -0,0 +1,4 @@
+# Reverts all system changes of `install-bootloader`
+def main [cfg] {
+    echo "Executing up task `install-bootloader` with config" $cfg
+}

configs/crystal/install-bootloader/up.nu

@@ -0,0 +1,4 @@
+# Applies all system changes of `install-bootloader`
+def main [cfg] {
+    echo "Executing up task `install-bootloader` with config" $cfg
+}

configs/crystal/install-desktop/down.nu

@@ -0,0 +1,4 @@
+# Reverts all system changes of `install-desktop`
+def main [cfg] {
+    echo "Executing up task `install-desktop` with config" $cfg
+}

configs/crystal/install-desktop/up.nu

@@ -0,0 +1,4 @@
+# Applies all system changes of `install-desktop`
+def main [cfg] {
+    echo "Executing up task `install-desktop` with config" $cfg
+}

configs/crystal/install-extra-packages/down.nu

@@ -0,0 +1,4 @@
+# Reverts all system changes of `install-extra-packages`
+def main [cfg] {
+    echo "Executing up task `install-extra-packages` with config" $cfg
+}

configs/crystal/install-extra-packages/up.nu

@@ -0,0 +1,4 @@
+# Applies all system changes of `install-extra-packages`
+def main [cfg] {
+    echo "Executing up task `install-extra-packages` with config" $cfg
+}

configs/crystal/install-flatpak/down.nu

@@ -0,0 +1,4 @@
+# Reverts all system changes of `install-flatpak`
+def main [cfg] {
+    echo "Executing up task `install-flatpak` with config" $cfg
+}

configs/crystal/install-flatpak/up.nu

@@ -0,0 +1,4 @@
+# Applies all system changes of `install-flatpak`
+def main [cfg] {
+    echo "Executing up task `install-flatpak` with config" $cfg
+}

configs/crystal/install-kernels/down.nu

@@ -0,0 +1,4 @@
+# Reverts all system changes of `install-kernels`
+def main [cfg] {
+    echo "Executing up task `install-kernels` with config" $cfg
+}

configs/crystal/install-kernels/up.nu

@@ -0,0 +1,4 @@
+# Applies all system changes of `install-kernels`
+def main [cfg] {
+    echo "Executing up task `install-kernels` with config" $cfg
+}

configs/crystal/install-timeshift/down.nu

@@ -0,0 +1,4 @@
+# Reverts all system changes of `install-timeshift`
+def main [cfg] {
+    echo "Executing up task `install-timeshift` with config" $cfg
+}

configs/crystal/install-timeshift/up.nu

@@ -0,0 +1,4 @@
+# Applies all system changes of `install-timeshift`
+def main [cfg] {
+    echo "Executing up task `install-timeshift` with config" $cfg
+}

configs/crystal/install-zramd/down.nu

@@ -0,0 +1,4 @@
+# Reverts all system changes of `install-zramd`
+def main [cfg] {
+    echo "Executing up task `install-zramd` with config" $cfg
+}

configs/crystal/install-zramd/up.nu

@@ -0,0 +1,4 @@
+# Applies all system changes of `install-zramd`
+def main [cfg] {
+    echo "Executing up task `install-zramd` with config" $cfg
+}

configs/crystal/scripts/configure-locale.nu

@@ -1,5 +0,0 @@
-
-def main [cfg] {
-    echo "Executing Task with config" $cfg                
-}
-        

configs/crystal/scripts/configure-network.nu

@@ -1,5 +0,0 @@
-
-def main [cfg] {
-    echo "Executing Task with config" $cfg                
-}
-        

configs/crystal/scripts/configure-unakite.nu

@@ -1,5 +0,0 @@
-
-def main [cfg] {
-    echo "Executing Task with config" $cfg                
-}
-        

configs/crystal/scripts/create-partitions.nu

@@ -1,5 +0,0 @@
-
-def main [cfg] {
-    echo "Executing Task with config" $cfg                
-}
-        

configs/crystal/scripts/install-base.nu

@@ -1,5 +0,0 @@
-
-def main [cfg] {
-    echo "Executing Task with config" $cfg                
-}
-        

configs/crystal/scripts/install-bootloader.nu

@@ -1,5 +0,0 @@
-
-def main [cfg] {
-    echo "Executing Task with config" $cfg                
-}
-        

configs/crystal/scripts/install-desktop.nu

@@ -1,5 +0,0 @@
-
-def main [cfg] {
-    echo "Executing Task with config" $cfg                
-}
-        

configs/crystal/scripts/install-extra-packages.nu

@@ -1,5 +0,0 @@
-
-def main [cfg] {
-    echo "Executing Task with config" $cfg                
-}
-        

configs/crystal/scripts/install-flatpak.nu

@@ -1,5 +0,0 @@
-
-def main [cfg] {
-    echo "Executing Task with config" $cfg                
-}
-        

configs/crystal/scripts/install-kernels.nu

@@ -1,5 +0,0 @@
-
-def main [cfg] {
-    echo "Executing Task with config" $cfg                
-}
-        

configs/crystal/scripts/install-timeshift.nu

@@ -1,5 +0,0 @@
-
-def main [cfg] {
-    echo "Executing Task with config" $cfg                
-}
-        

configs/crystal/scripts/install-zramd.nu

@@ -1,5 +0,0 @@
-
-def main [cfg] {
-    echo "Executing Task with config" $cfg                
-}
-        

configs/crystal/scripts/setup-root-user.nu

@@ -1,5 +0,0 @@
-
-def main [cfg] {
-    echo "Executing Task with config" $cfg                
-}
-        

configs/crystal/scripts/setup-users.nu

@@ -1,5 +0,0 @@
-
-def main [cfg] {
-    echo "Executing Task with config" $cfg                
-}
-        

configs/crystal/setup-root-user/down.nu

@@ -0,0 +1,4 @@
+# Reverts all system changes of `setup-root-user`
+def main [cfg] {
+    echo "Executing up task `setup-root-user` with config" $cfg
+}

configs/crystal/setup-root-user/up.nu

@@ -0,0 +1,4 @@
+# Applies all system changes of `setup-root-user`
+def main [cfg] {
+    echo "Executing up task `setup-root-user` with config" $cfg
+}

configs/crystal/setup-users/down.nu

@@ -0,0 +1,4 @@
+# Reverts all system changes of `setup-users`
+def main [cfg] {
+    echo "Executing up task `setup-users` with config" $cfg
+}

configs/crystal/setup-users/up.nu

@@ -0,0 +1,4 @@
+# Applies all system changes of `setup-users`
+def main [cfg] {
+    echo "Executing up task `setup-users` with config" $cfg
+}

src/lib.rs

@@ -1,7 +1,7 @@
 use config::Config;
 use error::{AppError, AppResult};
 use scripting::{
-    loader::{HookType, ScriptLoader},
+    loader::ScriptLoader,
     script::{NuScript, Script},
 };
 use tasks::*;
@@ -13,15 +13,10 @@ pub mod tasks;
 pub(crate) mod utils;
 pub use utils::generate_script_files;
 
-macro_rules! tasks {
-    ($($function:ident => $script:ident),+) => {
-       $(
-            #[tracing::instrument(level = "trace", skip(self))]
-            pub async fn $function(&self, cfg: <$script as crate::scripting::script::Script>::Args) -> AppResult<()> {
-                self.execute_task::<$script>(cfg).await
-            }
-        )+
-    }
+#[derive(Clone, Copy, Debug, PartialEq, PartialOrd)]
+pub enum TaskOperation {
+    Up,
+    Down,
 }
 
 pub struct TaskExecutor {
@@ -29,6 +24,20 @@ pub struct TaskExecutor {
     loader: ScriptLoader,
 }
 
+macro_rules! task_executors {
+    ($($function:ident => $task:ident),+) => {
+       $(
+            #[tracing::instrument(level = "trace", skip(self))]
+            pub async fn $function(&self, operation: TaskOperation, cfg: <$task as crate::tasks::Task>::Config) -> AppResult<()> {
+                match operation {
+                    TaskOperation::Up => self.execute_task::<<$task as crate::tasks::Task>::UpScript>(cfg).await,
+                    TaskOperation::Down => self.execute_task::<<$task as crate::tasks::Task>::DownScript>(cfg).await,
+                }
+            }
+        )+
+    }
+}
+
 impl TaskExecutor {
     /// Creates a new task executor with a given config
     pub fn with_config(config: Config) -> Self {
@@ -38,67 +47,66 @@ impl TaskExecutor {
         }
     }
 
-    tasks!(
-        setup_users => SetupUsersScript,
-        configure_network => ConfigureNetworkScript,
-        configure_unakite => ConfigureUnakiteScript,
-        create_partitions => CreatePartitionsScript,
-        install_base => InstallBaseScript,
-        install_bootloader => InstallBootloaderScript,
-        install_desktop => InstallDesktopScript,
-        install_extra_packages => InstallExtraPackagesScript,
-        install_flatpak => InstallFlatpakScript,
-        install_kernels => InstallKernelsScript,
-        install_timeshift => InstallTimeshiftScript,
-        install_zramd => InstallZRamDScript,
-        setup_root_user => SetupRootUserScript,
-        configure_locale => ConfigureLocaleScript
+    task_executors!(
+        setup_users => SetupUsersTask,
+        configure_network => ConfigureNetworkTask,
+        configure_unakite => ConfigureUnakiteTask,
+        create_partitions => CreatePartitionsTask,
+        install_base => InstallBaseTask,
+        install_bootloader => InstallBootloaderTask,
+        install_desktop => InstallDesktopTask,
+        install_extra_packages => InstallExtraPackagesTask,
+        install_flatpak => InstallFlatpakTask,
+        install_kernels => InstallKernelsTask,
+        install_timeshift => InstallTimeshiftTask,
+        install_zramd => InstallZRamDTask,
+        setup_root_user => SetupRootUserTask,
+        configure_locale => ConfigureLocaleTask
     );
 
     /// Installs the system from the given system configuration
     #[tracing::instrument(level = "trace", skip(self))]
     pub async fn install_from_config(&self) -> AppResult<()> {
         let config = self.config.clone().ok_or(AppError::MissingConfig)?;
-        self.create_partitions(config.partitions).await?;
-        self.install_base(()).await?;
-        self.install_kernels(config.kernels).await?;
-        self.install_bootloader(config.bootloader).await?;
-        self.configure_locale(config.locale).await?;
-        self.configure_network(config.network).await?;
+        self.create_partitions(TaskOperation::Up, config.partitions)
+            .await?;
+        self.install_base(TaskOperation::Up, ()).await?;
+        self.install_kernels(TaskOperation::Up, config.kernels)
+            .await?;
+        self.install_bootloader(TaskOperation::Up, config.bootloader)
+            .await?;
+        self.configure_locale(TaskOperation::Up, config.locale)
+            .await?;
+        self.configure_network(TaskOperation::Up, config.network)
+            .await?;
 
         if config.enable_zramd {
-            self.install_zramd(()).await?;
+            self.install_zramd(TaskOperation::Up, ()).await?;
         }
         if config.enable_timeshift {
-            self.install_timeshift(()).await?;
+            self.install_timeshift(TaskOperation::Up, ()).await?;
         }
         if config.enable_flatpak {
-            self.install_flatpak(()).await?;
+            self.install_flatpak(TaskOperation::Up, ()).await?;
         }
-        self.setup_users(config.users).await?;
-        self.setup_root_user(config.root_user).await?;
-        self.install_desktop(config.desktop).await?;
-        self.install_extra_packages(config.extra_packages).await?;
+        self.setup_users(TaskOperation::Up, config.users).await?;
+        self.setup_root_user(TaskOperation::Up, config.root_user)
+            .await?;
+        self.install_desktop(TaskOperation::Up, config.desktop)
+            .await?;
+        self.install_extra_packages(TaskOperation::Up, config.extra_packages)
+            .await?;
 
         if let Some(unakite) = config.unakite {
-            self.configure_unakite(unakite).await?;
+            self.configure_unakite(TaskOperation::Up, unakite).await?;
         }
 
         Ok(())
     }
 
     async fn execute_task<S: Script>(&self, args: S::Args) -> AppResult<()> {
-        if let Some(pre_hook) = self.loader.load_hook::<S>(HookType::Pre) {
-            self.execute(pre_hook, args.clone()).await?;
-        }
         let script = self.loader.load::<S>()?;
-
-        if let Some(post_hook) = self.loader.load_hook::<S>(HookType::Post) {
-            self.execute(script, args.clone()).await?;
-            self.execute(post_hook, args).await?;
-        } else {
-            self.execute(script, args.clone()).await?;
-        }
+        self.execute(script, args.clone()).await?;
 
         Ok(())
     }

src/scripting/loader.rs

@@ -7,12 +7,6 @@ use super::script::{NuScript, Script};
 /// A loader for nu script files
 pub struct ScriptLoader {
     script_dir: PathBuf,
-    hook_dir: PathBuf,
-}
-
-pub enum HookType {
-    Pre,
-    Post,
 }
 
 impl ScriptLoader {
@@ -20,14 +14,13 @@ impl ScriptLoader {
     pub fn new() -> Self {
         Self {
             script_dir: crate::utils::SCRIPT_PATH.to_owned(),
-            hook_dir: crate::utils::HOOK_PATH.to_owned(),
         }
     }
 
     /// Loads the given script file
     #[tracing::instrument(level = "trace", skip_all)]
     pub fn load<S: Script>(&self) -> AppResult<NuScript<S>> {
-        let script_path = self.script_dir.join(S::get_name());
+        let script_path = self.script_dir.join(S::name());
 
         if !script_path.exists() {
             Err(AppError::ScriptNotFound(script_path))
@@ -35,18 +28,4 @@ impl ScriptLoader {
             Ok(NuScript::new(script_path))
         }
     }
-
-    pub fn load_hook<S: Script>(&self, hook_type: HookType) -> Option<NuScript<S>> {
-        let script_name = match hook_type {
-            HookType::Pre => S::get_pre_hook(),
-            HookType::Post => S::get_post_hook(),
-        };
-        let script_path = self.hook_dir.join(script_name);
-
-        if !script_path.exists() {
-            None
-        } else {
-            Some(NuScript::new(script_path))
-        }
-    }
 }

src/scripting/script.rs

@@ -17,15 +17,7 @@ pub trait Script {
     /// Returns the (expected) name of the script file
     /// This function is used by the loader to load the associated file
     /// The name needs to include the file extension
-    fn get_name() -> &'static str;
-
-    /// Returns the name of the script file that get's executed before
-    /// the actual script. This has to be the full file name including the extension.
-    fn get_pre_hook() -> &'static str;
-
-    /// Returns the name of the script file that get's executed after
-    /// the actual script. This has to be the full file name including the extension.
-    fn get_post_hook() -> &'static str;
+    fn name() -> &'static str;
 }
 
 /// Script arguments that can be collected in a Vec to
@@ -89,32 +81,3 @@ impl<S: Script> NuScript<S> {
         fs::read_to_string(&self.path).await.map_err(AppError::from)
     }
 }
-
-/// Defines a script
-/// This macro doesn't accept a file extension for the script name
-/// as it is reused for the hook name
-#[macro_export]
-macro_rules! script {
-    ($script:ident {
-        file = $name:literal
-        args = $argtype:ident
-    }) => {
-        pub struct $script;
-
-        impl $crate::scripting::script::Script for $script {
-            type Args = $argtype;
-
-            fn get_name() -> &'static str {
-                concat!($name, ".nu")
-            }
-
-            fn get_pre_hook() -> &'static str {
-                concat!($name, ".pre.nu")
-            }
-
-            fn get_post_hook() -> &'static str {
-                concat!($name, ".post.nu")
-            }
-        }
-    };
-}

src/tasks/configure_locale.rs

@@ -1,10 +1,10 @@
 use embed_nu::rusty_value::*;
 use serde::Deserialize;
 
-use crate::script;
+use crate::task;
 
-script!(ConfigureLocaleScript {
-    file = "configure-locale"
+task!(ConfigureLocale {
+    name = "configure-locale"
     args = LocaleConfig
 });
 

src/tasks/configure_network.rs

@@ -1,10 +1,10 @@
 use embed_nu::rusty_value::*;
 use serde::Deserialize;
 
-use crate::script;
+use crate::task;
 
-script!(ConfigureNetworkScript {
-    file = "configure-network"
+task!(ConfigureNetwork {
+    name = "configure-network"
     args = NetworkConfig
 });
 

src/tasks/configure_unakite.rs

@@ -3,10 +3,10 @@ use std::path::PathBuf;
 use embed_nu::rusty_value::*;
 use serde::Deserialize;
 
-use crate::script;
+use crate::task;
 
-script!(ConfigureUnakiteScript {
-    file = "configure-unakite"
+task!(ConfigureUnakite {
+    name = "configure-unakite"
     args = UnakiteConfig
 });
 

src/tasks/create_partitions.rs

@@ -3,10 +3,10 @@ use std::path::PathBuf;
 use embed_nu::rusty_value::*;
 use serde::Deserialize;
 
-use crate::script;
+use crate::task;
 
-script!(CreatePartitionsScript {
-    file = "create-partitions"
+task!(CreatePartitions {
+    name = "create-partitions"
     args = PartitionsConfig
 });
 

src/tasks/install_base.rs

@@ -1,7 +1,7 @@
-use crate::script;
+use crate::task;
 
-script!(InstallBaseScript {
-    file = "install-base"
+task!(InstallBase {
+    name = "install-base"
     args = InstallBaseArgs
 });
 

src/tasks/install_bootloader.rs

@@ -3,10 +3,10 @@ use std::path::PathBuf;
 use embed_nu::rusty_value::*;
 use serde::Deserialize;
 
-use crate::script;
+use crate::task;
 
-script!(InstallBootloaderScript {
-    file = "install-bootloader"
+task!(InstallBootloader {
+    name = "install-bootloader"
     args = BootloaderConfig
 });
 

src/tasks/install_desktop.rs

@@ -1,10 +1,10 @@
 use embed_nu::rusty_value::*;
 use serde::Deserialize;
 
-use crate::script;
+use crate::task;
 
-script!(InstallDesktopScript {
-    file = "install-desktop"
+task!(InstallDesktop {
+    name = "install-desktop"
     args = DesktopConfig
 });
 

src/tasks/install_extra_packages.rs

@@ -1,7 +1,7 @@
-use crate::script;
+use crate::task;
 
-script!(InstallExtraPackagesScript {
-    file = "install-extra-packages"
+task!(InstallExtraPackages {
+    name = "install-extra-packages"
     args = ExtraPackages
 });
 

src/tasks/install_flatpak.rs

@@ -1,7 +1,7 @@
-use crate::script;
+use crate::task;
 
-script!(InstallFlatpakScript {
-    file = "install-flatpak"
+task!(InstallFlatpak {
+    name = "install-flatpak"
     args = FlatpakConfig
 });
 

src/tasks/install_kernels.rs

@@ -1,10 +1,10 @@
 use embed_nu::rusty_value::*;
 use serde::Deserialize;
 
-use crate::script;
+use crate::task;
 
-script!(InstallKernelsScript {
-    file = "install-kernels"
+task!(InstallKernels {
+    name = "install-kernels"
     args = KernelConfig
 });
 

src/tasks/install_timeshift.rs

@@ -1,7 +1,7 @@
-use crate::script;
+use crate::task;
 
-script!(InstallTimeshiftScript {
-    file = "install-timeshift"
+task!(InstallTimeshift {
+    name = "install-timeshift"
     args = TimeshiftConfig
 });
 

src/tasks/install_zramd.rs

@@ -1,7 +1,7 @@
-use crate::script;
+use crate::task;
 
-script!(InstallZRamDScript {
-    file = "install-zramd"
+task!(InstallZRamD {
+    name = "install-zramd"
     args = ZRamDConfig
 });
 

src/tasks/mod.rs

@@ -13,7 +13,7 @@ mod install_zramd;
 mod setup_root_user;
 mod setup_users;
 
-use std::path::{Path, PathBuf};
+use std::{fmt, path::PathBuf};
 
 pub use configure_locale::*;
 pub use configure_network::*;
@@ -30,57 +30,89 @@ pub use install_zramd::*;
 pub use setup_root_user::*;
 pub use setup_users::*;
 
-use crate::scripting::script::Script;
+use crate::scripting::script::{Script, ScriptArgs};
 
-pub struct TaskFiles {
-    script: String,
-    pre_hook: String,
-    post_hook: String,
+pub trait Task {
+    type Config: ScriptArgs + fmt::Debug + Clone;
+    type UpScript: Script<Args = Self::Config>;
+    type DownScript: Script<Args = Self::Config>;
+
+    fn name() -> &'static str;
 }
 
-impl TaskFiles {
-    pub fn script_path(&self, base: &Path) -> PathBuf {
-        base.join("scripts").join(&self.script)
-    }
+/// Defines a script
+/// This macro doesn't accept a file extension for the script name
+/// as it is reused for the hook name
+#[macro_export]
+macro_rules! task {
+    ($task:ident {
+        name = $name:literal
+        args = $argtype:ident
+    }) => {
+        paste::item! {
+            pub struct [<$task Task>];
 
-    pub fn pre_hook_path(&self, base: &Path) -> PathBuf {
-        base.join("hooks").join(&self.pre_hook)
-    }
+            impl $crate::tasks::Task for [<$task Task>] {
+                type Config = $argtype;
+                type UpScript = [<$task UpScript>];
+                type DownScript = [<$task DownScript>];
 
-    pub fn post_hook_path(&self, base: &Path) -> PathBuf {
-        base.join("hooks").join(&self.post_hook)
-    }
-}
+                #[inline]
+                fn name() -> &'static str {
+                    $name
+                }
+            }
+
+            pub struct [<$task UpScript>];
 
-macro_rules! __all_tasks {
-    ($($task:ident),+) => {
-        {
-            vec![$(
-                TaskFiles {
-                    script: $task::get_name().into(),
-                    pre_hook: $task::get_pre_hook().into(),
-                    post_hook: $task::get_post_hook().into(),
-                },
-            )+]
+            impl $crate::scripting::script::Script for [<$task UpScript>] {
+                type Args = $argtype;
+
+                #[inline]
+                fn name() -> &'static str {
+                    "up.nu"
+                }
+            }
+
+            pub struct [<$task DownScript>];
+
+            impl $crate::scripting::script::Script for [<$task DownScript>] {
+                type Args = $argtype;
+
+                #[inline]
+                fn name() -> &'static str {
+                    "down.nu"
+                }
+            }
         }
     };
 }
 
-pub fn all_tasks() -> Vec<TaskFiles> {
-    __all_tasks!(
-        ConfigureLocaleScript,
-        ConfigureNetworkScript,
-        ConfigureUnakiteScript,
-        CreatePartitionsScript,
-        InstallBaseScript,
-        InstallBootloaderScript,
-        InstallDesktopScript,
-        InstallExtraPackagesScript,
-        InstallFlatpakScript,
-        InstallKernelsScript,
-        InstallTimeshiftScript,
-        InstallZRamDScript,
-        SetupRootUserScript,
-        SetupUsersScript
-    )
+pub(crate) fn all_tasks() -> Vec<(&'static str, PathBuf, PathBuf)> {
+    macro_rules! task_scripts {
+        ($task:ident) => {{
+            let base = PathBuf::from($task::name());
+            (
+                $task::name(),
+                base.join(<$task as Task>::UpScript::name()),
+                base.join(<$task as Task>::DownScript::name()),
+            )
+        }};
+    }
+    vec![
+        task_scripts!(ConfigureLocaleTask),
+        task_scripts!(ConfigureNetworkTask),
+        task_scripts!(ConfigureUnakiteTask),
+        task_scripts!(CreatePartitionsTask),
+        task_scripts!(InstallBaseTask),
+        task_scripts!(InstallBootloaderTask),
+        task_scripts!(InstallDesktopTask),
+        task_scripts!(InstallExtraPackagesTask),
+        task_scripts!(InstallFlatpakTask),
+        task_scripts!(InstallKernelsTask),
+        task_scripts!(InstallTimeshiftTask),
+        task_scripts!(InstallZRamDTask),
+        task_scripts!(SetupRootUserTask),
+        task_scripts!(SetupUsersTask),
+    ]
 }

src/tasks/setup_root_user.rs

@@ -1,10 +1,10 @@
 use embed_nu::rusty_value::*;
 use serde::Deserialize;
 
-use crate::script;
+use crate::task;
 
-script!(SetupRootUserScript {
-    file = "setup-root-user"
+task!(SetupRootUser {
+    name = "setup-root-user"
     args = RootUserConfig
 });
 

src/tasks/setup_users.rs

@@ -1,10 +1,10 @@
 use embed_nu::rusty_value::*;
 use serde::Deserialize;
 
-use crate::script;
+use crate::task;
 
-script!(SetupUsersScript {
-    file = "setup-users"
+task!(SetupUsers {
+    name = "setup-users"
     args = UsersConfig
 });
 

src/utils.rs

@@ -4,7 +4,6 @@ use std::{env, path::PathBuf};
 use tokio::fs;
 
 use crate::error::AppResult;
-
 use crate::tasks::all_tasks;
 
 const DEFAULT_CONFIG_DIR: &str = "/etc";
@@ -12,53 +11,45 @@ const DEFAULT_CONFIG_DIR: &str = "/etc";
 lazy_static::lazy_static! {
     pub static ref CFG_PATH: PathBuf = env::var("TRM_CFG_PATH").map(PathBuf::from).unwrap_or_else(|_| PathBuf::from(DEFAULT_CONFIG_DIR).join("tourmaline"));
     pub static ref SCRIPT_PATH: PathBuf = CFG_PATH.join("scripts");
-    pub static ref HOOK_PATH: PathBuf = CFG_PATH.join("hooks");
 }
 
 pub async fn generate_script_files<P: AsRef<Path>>(output: P) -> AppResult<()> {
-    let script_path = output.as_ref().join("scripts");
-    let hook_path = output.as_ref().join("hooks");
-
-    if !script_path.exists() {
-        fs::create_dir_all(&script_path).await?;
-    }
-    if !hook_path.exists() {
-        fs::create_dir_all(&hook_path).await?;
-    }
+    let output = output.as_ref();
 
     let tasks = all_tasks();
 
-    for task in tasks {
-        fs::write(
-            task.script_path(output.as_ref()),
-            r#"
-def main [cfg] {
-    echo "Executing Task with config" $cfg                
-}
-        "#
-            .to_string(),
-        )
-        .await?;
+    for (name, up_script, down_script) in tasks {
+        let script_dir = output.join(&name);
+
+        if !script_dir.exists() {
+            fs::create_dir_all(&script_dir).await?;
+        }
+
         fs::write(
-            task.pre_hook_path(output.as_ref()),
-            r#"
-def main [cfg] {
-    echo "Executing before Task with task config: " $cfg                
-    echo "The global config is: " $TRM_CONFIG
-}
+            output.join(up_script),
+            format!(
+                r#"
+# Applies all system changes of `{name}`
+def main [cfg] {{
+    echo "Executing up task `{name}` with config" $cfg
+}}
         "#
-            .to_string(),
+            )
+            .trim(),
         )
         .await?;
+
         fs::write(
-            task.post_hook_path(output.as_ref()),
-            r#"
-def main [cfg] {
-    echo "Executing after Task with task config: " $cfg                
-    echo "The global config is: " $TRM_CONFIG
-}
+            output.join(down_script),
+            format!(
+                r#"
+# Reverts all system changes of `{name}`
+def main [cfg] {{
+    echo "Executing up task `{name}` with config" $cfg
+}}
         "#
-            .to_string(),
+            )
+            .trim(),
         )
         .await?;
     }