refactor: main into dockerMain

Signed-off-by: Alano Terblanche <[email protected]>

cmd/docker/docker.go

@@ -28,14 +28,20 @@ import (
 )
 
 func main() {
+	statusCode := dockerMain()
+	if statusCode != 0 {
+		os.Exit(statusCode)
+	}
+}
+
+func dockerMain() int {
 	ctx, cancelNotify := signal.NotifyContext(context.Background(), platformsignals.TerminationSignals...)
 	defer cancelNotify()
 
 	dockerCli, err := command.NewDockerCli(command.WithBaseContext(ctx))
 	if err != nil {
 		fmt.Fprintln(os.Stderr, err)
-		defer os.Exit(1)
-		return
+		return 1
 	}
 	logrus.SetOutput(dockerCli.Err())
 	otel.SetErrorHandler(debug.OTELErrorHandler)
@@ -48,20 +54,17 @@ func main() {
 			// StatusError should only be used for errors, and all errors should
 			// have a non-zero exit status, so never exit with 0
 			if sterr.StatusCode == 0 {
-				defer os.Exit(1)
-				return
+				return 1
 			}
-			defer os.Exit(sterr.StatusCode)
-			return
+			return sterr.StatusCode
 		}
 		if errdefs.IsCancelled(err) {
-			defer os.Exit(0)
-			return
+			return 0
 		}
 		fmt.Fprintln(dockerCli.Err(), err)
-		defer os.Exit(1)
-		return
+		return 1
 	}
+	return 0
 }
 
 func newDockerCommand(dockerCli *command.DockerCli) *cli.TopLevelCommand {