mirror of
https://github.com/yangjian102621/geekai.git
synced 2025-09-18 01:06:39 +08:00
122 lines
2.9 KiB
Go
122 lines
2.9 KiB
Go
package handler
|
|
|
|
// * +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
|
|
// * Copyright 2023 The Geek-AI Authors. All rights reserved.
|
|
// * Use of this source code is governed by a Apache-2.0 license
|
|
// * that can be found in the LICENSE file.
|
|
// * @Author yangjian102621@163.com
|
|
// * +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
|
|
|
|
import (
|
|
"geekai/core"
|
|
"geekai/core/types"
|
|
"geekai/service/oss"
|
|
"geekai/store/model"
|
|
"geekai/store/vo"
|
|
"geekai/utils"
|
|
"geekai/utils/resp"
|
|
"github.com/gin-gonic/gin"
|
|
"gorm.io/gorm"
|
|
"time"
|
|
)
|
|
|
|
type UploadHandler struct {
|
|
BaseHandler
|
|
uploaderManager *oss.UploaderManager
|
|
}
|
|
|
|
func NewUploadHandler(app *core.AppServer, db *gorm.DB, manager *oss.UploaderManager) *UploadHandler {
|
|
return &UploadHandler{BaseHandler: BaseHandler{App: app, DB: db}, uploaderManager: manager}
|
|
}
|
|
|
|
func (h *UploadHandler) Upload(c *gin.Context) {
|
|
file, err := h.uploaderManager.GetUploadHandler().PutFile(c, "file")
|
|
if err != nil {
|
|
resp.ERROR(c, err.Error())
|
|
return
|
|
}
|
|
|
|
logger.Info("upload file: %s", file.Name)
|
|
// cut the file name if it's too long
|
|
if len(file.Name) > 100 {
|
|
file.Name = file.Name[:90] + file.Ext
|
|
}
|
|
|
|
userId := h.GetLoginUserId(c)
|
|
res := h.DB.Create(&model.File{
|
|
UserId: int(userId),
|
|
Name: file.Name,
|
|
ObjKey: file.ObjKey,
|
|
URL: file.URL,
|
|
Ext: file.Ext,
|
|
Size: file.Size,
|
|
CreatedAt: time.Time{},
|
|
})
|
|
if res.Error != nil || res.RowsAffected == 0 {
|
|
resp.ERROR(c, "error with update database: "+res.Error.Error())
|
|
return
|
|
}
|
|
|
|
resp.SUCCESS(c, file)
|
|
}
|
|
|
|
func (h *UploadHandler) List(c *gin.Context) {
|
|
var data struct {
|
|
Urls []string `json:"urls"`
|
|
}
|
|
if err := c.ShouldBindJSON(&data); err != nil {
|
|
resp.ERROR(c, types.InvalidArgs)
|
|
return
|
|
}
|
|
|
|
userId := h.GetLoginUserId(c)
|
|
var items []model.File
|
|
var files = make([]vo.File, 0)
|
|
session := h.DB.Session(&gorm.Session{})
|
|
session = session.Where("user_id = ?", userId)
|
|
if len(data.Urls) > 0 {
|
|
session = session.Where("url IN ?", data.Urls)
|
|
}
|
|
session.Find(&items)
|
|
if len(items) > 0 {
|
|
for _, v := range items {
|
|
var file vo.File
|
|
err := utils.CopyObject(v, &file)
|
|
if err != nil {
|
|
logger.Error(err)
|
|
continue
|
|
}
|
|
file.CreatedAt = v.CreatedAt.Unix()
|
|
files = append(files, file)
|
|
}
|
|
}
|
|
|
|
resp.SUCCESS(c, files)
|
|
}
|
|
|
|
// Remove remove files
|
|
func (h *UploadHandler) Remove(c *gin.Context) {
|
|
userId := h.GetLoginUserId(c)
|
|
id := h.GetInt(c, "id", 0)
|
|
var file model.File
|
|
tx := h.DB.Where("user_id = ? AND id = ?", userId, id).First(&file)
|
|
if tx.Error != nil || file.Id == 0 {
|
|
resp.ERROR(c, "file not existed")
|
|
return
|
|
}
|
|
|
|
// remove database
|
|
tx = h.DB.Model(&model.File{}).Delete("id = ?", id)
|
|
if tx.Error != nil || tx.RowsAffected == 0 {
|
|
resp.ERROR(c, "failed to update database")
|
|
return
|
|
}
|
|
// remove files
|
|
objectKey := file.ObjKey
|
|
if objectKey == "" {
|
|
objectKey = file.URL
|
|
}
|
|
_ = h.uploaderManager.GetUploadHandler().Delete(objectKey)
|
|
resp.SUCCESS(c)
|
|
}
|