sparse checkout feature - Load the package after downloading the entire repository

pkg/client/client.go

@@ -52,6 +52,8 @@ type KpmClient struct {
 	settings settings.Settings
 	// The flag of whether to check the checksum of the package and update kcl.mod.lock.
 	noSumCheck bool
+	// The subpackage within a directory
+	subPackage string
 }
 
 // NewKpmClient will create a new kpm client with default settings.
@@ -500,6 +502,9 @@ func (c *KpmClient) Compile(kclPkg *pkg.KclPkg, kclvmCompiler *runner.Compiler)
 		if !filepath.IsAbs(dPath) {
 			dPath = filepath.Join(c.homePath, dPath)
 		}
+		if c.subPackage != "" {
+			dPath, _ = utils.FindFolder(dPath, c.subPackage)
+		}
 		kclvmCompiler.AddDepPath(dName, dPath)
 	}
 
@@ -1877,6 +1882,14 @@ func (c *KpmClient) DownloadDeps(deps *pkg.Dependencies, lockDeps *pkg.Dependenc
 	return &newDeps, nil
 }
 
+func (c *KpmClient) SetSubPackage (pkgName string) {
+	c.subPackage = pkgName
+}
+
+func (c *KpmClient) GetSubPackage () string {
+	return c.subPackage
+}
+
 // pullTarFromOci will pull a kcl package tar file from oci registry.
 func (c *KpmClient) pullTarFromOci(localPath string, ociOpts *opt.OciOptions) error {
 	absPullPath, err := filepath.Abs(localPath)

pkg/cmd/cmd_flags.go

@@ -17,6 +17,7 @@ const FLAG_DISABLE_NONE = "disable_none"
 const FLAG_ARGUMENT = "argument"
 const FLAG_OVERRIDES = "overrides"
 const FLAG_SORT_KEYS = "sort_keys"
+const FLAG_PACKAGE = "package"
 
 const FLAG_QUIET = "quiet"
 const FLAG_NO_SUM_CHECK = "no_sum_check"

pkg/cmd/cmd_run.go

@@ -81,6 +81,12 @@ func NewRunCmd(kpmcli *client.KpmClient) *cli.Command {
 				Aliases: []string{"k"},
 				Usage:   "sort result keys",
 			},
+
+			// KCL arg: --package
+			&cli.StringFlag{
+				Name:    FLAG_PACKAGE,
+				Usage:   "specify the package name",
+			},
 		},
 		Action: func(c *cli.Context) error {
 			return KpmRun(c, kpmcli)
@@ -103,6 +109,8 @@ func KpmRun(c *cli.Context, kpmcli *client.KpmClient) error {
 		}
 	}()
 
+	kpmcli.SetSubPackage(c.String(FLAG_PACKAGE))
+
 	kclOpts := CompileOptionFromCli(c)
 	kclOpts.SetNoSumCheck(c.Bool(FLAG_NO_SUM_CHECK))
 	runEntry, errEvent := runner.FindRunEntryFrom(c.Args().Slice())

pkg/utils/utils.go

@@ -599,3 +599,35 @@ func AbsTarPath(tarPath string) (string, error) {
 
 	return absTarPath, nil
 }
+
+// FindFolder will find the folder 'targetFolder' under the 'root' directory recursively. 
+func FindFolder(root, targetFolder string) (string, error) {
+	var result string
+
+	err := filepath.Walk(root, func(path string, info os.FileInfo, err error) error {
+		if err != nil {
+			return err
+		}
+
+		if info.IsDir() && strings.EqualFold(info.Name(), targetFolder) {
+			absPath, err := filepath.Abs(path)
+			if err != nil {
+				return err
+			}
+			result = absPath
+			return filepath.SkipAll
+		}
+
+		return nil
+	})
+
+	if err != nil {
+		return "", err
+	}
+
+	if result == "" {
+		return "", fmt.Errorf("folder '%s' not found", targetFolder)
+	}
+
+	return result, nil
+}