Refactor: Errors into custom export types and expose types

1 files changed, 82 insertions, 0 deletions

diff --git a/types/error.go b/types/error.go
new file mode 100644
index 0000000..607e6c6
--- /dev/null
+++ b/types/error.go
@@ -0,0 +1,82 @@
+package types
+
+import (
+	"errors"
+	"fmt"
+)
+
+type ErrorLevel int8
+
+const (
+	// All other errors
+	ERR_OTHER ErrorLevel = iota
+
+	// The error is just here as additional info
+	ERR_INFO
+)
+
+type WrappedErrorMessage struct {
+	Level   ErrorLevel
+	Message string
+	Err     error
+}
+
+func (e *WrappedErrorMessage) Unwrap() error {
+	return e.Err
+}
+
+func (e *WrappedErrorMessage) Cause() error {
+	causeErr := e.Err
+	for errors.Unwrap(causeErr) != nil {
+		causeErr = errors.Unwrap(causeErr)
+	}
+	return causeErr
+}
+
+func (e *WrappedErrorMessage) Traceback() string {
+	returnStr := fmt.Sprintf("%s\n%s", e.Message, "Traceback:")
+	causeErr := e.Err
+	for errors.Unwrap(causeErr) != nil {
+		causeErr = errors.Unwrap(causeErr)
+		var wrappedErr *WrappedErrorMessage
+
+		errorStr := causeErr.Error()
+
+		if errors.As(causeErr, &wrappedErr) {
+			errorStr = wrappedErr.Message
+		}
+		returnStr += fmt.Sprintf("\n - %s", errorStr)
+	}
+	return returnStr
+}
+
+func (e *WrappedErrorMessage) Error() string {
+	return fmt.Sprintf("Got error: %s, with cause: %s", e.Message, e.Err)
+}
+
+func GetErrorTraceback(err error) string {
+	var wrappedErr *WrappedErrorMessage
+
+	if errors.As(err, &wrappedErr) {
+		return wrappedErr.Traceback()
+	}
+	return err.Error()
+}
+
+func GetErrorCause(err error) error {
+	var wrappedErr *WrappedErrorMessage
+
+	if errors.As(err, &wrappedErr) {
+		return wrappedErr.Cause()
+	}
+	return err
+}
+
+func GetErrorLevel(err error) ErrorLevel {
+	var wrappedErr *WrappedErrorMessage
+
+	if errors.As(err, &wrappedErr) {
+		return wrappedErr.Level
+	}
+	return ERR_OTHER
+}