porter/cli/cmd/errors/error_handler.go

package errors

import (
	"fmt"
	"os"
	"time"

	"github.com/fatih/color"
	"github.com/getsentry/sentry-go"
	"github.com/porter-dev/porter/cli/cmd/config"
)

// SentryDSN is a global value for sentry's dsn. This should be removed
var SentryDSN string = ""

type errorHandler interface {
	HandleError(error)
}

type standardErrorHandler struct{}

// HandleError implements errorhandler for handling non-sentry errors
func (h *standardErrorHandler) HandleError(err error) {
	color.New(color.FgRed).Fprintf(os.Stderr, "error: %s\n", err.Error())
}

type sentryErrorHandler struct {
	cliConfig config.CLIConfig
}

// HandleError implements errorhandler for handling sentry errors
func (h *sentryErrorHandler) HandleError(err error) {
	if SentryDSN != "" {
		localHub := sentry.CurrentHub().Clone()

		localHub.ConfigureScope(func(scope *sentry.Scope) {
			scope.SetTags(map[string]string{
				"host":    h.cliConfig.Host,
				"project": fmt.Sprintf("%d", h.cliConfig.Project),
				"cluster": fmt.Sprintf("%d", h.cliConfig.Cluster),
			})
		})

		localHub.CaptureException(err)
		sentry.Flush(2 * time.Second)
	}

	color.New(color.FgRed).Fprintf(os.Stderr, "error: %s\n", err.Error())
}

// GetErrorHandler returns an errorhandler.
func GetErrorHandler(cliConf config.CLIConfig) errorHandler {
	if SentryDSN != "" {
		return &sentryErrorHandler{
			cliConfig: cliConf,
		}
	}

	return &standardErrorHandler{}
}