wip: venv temp

py/private/run.tmpl.sh

@@ -35,16 +35,31 @@ function python_location {
 }
 
 VENV_TOOL="$(rlocation {{VENV_TOOL}})"
-VIRTUAL_ENV="$(alocation "${RUNFILES_DIR}/{{ARG_VENV_NAME}}")"
-export VIRTUAL_ENV
+RUNFILES_VIRTUAL_ENV="$(alocation "${RUNFILES_DIR}/{{ARG_VENV_NAME}}")"
+
+TMP_VENV=$(mktemp -d -t venv.XXXXXX)
+VIRTUAL_ENV="$TMP_VENV/{{ARG_VENV_NAME}}"
+mkdir -p "${VIRTUAL_ENV}"
+
+# Remove preexisting runfiles virtualenv symlink if it exists
+rm "${RUNFILES_VIRTUAL_ENV}" || true
+# Create a symlink to the virtualenv in the runfiles dir
+ln -s "$VIRTUAL_ENV" "${RUNFILES_VIRTUAL_ENV}"
+
+# We need to rewrite the pth file to be relative to the runfiles dir
+PTH="$(cat "$(rlocation {{ARG_PTH_FILE}})")"
+PTH="${PTH//..\/..\/..\/../.}"
 
 "${VENV_TOOL}" \
     --location "${VIRTUAL_ENV}" \
     --python "$(python_location)" \
-    --pth-file "$(rlocation {{ARG_PTH_FILE}})" \
+    --pth-file <(echo "$PTH") \
+    --pth-entry-prefix "${RUNFILES_DIR}" \
     --collision-strategy "{{ARG_COLLISION_STRATEGY}}" \
     --venv-name "{{ARG_VENV_NAME}}"
 
+export VIRTUAL_ENV="${RUNFILES_VIRTUAL_ENV}"
+
 PATH="${VIRTUAL_ENV}/bin:${PATH}"
 export PATH
 

py/tools/py/src/pth.rs

@@ -66,7 +66,9 @@ impl PthFile {
             line.clear();
 
             match entry.file_name() {
+
                 Some(name) if name == "site-packages" => {
+                    println!("{:#?}", dest.join(entry.clone()));
                     let src_dir = dest
                         .join(entry)
                         .canonicalize()