From 1497d982d2c1b3e65f17c207cbb12ca7b66745d1 Mon Sep 17 00:00:00 2001
From: Gavin John <gavinnjohn@gmail.com>
Date: Sun, 25 Aug 2024 14:07:51 -0700
Subject: [PATCH] envision: autopatchelf after build

---
 pkgs/by-name/en/envision/autopatchelf.patch | 108 +++++++++++++++++++
 pkgs/by-name/en/envision/package.nix        | 110 +++++++++++++-------
 2 files changed, 181 insertions(+), 37 deletions(-)
 create mode 100644 pkgs/by-name/en/envision/autopatchelf.patch

diff --git a/pkgs/by-name/en/envision/autopatchelf.patch b/pkgs/by-name/en/envision/autopatchelf.patch
new file mode 100644
index 0000000000000..1d22d9813618e
--- /dev/null
+++ b/pkgs/by-name/en/envision/autopatchelf.patch
@@ -0,0 +1,108 @@
+diff --git a/src/builders/build_basalt.rs b/src/builders/build_basalt.rs
+index 237fc24..3c9c8ec 100644
+--- a/src/builders/build_basalt.rs
++++ b/src/builders/build_basalt.rs
+@@ -86,6 +86,13 @@ pub fn get_build_basalt_jobs(profile: &Profile, clean_build: bool) -> VecDeque<W
+             format!("{}/share/basalt/thirdparty", profile.prefix),
+         ]),
+     ));
++    jobs.push_back(WorkerJob::new_cmd(
++        None,
++        "autopatchelf".into(),
++        Some(vec![
++            build_dir.clone()
++        ]),
++    ));
+ 
+     jobs
+ }
+diff --git a/src/builders/build_libsurvive.rs b/src/builders/build_libsurvive.rs
+index 8ddd206..587de7f 100644
+--- a/src/builders/build_libsurvive.rs
++++ b/src/builders/build_libsurvive.rs
+@@ -60,6 +60,13 @@ pub fn get_build_libsurvive_jobs(profile: &Profile, clean_build: bool) -> VecDeq
+     }
+     jobs.push_back(cmake.get_build_job());
+     jobs.push_back(cmake.get_install_job());
++    jobs.push_back(WorkerJob::new_cmd(
++        None,
++        "autopatchelf".into(),
++        Some(vec![
++            build_dir.clone()
++        ]),
++    ));
+ 
+     jobs
+ }
+diff --git a/src/builders/build_monado.rs b/src/builders/build_monado.rs
+index fb049c5..e21a764 100644
+--- a/src/builders/build_monado.rs
++++ b/src/builders/build_monado.rs
+@@ -68,6 +68,13 @@ pub fn get_build_monado_jobs(profile: &Profile, clean_build: bool) -> VecDeque<W
+     }
+     jobs.push_back(cmake.get_build_job());
+     jobs.push_back(cmake.get_install_job());
++    jobs.push_back(WorkerJob::new_cmd(
++        None,
++        "autopatchelf".into(),
++        Some(vec![
++            build_dir.clone()
++        ]),
++    ));
+ 
+     jobs
+ }
+diff --git a/src/builders/build_opencomposite.rs b/src/builders/build_opencomposite.rs
+index bf45567..9fe0264 100644
+--- a/src/builders/build_opencomposite.rs
++++ b/src/builders/build_opencomposite.rs
+@@ -43,6 +43,13 @@ pub fn get_build_opencomposite_jobs(profile: &Profile, clean_build: bool) -> Vec
+         jobs.push_back(cmake.get_prepare_job());
+     }
+     jobs.push_back(cmake.get_build_job());
++    jobs.push_back(WorkerJob::new_cmd(
++        None,
++        "autopatchelf".into(),
++        Some(vec![
++            build_dir.clone()
++        ]),
++    ));
+ 
+     jobs
+ }
+diff --git a/src/builders/build_openhmd.rs b/src/builders/build_openhmd.rs
+index 5fe612c..5211229 100644
+--- a/src/builders/build_openhmd.rs
++++ b/src/builders/build_openhmd.rs
+@@ -55,6 +55,13 @@ pub fn get_build_openhmd_jobs(profile: &Profile, clean_build: bool) -> VecDeque<
+     }
+     jobs.push_back(cmake.get_build_job());
+     jobs.push_back(cmake.get_install_job());
++    jobs.push_back(WorkerJob::new_cmd(
++        None,
++        "autopatchelf".into(),
++        Some(vec![
++            build_dir.clone()
++        ]),
++    ));
+ 
+     jobs
+ }
+diff --git a/src/builders/build_wivrn.rs b/src/builders/build_wivrn.rs
+index 52bfb66..b28ab5c 100644
+--- a/src/builders/build_wivrn.rs
++++ b/src/builders/build_wivrn.rs
+@@ -52,6 +52,13 @@ pub fn get_build_wivrn_jobs(profile: &Profile, clean_build: bool) -> VecDeque<Wo
+     }
+     jobs.push_back(cmake.get_build_job());
+     jobs.push_back(cmake.get_install_job());
++    jobs.push_back(WorkerJob::new_cmd(
++        None,
++        "autopatchelf".into(),
++        Some(vec![
++            build_dir.clone()
++        ]),
++    ));
+ 
+     jobs
+ }
diff --git a/pkgs/by-name/en/envision/package.nix b/pkgs/by-name/en/envision/package.nix
index 665206d447405..90614a2118f26 100644
--- a/pkgs/by-name/en/envision/package.nix
+++ b/pkgs/by-name/en/envision/package.nix
@@ -3,43 +3,32 @@
   envision-unwrapped,
 }:
 
