From 8f7ca8f00fd548351b350eef79da5dcdddb2fecb Mon Sep 17 00:00:00 2001
From: hacker1024 <hacker1024@users.sourceforge.net>
Date: Sun, 30 Jul 2023 11:28:13 +1000
Subject: [PATCH] Wrap ROS Python scripts

---
 distros/build-ros-package/default.nix | 12 +++++++++++-
 1 file changed, 11 insertions(+), 1 deletion(-)

diff --git a/distros/build-ros-package/default.nix b/distros/build-ros-package/default.nix
index 388c89f67f..cbde57a3bf 100644
--- a/distros/build-ros-package/default.nix
+++ b/distros/build-ros-package/default.nix
@@ -11,7 +11,8 @@
 , dontWrapQtApps ? true
 , nativeBuildInputs ? [ ]
 , CXXFLAGS ? ""
-, passthru ? {}
+, postFixup ? ""
+, passthru ? { }
 , ...
 }@args:
 
@@ -36,6 +37,8 @@ else stdenv.mkDerivation) (args // {
 
   nativeBuildInputs = nativeBuildInputs ++ [ pythonPackages.setuptools ];
 
+  dontWrapPythonPrograms = true;
+
   buildPhase = ''
     runHook preBuild
 
@@ -53,4 +56,11 @@ else stdenv.mkDerivation) (args // {
 
     runHook postInstall
   '';
+
+  postFixup = ''
+    ${postFixup}
+    find "$out/lib" -mindepth 1 -maxdepth 1 -type d ! -name '${pythonPackages.python.libPrefix}' -print0 | while read -d '''''' d; do
+      wrapPythonProgramsIn "$d" "$out $pythonPath"
+    done
+  '';
 })