SCHISANDRA/schisandra-album-cloud-microservices
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

✨ code migration

Browse Source
This commit is contained in:
landaiqing
2024-11-15 02:02:19 +08:00
parent b2d753e832
commit 34c4690f80
85 changed files with 4349 additions and 421 deletions
Download Patch File Download Diff File Expand all files Collapse all files

44
app/core/api/common/captcha/generate/generate_basic_text_captcha.go Normal file
View File

@@ -0,0 +1,44 @@
package generate
import (
"context"
"encoding/json"
"time"
"github.com/redis/go-redis/v9"
"github.com/wenlng/go-captcha-assets/helper"
"github.com/wenlng/go-captcha/v2/click"
"schisandra-album-cloud-microservices/app/core/api/common/constant"
)
// GenerateBasicTextCaptcha generates a basic text captcha and saves it to redis.
func GenerateBasicTextCaptcha(capt click.Captcha, redis redis.Client, ctx context.Context) map[string]interface{} {
captData, err := capt.Generate()
if err != nil {
return nil
}
dotData := captData.GetData()
if dotData == nil {
return nil
}
var masterImageBase64, thumbImageBase64 string
masterImageBase64 = captData.GetMasterImage().ToBase64()
thumbImageBase64 = captData.GetThumbImage().ToBase64()
dotsByte, err := json.Marshal(dotData)
if err != nil {
return nil
}
key := helper.StringToMD5(string(dotsByte))
err = redis.Set(ctx, constant.UserCaptchaPrefix+key, dotsByte, time.Minute).Err()
if err != nil {
return nil
}
return map[string]interface{}{
"key": key,
"image": masterImageBase64,
"thumb": thumbImageBase64,
}
}

44
app/core/api/common/captcha/generate/generate_click_shape_captcha.go Normal file
View File

@@ -0,0 +1,44 @@
package generate
import (
"context"
"encoding/json"
"log"
"time"
"github.com/redis/go-redis/v9"
"github.com/wenlng/go-captcha-assets/helper"
"github.com/wenlng/go-captcha/v2/click"
"schisandra-album-cloud-microservices/app/core/api/common/constant"
)
// GenerateClickShapeCaptcha generate click shape captcha
func GenerateClickShapeCaptcha(click click.Captcha, redis redis.Client, ctx context.Context) map[string]interface{} {
captData, err := click.Generate()
if err != nil {
log.Fatalln(err)
}
dotData := captData.GetData()
if dotData == nil {
return nil
}
var masterImageBase64, thumbImageBase64 string
masterImageBase64 = captData.GetMasterImage().ToBase64()
thumbImageBase64 = captData.GetThumbImage().ToBase64()
dotsByte, err := json.Marshal(dotData)
if err != nil {
return nil
}
key := helper.StringToMD5(string(dotsByte))
err = redis.Set(ctx, constant.UserCaptchaPrefix+key, dotsByte, time.Minute).Err()
if err != nil {
return nil
}
return map[string]interface{}{
"key": key,
"image": masterImageBase64,
"thumb": thumbImageBase64,
}
}

42
app/core/api/common/captcha/generate/generate_rotate_captcha.go Normal file
View File

@@ -0,0 +1,42 @@
package generate
import (
"context"
"encoding/json"
"errors"
"time"
"github.com/redis/go-redis/v9"
"github.com/wenlng/go-captcha-assets/helper"
"github.com/wenlng/go-captcha/v2/rotate"
"schisandra-album-cloud-microservices/app/core/api/common/constant"
)
// GenerateRotateCaptcha generate rotate captcha
func GenerateRotateCaptcha(captcha rotate.Captcha, redis *redis.Client, ctx context.Context) (map[string]interface{}, error) {
captchaData, err := captcha.Generate()
if err != nil {
return nil, err
}
blockData := captchaData.GetData()
if blockData == nil {
return nil, errors.New("captcha data is nil")
}
masterImageBase64 := captchaData.GetMasterImage().ToBase64()
thumbImageBase64 := captchaData.GetThumbImage().ToBase64()
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": thumbImageBase64,
}, nil
}

50
app/core/api/common/captcha/generate/generate_slide_basic_captcha.go Normal file
View File

@@ -0,0 +1,50 @@
package generate
import (
"context"
"encoding/json"
"errors"
"time"
"github.com/redis/go-redis/v9"
"github.com/wenlng/go-captcha-assets/helper"
"github.com/wenlng/go-captcha/v2/slide"
"schisandra-album-cloud-microservices/app/core/api/common/constant"
)
// 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, errors.New("block data is nil")
}
var masterImageBase64, tileImageBase64 string
masterImageBase64 = captData.GetMasterImage().ToBase64()
tileImageBase64 = captData.GetTileImage().ToBase64()
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
}

50
app/core/api/common/captcha/generate/generate_slide_region_captcha.go Normal file
View File

