package wallet

import (
	"errors"

	"hfb_sys/backend/internal/middleware"
	"hfb_sys/backend/pkg/response"

	"github.com/gin-gonic/gin"
)

type Handler struct {
	service *Service
}

func NewHandler(service *Service) *Handler {
	return &Handler{service: service}
}

func (h *Handler) Balance(c *gin.Context) {
	userID, ok := currentUserID(c)
	if !ok {
		response.Unauthorized(c, "缺少用户上下文")
		return
	}
	account, err := h.service.Account(userID)
	if err != nil {
		writeWalletError(c, err)
		return
	}
	response.OK(c, account)
}

func (h *Handler) Ledger(c *gin.Context) {
	userID, ok := currentUserID(c)
	if !ok {
		response.Unauthorized(c, "缺少用户上下文")
		return
	}
	items, err := h.service.Ledger(userID)
	if err != nil {
		writeWalletError(c, err)
		return
	}
	response.OK(c, gin.H{"items": items})
}

func currentUserID(c *gin.Context) (uint64, bool) {
	value, ok := c.Get(middleware.ContextUserID)
	if !ok {
		return 0, false
	}
	userID, ok := value.(uint64)
	return userID, ok
}

func writeWalletError(c *gin.Context, err error) {
	switch {
	case errors.Is(err, ErrDependencyUnavailable):
		response.ServiceUnavailable(c, "数据库未连接")
	default:
		response.ServiceUnavailable(c, "钱包服务暂时不可用")
	}
}