package service

import (
	"blazing/common/data"
	"blazing/common/utils"
	"blazing/cool"
	"blazing/modules/config/model"
	"context"
	"encoding/json"

	"github.com/gogf/gf/v2/database/gdb"
	"github.com/gogf/gf/v2/frame/g"
	"github.com/gogf/gf/v2/util/gconv"
	"github.com/gogf/gf/v2/util/grand"
)

type ShinyService struct {
	*cool.Service
}

func NewShinyService() *ShinyService {
	return &ShinyService{
		&cool.Service{
			Model: model.NewColorfulSkin(),
			InsertParam: func(ctx context.Context) g.MapStrAny {
				admin := cool.GetAdmin(ctx)
				userId := admin.UserId
				return g.MapStrAny{
					"author": userId,
				}
			},
		},
	}
}
func (s *ShinyService) ModifyBefore(ctx context.Context, method string, param g.MapStrAny) (err error) {
	var t data.GlowFilter

	r := json.Unmarshal([]byte(gconv.String(param["color"])), &t)
	if r != nil {
		return r
	}
	return nil
}
func (s *ShinyService) RandShiny(id uint32) *data.GlowFilter {
	var ret []model.ColorfulSkin

	// 执行 Raw SQL 并扫描返回值
	cool.DBM(s.Model).Wheref(`bind_elf_ids @> ?::jsonb`, id).Wheref(`jsonb_typeof(bind_elf_ids) = ?`, "array").Cache(gdb.CacheOption{
		// Duration: time.Hour,

		Force: false,
	}).Scan(&ret)

	for _, v := range ret {
		//print(v.ID)

		id := v.ID

		if grand.Meet(int(v.ElfProbability), 1000) {
			var t data.GlowFilter

			r := json.Unmarshal([]byte(v.Color), &t)
			if r == nil {
				m := cool.DBM(s.Model).Where("id", id)
				m.Increment("refresh_count", 1)
				return &t
			}

		}
	}

	return nil
}
func (s *ShinyService) FixShiny(id uint32) *data.GlowFilter {
	var ret []model.ColorfulSkin

	// 执行 Raw SQL 并扫描返回值
	cool.DBM(s.Model).Wheref(`bind_elf_ids @> ?::jsonb`, id).Wheref(`jsonb_typeof(bind_elf_ids) = ?`, "array").Cache(gdb.CacheOption{
		// Duration: time.Hour,

		Force: false,
	}).Scan(&ret)

	rets := utils.ToMap(ret, func(t model.ColorfulSkin) uint32 {
		return uint32(t.ID)
	})

	for _, v := range rets {
		//print(v.ID)

		id := v.ID

		var t data.GlowFilter

		r := json.Unmarshal([]byte(v.Color), &t)
		if r == nil {
			m := cool.DBM(s.Model).Where("id", id)
			m.Increment("refresh_count", 1)
			return &t
		}

	}

	return nil
}