@@ -0,0 +1,50 @@
package generate
import (
"context"
"encoding/json"
"time"
"github.com/redis/go-redis/v9"
"github.com/wenlng/go-captcha-assets/helper"
"github.com/wenlng/go-captcha/v2/slide"
"schisandra-album-cloud-microservices/app/core/api/common/constant"
)
// GenerateSlideRegionCaptcha generate slide region captcha
func GenerateSlideRegionCaptcha(slide slide.Captcha, redis redis.Client, ctx context.Context) map[string]interface{} {
captData, err := slide.Generate()
if err != nil {
return nil
}
blockData := captData.GetData()
if blockData == nil {
return nil
}
var masterImageBase64, tileImageBase64 string
masterImageBase64 = captData.GetMasterImage().ToBase64()
tileImageBase64 = captData.GetTileImage().ToBase64()
blockByte, err := json.Marshal(blockData)
if err != nil {
return nil
}
key := helper.StringToMD5(string(blockByte))
err = redis.Set(ctx, constant.UserCaptchaPrefix+key, blockByte, time.Minute).Err()
if err != nil {
return nil
}
return map[string]interface{}{
"code": 0,
"key": key,
"image": masterImageBase64,
"tile": tileImageBase64,
"tile_width": blockData.Width,
"tile_height": blockData.Height,
"tile_x": blockData.TileX,
"tile_y": blockData.TileY,
}
}

48
app/core/api/common/captcha/verify/verify_basic_text_captcha.go Normal file
View File

@@ -0,0 +1,48 @@
package verify
import (
"context"
"encoding/json"
"fmt"
"strconv"
"strings"
"github.com/redis/go-redis/v9"
"github.com/wenlng/go-captcha/v2/click"
"schisandra-album-cloud-microservices/app/core/api/common/constant"
)
// VerifyBasicTextCaptcha verify basic text captcha
func VerifyBasicTextCaptcha(dots string, key string, redis *redis.Client, ctx context.Context) bool {
cacheDataByte, err := redis.Get(ctx, constant.UserCaptchaPrefix+key).Bytes()
if len(cacheDataByte) == 0 || err != nil {
return false
}
src := strings.Split(dots, ",")
var dct map[int]*click.Dot
if err := json.Unmarshal(cacheDataByte, &dct); err != nil {
return false
}
chkRet := false
if (len(dct) * 2) == len(src) {
for i := 0; i < len(dct); i++ {
dot := dct[i]
j := i * 2
k := i*2 + 1
sx, _ := strconv.ParseFloat(fmt.Sprintf("%v", src[j]), 64)
sy, _ := strconv.ParseFloat(fmt.Sprintf("%v", src[k]), 64)
chkRet = click.CheckPoint(int64(sx), int64(sy), int64(dot.X), int64(dot.Y), int64(dot.Width), int64(dot.Height), 0)
if !chkRet {
break
}
}
}
if chkRet {
return true
}
return false
}

8
app/core/api/common/constant/oauth_type.go Normal file
View File

@@ -0,0 +1,8 @@
package constant
const (
OAuthSourceWechat = "wechat"
OAuthSourceQQ = "qq"
OAuthSourceGithub = "github"
OAuthSourceGitee = "gitee"
)

17
app/core/api/common/constant/redis_prefix.go
View File

@@ -1,9 +1,26 @@
package constant
// 用户相关的redis key
const (
UserClientPrefix string = "user:client:"
UserSessionPrefix string = "user:session:"
UserCaptchaPrefix string = "user:captcha:"
UserTokenPrefix string = "user:token:"
UserSmsRedisPrefix string = "user:sms:"
UserQrcodePrefix = "user:qrcode:"
)
// 评论相关的redis key
const (
CommentSubmitCaptchaPrefix = "comment:submit:captcha:"
CommentOfflineMessagePrefix = "comment:offline:message:"
CommentLikeLockPrefix = "comment:like:lock:"
CommentDislikeLockPrefix = "comment:dislike:lock:"
CommentLikeListPrefix = "comment:like:list:"
CommentUserListPrefix = "comment:user:list:"
)
// 系统相关的redis key
const (
SystemApiNonceRedisKey = "system:controller:nonce:"
)

20
app/core/api/common/utils/gen_validate_code.go Normal file
View File

@@ -0,0 +1,20 @@
package utils
import (
"fmt"
"math/rand"
"strings"
"time"
)
func GenValidateCode(width int) string {
numeric := [10]byte{0, 1, 2, 3, 4, 5, 6, 7, 8, 9}
r := len(numeric)
rand.New(rand.NewSource(time.Now().UnixNano()))
var sb strings.Builder
for i := 0; i < width; i++ {
fmt.Fprintf(&sb, "%d", numeric[rand.Intn(r)])
}
return sb.String()
}