soul-yongping/soul-api/internal/handler/admin_dashboard.go

package handler

import (
	"encoding/json"
	"net/http"
	"sync"

	"soul-api/internal/database"
	"soul-api/internal/model"
	"soul-api/internal/wechat"

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

var paidStatuses = []string{"paid", "completed", "success"}

// AdminDashboardStats GET /api/admin/dashboard/stats
// 轻量聚合：总用户、付费订单数、付费用户数、总营收、转化率（无订单/用户明细）
func AdminDashboardStats(c *gin.Context) {
	db := database.DB()
	var (
		totalUsers     int64
		paidOrderCount int64
		totalRevenue   float64
		paidUserCount  int64
	)
	var wg sync.WaitGroup
	wg.Add(4)
	go func() { defer wg.Done(); db.Model(&model.User{}).Count(&totalUsers) }()
	go func() { defer wg.Done(); db.Model(&model.Order{}).Where("status IN ?", paidStatuses).Count(&paidOrderCount) }()
	go func() {
		defer wg.Done()
		db.Model(&model.Order{}).Where("status IN ?", paidStatuses).
			Select("COALESCE(SUM(amount), 0)").Scan(&totalRevenue)
	}()
	go func() {
		defer wg.Done()
		db.Table("orders").Where("status IN ?", paidStatuses).
			Select("COUNT(DISTINCT user_id)").Scan(&paidUserCount)
	}()
	wg.Wait()
	conversionRate := 0.0
	if totalUsers > 0 && paidUserCount > 0 {
		conversionRate = float64(paidUserCount) / float64(totalUsers) * 100
	}
	c.JSON(http.StatusOK, gin.H{
		"success":        true,
		"totalUsers":     totalUsers,
		"paidOrderCount": paidOrderCount,
		"paidUserCount":  paidUserCount,
		"totalRevenue":   totalRevenue,
		"conversionRate": conversionRate,
	})
}

// AdminDashboardRecentOrders GET /api/admin/dashboard/recent-orders
func AdminDashboardRecentOrders(c *gin.Context) {
	db := database.DB()
	var recentOrders []model.Order
	db.Where("status IN ?", paidStatuses).Order("created_at DESC").Limit(10).Find(&recentOrders)
	c.JSON(http.StatusOK, gin.H{"success": true, "recentOrders": buildRecentOrdersOut(db, recentOrders)})
}

// AdminDashboardNewUsers GET /api/admin/dashboard/new-users
func AdminDashboardNewUsers(c *gin.Context) {
	db := database.DB()
	var newUsers []model.User
	db.Model(&model.User{}).Order("created_at DESC").Limit(10).Find(&newUsers)
	c.JSON(http.StatusOK, gin.H{"success": true, "newUsers": buildNewUsersOut(newUsers)})
}

// AdminDashboardOverview GET /api/admin/dashboard/overview
// 数据概览：总用户、付费订单数、付费用户数、总营收、转化率、最近订单、新用户
// 优化：6 组查询并行执行，减少总耗时
func AdminDashboardOverview(c *gin.Context) {
	db := database.DB()

	var (
		totalUsers     int64
		paidOrderCount int64
		totalRevenue   float64
		paidUserCount  int64
		recentOrders   []model.Order
		newUsers       []model.User
	)
	var wg sync.WaitGroup
	wg.Add(6)

	go func() {
		defer wg.Done()
		db.Model(&model.User{}).Count(&totalUsers)
	}()
	go func() {
		defer wg.Done()
		db.Model(&model.Order{}).Where("status IN ?", paidStatuses).Count(&paidOrderCount)
	}()
	go func() {
		defer wg.Done()
		db.Model(&model.Order{}).Where("status IN ?", paidStatuses).
			Select("COALESCE(SUM(amount), 0)").Scan(&totalRevenue)
	}()
	go func() {
		defer wg.Done()
		db.Table("orders").Where("status IN ?", paidStatuses).
			Select("COUNT(DISTINCT user_id)").Scan(&paidUserCount)
	}()
	go func() {
		defer wg.Done()
		db.Where("status IN ?", paidStatuses).
			Order("created_at DESC").Limit(5).Find(&recentOrders)
	}()
	go func() {
		defer wg.Done()
		db.Model(&model.User{}).Order("created_at DESC").Limit(10).Find(&newUsers)
	}()
	wg.Wait()

	conversionRate := 0.0
	if totalUsers > 0 && paidUserCount > 0 {
		conversionRate = float64(paidUserCount) / float64(totalUsers) * 100
	}

	recentOut := buildRecentOrdersOut(db, recentOrders)
	newOut := buildNewUsersOut(newUsers)

	c.JSON(http.StatusOK, gin.H{
		"success":         true,
		"totalUsers":      totalUsers,
		"paidOrderCount":  paidOrderCount,
		"paidUserCount":   paidUserCount,
		"totalRevenue":    totalRevenue,
		"conversionRate":  conversionRate,
		"recentOrders":    recentOut,
		"newUsers":        newOut,
	})
}

func dashStr(s *string) string {
	if s == nil || *s == "" {
		return ""
	}
	return *s
}

func buildRecentOrdersOut(db *gorm.DB, recentOrders []model.Order) []gin.H {
	if len(recentOrders) == 0 {
		return nil
	}
	userIDs := make(map[string]bool)
	for _, o := range recentOrders {
		if o.UserID != "" {
			userIDs[o.UserID] = true
		}
	}
	ids := make([]string, 0, len(userIDs))
	for id := range userIDs {
		ids = append(ids, id)
	}
	var users []model.User
	db.Where("id IN ?", ids).Find(&users)
	userMap := make(map[string]*model.User)
	for i := range users {
		userMap[users[i].ID] = &users[i]
	}
	out := make([]gin.H, 0, len(recentOrders))
	for _, o := range recentOrders {
		b, _ := json.Marshal(o)
		var m map[string]interface{}
		_ = json.Unmarshal(b, &m)
		if u := userMap[o.UserID]; u != nil {
			m["userNickname"] = dashStr(u.Nickname)
			m["userAvatar"] = dashStr(u.Avatar)
		} else {
			m["userNickname"] = ""
			m["userAvatar"] = ""
		}
		out = append(out, m)
	}
	return out
}

// AdminDashboardMerchantBalance GET /api/admin/dashboard/merchant-balance
// 查询微信商户号实时余额（可用余额、待结算余额），用于看板展示
// 注意：普通商户可能需向微信申请开通权限，未开通时返回 error
func AdminDashboardMerchantBalance(c *gin.Context) {
	bal, err := wechat.QueryMerchantBalance("BASIC")
	if err != nil {
		c.JSON(http.StatusOK, gin.H{
			"success": false,
			"error":   err.Error(),
			"message": "查询商户余额失败，可能未开通权限（请联系微信支付运营申请）",
		})
		return
	}
	c.JSON(http.StatusOK, gin.H{
		"success":         true,
		"availableAmount": bal.AvailableAmount, // 单位：分
		"pendingAmount":   bal.PendingAmount,   // 单位：分
	})
}

func buildNewUsersOut(newUsers []model.User) []gin.H {
	out := make([]gin.H, 0, len(newUsers))
	for _, u := range newUsers {
		out = append(out, gin.H{
			"id":           u.ID,
			"nickname":     dashStr(u.Nickname),
			"phone":        dashStr(u.Phone),
			"referralCode": dashStr(u.ReferralCode),
			"createdAt":    u.CreatedAt,
		})
	}
	return out
}