troubleshoot/pkg/types/types.go

package types

import (
	"fmt"
)

type NotFoundError struct {
	Name string
}

func (e *NotFoundError) Error() string {
	return e.Name + ": not found"
}

type ExitError interface {
	Error() string
	ExitStatus() int
}

type ExitCodeError struct {
	Msg  string
	Code int
	Err  error
}

type ExitCodeWarning struct {
	Msg string
}

func (e *ExitCodeError) Error() string {
	return e.Msg
}

func (e *ExitCodeError) Unwrap() error {
	return e.Err
}

func (e *ExitCodeError) ExitStatus() int {
	return e.Code
}

func NewExitCodeError(exitCode int, theErr error) *ExitCodeError {
	useErr := ""
	if theErr != nil {
		useErr = theErr.Error()
	}
	return &ExitCodeError{Msg: useErr, Code: exitCode, Err: theErr}
}

func NewExitCodeWarning(theErrMsg string) *ExitCodeWarning {
	return &ExitCodeWarning{Msg: theErrMsg}
}

func (e *ExitCodeWarning) Warning() string {
	return fmt.Sprintf("Warning: %s", e.Msg)
}