build: support statuscode response for print requests

Signed-off-by: Tonis Tiigi <tonistiigi@gmail.com>
crazy-max · Apr 15, 2024 · 46c44c5 · 46c44c5
1 parent df7a3db
commit 46c44c5
Show file tree

Hide file tree

Showing 6 changed files with 156 additions and 131 deletions.
diff --git a/build/build.go b/build/build.go
@@ -88,8 +88,9 @@ type Options struct {
 }
 
 type PrintFunc struct {
-	Name   string
-	Format string
+	Name         string
+	Format       string
+	IgnoreStatus bool
 }
 
 type Inputs struct {

diff --git a/commands/build.go b/commands/build.go
@@ -874,6 +874,11 @@ func printResult(f *controllerapi.PrintFunc, res map[string]string) error {
 			log.Printf("%s %+v", f, res)
 		}
 	}
+	if v, ok := res["result.statuscode"]; !f.IgnoreStatus && ok {
+		if n, err := strconv.Atoi(v); err == nil && n != 0 {
+			os.Exit(n)
+		}
+	}
 	return nil
 }
 

diff --git a/controller/build/build.go b/controller/build/build.go
@@ -161,8 +161,9 @@ func RunBuild(ctx context.Context, dockerCli command.Cli, in controllerapi.Build
 
 	if in.PrintFunc != nil {
 		opts.PrintFunc = &build.PrintFunc{
-			Name:   in.PrintFunc.Name,
-			Format: in.PrintFunc.Format,
+			Name:         in.PrintFunc.Name,
+			Format:       in.PrintFunc.Format,
+			IgnoreStatus: in.PrintFunc.IgnoreStatus,
 		}
 	}