package api

import (
	"net/http"
	"time"

	"github.com/gin-gonic/gin"
	"github.com/openteam/server/internal/pkg/resp"
	"github.com/openteam/server/internal/store"
)

// UserProfile GET /api/v1/user/profile
func (h *Handler) UserProfile(c *gin.Context) {
	u, _ := userFromContext(c)
	resp.OK(c, gin.H{"user": h.publicUser(u)})
}

// UserBalance GET /api/v1/user/balance — 余额 + 近 30 日消耗。
func (h *Handler) UserBalance(c *gin.Context) {
	u, _ := userFromContext(c)
	var spent float64
	h.a.DB.Model(&store.UsageLog{}).
		Where("user_id = ? AND status = ? AND created_at >= ?", u.ID, store.UsageStatusSuccess, time.Now().Add(-30*24*time.Hour)).
		Select("COALESCE(SUM(cost),0)").Scan(&spent)
	resp.OK(c, gin.H{
		"balance":          u.Balance,
		"spent_last_30d":   spent,
		"today":            h.todayUsage(c, u.ID),
		"models_available": h.availableModelCount(),
	})
}

func (h *Handler) availableModelCount() int64 {
	var n int64
	h.a.DB.Model(&store.Model{}).Where("enabled = ?", true).Count(&n)
	return n
}

func (h *Handler) todayUsage(c *gin.Context, userID uint64) gin.H {
	var requests int64
	var tokens int64
	var cost float64
	today := time.Now().UTC().Format("2006-01-02")
	h.a.DB.Model(&store.UsageDaily{}).
		Where("user_id = ? AND date = ?", userID, today).
		Select("COALESCE(SUM(requests),0), COALESCE(SUM(input_tokens+output_tokens+cache_read_tokens),0), COALESCE(SUM(cost),0)").
		Row().Scan(&requests, &tokens, &cost)
	return gin.H{"requests": requests, "tokens": tokens, "cost": cost}
}

// UserModels GET /api/v1/user/models — 控制台可用模型列表（无需 API Key）。
func (h *Handler) UserModels(c *gin.Context) {
	var ms []store.Model
	if err := h.a.DB.Where("enabled = ?", true).Order("sort ASC, id ASC").Find(&ms).Error; err != nil {
		resp.Fail(c, http.StatusInternalServerError, "failed to load models")
		return
	}
	out := make([]string, 0, len(ms))
	for _, m := range ms {
		out = append(out, m.Name)
	}
	resp.OK(c, gin.H{"items": out})
}