package storage

import (
	"context"
	"fmt"
	"strings"
	"time"

	"go.mongodb.org/mongo-driver/v2/bson"
	"go.mongodb.org/mongo-driver/v2/mongo"
)

func (r *MySQLRepository) listMongoGiftAmounts(ctx context.Context, db *mongo.Database,
	start, end time.Time, statTimezone string, luckyGift bool) ([]mongoGiftAmount, error) {
	filter := bson.D{
		{Key: "refunded", Value: bson.D{{Key: "$ne", Value: true}}},
		{Key: "userId", Value: bson.D{{Key: "$ne", Value: nil}}},
		{Key: "createTime", Value: bson.D{{Key: "$gte", Value: start}, {Key: "$lt", Value: end}}},
		{Key: "sysOrigin", Value: bson.D{{Key: "$in", Value: r.cfg.Dashboard.SysOrigins}}},
	}
	amountPath := "$giftValue.actualAmount"
	unwindAcceptUsers := false
	if luckyGift {
		filter = append(filter, bson.E{Key: "luckyGift", Value: true})
		amountPath = "$acceptUsers.targetAmount"
		unwindAcceptUsers = true
	} else {
		filter = append(filter,
			bson.E{Key: "luckyGift", Value: bson.D{{Key: "$ne", Value: true}}},
			bson.E{Key: "originId", Value: bson.D{{Key: "$regex", Value: "^\\d+$"}}},
			bson.E{Key: "$or", Value: bson.A{
				bson.D{{Key: "dynamicContentId", Value: bson.D{{Key: "$exists", Value: false}}}},
				bson.D{{Key: "dynamicContentId", Value: nil}},
				bson.D{{Key: "dynamicContentId", Value: ""}},
			}},
		)
	}
	pipeline := mongo.Pipeline{
		bson.D{{Key: "$match", Value: filter}},
	}
	if unwindAcceptUsers {
		pipeline = append(pipeline, bson.D{{Key: "$unwind", Value: "$acceptUsers"}})
	}
	pipeline = append(pipeline,
		bson.D{{Key: "$project", Value: bson.D{
			{Key: "userId", Value: "$userId"},
			{Key: "amount", Value: amountPath},
			{Key: "day", Value: bson.D{{Key: "$dateToString", Value: bson.D{
				{Key: "format", Value: "%Y-%m-%d"},
				{Key: "date", Value: "$createTime"},
				{Key: "timezone", Value: statTimezone},
			}}}},
		}}},
		bson.D{{Key: "$group", Value: bson.D{
			{Key: "_id", Value: bson.D{{Key: "userId", Value: "$userId"}, {Key: "day", Value: "$day"}}},
			{Key: "amount", Value: bson.D{{Key: "$sum", Value: "$amount"}}},
		}}},
		bson.D{{Key: "$project", Value: bson.D{
			{Key: "_id", Value: 0},
			{Key: "userId", Value: "$_id.userId"},
			{Key: "day", Value: "$_id.day"},
			{Key: "amount", Value: "$amount"},
		}}},
	)
	cursor, err := db.Collection(giftRunningWaterCollection).Aggregate(ctx, pipeline)
	if err != nil {
		return nil, err
	}
	defer cursor.Close(ctx)
	var result []mongoGiftAmount
	for cursor.Next(ctx) {
		var doc bson.M
		if err := cursor.Decode(&doc); err != nil {
			return nil, err
		}
		userID := bsonInt64(doc["userId"])
		day := strings.TrimSpace(fmt.Sprint(doc["day"]))
		amount := bsonDecimal(doc["amount"])
		if userID == 0 || day == "" {
			continue
		}
		result = append(result, mongoGiftAmount{userID: userID, day: day, amount: amount})
	}
	return result, cursor.Err()
}