use pushd and popd to tackle venv

2 files changed, 11 insertions, 14 deletions

diff --git a/xmake/scripts/virtualenvs/register-virtualenvs.sh b/xmake/scripts/virtualenvs/register-virtualenvs.sh
index 15ed63bcd..1f389b99e 100644
--- a/xmake/scripts/virtualenvs/register-virtualenvs.sh
+++ b/xmake/scripts/virtualenvs/register-virtualenvs.sh
@@ -21,6 +21,7 @@
 if test "${XMAKE_ROOTDIR}"; then
     export XMAKE_EXE=${XMAKE_ROOTDIR}/xmake
 else
+    local -i XMAKE_ROOTDIR=~/.local/bin
     export XMAKE_EXE=xmake
 fi
 
@@ -69,22 +70,20 @@ function xrepo {
                         unset XMAKE_PROMPT_BACKUP
                         unset XMAKE_ENV_BACKUP
                     fi
-                    local currentTest="$("$XMAKE_EXE" lua --quiet private.xrepo.action.env)" || return 1
-                    if [ ! -z "$currentTest" ]; then
-                        echo "error: corrupt xmake.lua detected in the current directory!"
-                        return 1
-                    fi
-                    "$XMAKE_EXE" lua private.xrepo.action.env.info config $bnd || return 1
-                    local prompt="$("$XMAKE_EXE" lua --quiet private.xrepo.action.env.info prompt $bnd)" || return 1
+                    pushd ${XMAKE_ROOTDIR} 1>/dev/null
+                    "$XMAKE_EXE" lua private.xrepo.action.env.info config $bnd || (popd 1>/dev/null && return 1)
+                    local prompt="$("$XMAKE_EXE" lua --quiet private.xrepo.action.env.info prompt $bnd)" || (popd 1>/dev/null && return 1)
                     if [ -z "${prompt+x}" ]; then
+                        popd 1>/dev/null
                         echo "error: invalid environment!"
                         return 1
                     fi
-                    local activateCommand="$("$XMAKE_EXE" lua --quiet private.xrepo.action.env.info script.bash $bnd)" || return 1
+                    local activateCommand="$("$XMAKE_EXE" lua --quiet private.xrepo.action.env.info script.bash $bnd)" || (popd 1>/dev/null && return 1)
                     export XMAKE_ENV_BACKUP="$("$XMAKE_EXE" lua private.xrepo.action.env.info envfile $bnd)"
                     export XMAKE_PROMPT_BACKUP="${PS1}"
                     "$XMAKE_EXE" lua --quiet private.xrepo.action.env.info backup.bash $bnd 1>"$XMAKE_ENV_BACKUP"
                     eval "$activateCommand"
+                    popd 1>/dev/null
                     PS1="${prompt} $PS1"
                 else
                     "$XMAKE_EXE" lua private.xrepo "$@"
diff --git a/xmake/scripts/xrepo-hook.psm1 b/xmake/scripts/xrepo-hook.psm1
index b05557efa..c0a6aad94 100644
--- a/xmake/scripts/xrepo-hook.psm1
+++ b/xmake/scripts/xrepo-hook.psm1
@@ -15,12 +15,6 @@ function Enter-XrepoEnvironment {
     );
 
     begin {
-        $currentTest = (& $Env:XMAKE_EXE lua --quiet private.xrepo.action.env | Out-String);
-        if (-not $? -or $currentTest) {
-            Write-Host "error: corrupt xmake.lua detected in the current directory!";
-            Exit 1;
-        }
-
         $script:xrepoOldEnvs = (Get-ChildItem -Path Env:);
 
         if (-not $bnd) {
@@ -39,8 +33,10 @@ function Enter-XrepoEnvironment {
 
             $activateCommand = (& $Env:XMAKE_EXE lua --quiet private.xrepo.action.env.info script.powershell | Out-String);
         } else {
+            Push-Location $Env:XMAKE_ROOTDIR;
             & $Env:XMAKE_EXE lua private.xrepo.action.env.info config $bnd;
             if (-not $?) {
+                Pop-Location;
                 Exit 1;
             }
 
@@ -48,6 +44,7 @@ function Enter-XrepoEnvironment {
             $xrepoPrompt = (& $Env:XMAKE_EXE lua --quiet private.xrepo.action.env.info prompt $bnd | Out-String);
             $Env:XMAKE_COLORTERM = $xmakeColorTermBackup;
             if (-not $xrepoPrompt.StartsWith("[")) {
+                Pop-Location;
                 Write-Host "error: invalid environment!";
                 Exit 1;
             }
@@ -58,6 +55,7 @@ function Enter-XrepoEnvironment {
         Write-Verbose "[xrepo env script.powershell]`n$activateCommand";
         Invoke-Expression -Command $activateCommand;
 
+        if ($bnd) { Pop-Location; }
         $Env:XMAKE_PROMPT_MODIFIER = $xrepoPrompt.Trim() + " ";
     }
     process {}