pkgset: add :main selector

Fixes #50
loov · Mar 24, 2021 · d33a4f6 · d33a4f6
1 parent 8a5165e
commit d33a4f6
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 0 deletions.
diff --git a/pkgset/calc.go b/pkgset/calc.go
@@ -175,6 +175,13 @@ func Calc(parentContext context.Context, expr []string) (Set, error) {
 
 				return Subtract(set, Sources(set)), nil
 
+			case "main":
+				set, err := eval(ctx, e.Expr)
+				if err != nil {
+					return nil, err
+				}
+				return Main(set), nil
+
 			default:
 				return nil, fmt.Errorf("unknown selector %v: %v", e.Selector, e)
 			}

diff --git a/pkgset/set.go b/pkgset/set.go
@@ -278,3 +278,14 @@ func Dependencies(a Set) Set {
 	}
 	return result
 }
+
+// Main returns main pacakges.
+func Main(a Set) Set {
+	rs := Set{}
+	for pid, pkg := range a {
+		if pkg.Name == "main" {
+			rs[pid] = pkg
+		}
+	}
+	return rs
+}