Merge pull request #386 from zong-zhe/fix-run-bug

fix: fix wrong root path with 'kcl run'

pkg/client/client_test.go

@@ -1760,6 +1760,7 @@ func TestRunLocalWithoutArgs(t *testing.T) {
 		{"run_5", true, "", "kcl_6: KCL 6\na: sub6\nkcl_7: KCL 7\nb: sub7"},
 		{filepath.Join("run_6", "main"), true, "", "The_sub_kcl_program: Hello Sub World!\nThe_first_kcl_program: Hello World!"},
 		{"run_7", true, "", "hello: Hello World!\nThe_first_kcl_program: Hello World!"},
+		{filepath.Join("run_8", "sub"), true, "", "sub: Hello Sub !"},
 	}
 
 	for _, test := range tests {

pkg/client/run.go

@@ -461,6 +461,18 @@ func (o *RunOptions) getRootPkgSource() (*downloader.Source, error) {
 				}
 			}
 		}
+
+		if rootPkgSource.IsLocalKPath() || rootPkgSource.IsDir() {
+			rootPath, err = rootPkgSource.FindRootPath()
+			if err != nil {
+				return nil, err
+			}
+
+			rootPkgSource, err = downloader.NewSourceFromStr(rootPath)
+			if err != nil {
+				return nil, err
+			}
+		}
 	}
 
 	if rootPkgSource == nil {

pkg/client/test_data/test_run_options/no_args/run_8/kcl.mod

@@ -0,0 +1,5 @@
+[package]
+name = "run_7"
+edition = "v0.9.0"
+version = "0.0.1"
+

pkg/client/test_data/test_run_options/no_args/run_8/main.k

@@ -0,0 +1 @@
+hello = 'Hello World!'

pkg/client/test_data/test_run_options/no_args/run_8/sub/main.k

@@ -0,0 +1 @@
+sub = 'Hello Sub !'

pkg/client/visitor.go

@@ -72,7 +72,7 @@ func (vpv *VirtualPkgVisitor) Visit(s *downloader.Source, v visitFunc) error {
 		return fmt.Errorf("source is not local")
 	}
 
-	sourcePath, err := s.FindRootPath()
+	sourcePath, err := s.ToFilePath()
 	if err != nil {
 		return err
 	}
@@ -244,7 +244,7 @@ func NewVisitor(source downloader.Source, kpmcli *KpmClient) Visitor {
 	} else if source.IsLocalTarPath() || source.IsLocalTgzPath() {
 		return NewArchiveVisitor(NewPkgVisitor(kpmcli))
 	} else if source.IsLocalPath() {
-		rootPath, err := source.FindRootPath()
+		rootPath, err := source.ToFilePath()
 		if err != nil {
 			return nil
 		}

pkg/downloader/source.go

@@ -109,6 +109,9 @@ func (local *Local) IsLocalKPath() bool {
 }
 
 func (local *Local) IsDir() bool {
+	if local == nil {
+		return false
+	}
 	fileInfo, err := os.Stat(local.Path)
 	if err != nil {
 		return false