Mostly replace profile scripts with Rust

src/action/common/configure_shell_profile.rs

@@ -9,8 +9,8 @@ use std::path::{Path, PathBuf};
 use tokio::task::JoinSet;
 use tracing::{span, Instrument, Span};
 
-const PROFILE_NIX_FILE_SHELL: &str = "/nix/var/nix/profiles/default/etc/profile.d/nix-daemon.sh";
-const PROFILE_NIX_FILE_FISH: &str = "/nix/var/nix/profiles/default/etc/profile.d/nix-daemon.fish";
+pub(crate) const PROFILE_NIX_FILE_SHELL: &str = "/nix/nix-installer.d/profile.sh";
+pub(crate) const PROFILE_NIX_FILE_FISH: &str = "/nix/nix-installer.d/profile.fish";
 
 /**
 Configure any detected shell profiles to include Nix support
@@ -31,14 +31,47 @@ impl ConfigureShellProfile {
         let mut create_directories = Vec::default();
 
         let shell_buf = format!(
-            "\n\
-            # Nix\n\
-            if [ -e '{PROFILE_NIX_FILE_SHELL}' ]; then\n\
-            {inde}. '{PROFILE_NIX_FILE_SHELL}'\n\
-            fi\n\
-            # End Nix\n
-        \n",
-            inde = "    ", // indent
+            r#"
+
+# Begin Nix
+if [ -f '{PROFILE_NIX_FILE_SHELL}' ]; then
+    . '{PROFILE_NIX_FILE_SHELL}'
+fi
+# End Nix
+
+"#
+        );
+
+        create_directories.push(
+            CreateDirectory::plan("/nix/nix-installer.d", None, None, 0o0755, false)
+                .await
+                .map_err(Self::error)?,
+        );
+
+        create_or_insert_files.push(
+            CreateOrInsertIntoFile::plan(
+                PROFILE_NIX_FILE_SHELL,
+                None,
+                None,
+                0o644,
+                include_str!("./profiles/profile.sh").to_string(),
+                create_or_insert_into_file::Position::Beginning,
+            )
+            .await
+            .map_err(Self::error)?,
+        );
+
+        create_or_insert_files.push(
+            CreateOrInsertIntoFile::plan(
+                PROFILE_NIX_FILE_FISH,
+                None,
+                None,
+                0o644,
+                include_str!("./profiles/profile.fish").to_string(),
+                create_or_insert_into_file::Position::Beginning,
+            )
+            .await
+            .map_err(Self::error)?,
         );
 
         for profile_target in locations.bash.iter().chain(locations.zsh.iter()) {
@@ -67,14 +100,15 @@ impl ConfigureShellProfile {
         }
 
         let fish_buf = format!(
-            "\n\
-            # Nix\n\
-            if test -e '{PROFILE_NIX_FILE_FISH}'\n\
-            {inde}. '{PROFILE_NIX_FILE_FISH}'\n\
-            end\n\
-            # End Nix\n\
-        \n",
-            inde = "    ", // indent
+            r#"
+
+# Begin Nix
+if [ -f {PROFILE_NIX_FILE_FISH} ]; then
+    . {PROFILE_NIX_FILE_FISH}
+fi
+# End Nix
+
+"#
         );
 
         for fish_prefix in &locations.fish.confd_prefixes {

src/action/common/profiles/profile.fish

@@ -0,0 +1,6 @@
+if [ -f /nix/nix-installer ] && [ -x /nix/nix-installer ] && [ -z "${__ETC_PROFILE_NIX_SOURCED:-}" ]; then
+    /nix/nix-installer export --format null-separated \
+        | while read --null key
+            read --export --null "$key"
+        end
+fi

src/action/common/profiles/profile.sh

@@ -0,0 +1,11 @@
+# shellcheck shell=sh
+if [ -f /nix/nix-installer ] && [ -x /nix/nix-installer ] && [ -z "${__ETC_PROFILE_NIX_SOURCED:-}" ]; then
+    NIX_INSTALLER_EXPORT_DATA=$(/nix/nix-installer export --verbose --verbose --format space-newline-separated);
+    while read -r nix_installer_export_key nix_installer_export_value; do
+        export "$nix_installer_export_key=$nix_installer_export_value";
+    done <<DATA_INPUT
+$NIX_INSTALLER_EXPORT_DATA
+DATA_INPUT
+
+    unset NIX_INSTALLER_EXPORT_DATA nix_installer_export_key nix_installer_export_value
+fi

src/action/macos/configure_remote_building.rs

@@ -1,10 +1,10 @@
-use crate::action::base::{create_or_insert_into_file, CreateOrInsertIntoFile};
-use crate::action::{Action, ActionDescription, ActionError, ActionTag, StatefulAction};
-
 use std::path::Path;
+
 use tracing::{span, Instrument, Span};
 
-const PROFILE_NIX_FILE_SHELL: &str = "/nix/var/nix/profiles/default/etc/profile.d/nix-daemon.sh";
+use crate::action::base::{create_or_insert_into_file, CreateOrInsertIntoFile};
+use crate::action::common::configure_shell_profile::PROFILE_NIX_FILE_SHELL;
+use crate::action::{Action, ActionDescription, ActionError, ActionTag, StatefulAction};
 
 /**
 Configure macOS's zshenv to load the Nix environment when ForceCommand is used.
@@ -20,12 +20,14 @@ impl ConfigureRemoteBuilding {
     pub async fn plan() -> Result<StatefulAction<Self>, ActionError> {
         let shell_buf = format!(
             r#"
+
 # Set up Nix only on SSH connections
 # See: https://github.com/DeterminateSystems/nix-installer/pull/714
-if [ -e '{PROFILE_NIX_FILE_SHELL}' ] && [ -n "${{SSH_CONNECTION}}" ] && [ "${{SHLVL}}" -eq 1 ]; then
+if [ -n "${{SSH_CONNECTION}}" ] && [ "${{SHLVL}}" -eq 1 ] && [ -f '{PROFILE_NIX_FILE_SHELL}' ]; then
     . '{PROFILE_NIX_FILE_SHELL}'
 fi
 # End Nix
+
 "#
         );
 

src/cli/mod.rs

@@ -49,6 +49,7 @@ impl CommandExecute for NixInstallerCli {
             NixInstallerSubcommand::Install(install) => install.execute().await,
             NixInstallerSubcommand::Repair(restore_shell) => restore_shell.execute().await,
             NixInstallerSubcommand::Uninstall(revert) => revert.execute().await,
+            NixInstallerSubcommand::Export(export) => export.execute().await,
         }
     }
 }