70 lines
2.2 KiB
Go
70 lines
2.2 KiB
Go
package utils
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"fmt"
|
|
"os"
|
|
"path/filepath"
|
|
|
|
"github.com/samber/lo"
|
|
"github.com/xmdhs/authlib-skin/config"
|
|
"github.com/xmdhs/authlib-skin/db/ent"
|
|
"github.com/xmdhs/authlib-skin/db/ent/texture"
|
|
"github.com/xmdhs/authlib-skin/db/ent/usertexture"
|
|
)
|
|
|
|
func DelTexture(ctx context.Context, userProfileID int, textureType string, client *ent.Client, config config.Config) error {
|
|
// 查找此用户该类型下是否已经存在皮肤
|
|
tl, err := client.UserTexture.Query().Where(usertexture.And(
|
|
usertexture.UserProfileID(userProfileID),
|
|
usertexture.Type(textureType),
|
|
)).All(ctx)
|
|
if err != nil {
|
|
return fmt.Errorf("DelTexture: %w", err)
|
|
}
|
|
if len(tl) == 0 {
|
|
return nil
|
|
}
|
|
// 若存在，查找是否被引用
|
|
for _, v := range tl {
|
|
c, err := client.UserTexture.Query().Where(usertexture.TextureID(v.TextureID)).Count(ctx)
|
|
if err != nil {
|
|
return fmt.Errorf("DelTexture: %w", err)
|
|
}
|
|
if c == 1 {
|
|
// 若没有其他用户使用该皮肤，删除文件和记录
|
|
t, err := client.Texture.Query().Where(texture.ID(v.TextureID)).Only(ctx)
|
|
if err != nil {
|
|
var nf *ent.NotFoundError
|
|
if errors.As(err, &nf) {
|
|
continue
|
|
}
|
|
return fmt.Errorf("DelTexture: %w", err)
|
|
}
|
|
path := filepath.Join(config.TexturePath, t.TextureHash[:2], t.TextureHash[2:4], t.TextureHash)
|
|
err = os.Remove(path)
|
|
if err != nil && !errors.Is(err, os.ErrNotExist) {
|
|
return fmt.Errorf("DelTexture: %w", err)
|
|
}
|
|
// Texture 表中删除记录
|
|
err = client.Texture.DeleteOneID(v.TextureID).Exec(ctx)
|
|
if err != nil {
|
|
return fmt.Errorf("DelTexture: %w", err)
|
|
}
|
|
}
|
|
}
|
|
ids := lo.Map[*ent.UserTexture, int](tl, func(item *ent.UserTexture, index int) int {
|
|
return item.ID
|
|
})
|
|
// 中间表删除记录
|
|
// UserProfile 上没有于此相关的字段，所以无需操作
|
|
_, err = client.UserTexture.Delete().Where(usertexture.IDIn(ids...)).Exec(ctx)
|
|
// 小概率皮肤上传后，高并发时被此处清理。问题不大重新上传一遍就行。
|
|
// 条件为使用一个独一无二的皮肤的用户，更换皮肤时，另一个用户同时更换自己的皮肤到这个独一无二的皮肤上。
|
|
if err != nil {
|
|
return fmt.Errorf("DelTexture: %w", err)
|
|
}
|
|
return nil
|
|
}
|