frostfs-node/cmd/internal/common/exit.go

package common

import (
	"errors"
	"fmt"
	"os"

	sdkstatus "git.frostfs.info/TrueCloudLab/frostfs-sdk-go/client/status"
	"github.com/spf13/cobra"
)

// ExitOnErr prints error and exits with a code that matches
// one of the common errors from sdk library. If no errors
// found, exits with 1 code.
// Does nothing if passed error in nil.
func ExitOnErr(cmd *cobra.Command, errFmt string, err error) {
	if err == nil {
		return
	}

	if errFmt != "" {
		err = fmt.Errorf(errFmt, err)
	}

	const (
		_ = iota
		internal
		aclDenied
		apemanagerDenied
	)

	var (
		code int

		internalErr   = new(sdkstatus.ServerInternal)
		accessErr     = new(sdkstatus.ObjectAccessDenied)
		apemanagerErr = new(sdkstatus.APEManagerAccessDenied)
	)

	switch {
	case errors.As(err, &internalErr):
		code = internal
	case errors.As(err, &accessErr):
		code = aclDenied
		err = fmt.Errorf("%w: %s", err, accessErr.Reason())
	case errors.As(err, &apemanagerErr):
		code = apemanagerDenied
		err = fmt.Errorf("%w: %s", err, apemanagerErr.Reason())
	default:
		code = internal
	}

	cmd.PrintErrln(err)
	for p := cmd; p != nil; p = p.Parent() {
		if p.PersistentPostRun != nil {
			p.PersistentPostRun(cmd, nil)
			if !cobra.EnableTraverseRunHooks {
				break
			}
		}
	}
	os.Exit(code)
}