init push

backend/handler/base.go

@@ -0,0 +1,65 @@
+package handler
+
+import (
+	"fmt"
+	"log/slog"
+	"net/http"
+
+	"github.com/google/uuid"
+	"github.com/labstack/echo/v4"
+	"go.opentelemetry.io/otel/attribute"
+	"go.opentelemetry.io/otel/trace"
+
+	"github.com/chaitin/panda-wiki/config"
+	"github.com/chaitin/panda-wiki/domain"
+	"github.com/chaitin/panda-wiki/log"
+	"github.com/chaitin/panda-wiki/middleware"
+	"github.com/chaitin/panda-wiki/pkg/captcha"
+)
+
+type BaseHandler struct {
+	Router              *echo.Echo
+	baseLogger          *log.Logger
+	config              *config.Config
+	ShareAuthMiddleware *middleware.ShareAuthMiddleware
+	V1Auth              middleware.AuthMiddleware
+	Captcha             *captcha.Captcha
+}
+
+func NewBaseHandler(echo *echo.Echo, logger *log.Logger, config *config.Config, v1Auth middleware.AuthMiddleware, shareAuthMiddleware *middleware.ShareAuthMiddleware, cap *captcha.Captcha) *BaseHandler {
+	return &BaseHandler{
+		Router:              echo,
+		baseLogger:          logger.WithModule("http_base_handler"),
+		config:              config,
+		ShareAuthMiddleware: shareAuthMiddleware,
+		V1Auth:              v1Auth,
+		Captcha:             cap,
+	}
+}
+
+func (h *BaseHandler) NewResponseWithData(c echo.Context, data any) error {
+	return c.JSON(http.StatusOK, domain.PWResponse{
+		Success: true,
+		Data:    data,
+	})
+}
+
+func (h *BaseHandler) NewResponseWithErrCode(c echo.Context, resp domain.PWResponseErrCode) error {
+	return c.JSON(http.StatusOK, resp)
+}
+
+func (h *BaseHandler) NewResponseWithError(c echo.Context, msg string, err error) error {
+	traceID := ""
+	if h.config.GetBool("apm.enabled") {
+		span := trace.SpanFromContext(c.Request().Context())
+		traceID = span.SpanContext().TraceID().String()
+		span.SetAttributes(attribute.String("error", fmt.Sprintf("%+v", err)), attribute.String("msg", msg))
+	} else {
+		traceID = uuid.New().String()
+	}
+	h.baseLogger.LogAttrs(c.Request().Context(), slog.LevelError, msg, slog.String("trace_id", traceID), slog.Any("error", err))
+	return c.JSON(http.StatusOK, domain.PWResponse{
+		Success: false,
+		Message: fmt.Sprintf("%s [trace_id: %s]", msg, traceID),
+	})
+}