Improve Subsystem paths handling on Windows hosts

Signed-off-by: Arthur Sengileyev <[email protected]>

Author: arixmkii

cmd/limactl/shell.go

@@ -8,14 +8,14 @@ import (
 	"fmt"
 	"os"
 	"os/exec"
-	"path"
 	"runtime"
 	"strconv"
 	"strings"
 
 	"al.essio.dev/pkg/shellescape"
 	"github.com/coreos/go-semver/semver"
 	"github.com/lima-vm/lima/pkg/ioutilx"
+	"github.com/lima-vm/lima/pkg/limayaml"
 	"github.com/lima-vm/lima/pkg/sshutil"
 	"github.com/lima-vm/lima/pkg/store"
 	"github.com/mattn/go-isatty"
@@ -93,10 +93,10 @@ func shellAction(cmd *cobra.Command, args []string) error {
 	if workDir != "" {
 		changeDirCmd = fmt.Sprintf("cd %s || exit 1", shellescape.Quote(workDir))
 		// FIXME: check whether y.Mounts contains the home, not just len > 0
-	} else if len(inst.Config.Mounts) > 0 {
+	} else if len(inst.Config.Mounts) > 0 || inst.VMType == limayaml.WSL2 {
 		hostCurrentDir, err := os.Getwd()
 		if err == nil && runtime.GOOS == "windows" {
-			hostCurrentDir, err = mountDirFromWindowsDir(hostCurrentDir)
+			hostCurrentDir, err = mountDirFromWindowsDir(inst, hostCurrentDir)
 		}
 		if err == nil {
 			changeDirCmd = fmt.Sprintf("cd %s", shellescape.Quote(hostCurrentDir))
@@ -106,7 +106,7 @@ func shellAction(cmd *cobra.Command, args []string) error {
 		}
 		hostHomeDir, err := os.UserHomeDir()
 		if err == nil && runtime.GOOS == "windows" {
-			hostHomeDir, err = mountDirFromWindowsDir(hostHomeDir)
+			hostHomeDir, err = mountDirFromWindowsDir(inst, hostHomeDir)
 		}
 		if err == nil {
 			changeDirCmd = fmt.Sprintf("%s || cd %s", changeDirCmd, shellescape.Quote(hostHomeDir))
@@ -198,12 +198,12 @@ func shellAction(cmd *cobra.Command, args []string) error {
 	return sshCmd.Run()
 }
 
-func mountDirFromWindowsDir(dir string) (string, error) {
-	dir, err := ioutilx.WindowsSubsystemPath(dir)
-	if err == nil && !strings.HasPrefix(dir, "/mnt/") {
-		dir = path.Join("/mnt", dir)
+func mountDirFromWindowsDir(inst *store.Instance, dir string) (string, error) {
+	if inst.VMType == limayaml.WSL2 {
+		distroName := "lima-" + inst.Name
+		return ioutilx.WindowsSubsystemPathForLinux(dir, distroName)
 	}
-	return dir, err
+	return ioutilx.WindowsSubsystemPath(dir)
 }
 
 func shellBashComplete(cmd *cobra.Command, _ []string, _ string) ([]string, cobra.ShellCompDirective) {

pkg/ioutilx/ioutilx.go

@@ -53,7 +53,16 @@ func WindowsSubsystemPath(orig string) (string, error) {
 	out, err := exec.Command("cygpath", filepath.ToSlash(orig)).CombinedOutput()
 	if err != nil {
 		logrus.WithError(err).Errorf("failed to convert path to mingw, maybe not using Git ssh?")
-		return orig, err
+		return "", err
+	}
+	return strings.TrimSpace(string(out)), nil
+}
+
+func WindowsSubsystemPathForLinux(orig, distro string) (string, error) {
+	out, err := exec.Command("wsl", "-d", distro, "--exec", "wslpath", filepath.ToSlash(orig)).CombinedOutput()
+	if err != nil {
+		logrus.WithError(err).Errorf("failed to convert path to mingw, maybe wsl command is not operational?")
+		return "", err
 	}
 	return strings.TrimSpace(string(out)), nil
 }

pkg/osutil/user.go

@@ -7,6 +7,7 @@ import (
 	"fmt"
 	"os/exec"
 	"os/user"
+	"path"
 	"path/filepath"
 	"regexp"
 	"runtime"
@@ -149,7 +150,19 @@ func LimaUser(limaVersion string, warn bool) *user.User {
 			if err != nil {
 				logrus.Debug(err)
 			} else {
-				home += ".linux"
+				// Trim mount prefix within Subsystem
+				// cygwin/msys2 cygpath could have prefix for drive mounts configured via /etc/fstab
+				// wsl wslpath could have prefix for drive mounts configured via [automount] section in wsl.conf
+				drivePath, err := ioutilx.WindowsSubsystemPath(filepath.VolumeName(limaUser.HomeDir) + "/")
+				if err != nil {
+					logrus.Debug(err)
+				} else {
+					prefix := path.Dir(strings.TrimSuffix(drivePath, "/"))
+					if prefix != "/" {
+						home = strings.TrimPrefix(home, prefix)
+					}
+					home += ".linux"
+				}
 			}
 			if home == "" {
 				drive := filepath.VolumeName(limaUser.HomeDir)
@@ -160,10 +173,6 @@ func LimaUser(limaVersion string, warn bool) *user.User {
 				home += ".linux"
 			}
 			if !regexPath.MatchString(limaUser.HomeDir) {
-				// Trim prefix of well known default mounts
-				if strings.HasPrefix(home, "/mnt/") {
-					home = strings.TrimPrefix(home, "/mnt")
-				}
 				warning := fmt.Sprintf("local home %q is not a valid Linux path (must match %q); using %q home instead",
 					limaUser.HomeDir, regexPath.String(), home)
 				warnings = append(warnings, warning)