add packages function

Author: Daishan Peng

conf/conf.go

@@ -15,6 +15,7 @@ type Conf struct {
 	Package   string            `yaml:"package,omitempty"`
 	Imports   []Import          `yaml:"import,omitempty"`
 	Excludes  []string          `yaml:"exclude,omitempty"`
+	Packages  []string          `yaml:"packages,omitempty"`
 	ImportMap map[string]Import `yaml:"-"`
 	confFile  string            `yaml:"-"`
 	yamlType  bool              `yaml:"-"`
@@ -25,7 +26,7 @@ type Import struct {
 	Version string `yaml:"version,omitempty"`
 	Repo    string `yaml:"repo,omitempty"`
 	Update  bool   `yaml:"-"`
-	Options `yaml:"-"`
+	Options `yaml:",inline"`
 }
 
 type Options struct {
@@ -79,6 +80,11 @@ func Parse(path string) (*Conf, error) {
 			continue
 		}
 
+		if strings.HasPrefix(fields[0], "package=") {
+			trashConf.Packages = append(trashConf.Packages, strings.TrimPrefix(fields[0], "package="))
+			continue
+		}
+
 		// Otherwise it's an import pattern
 		packageImport := Import{}
 		packageImport.Package = fields[0] // at least 1 field at this point: trimmed the line and skipped empty

trash.go

@@ -983,6 +983,10 @@ func cleanup(update bool, dir, targetDir string, trashConf *conf.Conf) error {
 	if err := removeExcludes(trashConf.Excludes, targetDir); err != nil {
 		logrus.Errorf("Error removing excluded dirs: %v", err)
 	}
+	for _, im := range trashConf.Packages {
+		logrus.Infof("Must include package %s", im)
+		imports[im] = true
+	}
 	if err := removeUnusedImports(imports, targetDir, updatePackages); err != nil {
 		logrus.Errorf("Error removing unused dirs: %v", err)
 	}