Refactor

pkg/schemaserver/handlers/error.go

@@ -0,0 +1,65 @@
+package handlers
+
+import (
+	"net/http"
+	"net/url"
+
+	"github.com/rancher/steve/pkg/schemaserver/httperror"
+	"github.com/rancher/steve/pkg/schemaserver/types"
+	"github.com/rancher/wrangler/pkg/schemas/validation"
+	"github.com/sirupsen/logrus"
+)
+
+func ErrorHandler(request *types.APIRequest, err error) {
+	if err == validation.ErrComplete {
+		return
+	}
+
+	if ec, ok := err.(validation.ErrorCode); ok {
+		err = httperror.NewAPIError(ec, "")
+	}
+
+	var error *httperror.APIError
+	if apiError, ok := err.(*httperror.APIError); ok {
+		if apiError.Cause != nil {
+			url, _ := url.PathUnescape(request.Request.URL.String())
+			if url == "" {
+				url = request.Request.URL.String()
+			}
+			logrus.Errorf("API error response %v for %v %v. Cause: %v", apiError.Code.Status, request.Request.Method,
+				url, apiError.Cause)
+		}
+		error = apiError
+	} else {
+		logrus.Errorf("Unknown error: %v", err)
+		error = &httperror.APIError{
+			Code:    validation.ServerError,
+			Message: err.Error(),
+		}
+	}
+
+	if error.Code.Status == http.StatusNoContent {
+		request.Response.WriteHeader(http.StatusNoContent)
+		return
+	}
+
+	data := toError(error)
+	request.WriteResponse(error.Code.Status, data)
+}
+
+func toError(apiError *httperror.APIError) types.APIObject {
+	e := map[string]interface{}{
+		"type":    "error",
+		"status":  apiError.Code.Status,
+		"code":    apiError.Code.Code,
+		"message": apiError.Message,
+	}
+	if apiError.FieldName != "" {
+		e["fieldName"] = apiError.FieldName
+	}
+
+	return types.APIObject{
+		Type:   "error",
+		Object: e,
+	}
+}