feat(core): add option to return specific error codes

Signed-off-by: Patrik Cyvoct <[email protected]>

Author: Sh4d1

internal/core/bootstrap.go

@@ -125,6 +125,11 @@ func Bootstrap(config *BootstrapConfig) (exitCode int, result interface{}, err e
 		if _, ok := err.(*interactive.InterruptError); ok {
 			return 130, nil, err
 		}
+		if cliErr, ok := err.(*CliError); ok {
+			if cliErr.Code != 0 {
+				return cliErr.Code, nil, nil
+			}
+		}
 		printErr := meta.Printer.Print(err, nil)
 		if printErr != nil {
 			_, _ = fmt.Fprintln(os.Stderr, err)

internal/core/bootstrap_test.go

@@ -27,4 +27,21 @@ func TestInterruptError(t *testing.T) {
 		Cmd:             "scw test interrupt error",
 		Check:           TestCheckExitCode(130),
 	}))
+	t.Run("exit-code", Test(&TestConfig{
+		Commands: NewCommands(
+			&Command{
+				Namespace: "test",
+				Resource:  "code",
+				Verb:      "error",
+				ArgsType:  reflect.TypeOf(args.RawArgs{}),
+				Run: func(ctx context.Context, argsI interface{}) (i interface{}, e error) {
+					return nil, &CliError{Code: 99}
+				},
+			},
+		),
+		UseE2EClient:    true,
+		DisableParallel: true, // because e2e client is used
+		Cmd:             "scw test code error",
+		Check:           TestCheckExitCode(99),
+	}))
 }

internal/core/error.go

@@ -23,6 +23,10 @@ type CliError struct {
 
 	Details string
 	Hint    string
+
+	// Code allows to return an error code from the main binary. If different than 0,
+	// it does not print the error, only exits with this code.
+	Code int
 }
 
 func (s *CliError) Error() string {