lan
d9aa4b46c3
- Introduce interfaces for all major services (JWT, PostAI, Comment, Message, Notification, QRCodeLogin, Upload, Vote, etc.) to support dependency inversion. - Move query parameters from `internal/dto` to a new `internal/query` package to separate request payloads from data transfer objects. - Refactor `internal/router` to use embedded `RouterDeps` for cleaner dependency management. - Decouple handlers from repositories by injecting services instead of direct repository access, ensuring proper layering. - Improve database initialization by moving it from `internal/model` to `internal/database`. - Optimize message decryption by implementing a more efficient `BatchDecrypt` method in `MessageEncryptor` using a worker pool. - Enhance error handling and security by implementing fail-fast checks for encryption key length during startup. - Clean up unused code, including the `avatar` package and several unused DTOs.
95 lines
2.0 KiB
Go
95 lines
2.0 KiB
Go
package handler
|
|
|
|
import (
|
|
"github.com/gin-gonic/gin"
|
|
|
|
"with_you/internal/pkg/response"
|
|
"with_you/internal/service"
|
|
)
|
|
|
|
// UploadHandler 上传处理器
|
|
type UploadHandler struct {
|
|
uploadService service.UploadService
|
|
}
|
|
|
|
// NewUploadHandler 创建上传处理器
|
|
func NewUploadHandler(uploadService service.UploadService) *UploadHandler {
|
|
return &UploadHandler{
|
|
uploadService: uploadService,
|
|
}
|
|
}
|
|
|
|
// UploadImage 上传图片
|
|
func (h *UploadHandler) UploadImage(c *gin.Context) {
|
|
userID := c.GetString("user_id")
|
|
if userID == "" {
|
|
response.Unauthorized(c, "")
|
|
return
|
|
}
|
|
|
|
file, err := c.FormFile("image")
|
|
if err != nil {
|
|
response.BadRequest(c, "image file is required")
|
|
return
|
|
}
|
|
|
|
url, previewURL, previewURLLarge, err := h.uploadService.UploadImage(c.Request.Context(), file)
|
|
if err != nil {
|
|
response.InternalServerError(c, "failed to upload image")
|
|
return
|
|
}
|
|
|
|
response.Success(c, gin.H{
|
|
"url": url,
|
|
"preview_url": previewURL,
|
|
"preview_url_large": previewURLLarge,
|
|
})
|
|
}
|
|
|
|
// UploadAvatar 上传头像
|
|
func (h *UploadHandler) UploadAvatar(c *gin.Context) {
|
|
userID := c.GetString("user_id")
|
|
if userID == "" {
|
|
response.Unauthorized(c, "")
|
|
return
|
|
}
|
|
|
|
file, err := c.FormFile("image")
|
|
|
|
if err != nil {
|
|
response.BadRequest(c, "avatar file is required")
|
|
return
|
|
}
|
|
|
|
url, err := h.uploadService.UploadAvatar(c.Request.Context(), userID, file)
|
|
if err != nil {
|
|
response.InternalServerError(c, "failed to upload avatar")
|
|
return
|
|
}
|
|
|
|
response.Success(c, gin.H{"url": url})
|
|
}
|
|
|
|
// UploadCover 上传头图（个人主页封面）
|
|
func (h *UploadHandler) UploadCover(c *gin.Context) {
|
|
userID := c.GetString("user_id")
|
|
if userID == "" {
|
|
response.Unauthorized(c, "")
|
|
return
|
|
}
|
|
|
|
file, err := c.FormFile("image")
|
|
if err != nil {
|
|
response.BadRequest(c, "image file is required")
|
|
return
|
|
}
|
|
|
|
url, err := h.uploadService.UploadCover(c.Request.Context(), userID, file)
|
|
if err != nil {
|
|
response.InternalServerError(c, "failed to upload cover")
|
|
return
|
|
}
|
|
|
|
response.Success(c, gin.H{"url": url})
|
|
}
|