first commit

bff/internal/response/middleware.go

@@ -0,0 +1,100 @@
+package response
+
+import (
+	"bytes"
+	"encoding/json"
+	"net/http"
+	"strings"
+
+	"google.golang.org/grpc/codes"
+	"google.golang.org/grpc/status"
+)
+
+type bodyWriter struct {
+	http.ResponseWriter
+	status int
+	buf    bytes.Buffer
+}
+
+func (w *bodyWriter) WriteHeader(statusCode int) {
+	w.status = statusCode
+}
+
+func (w *bodyWriter) Write(b []byte) (int, error) {
+	return w.buf.Write(b)
+}
+
+func Wrap(next http.HandlerFunc) http.HandlerFunc {
+	return func(w http.ResponseWriter, r *http.Request) {
+		bw := &bodyWriter{ResponseWriter: w, status: http.StatusOK}
+		next(bw, r)
+
+		ctrl := BaseController{}
+		if bw.status >= http.StatusBadRequest {
+			ctrl.Error(w, mapHTTPError(bw.status, bw.buf.String()))
+			return
+		}
+
+		raw := bytes.TrimSpace(bw.buf.Bytes())
+		var data any
+		if len(raw) > 0 {
+			if err := json.Unmarshal(raw, &data); err != nil {
+				ctrl.Fail(w)
+				return
+			}
+		}
+		ctrl.Ok(w, data)
+	}
+}
+
+func mapHTTPError(httpStatus int, body string) Error {
+	msg := strings.TrimSpace(body)
+	msg = strings.TrimPrefix(msg, "rpc error: ")
+	if i := strings.Index(msg, "desc = "); i >= 0 {
+		msg = strings.TrimSpace(msg[i+len("desc = "):])
+	}
+	if msg == "" {
+		msg = Fail.GetMsg()
+	}
+
+	switch httpStatus {
+	case http.StatusBadRequest:
+		return Define(ErrorMissingParams, msg)
+	case http.StatusUnauthorized:
+		return Define(ErrorNoLogin, msg)
+	case http.StatusForbidden:
+		return Define(ErrorAuthority, msg)
+	case http.StatusNotFound:
+		return Define(ErrorDataNotExist, msg)
+	default:
+		return Define(Fail, msg)
+	}
+}
+
+func FromGRPC(err error) Error {
+	if err == nil {
+		return OK
+	}
+	st, ok := status.FromError(err)
+	if !ok {
+		return Define(Fail, err.Error())
+	}
+	msg := st.Message()
+	if msg == "" {
+		msg = st.Code().String()
+	}
+	switch st.Code() {
+	case codes.InvalidArgument:
+		return Define(ErrorMissingParams, msg)
+	case codes.Unauthenticated:
+		return Define(ErrorNoLogin, msg)
+	case codes.PermissionDenied:
+		return Define(ErrorAuthority, msg)
+	case codes.NotFound:
+		return Define(ErrorDataNotExist, msg)
+	case codes.AlreadyExists:
+		return Define(ErrorDataIsExist, msg)
+	default:
+		return Define(Fail, msg)
+	}
+}