Gmarker689
ecb74a2aed
后端:
- 异步生成: POST /api/v1/generate 立即返回 taskId，后台执行管线
- 任务轮询: GET /api/v1/tasks/:id + GET /api/v1/tasks/:id/assets
- 图片保存: 生成图片写入 ../generation/{projectId}/{taskId}/，静态服务
- 图片编辑: POST /api/v1/images/edit (multipart/form-data)
- JWT 中间件: mildware/auth.go 保护生成/编辑端点
- config.yml 清空敏感默认值，交由 .env 控制
- ImageGenConfig 新增 Quality 字段
前端:
- api/generate.ts: 对接真实 API (submitGenerate + poll getTask/getAssets)
- api/types.ts: 新增 GenerateResponse, AssetsResponse, Task 类型
- stores/generation.ts: 异步提交→轮询进度→获取素材→完成
- stores/task.ts: 默认分辨率 256→1024
- GenerateForm: 分辨率范围 1024-1536
- GeneratePage: 显示状态文本，完成后可查看结果/继续生成
- ResultPage: 从 store 读取，下载功能实现
80 lines
2.0 KiB
Go
Executable File
80 lines
2.0 KiB
Go
Executable File
package service
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"time"
|
|
|
|
"gen2d/internal/db"
|
|
"gen2d/internal/model"
|
|
|
|
"github.com/golang-jwt/jwt/v5"
|
|
"golang.org/x/crypto/bcrypt"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
// AuthService 用户认证服务。
|
|
type AuthService struct {
|
|
jwtSecret []byte
|
|
jwtExpire time.Duration
|
|
}
|
|
|
|
// NewAuthService 创建 AuthService 实例。
|
|
func NewAuthService(jwtSecret string, jwtExpire int64) *AuthService {
|
|
return &AuthService{
|
|
jwtSecret: []byte(jwtSecret),
|
|
jwtExpire: time.Duration(jwtExpire) * time.Second,
|
|
}
|
|
}
|
|
|
|
// Register 注册新用户，密码使用 bcrypt 加密存储。
|
|
func (s *AuthService) Register(ctx context.Context, username, password, email string) (*model.User, error) {
|
|
hashed, err := bcrypt.GenerateFromPassword([]byte(password), bcrypt.DefaultCost)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
user := model.User{
|
|
Username: username,
|
|
Password: string(hashed),
|
|
Email: email,
|
|
}
|
|
|
|
if err := db.DB.WithContext(ctx).Create(&user).Error; err != nil {
|
|
if errors.Is(err, gorm.ErrDuplicatedKey) {
|
|
return nil, errors.New("用户名已存在")
|
|
}
|
|
return nil, err
|
|
}
|
|
|
|
return &user, nil
|
|
}
|
|
|
|
// Login 校验用户名密码，成功返回 JWT token 及用户信息。
|
|
func (s *AuthService) Login(ctx context.Context, username, password string) (string, int64, *model.User, error) {
|
|
var user model.User
|
|
if err := db.DB.WithContext(ctx).Where("username = ?", username).First(&user).Error; err != nil {
|
|
if errors.Is(err, gorm.ErrRecordNotFound) {
|
|
return "", 0, nil, errors.New("用户名或密码错误")
|
|
}
|
|
return "", 0, nil, err
|
|
}
|
|
|
|
if err := bcrypt.CompareHashAndPassword([]byte(user.Password), []byte(password)); err != nil {
|
|
return "", 0, nil, errors.New("用户名或密码错误")
|
|
}
|
|
|
|
expiresAt := time.Now().Add(s.jwtExpire)
|
|
claims := jwt.MapClaims{
|
|
"sub": user.ID,
|
|
"exp": expiresAt.Unix(),
|
|
"iat": time.Now().Unix(),
|
|
}
|
|
token, err := jwt.NewWithClaims(jwt.SigningMethodHS256, claims).SignedString(s.jwtSecret)
|
|
if err != nil {
|
|
return "", 0, nil, err
|
|
}
|
|
|
|
return token, int64(s.jwtExpire.Seconds()), &user, nil
|
|
}
|