From 463bdf0978597fbb6fe83d16d53965ebdc884164 Mon Sep 17 00:00:00 2001
From: patryk4815 <patryk.sondej@grupawp.pl>
Date: Thu, 29 Aug 2024 23:09:13 +0200
Subject: [PATCH] nix/portable: add pycache (#2402)

---
 nix/portable.nix | 9 +++++----
 1 file changed, 5 insertions(+), 4 deletions(-)

diff --git a/nix/portable.nix b/nix/portable.nix
index a312235fa..f977dc064 100644
--- a/nix/portable.nix
+++ b/nix/portable.nix
@@ -20,10 +20,8 @@ let
     #!/bin/sh
     dir="$(cd -- "$(dirname "$(dirname "$(realpath "$0")")")" >/dev/null 2>&1 ; pwd -P)"
     export PYTHONHOME="$dir"
-    export PYTHONPYCACHEPREFIX="$dir/cache/"
     exec "$dir/lib/${ldName}" "$dir/exe/gdb" --quiet --early-init-eval-command="set auto-load safe-path /" --command=$dir/exe/gdbinit.py "$@"
   '';
-  # for cache: pwndbg --eval-command="py import compileall; compileall.compile_dir('/usr/lib/pwndbg/'); exit()"
 
   portable =
     pkgs.runCommand "portable-${pwndbg.name}"
@@ -33,14 +31,13 @@ let
           version = pwndbg.version;
           architecture = gdb.stdenv.targetPlatform.system;
         };
-        nativeBuildInputs = [ pkgs.makeWrapper ];
+        nativeBuildInputs = [ pkgs.makeWrapper pkgs.proot ];
       }
       ''
         mkdir -p $out/pwndbg/bin/
         mkdir -p $out/pwndbg/lib/
         mkdir -p $out/pwndbg/exe/
         mkdir -p $out/pwndbg/share/gdb/
-        mkdir -p $out/pwndbg/cache/
         touch $out/pwndbg/exe/.skip-venv
 
         cp -rf ${gdbBundledLib}/exe/* $out/pwndbg/exe/
@@ -58,6 +55,10 @@ let
 
         # fix python "subprocess.py" to use "/bin/sh" and not the nix'ed version, otherwise "gdb-pt-dump" is broken
         substituteInPlace $out/pwndbg/lib/${python3.libPrefix}/subprocess.py --replace "'${pkgs.bash}/bin/sh'" "'/bin/sh'"
+
+        # build pycache
+        chmod -R +w $out/pwndbg/lib/${python3.libPrefix}/site-packages/pwndbg
+        SOURCE_DATE_EPOCH=0 proot -b $out/pwndbg:/usr/lib/pwndbg ${pwndbgVenv}/bin/python3 -c "import compileall; compileall.compile_dir('/usr/lib/pwndbg/', force=True);"
       '';
 in
 portable