Support ko build ./...

.github/workflows/modules-integration-test.yaml

@@ -8,6 +8,7 @@ jobs:
   test:
     name: Module Tests
     strategy:
+      fail-fast: false
       matrix:
         go-version: [1.16.x, 1.17.x]
     runs-on: 'ubuntu-latest'

main.go

@@ -29,6 +29,6 @@ func main() {
 	ctx, stop := signal.NotifyContext(context.Background(), os.Interrupt)
 	defer stop()
 	if err := commands.Root.ExecuteContext(ctx); err != nil {
-		log.Fatal("error during command execution:", err)
+		log.Fatal("error during command execution: ", err)
 	}
 }

pkg/commands/build.go

@@ -15,10 +15,13 @@
 package commands
 
 import (
+	"errors"
 	"fmt"
+	"path/filepath"
 
 	"github.com/google/ko/pkg/commands/options"
 	"github.com/spf13/cobra"
+	"golang.org/x/tools/go/packages"
 )
 
 // addBuild augments our CLI surface with build.
@@ -61,9 +64,13 @@ func addBuild(topLevel *cobra.Command) {
 			if err := options.Validate(po, bo); err != nil {
 				return fmt.Errorf("validating options: %w", err)
 			}
-
 			ctx := cmd.Context()
 
+			importpaths, err := importPaths(bo.WorkingDirectory, args)
+			if err != nil {
+				return fmt.Errorf("resolving import paths: %w", err)
+			}
+
 			bo.InsecureRegistry = po.InsecureRegistry
 			builder, err := makeBuilder(ctx, bo)
 			if err != nil {
@@ -74,7 +81,7 @@ func addBuild(topLevel *cobra.Command) {
 				return fmt.Errorf("error creating publisher: %w", err)
 			}
 			defer publisher.Close()
-			images, err := publishImages(ctx, args, publisher, builder)
+			images, err := publishImages(ctx, importpaths, publisher, builder)
 			if err != nil {
 				return fmt.Errorf("failed to publish images: %w", err)
 			}
@@ -88,3 +95,27 @@ func addBuild(topLevel *cobra.Command) {
 	options.AddBuildOptions(build, bo)
 	topLevel.AddCommand(build)
 }
+
+// importPaths resolves a list of importpath strings that may contain wildcards
+// like "./cmd/..." or "./...", and returns the 'package main' packages matched
+// by those importpaths.
+func importPaths(dir string, args []string) ([]string, error) {
+	dir = filepath.Clean(dir)
+	if dir == "." {
+		dir = ""
+	}
+	ps, err := packages.Load(&packages.Config{Dir: dir, Mode: packages.NeedName}, args...)
+	if err != nil {
+		return nil, fmt.Errorf("loading packages: %w", err)
+	}
+	out := []string{}
+	for _, p := range ps {
+		if p.Name == "main" {
+			out = append(out, p.String())
+		}
+	}
+	if len(out) == 0 {
+		return nil, errors.New("no package main packages matched")
+	}
+	return out, nil
+}