-buildFHSEnv {
-  name = "envision";
-
-  extraOutputsToInstall = [ "dev" ];
-
-  strictDeps = true;
-
-  # TODO: I'm pretty suspicious of this list of additonal required dependencies. Are they all really needed?
-  targetPkgs =
-    pkgs:
-    [ pkgs.envision-unwrapped ]
-    ++ (with pkgs; [
-      stdenv.cc.libc
-      gcc
-    ])
-    ++ (
-      # OpenHMD dependencies
-      pkgs.openhmd.buildInputs ++ pkgs.openhmd.nativeBuildInputs
-    )
-    ++ (
-      # OpenComposite dependencies
-      pkgs.opencomposite.buildInputs ++ pkgs.opencomposite.nativeBuildInputs
-    )
-    ++ (
-      # Monado dependencies
+let
+  runtimeBuildDeps =
+    pkgs':
+    with pkgs';
+    (
       (
-        pkgs.monado.buildInputs
-        ++ pkgs.monado.nativeBuildInputs
-        ++ (with pkgs; [
+        # OpenHMD dependencies
+        openhmd.buildInputs ++ openhmd.nativeBuildInputs
+      )
+      ++ (
+        # OpenComposite dependencies
+        opencomposite.buildInputs ++ opencomposite.nativeBuildInputs ++ [ boost ]
+      )
+      ++ (
+        # Monado dependencies
+        monado.buildInputs
+        ++ monado.nativeBuildInputs
+        ++ [
           # Additional dependencies required by Monado when built using Envision
-          mesa
-          shaderc
+          mesa # TODO: Does this really need "mesa-common-dev"?
           xorg.libX11
           xorg.libxcb
           xorg.libXrandr
           xorg.libXrender
           xorg.xorgproto
+
           # Additional dependencies required for Monado WMR support
           bc
           fmt
@@ -51,7 +40,17 @@ buildFHSEnv {
           lz4.dev
           tbb
           libxkbcommon
-        ])
+
+          # Not required for build, but autopatchelf requires them
+          glibc
+          SDL2
+          bluez
+          librealsense
+          onnxruntime
+          libusb1
+          libjpeg
+          libGL
+        ]
       )
     )
     ++ (
@@ -63,13 +62,47 @@ buildFHSEnv {
       ++ pkgs.wivrn.nativeBuildInputs
       ++ (with pkgs; [
         glib
+        avahi
+        cmake
+        cli11
+        ffmpeg
+        git
+        gst_all_1.gstreamer
+        gst_all_1.gst-plugins-base
         libmd
         ninja
-      ])
-      ++ (with pkgs; [
-        android-tools # For adb installing WiVRn APKs
-      ])
-    );
+        nlohmann_json
+        openxr-loader
+        pipewire
+        systemdLibs # udev
+        vulkan-loader
+        vulkan-headers
+        x264
+      ]
+    )
+  );
+in
+buildFHSEnv rec {
+  name = "envision";
+
+  extraOutputsToInstall = [
+    "dev"
+    "lib"
+  ];
+
+  strictDeps = true;
+
+  targetPkgs =
+    pkgs':
+    [
+      (pkgs'.envision-unwrapped.overrideAttrs (oldAttrs: {
+        patches = (oldAttrs.patches or [ ]) ++ [ ./autopatchelf.patch ]; # Adds an envision build step to run autopatchelf
+      }))
+      pkgs'.autopatchelf
+      pkgs'.gcc
+      pkgs'.android-tools # For adb installing WiVRn APKs
+    ]
+    ++ (runtimeBuildDeps pkgs');
 
   profile = ''
     export CMAKE_LIBRARY_PATH=/usr/lib
@@ -81,7 +114,10 @@ buildFHSEnv {
     ln -s ${envision-unwrapped}/share $out/share
   '';
 
-  runScript = "envision";
+  # Putting libgcc.lib in runtimeBuildDeps causes error "ld: cannot find crt1.o: No such file or directory"
+  runScript = "env libs=${
+    lib.makeLibraryPath ((runtimeBuildDeps pkgs) ++ [ pkgs.libgcc.lib ])
+  } envision --skip-dependency-check";
 
   meta = envision-unwrapped.meta // {
     description = "${envision-unwrapped.meta.description} (with build environment)";