From 6245ab9899eda00be24bfeef5f81948a92916df8 Mon Sep 17 00:00:00 2001
From: notgne2 <gen2@gen2.space>
Date: Sun, 8 Jan 2023 14:13:51 -0700
Subject: [PATCH] Use more complex activation wrapper and add workaround for
 deploy-rs#185

---
 flake.nix | 136 +++++++++++++++++++++++++++++++-----------------------
 1 file changed, 78 insertions(+), 58 deletions(-)

diff --git a/flake.nix b/flake.nix
index 9b8e6eb4..22ad4817 100644
--- a/flake.nix
+++ b/flake.nix
@@ -27,64 +27,84 @@
         ];
       };
     in
-    {
-      deploy-rs = {
-
-        deploy-rs = final.rustPlatform.buildRustPackage (darwinOptions // {
-          pname = "deploy-rs";
-          version = "0.1.0";
-
-          src = ./.;
-
-          cargoLock.lockFile = ./Cargo.lock;
-        }) // { meta.description = "A Simple multi-profile Nix-flake deploy tool"; };
-
-        lib = rec {
-
-          setActivate = builtins.trace
-            "deploy-rs#lib.setActivate is deprecated, use activate.noop, activate.nixos or activate.custom instead"
-            activate.custom;
-
-          activate = rec {
-            custom =
-              {
-                __functor = customSelf: base: activate:
-                  final.buildEnv {
-                    name = ("activatable-" + base.name);
-                    paths =
-                      [
-                        base
-                        (final.writeTextFile {
-                          name = base.name + "-activate-path";
-                          text = ''
-                            #!${final.runtimeShell}
-                            set -euo pipefail
-
-                            if [[ "''${DRY_ACTIVATE:-}" == "1" ]]
-                            then
-                                ${customSelf.dryActivate or "echo ${final.writeScript "activate" activate}"}
-                            elif [[ "''${BOOT:-}" == "1" ]]
-                            then
-                                ${customSelf.boot or "echo ${final.writeScript "activate" activate}"}
-                            else
-                                ${activate}
-                            fi
-                          '';
-                          executable = true;
-                          destination = "/deploy-rs-activate";
-                        })
-                        (final.writeTextFile {
-                            name = base.name + "-activate-rs";
-                            text = ''
-                            #!${final.runtimeShell}
-                            exec ${self.packages.${system}.default}/bin/activate "$@"
-                          '';
-                          executable = true;
-                          destination = "/activate-rs";
-                        })
-                      ];
-                  };
-              };
+        {
+          deploy-rs = {
+
+            deploy-rs = final.rustPlatform.buildRustPackage (darwinOptions // {
+              pname = "deploy-rs";
+              version = "0.1.0";
+
+              src = ./.;
+
+              cargoLock.lockFile = ./Cargo.lock;
+            }) // { meta.description = "A Simple multi-profile Nix-flake deploy tool"; };
+
+            lib = rec {
+
+              setActivate = builtins.trace
+                "deploy-rs#lib.setActivate is deprecated, use activate.noop, activate.nixos or activate.custom instead"
+                activate.custom;
+
+              activate = rec {
+                custom =
+                  {
+                  __functor = customSelf: base: activate:
+                    base.overrideAttrs (oldAttrs: {
+                      name = "activatable-${base.name}";
+                      buildCommand = ''
+                        set -euo pipefail
+                        ${nixpkgs.lib.concatStringsSep "\n" (map (outputName:
+                          let
+                            activatePath = final.writeTextFile {
+                              name = base.name + "-activate-path";
+                              text = ''
+                                #!${final.runtimeShell}
+                                set -euo pipefail
+
+                                if [[ "''${DRY_ACTIVATE:-}" == "1" ]]
+                                then
+                                    ${customSelf.dryActivate or "echo ${final.writeScript "activate" activate}"}
+                                elif [[ "''${BOOT:-}" == "1" ]]
+                                then
+                                    ${customSelf.boot or "echo ${final.writeScript "activate" activate}"}
+                                else
+                                    ${activate}
+                                fi
+                              '';
+                              executable = true;
+                            };
+
+                            activateRs = final.writeTextFile {
+                              name = base.name + "-activate-rs";
+                              text = ''
+                                #!${final.runtimeShell}
+                                exec ${self.packages.${system}.default}/bin/activate "$@"
+                              '';
+                              executable = true;
+                            };
+                          in (''
+                            ${final.coreutils}/bin/mkdir "''$${outputName}"
+
+                            echo "Linking activation components in ${outputName}"
+                            ${final.coreutils}/bin/ln -s "${activatePath}" "''$${outputName}/deploy-rs-activate"
+                            ${final.coreutils}/bin/ln -s "${activateRs}" "''$${outputName}/activate-rs"
+
+                            echo "Linking output contents of ${outputName}"
+                            ${final.findutils}/bin/find "${base.${outputName}}" -maxdepth 1 | while read -r file; do
+                              ${final.coreutils}/bin/ln -s "$file" "''$${outputName}/$(${final.coreutils}/bin/basename "$file")"
+                            done
+                          '' + nixpkgs.lib.optionalString
+                            (outputName == "out") ''
+                              # Workaround for https://github.com/serokell/deploy-rs/issues/185
+                              if [ -x "${base.${outputName}}/prepare-root" ]; then
+                                echo "Copying prepare-root"
+                                rm "$out/prepare-root" || :
+                                cp "${base.${outputName}}/prepare-root" "$out/prepare-root"
+                              fi
+                            '')) (base.outputs or [ "out" ]))}
+                      '';
+                    });
+                };
 
             nixos = base:
               (custom // {