hfb_sys/backend/internal/modules/invitation/handler.go

package invitation

import (
	"errors"
	"net/http"
	"strconv"
	"strings"

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

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

type Handler struct {
	service *Service
}

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

// GetInviteInfo 公开接口：根据邀请码查询邀请人简要信息（用于落地注册页）
func (h *Handler) GetInviteInfo(c *gin.Context) {
	code := strings.TrimSpace(c.Param("code"))
	if code == "" {
		code = strings.TrimSpace(c.Query("code"))
	}
	info, err := h.service.GetInviteCodeInfo(c.Request.Context(), code)
	if err != nil {
		writeInvitationError(c, err)
		return
	}
	response.OK(c, info)
}

// UserSummary 个人中心邀请返利概览
func (h *Handler) UserSummary(c *gin.Context) {
	userID, ok := currentUserID(c)
	if !ok {
		response.Unauthorized(c, "缺少用户上下文")
		return
	}
	siteID := parseSiteID(c)
	summary, err := h.service.GetUserSummary(c.Request.Context(), userID, siteID)
	if err != nil {
		writeInvitationError(c, err)
		return
	}
	response.OK(c, summary)
}

// ListMyInvitedUsers 我的邀请好友列表
func (h *Handler) ListMyInvitedUsers(c *gin.Context) {
	userID, ok := currentUserID(c)
	if !ok {
		response.Unauthorized(c, "缺少用户上下文")
		return
	}
	page, pageSize := parsePagination(c)
	items, total, err := h.service.ListMyInvitedUsers(c.Request.Context(), userID, page, pageSize)
	if err != nil {
		writeInvitationError(c, err)
		return
	}
	response.OK(c, gin.H{
		"items":     items,
		"total":     total,
		"page":      page,
		"page_size": pageSize,
	})
}

// ListMyRebates 我的返利流水
func (h *Handler) ListMyRebates(c *gin.Context) {
	userID, ok := currentUserID(c)
	if !ok {
		response.Unauthorized(c, "缺少用户上下文")
		return
	}
	page, pageSize := parsePagination(c)
	items, total, err := h.service.ListMyRebates(c.Request.Context(), userID, page, pageSize)
	if err != nil {
		writeInvitationError(c, err)
		return
	}
	response.OK(c, gin.H{
		"items":     items,
		"total":     total,
		"page":      page,
		"page_size": pageSize,
	})
}

// AdminGetConfig 管理员获取返利比例配置
func (h *Handler) AdminGetConfig(c *gin.Context) {
	siteID := parseAdminSiteID(c)
	cfg, err := h.service.GetRebateConfig(c.Request.Context(), siteID)
	if err != nil {
		writeInvitationError(c, err)
		return
	}
	response.OK(c, cfg)
}

// AdminUpdateConfig 管理员更新返利比例配置
func (h *Handler) AdminUpdateConfig(c *gin.Context) {
	siteID := parseAdminSiteID(c)
	var req UpdateConfigRequest
	if err := c.ShouldBindJSON(&req); err != nil {
		response.BadRequest(c, "比例参数格式不正确，必须在0到1之间")
		return
	}
	cfg, err := h.service.UpdateRebateConfig(c.Request.Context(), siteID, req)
	if err != nil {
		writeInvitationError(c, err)
		return
	}
	response.OK(c, cfg)
}

// AdminInvitations 管理员查询全量邀请关系
func (h *Handler) AdminInvitations(c *gin.Context) {
	var query AdminInvitationQuery
	if err := c.ShouldBindQuery(&query); err != nil {
		response.BadRequest(c, "查询参数不正确")
		return
	}
	items, total, err := h.service.ListAdminInvitations(c.Request.Context(), query)
	if err != nil {
		writeInvitationError(c, err)
		return
	}
	response.OK(c, gin.H{
		"items":     items,
		"total":     total,
		"page":      query.Page,
		"page_size": query.PageSize,
	})
}

// AdminRebates 管理员查询全量返利流水
func (h *Handler) AdminRebates(c *gin.Context) {
	var query AdminRebateLedgerQuery
	if err := c.ShouldBindQuery(&query); err != nil {
		response.BadRequest(c, "查询参数不正确")
		return
	}
	items, total, err := h.service.ListAdminRebateLedgers(c.Request.Context(), query)
	if err != nil {
		writeInvitationError(c, err)
		return
	}
	response.OK(c, gin.H{
		"items":     items,
		"total":     total,
		"page":      query.Page,
		"page_size": query.PageSize,
	})
}

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 parsePagination(c *gin.Context) (int, int) {
	page, _ := strconv.Atoi(c.DefaultQuery("page", "1"))
	pageSize, _ := strconv.Atoi(c.DefaultQuery("page_size", "20"))
	if page < 1 {
		page = 1
	}
	if pageSize < 1 {
		pageSize = 20
	}
	if pageSize > 100 {
		pageSize = 100
	}
	return page, pageSize
}

func parseSiteID(c *gin.Context) uint64 {
	siteCode := sitecode.FromRequest(c)
	if strings.TrimSpace(siteCode) == "" {
		return 0
	}
	// 可从上下文或按 siteCode 获取 siteID
	return 0
}

func parseAdminSiteID(c *gin.Context) uint64 {
	raw := c.Query("distribution_site_id")
	if raw == "" {
		raw = c.Param("id")
	}
	if raw == "" {
		raw = c.Param("site_id")
	}
	if raw != "" {
		id, _ := strconv.ParseUint(raw, 10, 64)
		return id
	}
	return 0
}

func writeInvitationError(c *gin.Context, err error) {
	response.RecordError(c, err)
	switch {
	case errors.Is(err, ErrDependencyUnavailable):
		response.ServiceUnavailable(c, "数据库服务不可用")
	case errors.Is(err, ErrInvalidInviteCode):
		response.BadRequest(c, "邀请码不存在或无效")
	case errors.Is(err, ErrSelfInviteNotAllowed):
		response.BadRequest(c, "不能自己邀请自己")
	case errors.Is(err, ErrAlreadyInvited):
		response.BadRequest(c, "用户已绑定过邀请人")
	case errors.Is(err, ErrInviterNotVerified):
		response.BadRequest(c, "邀请人尚未完成实名认证")
	default:
		response.Error(c, http.StatusInternalServerError, "invitation_error", err.Error())
	}
}