feat: mock invoke EnvKeys

internal/env/path.go

@@ -69,8 +69,6 @@ func (p *Paths) ToBinPaths() (*Paths, error) {
 // NewPaths returns a new Paths.
 // from is the source of the paths.
 // If from is OsPaths, it returns the paths from the environment variable PATH.
-// If from is PreviousPaths, it returns the paths from the environment variable __VFOX_PREVIOUS_PATHS
-// If from is neither OsPaths nor PreviousPaths, it returns an empty Paths.
 func NewPaths(from PathFrom) *Paths {
 	var paths []string
 	switch from {

internal/package.go

@@ -32,7 +32,7 @@ type LocationPackage struct {
 	location Location
 }
 
-func NewLocationPackage(version Version, sdk *Sdk, location Location) (*LocationPackage, error) {
+func newLocationPackage(version Version, sdk *Sdk, location Location) (*LocationPackage, error) {
 	var mockPath string
 	switch location {
 	case OriginalLocation:
@@ -112,6 +112,19 @@ func (l *LocationPackage) Link() (*Package, error) {
 	return targetPackage, nil
 }
 
+// checkPackageValid checks if the package is valid
+func checkPackageValid(p *Package) bool {
+	if !util.FileExists(p.Main.Path) {
+		return false
+	}
+	for _, a := range p.Additions {
+		if !util.FileExists(a.Path) {
+			return false
+		}
+	}
+	return true
+}
+
 type Package struct {
 	Main      *Info
 	Additions []*Info

internal/sdk.go

@@ -261,7 +261,7 @@ func (b *Sdk) Available(args []string) ([]*Package, error) {
 }
 
 func (b *Sdk) ToLinkPackage(version Version, location Location) error {
-	linkPackage, err := b.GetLinkPackage(version, ShellLocation)
+	linkPackage, err := b.GetLinkPackage(version, location)
 	if err != nil {
 		return err
 	}
@@ -271,10 +271,13 @@ func (b *Sdk) ToLinkPackage(version Version, location Location) error {
 
 // GetLinkPackage will make symlink according Location and return the sdk package.
 func (b *Sdk) GetLinkPackage(version Version, location Location) (*LocationPackage, error) {
-	return NewLocationPackage(version, b, location)
+	return newLocationPackage(version, b, location)
 }
 
-func (b *Sdk) EnvKeysWithoutLink(version Version, location Location) (*env.Envs, error) {
+// MockEnvKeys It just simulates to get the environment configuration information,
+// if the corresponding location of the package does not exist, then it will return
+// the empty environment information, without calling the EnvKeys hook.
+func (b *Sdk) MockEnvKeys(version Version, location Location) (*env.Envs, error) {
 	label := b.label(version)
 	if !b.CheckExists(version) {
 		return nil, fmt.Errorf("%s is not installed", label)
@@ -284,6 +287,14 @@ func (b *Sdk) EnvKeysWithoutLink(version Version, location Location) (*env.Envs,
 		return nil, err
 	}
 	sdkPackage := linkPackage.ConvertLocation()
+	if !checkPackageValid(sdkPackage) {
+		logger.Debugf("Package is invalid: %v\n", sdkPackage)
+		return &env.Envs{
+			Variables: make(env.Vars),
+			Paths:     env.NewPaths(env.EmptyPaths),
+			BinPaths:  env.NewPaths(env.EmptyPaths),
+		}, nil
+	}
 	keys, err := b.Plugin.EnvKeys(sdkPackage)
 	if err != nil {
 		return nil, fmt.Errorf("plugin [EnvKeys] error: err:%w", err)
@@ -305,6 +316,7 @@ func (b *Sdk) EnvKeys(version Version, location Location) (*env.Envs, error) {
 	if err != nil {
 		return nil, err
 	}
+
 	keys, err := b.Plugin.EnvKeys(sdkPackage)
 	if err != nil {
 		return nil, fmt.Errorf("plugin [EnvKeys] error: err:%w", err)
@@ -700,7 +712,7 @@ func (b *Sdk) ClearCurrentEnv() error {
 	}
 
 	if current != "" {
-		envKeys, err := b.EnvKeysWithoutLink(current, GlobalLocation)
+		envKeys, err := b.MockEnvKeys(current, GlobalLocation)
 		if err != nil {
 			return err
 		}
@@ -715,10 +727,11 @@ func (b *Sdk) ClearCurrentEnv() error {
 
 		envManager := b.sdkManager.EnvManager
 		fmt.Println("Cleaning up env config...")
+		envKeys.Paths.Add(filepath.Join(b.InstallPath, "current"))
 		_ = envManager.Remove(envKeys)
 		_ = envManager.Flush()
 
-		envKeys, err = b.EnvKeysWithoutLink(current, OriginalLocation)
+		envKeys, err = b.MockEnvKeys(current, OriginalLocation)
 		if err != nil {
 			return err
 		}

internal/util/file.go

@@ -26,11 +26,30 @@ import (
 )
 
 func FileExists(filename string) bool {
-	_, err := os.Stat(filename)
-	if os.IsNotExist(err) {
+	fileInfo, err := os.Lstat(filename)
+	if err != nil {
+		if os.IsNotExist(err) {
+			return false
+		}
 		return false
 	}
-	return err == nil
+
+	if fileInfo.Mode()&os.ModeSymlink != 0 {
+		target, err := os.Readlink(filename)
+		if err != nil {
+			return false
+		}
+		_, err = os.Stat(target)
+		if err != nil {
+			if os.IsNotExist(err) {
+				return false
+			}
+			return false
+		}
+		return true
+	}
+
+	return true
 }
 
 func CopyFile(src, dst string) error {