package generate

import (
	"context"
	"encoding/json"
	"schisandra-album-cloud-microservices/common/constant"
	"time"

	"github.com/redis/go-redis/v9"
	"github.com/wenlng/go-captcha-assets/helper"
	"github.com/wenlng/go-captcha/v2/slide"
)

// GenerateSlideBasicCaptcha generate slide basic captcha
func GenerateSlideBasicCaptcha(slide slide.Captcha, redis *redis.Client, ctx context.Context) (map[string]interface{}, error) {
	captData, err := slide.Generate()
	if err != nil {
		return nil, err
	}
	blockData := captData.GetData()
	if blockData == nil {
		return nil, nil
	}
	var masterImageBase64, tileImageBase64 string
	masterImageBase64, err = captData.GetMasterImage().ToBase64()
	if err != nil {
		return nil, err
	}

	tileImageBase64, err = captData.GetTileImage().ToBase64()
	if err != nil {
		return nil, err
	}

	dotsByte, err := json.Marshal(blockData)
	if err != nil {
		return nil, err
	}
	key := helper.StringToMD5(string(dotsByte))
	err = redis.Set(ctx, constant.UserCaptchaPrefix+key, dotsByte, time.Minute).Err()
	if err != nil {
		return nil, err
	}
	return map[string]interface{}{
		"key":          key,
		"image":        masterImageBase64,
		"thumb":        tileImageBase64,
		"thumb_width":  blockData.Width,
		"thumb_height": blockData.Height,
		"thumb_x":      blockData.TileX,
		"thumb_y":      blockData.TileY,
	}, nil

}