feat(SPV-847): add option to wrap defined errors with cause (#654)

models/errors.go

@@ -1,17 +1,27 @@
 package models
 
+import (
+	"github.com/pkg/errors"
+)
+
 type ExtendedError interface {
 	error
 	GetCode() string
 	GetMessage() string
 	GetStatusCode() int
+	StackTrace() (trace errors.StackTrace)
 }
 
 // SPVError is extended error which holds information about http status and code that should be returned
 type SPVError struct {
 	Code       string
 	Message    string
 	StatusCode int
+	cause      error
+}
+
+type stackTracer interface {
+	StackTrace() errors.StackTrace
 }
 
 // ResponseError is an error which will be returned in HTTP response
@@ -41,3 +51,31 @@ func (e SPVError) GetMessage() string {
 func (e SPVError) GetStatusCode() int {
 	return e.StatusCode
 }
+
+// StackTrace returns the error's stack trace.
+func (e SPVError) StackTrace() (trace errors.StackTrace) {
+	err, ok := e.cause.(stackTracer)
+	if !ok {
+		return
+	}
+
+	trace = err.StackTrace()
+
+	return
+}
+
+func (e SPVError) Unwrap() error {
+	return e.cause
+}
+
+func (e SPVError) Wrap(err error) SPVError {
+	e.cause = err
+	return e
+}
+
+func (e SPVError) WithTrace(err error) SPVError {
+	if st := stackTracer(nil); !errors.As(e.cause, &st) {
+		return e.Wrap(errors.WithStack(err))
+	}
+	return e.Wrap(err)
+}